/
module.go
120 lines (100 loc) · 2.45 KB
/
module.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
package mod
import (
"fmt"
"os"
"path/filepath"
"sync"
"github.com/spf13/afero"
)
type Module struct {
// The name of module joined by forward slash(/). e.g. "github.com/anz-bank/foo"
Name string
// The absolute path to the module.
// e.g. "/Users/username/go/pkg/mod/github.com/anz-bank/foo@v1.1.0" on Linux and macOS
// "C:\Users\username\go\pkg\mod\github.com\anz-bank\foo@v1.1.0" on Windows
Dir string
// The version of the module. e.g. "v1.1.0"
Version string
}
type Modules []*Module
var modules Modules
var manager DependencyManager = &goModules{}
var mode Mode = Mode{sync.RWMutex{}, GoModulesMode}
type Mode struct {
sync.RWMutex
modeType ModeType
}
type ModeType string
const (
GitHubMode ModeType = "github"
GoModulesMode ModeType = "go modules"
)
const MasterBranch = "master"
type DependencyManager interface {
// Download external dependency to local directory
Get(filename, ver string, m *Modules) (*Module, error)
// Find dependency in m *Modules
Find(filename, ver string, m *Modules) *Module
// Load local cache into m *Modules
Load(m *Modules) error
}
func (m *Modules) Add(v *Module) {
*m = append(*m, v)
}
func (m *Modules) Len() int {
return len(*m)
}
func Config(m ModeType, goModopt GoModulesOptions, githubOpt GitHubOptions) error {
mode.Lock()
defer mode.Unlock()
mode.modeType = m
switch m {
case GitHubMode:
gh, err := newGitHubMgr(githubOpt)
if err != nil {
return err
}
manager = gh
case GoModulesMode:
gm := &goModules{}
if err := gm.Init(goModopt); err != nil {
return err
}
manager = gm
default:
return fmt.Errorf("unknown mode type %s", m)
}
return nil
}
func Retrieve(name string, ver string) (*Module, error) {
if modules.Len() == 0 {
if err := manager.Load(&modules); err != nil {
return nil, fmt.Errorf("error loading modules: %s", err.Error())
}
}
if ver != MasterBranch || (mode.modeType == GitHubMode && ver != "") {
mod := manager.Find(name, ver, &modules)
if mod != nil {
return mod, nil
}
}
return manager.Get(name, ver, &modules)
}
func hasPathPrefix(prefix, s string) bool {
prefix = filepath.Clean(prefix)
s = filepath.Clean(s)
if len(s) > len(prefix) {
return s[len(prefix)] == filepath.Separator && s[:len(prefix)] == prefix
}
return s == prefix
}
func FileExists(fs afero.Fs, filename string, isDir bool) bool {
info, err := fs.Stat(filename)
if os.IsNotExist(err) {
return false
}
if isDir {
return info.IsDir()
}
return !info.IsDir()
}