136 lines
2.7 KiB
Go
136 lines
2.7 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
|
|
debug bool
|
|
)
|
|
|
|
var (
|
|
stdLibRegexp = regexp.MustCompile(`^\w+(/\w+)*$`)
|
|
sourceRegexp = regexp.MustCompile(`^[^.].*\.go$`)
|
|
log = logging.NewConsole()
|
|
imports = map[string]bool{}
|
|
fset = &token.FileSet{}
|
|
)
|
|
|
|
func debugf(format string, args ...interface{}) {
|
|
if debug {
|
|
fmt.Printf(format, args...)
|
|
}
|
|
}
|
|
|
|
func debugln(args ...interface{}) {
|
|
if debug {
|
|
fmt.Println(args...)
|
|
}
|
|
}
|
|
|
|
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):]
|
|
}
|
|
|
|
func walkFile(path string, info os.FileInfo, err error) error {
|
|
if ignores[path] {
|
|
return filepath.SkipDir
|
|
}
|
|
|
|
if !sourceRegexp.MatchString(path) {
|
|
return nil
|
|
}
|
|
|
|
debugln(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) {
|
|
debugln("standard lib:", importPath)
|
|
continue
|
|
} else if strings.HasPrefix(importPath, project) {
|
|
debugln("internal import:", importPath)
|
|
continue
|
|
} else if strings.HasPrefix(importPath, "golang.org/") {
|
|
debugln("extended lib:", importPath)
|
|
continue
|
|
}
|
|
debugln("import:", importPath)
|
|
imports[importPath] = true
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
var ignores = map[string]bool{}
|
|
|
|
func main() {
|
|
var ignoreLine string
|
|
var noVendor bool
|
|
flag.StringVar(&ignoreLine, "i", "", "comma-separated list of directories to ignore")
|
|
flag.BoolVar(&noVendor, "nv", false, "ignore the vendor directory")
|
|
flag.BoolVar(&debug, "v", false, "log debugging information")
|
|
flag.Parse()
|
|
|
|
if noVendor {
|
|
ignores["vendor"] = true
|
|
}
|
|
|
|
for _, word := range strings.Split(ignoreLine, ",") {
|
|
ignores[strings.TrimSpace(word)] = true
|
|
}
|
|
|
|
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)
|
|
}
|
|
}
|