Keep track of files in a package

package.go

@@ -14,6 +14,9 @@ type Package struct {
 	// Name of the package
 	Name string
 
+	// Files in this package
+	Files []string
+
 	// OutRefs and InRefs are slices of references. For OutRefs
 	// the Ref is to an identifier in another package. For InRefs
 	// the Ref is to an identifier within this package.  Most

packagegraph.go

@@ -177,6 +177,9 @@ func (pg *PackageGraph) loadPackage(prog *loader.Program, loadpath string, pi *l
 
 	// Iterate over all files in that package.
 	for _, f := range pi.Files {
+		// Add that file to the package's file list
+		pkg.Files = append(pkg.Files, corpus.Rel(prog.Fset.File(f.Package).Name()))
+
 		// Iterate over all imports in that file
 		for _, imported := range f.Imports {
 			// Find the import's load-path and load that

simple_test.go

@@ -20,6 +20,7 @@ func TestSimplePackage(t *testing.T) {
 	assert.Contains(t, pg.Packages, "fmt")
 	pkg := pg.Packages[pkgpath]
 	assert.Empty(t, pkg.InRefs)
+	assert.Contains(t, pkg.Files, filepath)
 
 	testutils.AssertPresenceOfRef(t, pg.Packages["fmt"], "fmt", pkg, "fmt", goref.Import, true)