go/packages: add support for file= queries on ad-hoc packages

This will do a go list using a filename directly if listing using
the directory failed.

Fixes golang/go#32587

Change-Id: Id9993968f0ebc18a455132e0f1468356416a66dd
Reviewed-on: https://go-review.googlesource.com/c/tools/+/182465
Run-TryBot: Michael Matloob <matloob@golang.org>
Reviewed-by: Rebecca Stambler <rstambler@golang.org>
This commit is contained in:
Michael Matloob 2019-06-14 13:17:31 -04:00
parent 7653e6d3d5
commit 4bb205a206
2 changed files with 49 additions and 0 deletions

View File

@ -231,6 +231,13 @@ func runContainsQueries(cfg *Config, driver driver, response *responseDeduper, q
if err != nil {
return err
}
if len(dirResponse.Roots) == 0 {
// Couldn't find a package for the directory. Try to load the file as an ad-hoc package.
dirResponse, err = driver(cfg, query)
if err != nil {
return err
}
}
isRoot := make(map[string]bool, len(dirResponse.Roots))
for _, root := range dirResponse.Roots {
isRoot[root] = true

View File

@ -1854,6 +1854,48 @@ func testMissingDependency(t *testing.T, exporter packagestest.Exporter) {
}
}
func TestAdHocContains(t *testing.T) { packagestest.TestAll(t, testAdHocContains) }
func testAdHocContains(t *testing.T, exporter packagestest.Exporter) {
exported := packagestest.Export(t, exporter, []packagestest.Module{{
Name: "golang.org/fake",
Files: map[string]interface{}{
"a/a.go": `package a;`,
}}})
defer exported.Cleanup()
tmpfile, err := ioutil.TempFile("", "adhoc*.go")
filename := tmpfile.Name()
if err != nil {
t.Fatal(err)
}
fmt.Fprint(tmpfile, `package main; import "fmt"; func main() { fmt.Println("time for coffee") }`)
if err := tmpfile.Close(); err != nil {
t.Fatal(err)
}
defer func() {
if err := os.Remove(filename); err != nil {
t.Fatal(err)
}
}()
exported.Config.Mode = packages.NeedImports | packages.NeedFiles
pkgs, err := packages.Load(exported.Config, filename)
if err != nil {
t.Fatal(err)
}
if len(pkgs) != 1 && pkgs[0].PkgPath != "command-line-arguments" {
t.Fatalf("packages.Load: want [command-line-arguments], got %v", pkgs)
}
pkg := pkgs[0]
if _, ok := pkg.Imports["fmt"]; !ok || len(pkg.Imports) != 1 {
t.Fatalf("Imports of loaded package: want [fmt], got %v", pkg.Imports)
}
if len(pkg.GoFiles) != 1 || pkg.GoFiles[0] != filename {
t.Fatalf("GoFiles of loaded packge: want [%s], got %v", filename, pkg.GoFiles)
}
}
func errorMessages(errors []packages.Error) []string {
var msgs []string
for _, err := range errors {