forked from mirror/pkger
49 lines
762 B
Go
49 lines
762 B
Go
|
package mem
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"io"
|
||
|
|
"path/filepath"
|
||
|
|
|
||
|
|
"github.com/markbates/pkger/pkging"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Add copies the pkging.File into the *Pkger
|
||
|
|
func (fx *Pkger) Add(files ...pkging.File) error {
|
||
|
|
for _, f := range files {
|
||
|
|
info, err := f.Stat()
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if f.Path().Pkg == fx.Here.ImportPath {
|
||
|
|
dir := filepath.Dir(f.Name())
|
||
|
|
if dir != "/" {
|
||
|
|
if err := fx.MkdirAll(dir, 0755); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
mf := &File{
|
||
|
|
Here: f.Info(),
|
||
|
|
info: pkging.NewFileInfo(info),
|
||
|
|
path: f.Path(),
|
||
|
|
pkging: fx,
|
||
|
|
}
|
||
|
|
|
||
|
|
if !info.IsDir() {
|
||
|
|
bb := &bytes.Buffer{}
|
||
|
|
_, err = io.Copy(bb, f)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
mf.data = bb.Bytes()
|
||
|
|
}
|
||
|
|
|
||
|
|
fx.files.Store(mf.Path(), mf)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|