pkger/index.go

package pkger

import (
	"fmt"
	"os"
	"path/filepath"
	"strings"

	"github.com/gobuffalo/here"
	"github.com/markbates/pkger/pkgs"
)

type index struct {
	Pkg   string
	Files map[Path]*File
}

func (i index) Walk(pt Path, wf WalkFunc) error {

	if len(pt.Pkg) == 0 {
		pt.Pkg = i.Pkg
	}
	if len(i.Files) > 0 {
		for k, v := range i.Files {
			if k.Pkg != pt.Pkg {
				continue
			}
			if err := wf(k, v.info, nil); err != nil {
				return err
			}
		}
	}

	var info here.Info
	var err error
	if pt.Pkg == "." {
		info, err = pkgs.Current()
		if err != nil {
			return err
		}
		pt.Pkg = info.ImportPath
	}
	if info.IsZero() {
		info, err = pkgs.Pkg(pt.Pkg)
		if err != nil {
			return fmt.Errorf("%s: %s", pt, err)
		}
	}

	err = filepath.Walk(info.Dir, func(path string, fi os.FileInfo, err error) error {
		if err != nil {
			return err
		}

		path = strings.TrimPrefix(path, info.Dir)
		pt, err := Parse(fmt.Sprintf("%s:%s", pt.Pkg, path))
		if err != nil {
			return err
		}
		return wf(pt, NewFileInfo(fi), err)
	})

	return err
}

func (i index) Open(pt Path) (*File, error) {
	if len(pt.Pkg) == 0 {
		pt.Pkg = i.Pkg
	}

	f, ok := i.Files[pt]
	if !ok {
		return i.openDisk(pt)
	}
	return &File{
		info: f.info,
		path: f.path,
		data: f.data,
		her:  f.her,
		index: &index{
			Files: map[Path]*File{},
		},
	}, nil
}

func (i index) openDisk(pt Path) (*File, error) {
	if len(pt.Pkg) == 0 {
		pt.Pkg = i.Pkg
	}
	info, err := pkgs.Pkg(pt.Pkg)
	if err != nil {
		return nil, err
	}
	fp := info.Dir
	if len(pt.Name) > 0 {
		fp = filepath.Join(fp, pt.Name)
	}

	fi, err := os.Stat(fp)
	if err != nil {
		return nil, err
	}
	f := &File{
		info: WithName(pt.Name, NewFileInfo(fi)),
		her:  info,
		path: pt,
		index: &index{
			Files: map[Path]*File{},
		},
	}
	return f, nil
}

var rootIndex = &index{
	Files: map[Path]*File{},
}