feat: In doc, add newPkgDataFromMemPkg

Signed-off-by: Jeff Thompson <[email protected]>

Author: jefft0

gnovm/pkg/doc/pkg.go

@@ -6,9 +6,11 @@ import (
 	"go/doc"
 	"go/parser"
 	"go/token"
-	"os"
 	"path/filepath"
 	"strings"
+
+	"github.com/gnolang/gno/gnovm"
+	"github.com/gnolang/gno/gnovm/pkg/gnolang"
 )
 
 type pkgData struct {
@@ -36,54 +38,55 @@ type symbolData struct {
 }
 
 func newPkgData(dir bfsDir, unexported bool) (*pkgData, error) {
-	files, err := os.ReadDir(dir.dir)
+	memPkg, err := gnolang.ReadMemPackage(dir.dir, dir.importPath)
 	if err != nil {
-		return nil, fmt.Errorf("commands/doc: open %q: %w", dir.dir, err)
+		return nil, fmt.Errorf("commands/doc: read files %q: %w", dir.dir, err)
 	}
+	return newPkgDataFromMemPkg(memPkg, unexported)
+}
+
+func newPkgDataFromMemPkg(memPkg *gnovm.MemPackage, unexported bool) (*pkgData, error) {
 	pkg := &pkgData{
-		dir:  dir,
+		dir: bfsDir{
+			importPath: memPkg.Name,
+			dir:        memPkg.Path,
+		},
 		fset: token.NewFileSet(),
 	}
-	for _, file := range files {
-		n := file.Name()
+	for _, file := range memPkg.Files {
+		n := file.Name
 		// Ignore files with prefix . or _ like go tools do.
 		// Ignore _filetest.gno, but not _test.gno, as we use those to compute
 		// examples.
-		if file.IsDir() ||
-			!strings.HasSuffix(n, ".gno") ||
+		if !strings.HasSuffix(n, ".gno") ||
 			strings.HasPrefix(n, ".") ||
 			strings.HasPrefix(n, "_") ||
 			strings.HasSuffix(n, "_filetest.gno") {
 			continue
 		}
-		fullPath := filepath.Join(dir.dir, n)
-		err := pkg.parseFile(fullPath, unexported)
+		fullPath := filepath.Join(memPkg.Path, n)
+		err := pkg.parseFile(n, file.Body, unexported)
 		if err != nil {
 			return nil, fmt.Errorf("commands/doc: parse file %q: %w", fullPath, err)
 		}
 	}
 
 	if len(pkg.files) == 0 {
-		return nil, fmt.Errorf("commands/doc: no valid gno files in %q", dir.dir)
+		return nil, fmt.Errorf("commands/doc: no valid gno files in %q", memPkg.Path)
 	}
 	pkgName := pkg.files[0].Name.Name
 	for _, file := range pkg.files[1:] {
 		if file.Name.Name != pkgName {
-			return nil, fmt.Errorf("commands/doc: multiple packages (%q / %q) in dir %q", pkgName, file.Name.Name, dir.dir)
+			return nil, fmt.Errorf("commands/doc: multiple packages (%q / %q) in dir %q", pkgName, file.Name.Name, memPkg.Path)
 		}
 	}
 	pkg.name = pkgName
 
 	return pkg, nil
 }
 
-func (pkg *pkgData) parseFile(fileName string, unexported bool) error {
-	f, err := os.Open(fileName)
-	if err != nil {
-		return err
-	}
-	defer f.Close()
-	astf, err := parser.ParseFile(pkg.fset, filepath.Base(fileName), f, parser.ParseComments)
+func (pkg *pkgData) parseFile(fileName string, body string, unexported bool) error {
+	astf, err := parser.ParseFile(pkg.fset, fileName, body, parser.ParseComments)
 	if err != nil {
 		return err
 	}