pkger/parser/parser.go

package parser

import (
	"fmt"
	"go/parser"
	"go/token"
	"os"

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

// var DefaultIgnoredFolders = []string{".", "_", "vendor", "node_modules", "_fixtures", "testdata"}

func Parse(her here.Info) (Decls, error) {
	src, err := fromSource(her)
	if err != nil {
		return nil, err
	}

	return src, nil
}

func fromSource(her here.Info) (Decls, error) {
	root := her.Dir
	fi, err := os.Stat(root)
	if err != nil {
		return nil, err
	}
	if !fi.IsDir() {
		return nil, fmt.Errorf("%q is not a directory", root)
	}

	fset := token.NewFileSet()

	pkgs, err := parser.ParseDir(fset, root, nil, 0)
	if err != nil {
		return nil, err
	}

	var decls Decls

	for _, pkg := range pkgs {
		for name, pf := range pkg.Files {
			f := &file{
				fset:     fset,
				astFile:  pf,
				filename: name,
			}

			x, err := f.find()
			if err != nil {
				return nil, err
			}
			decls = append(decls, x...)
		}
	}

	return decls, nil
}