89 lines
1.6 KiB
Go
89 lines
1.6 KiB
Go
package here
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
// Dir attempts to gather info for the requested directory.
|
|
func (h Here) Dir(p string) (Info, error) {
|
|
i, err := h.cache(p, func(p string) (Info, error) {
|
|
var i Info
|
|
|
|
fi, err := os.Stat(p)
|
|
if err != nil {
|
|
return i, err
|
|
}
|
|
|
|
if !fi.IsDir() {
|
|
p = filepath.Dir(p)
|
|
}
|
|
|
|
pwd, err := os.Getwd()
|
|
if err != nil {
|
|
return i, err
|
|
}
|
|
|
|
defer os.Chdir(pwd)
|
|
|
|
os.Chdir(p)
|
|
|
|
b, err := run("go", "list", "-json")
|
|
// go: cannot find main module; see 'go help modules'
|
|
// build .: cannot find module for path .
|
|
// no Go files in
|
|
if err != nil {
|
|
if nonGoDirRx.MatchString(err.Error()) {
|
|
return fromNonGoDir(p)
|
|
}
|
|
return i, err
|
|
}
|
|
|
|
if err := json.Unmarshal(b, &i); err != nil {
|
|
return i, err
|
|
}
|
|
|
|
return i, nil
|
|
})
|
|
|
|
if err != nil {
|
|
return i, fmt.Errorf("here.Dir: %s: %w", p, err)
|
|
}
|
|
|
|
return h.cache(i.ImportPath, func(p string) (Info, error) {
|
|
return i, nil
|
|
})
|
|
}
|
|
|
|
// Dir attempts to gather info for the requested directory.
|
|
func Dir(p string) (Info, error) {
|
|
return New().Dir(p)
|
|
}
|
|
|
|
func fromNonGoDir(dir string) (Info, error) {
|
|
i := Info{
|
|
Dir: dir,
|
|
Name: filepath.Base(dir),
|
|
}
|
|
|
|
b, err := run("go", "list", "-json", "-m")
|
|
if err != nil {
|
|
if nonGoDirRx.MatchString(err.Error()) {
|
|
return i, nil
|
|
}
|
|
return i, fmt.Errorf("here.nonGoDir: %s: %w", dir, err)
|
|
}
|
|
|
|
if err := json.Unmarshal(b, &i.Module); err != nil {
|
|
return i, fmt.Errorf("here.nonGoDir: %s: %w", dir, err)
|
|
}
|
|
|
|
if i.ImportPath == "" && i.Module.Path != "command-line-arguments" {
|
|
i.ImportPath = i.Module.Path
|
|
}
|
|
|
|
return i, nil
|
|
}
|