mirror of https://github.com/markbates/pkger.git
45 lines
808 B
Go
45 lines
808 B
Go
|
package mem
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"compress/gzip"
|
||
|
|
"encoding/hex"
|
||
|
|
"encoding/json"
|
||
|
|
|
||
|
|
"github.com/markbates/pkger/pkging"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (pkg *Pkger) MarshalEmbed() ([]byte, error) {
|
||
|
|
bb := &bytes.Buffer{}
|
||
|
|
gz := gzip.NewWriter(bb)
|
||
|
|
defer gz.Close()
|
||
|
|
if err := json.NewEncoder(gz).Encode(pkg); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if err := gz.Close(); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
s := hex.EncodeToString(bb.Bytes())
|
||
|
|
return []byte(s), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func UnmarshalEmbed(in []byte) (pkging.Pkger, error) {
|
||
|
|
b := make([]byte, len(in))
|
||
|
|
if _, err := hex.Decode(b, in); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
gz, err := gzip.NewReader(bytes.NewReader(b))
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
defer gz.Close()
|
||
|
|
|
||
|
|
pkg := &Pkger{}
|
||
|
|
if err := json.NewDecoder(gz).Decode(pkg); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return pkg, nil
|
||
|
|
}
|