forked from cespare/deplist
-
Notifications
You must be signed in to change notification settings - Fork 0
/
deplist.go
123 lines (108 loc) · 1.97 KB
/
deplist.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
package main
import (
"flag"
"fmt"
"go/build"
"log"
"os"
"sort"
)
func usage() {
fmt.Printf(`Usage:
%s [flags] [pkg]
where pkg is the name of a Go package (e.g., github.com/cespare/deplist).
If no package name is given, the current directory is used.
Flags:
`, os.Args[0])
flag.PrintDefaults()
}
type opts uint
const (
optTestImports opts = 1 << iota
optStd
)
type context struct {
soFar map[string]struct{}
ctx build.Context
}
func (c *context) find(name, dir string, o opts) error {
if name == "C" {
return nil
}
pkg, err := c.ctx.Import(name, dir, 0)
if err != nil {
return err
}
if pkg.Goroot && o&optStd == 0 {
return nil
}
if name != "." {
c.soFar[pkg.ImportPath] = struct{}{}
}
imports := pkg.Imports
if o&optTestImports != 0 {
imports = append(imports, pkg.TestImports...)
}
for _, imp := range imports {
if _, ok := c.soFar[imp]; !ok {
if err := c.find(imp, pkg.Dir, o); err != nil {
return err
}
}
}
return nil
}
func findDeps(name, dir, gopath string, o opts) ([]string, error) {
ctx := build.Default
if gopath != "" {
ctx.GOPATH = gopath
}
c := &context{
soFar: make(map[string]struct{}),
ctx: ctx,
}
if err := c.find(name, dir, o); err != nil {
return nil, err
}
var deps []string
for p := range c.soFar {
if p != name {
deps = append(deps, p)
}
}
sort.Strings(deps)
return deps, nil
}
func main() {
testImports := flag.Bool("t", false, "Include test dependencies")
std := flag.Bool("std", false, "Include standard library dependencies")
flag.Usage = usage
flag.Parse()
pkg := "."
switch flag.NArg() {
case 1:
pkg = flag.Arg(0)
case 0:
default:
usage()
os.Exit(1)
}
var o opts
if *testImports {
o |= optTestImports
}
if *std {
o |= optStd
}
cwd, err := os.Getwd()
if err != nil {
log.Fatalln("Couldn't determine working directory:", err)
}
deps, err := findDeps(pkg, cwd, "", o)
if err != nil {
log.Fatal(err)
}
for _, dep := range deps {
fmt.Println(dep)
}
}