-
Notifications
You must be signed in to change notification settings - Fork 567
/
util.go
126 lines (115 loc) · 2.99 KB
/
util.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
package pps
import (
"fmt"
"sort"
"strings"
"github.com/pachyderm/pachyderm/src/client/pfs"
"gopkg.in/src-d/go-git.v4"
)
// VisitInput visits each input recursively in ascending order (root last)
func VisitInput(input *Input, f func(*Input)) {
switch {
case input.Cross != nil:
for _, input := range input.Cross {
VisitInput(input, f)
}
case input.Union != nil:
for _, input := range input.Union {
VisitInput(input, f)
}
}
f(input)
}
// InputName computes the name of an Input.
func InputName(input *Input) string {
switch {
case input.Atom != nil:
return input.Atom.Name
case input.Pfs != nil:
return input.Pfs.Name
case input.Cross != nil:
if len(input.Cross) > 0 {
return InputName(input.Cross[0])
}
case input.Union != nil:
if len(input.Union) > 0 {
return InputName(input.Union[0])
}
}
return ""
}
// SortInput sorts an Input.
func SortInput(input *Input) {
VisitInput(input, func(input *Input) {
SortInputs := func(inputs []*Input) {
sort.SliceStable(inputs, func(i, j int) bool { return InputName(inputs[i]) < InputName(inputs[j]) })
}
switch {
case input.Cross != nil:
SortInputs(input.Cross)
case input.Union != nil:
SortInputs(input.Union)
}
})
}
// InputBranches returns the branches in an Input.
func InputBranches(input *Input) []*pfs.Branch {
var result []*pfs.Branch
VisitInput(input, func(input *Input) {
if input.Atom != nil {
result = append(result, &pfs.Branch{
Repo: &pfs.Repo{Name: input.Atom.Repo},
Name: input.Atom.Branch,
})
}
if input.Pfs != nil {
result = append(result, &pfs.Branch{
Repo: &pfs.Repo{Name: input.Pfs.Repo},
Name: input.Pfs.Branch,
})
}
if input.Cron != nil {
result = append(result, &pfs.Branch{
Repo: &pfs.Repo{Name: input.Cron.Repo},
Name: "master",
})
}
if input.Git != nil {
result = append(result, &pfs.Branch{
Repo: &pfs.Repo{Name: input.Git.Name},
Name: input.Atom.Branch,
})
}
})
return result
}
// ValidateGitCloneURL returns an error if the provided URL is invalid
func ValidateGitCloneURL(url string) error {
exampleURL := "https://github.com/org/foo.git"
if url == "" {
return fmt.Errorf("clone URL is missing (example clone URL %v)", exampleURL)
}
// Use the git client's validator to make sure its a valid URL
o := &git.CloneOptions{
URL: url,
}
if err := o.Validate(); err != nil {
return err
}
// Make sure its the type that we want. Of the following we
// only accept the 'clone' type of url:
// git_url: "git://github.com/sjezewski/testgithook.git",
// ssh_url: "git@github.com:sjezewski/testgithook.git",
// clone_url: "https://github.com/sjezewski/testgithook.git",
// svn_url: "https://github.com/sjezewski/testgithook",
invalidErr := fmt.Errorf("clone URL is missing .git suffix (example clone URL %v)", exampleURL)
if !strings.HasSuffix(url, ".git") {
// svn_url case
return invalidErr
}
if !strings.HasPrefix(url, "https://") {
// git_url or ssh_url cases
return invalidErr
}
return nil
}