forked from constabulary/gb
/
package.go
176 lines (158 loc) · 4.61 KB
/
package.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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
package gb
import (
"fmt"
"go/build"
"path/filepath"
"strings"
)
// Package represents a resolved package from the Project with respect to the Context.
type Package struct {
*Context
*build.Package
Scope string // scope: build, test, etc
ExtraIncludes string // hook for test
Stale bool // is the package out of date wrt. its cached copy
Standard bool // is this package part of the standard library
}
// NewPackage creates a resolved Package.
func NewPackage(ctx *Context, p *build.Package) *Package {
pkg := Package{
Context: ctx,
Package: p,
}
// seed pkg.c so calling result never blocks
pkg.Stale = isStale(&pkg)
return &pkg
}
// isMain returns true if this is a command, a main package.
func (p *Package) isMain() bool {
return p.Name == "main" || strings.HasSuffix(p.ImportPath, "testmain") && p.Scope == "test"
}
// Imports returns the Pacakges that this Package depends on.
func (p *Package) Imports() []*Package {
pkgs := make([]*Package, 0, len(p.Package.Imports))
for _, i := range p.Package.Imports {
if p.shouldignore(i) {
continue
}
pkg, ok := p.pkgs[i]
if !ok {
panic("could not locate package: " + i)
}
pkgs = append(pkgs, pkg)
}
return pkgs
}
func (p *Package) String() string {
return fmt.Sprintf("%v", struct {
Name, ImportPath, Dir string
}{
p.Name, p.ImportPath, p.Dir,
})
}
// Complete indicates if this is a pure Go package
func (p *Package) Complete() bool {
// If we're giving the compiler the entire package (no C etc files), tell it that,
// so that it can give good error messages about forward declarations.
// Exceptions: a few standard packages have forward declarations for
// pieces supplied behind-the-scenes by package runtime.
extFiles := len(p.CgoFiles) + len(p.CFiles) + len(p.CXXFiles) + len(p.MFiles) + len(p.SFiles) + len(p.SysoFiles) + len(p.SwigFiles) + len(p.SwigCXXFiles)
if p.Standard {
switch p.ImportPath {
case "bytes", "net", "os", "runtime/pprof", "sync", "time":
extFiles++
}
}
return extFiles == 0
}
// Objdir returns the destination for object files compiled for this Package.
func (pkg *Package) Objdir() string {
switch pkg.Scope {
case "test":
ip := strings.TrimSuffix(filepath.FromSlash(pkg.ImportPath), "_test")
return filepath.Join(pkg.Workdir(), ip, "_test", filepath.Dir(filepath.FromSlash(pkg.ImportPath)))
default:
return filepath.Join(pkg.Workdir(), filepath.Dir(filepath.FromSlash(pkg.ImportPath)))
}
}
// Binfile returns the destination of the compiled target of this command.
func (pkg *Package) Binfile() string {
// TODO(dfc) should have a check for package main, or should be merged in to objfile.
var target string
switch pkg.Scope {
case "test":
target = filepath.Join(pkg.Workdir(), filepath.FromSlash(pkg.ImportPath), "_test", binname(pkg))
default:
target = filepath.Join(pkg.Bindir(), binname(pkg))
}
// if this is a cross compile, add -$GOOS-$GOARCH
if pkg.isCrossCompile() {
target += "-" + pkg.gotargetos + "-" + pkg.gotargetarch
}
if pkg.GOOS == "windows" {
target += ".exe"
}
return target
}
// loadPackage recursively resolves path and its imports and if successful
// stores those packages in the Context's internal package cache.
func loadPackage(c *Context, stack []string, path string) (*Package, error) {
if build.IsLocalImport(path) {
// sanity check
return nil, fmt.Errorf("%q is not a valid import path", path)
}
if pkg, ok := c.pkgs[path]; ok {
// already loaded, just return
return pkg, nil
}
push := func(path string) {
stack = append(stack, path)
}
pop := func(path string) {
stack = stack[:len(stack)-1]
}
onStack := func(path string) bool {
for _, p := range stack {
if p == path {
return true
}
}
return false
}
p, err := c.Context.Import(path, c.Projectdir(), 0)
if err != nil {
return nil, err
}
// everything depends on runtime, except the runtime itself.
// TODO(dfc) see if this can be made more selective by adding
// runtime as a dependency of some select packages.
standard := p.Goroot && p.ImportPath != "" && !strings.Contains(p.ImportPath, ".")
if standard && p.ImportPath != "runtime" {
p.Imports = append(p.Imports, "runtime")
}
push(path)
var stale bool
for _, i := range p.Imports {
if c.shouldignore(i) {
continue
}
if onStack(i) {
push(i)
return nil, fmt.Errorf("import cycle detected: %s", strings.Join(stack, " -> "))
}
pkg, err := loadPackage(c, stack, i)
if err != nil {
return nil, err
}
stale = stale || pkg.Stale
}
pop(path)
pkg := Package{
Context: c,
Package: p,
Standard: standard,
}
pkg.Stale = stale || isStale(&pkg)
c.pkgs[path] = &pkg
return &pkg, nil
}