-
Notifications
You must be signed in to change notification settings - Fork 4
/
check.go
149 lines (137 loc) · 3.67 KB
/
check.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
// Package pkg has the main brains of Effrit.
// @package_owner = @skarlso
package pkg
import (
"bufio"
"context"
"encoding/json"
"errors"
"fmt"
"io/ioutil"
"os"
"path/filepath"
"sort"
"strings"
"github.com/google/go-github/github"
"golang.org/x/oauth2"
)
const (
// EffritFileName sets the name of the file Effrit is using to save package data.
EffritFileName = ".effrit_package_data.json"
// CommentTag is the name of the tag.
CommentTag = "@package_owner"
// CommentFormat is the format used to parse a line for owner name.
CommentFormat = "// " + CommentTag + " = %s"
)
// Check will run an analysis of packages and detect if a new dependency has been added
// to package.
func Check(projectName string, parallel int, owner, repo string, prNumber int) error {
if _, err := os.Stat(EffritFileName); err != nil {
if os.IsNotExist(err) {
return errors.New(EffritFileName)
}
}
var packages = struct {
Packages []Package `json:"packages"`
}{
Packages: make([]Package, 0),
}
data, err := ioutil.ReadFile(EffritFileName)
if err != nil {
return err
}
if err := json.Unmarshal(data, &packages); err != nil {
return err
}
// Construct a map of package names and their imports.
packageMap := make(map[string][]string)
for _, p := range packages.Packages {
sort.Strings(p.DependedOnByNames)
packageMap[p.FullName] = p.DependedOnByNames
}
pkgs, err := Scan(projectName, parallel)
if err != nil {
return err
}
pkgs.Dump()
packages.Packages = make([]Package, 0)
data, _ = ioutil.ReadFile(EffritFileName)
_ = json.Unmarshal(data, &packages)
ownersToContact := make(map[string]string)
// Compare the new result with the old result's map data.
for _, p := range packages.Packages {
dependents := packageMap[p.FullName]
for _, dep := range dependents {
if !contains(p.DependedOnByNames, dep) {
owner, err := getOwnerForFile(p.Dir, p.GoFiles)
if err != nil {
return err
}
ownersToContact[owner] = dep
break
}
}
}
if len(ownersToContact) > 0 {
fmt.Print("Contacting owners about package dependency changes...")
if err := contactOwners(ownersToContact, owner, repo, prNumber); err != nil {
return err
}
fmt.Println("done.")
}
return nil
}
func getOwnerForFile(dir string, files []string) (string, error) {
// we check until we find an owner tag for this package.
for _, f := range files {
file, err := os.Open(filepath.Join(dir, f))
if err != nil {
return "", err
}
fs := bufio.NewScanner(file)
for fs.Scan() {
line := fs.Text()
if strings.Contains(line, CommentTag) {
var owner string
_, _ = fmt.Sscanf(line, CommentFormat, &owner)
_ = file.Close()
return owner, nil
}
}
_ = file.Close()
}
return "", nil
}
func contactOwners(notify map[string]string, owner, repo string, n int) error {
token := os.Getenv("EFFRIT_GITHUB_TOKEN")
if len(token) < 1 {
return errors.New("please define EFFRIT_GITHUB_TOKEN in order to allow effrit to create comments")
}
ctx := context.Background()
ts := oauth2.StaticTokenSource(
&oauth2.Token{AccessToken: token},
)
tc := oauth2.NewClient(ctx, ts)
client := github.NewClient(tc)
pr, _, _ := client.PullRequests.Get(ctx, owner, repo, n)
c := "# Dependency Changed\n"
for o, dep := range notify {
c += fmt.Sprintf("%s please review your package. **%s** has been added as a new dependency\n", o, dep)
}
comment := github.IssueComment{
AuthorAssociation: pr.AuthorAssociation,
Body: &c,
}
if _, _, err := client.Issues.CreateComment(ctx, owner, repo, n, &comment); err != nil {
return err
}
return nil
}
func contains(a []string, elem string) bool {
for _, e := range a {
if e == elem {
return true
}
}
return false
}