kjuulh/moq
1
0
Fork 0
Code Issues 1 Pull Requests Projects Releases Wiki Activity

Merge pull request #80 from breml/abs-path-modules

Browse Source 
Import package by files if outside of GOPATH
This commit is contained in:
Mat Ryer 2019-01-21 17:22:14 +00:00 committed by GitHub
commit 055cc3eebc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pkg/moq/moq.go
View File

@ -10,6 +10,7 @@ import (
"go/token"
"go/types"
"io"
"io/ioutil"
"os"
"path"
"path/filepath"
@ -234,7 +235,19 @@ func pkgInfoFromPath(src string) (*loader.PackageInfo, error) {
ParserMode: parser.SpuriousErrors,
Cwd: src,
}
conf.Import(pkgFull)
if strings.HasPrefix(pkgFull, string(filepath.Separator)) {
files, err := ioutil.ReadDir(pkgFull)
if err != nil {
return nil, err
}
for _, file := range files {
if !file.IsDir() && strings.HasSuffix(file.Name(), ".go") && !strings.HasSuffix(file.Name(), "_test.go") {
conf.CreateFromFilenames(abs, file.Name())
}
}
} else {
conf.Import(pkgFull)
}
lprog, err := conf.Load()
if err != nil {
return nil, err