forked from mirror/pkger
154 lines
2.5 KiB
Go
154 lines
2.5 KiB
Go
package hdfs
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/markbates/pkger/fs"
|
|
"github.com/markbates/pkger/here"
|
|
"github.com/markbates/pkger/internal/maps"
|
|
)
|
|
|
|
var _ fs.FileSystem = &FS{}
|
|
|
|
type FS struct {
|
|
infos *maps.Infos
|
|
paths *maps.Paths
|
|
current here.Info
|
|
}
|
|
|
|
func New() (*FS, error) {
|
|
f := &FS{
|
|
infos: &maps.Infos{},
|
|
paths: &maps.Paths{},
|
|
}
|
|
|
|
var err error
|
|
f.current, err = here.Current()
|
|
return f, err
|
|
}
|
|
|
|
func (fx *FS) Create(name string) (fs.File, error) {
|
|
name, err := fx.locate(name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
f, err := os.Create(name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewFile(fx, f)
|
|
}
|
|
|
|
func (f *FS) Current() (here.Info, error) {
|
|
return f.current, nil
|
|
}
|
|
|
|
func (f *FS) Info(p string) (here.Info, error) {
|
|
info, ok := f.infos.Load(p)
|
|
if ok {
|
|
return info, nil
|
|
}
|
|
|
|
info, err := here.Package(p)
|
|
if err != nil {
|
|
return info, err
|
|
}
|
|
f.infos.Store(p, info)
|
|
return info, nil
|
|
}
|
|
|
|
func (f *FS) MkdirAll(p string, perm os.FileMode) error {
|
|
p, err := f.locate(p)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return os.MkdirAll(p, perm)
|
|
}
|
|
|
|
func (fx *FS) Open(name string) (fs.File, error) {
|
|
name, err := fx.locate(name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
f, err := os.Open(name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewFile(fx, f)
|
|
}
|
|
|
|
func (f *FS) Parse(p string) (fs.Path, error) {
|
|
return f.paths.Parse(p)
|
|
}
|
|
|
|
func (f *FS) ReadFile(s string) ([]byte, error) {
|
|
s, err := f.locate(s)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return ioutil.ReadFile(s)
|
|
}
|
|
|
|
func (f *FS) Stat(name string) (os.FileInfo, error) {
|
|
name, err := f.locate(name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return os.Stat(name)
|
|
}
|
|
|
|
func (f *FS) Walk(p string, wf filepath.WalkFunc) error {
|
|
fp, err := f.locate(p)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
pt, err := f.Parse(p)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = filepath.Walk(fp, func(path string, fi os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
path = strings.TrimPrefix(path, fp)
|
|
pt, err := f.Parse(fmt.Sprintf("%s:%s", pt.Pkg, path))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return wf(pt.String(), fs.WithName(path, fs.NewFileInfo(fi)), nil)
|
|
})
|
|
|
|
return err
|
|
}
|
|
|
|
func (f *FS) locate(p string) (string, error) {
|
|
pt, err := f.Parse(p)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
var info here.Info
|
|
if pt.Pkg == "." {
|
|
info, err = f.Current()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
pt.Pkg = info.ImportPath
|
|
}
|
|
|
|
if info.IsZero() {
|
|
info, err = f.Info(pt.Pkg)
|
|
if err != nil {
|
|
return "", fmt.Errorf("%s: %s", pt, err)
|
|
}
|
|
}
|
|
fp := filepath.Join(info.Dir, pt.Name)
|
|
return fp, nil
|
|
}
|