111 lines
2.2 KiB
Go
111 lines
2.2 KiB
Go
// showimp is a utility for displaying the imports in a package.
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"go/parser"
|
|
"go/token"
|
|
"os"
|
|
"path/filepath"
|
|
"regexp"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/kisom/goutils/die"
|
|
"github.com/kisom/goutils/logging"
|
|
)
|
|
|
|
var (
|
|
gopath string
|
|
project string
|
|
)
|
|
|
|
var (
|
|
stdLibRegexp = regexp.MustCompile(`^\w+(/\w+)*$`)
|
|
sourceRegexp = regexp.MustCompile(`^[^.].*\.go$`)
|
|
log = logging.Init()
|
|
)
|
|
|
|
func init() {
|
|
gopath = os.Getenv("GOPATH")
|
|
if gopath == "" {
|
|
fmt.Fprintf(os.Stderr, "GOPATH isn't set, can't proceed.")
|
|
os.Exit(1)
|
|
}
|
|
gopath += "/src/"
|
|
|
|
wd, err := os.Getwd()
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "Unable to establish working directory: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
if !strings.HasPrefix(wd, gopath) {
|
|
fmt.Fprintf(os.Stderr, "Can't determine my location in the GOPATH.\n")
|
|
fmt.Fprintf(os.Stderr, "Working directory is %s\n", wd)
|
|
fmt.Fprintf(os.Stderr, "Go source path is %s\n", gopath)
|
|
os.Exit(1)
|
|
}
|
|
|
|
project = wd[len(gopath):]
|
|
}
|
|
|
|
var (
|
|
imports = map[string]bool{}
|
|
fset = &token.FileSet{}
|
|
)
|
|
|
|
func walkFile(path string, info os.FileInfo, err error) error {
|
|
if !sourceRegexp.MatchString(path) {
|
|
return nil
|
|
}
|
|
|
|
log.Debug(path)
|
|
|
|
f, err := parser.ParseFile(fset, path, nil, parser.ImportsOnly)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, importSpec := range f.Imports {
|
|
importPath := strings.Trim(importSpec.Path.Value, `"`)
|
|
if stdLibRegexp.MatchString(importPath) {
|
|
log.Debug("standard lib:", importPath)
|
|
continue
|
|
} else if strings.HasPrefix(importPath, project) {
|
|
log.Debug("internal import:", importPath)
|
|
continue
|
|
} else if strings.HasPrefix(importPath, "golang.org/") {
|
|
log.Debug("extended lib:", importPath)
|
|
continue
|
|
}
|
|
log.Debug("import:", importPath)
|
|
imports[importPath] = true
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func main() {
|
|
verbose := flag.Bool("v", false, "log debugging information")
|
|
flag.Parse()
|
|
|
|
if *verbose {
|
|
log.SetLevel(logging.LevelDebug)
|
|
}
|
|
err := filepath.Walk(".", walkFile)
|
|
die.If(err)
|
|
|
|
fmt.Println("External imports:")
|
|
importList := make([]string, 0, len(imports))
|
|
for imp := range imports {
|
|
importList = append(importList, imp)
|
|
}
|
|
sort.Strings(importList)
|
|
|
|
for _, imp := range importList {
|
|
fmt.Println("\t", imp)
|
|
}
|
|
}
|