forked from drone/go-scm
-
Notifications
You must be signed in to change notification settings - Fork 86
/
main.go
72 lines (63 loc) · 1.41 KB
/
main.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
package main
import (
"context"
"fmt"
"os"
"strconv"
"github.com/ghodss/yaml"
"github.com/jenkins-x/go-scm/scm"
"github.com/pkg/errors"
"github.com/jenkins-x/go-scm/scm/factory"
"github.com/jenkins-x/go-scm/scm/factory/examples/helpers"
)
func main() {
args := os.Args
if len(args) < 2 {
fmt.Printf("usage: org/repo prNumber")
return
}
repo := args[1]
client, err := factory.NewClientFromEnvironment()
if err != nil {
helpers.Fail(err)
return
}
ctx := context.Background()
if len(args) < 3 {
fmt.Printf("Getting PRs\n")
prs, _, err := client.PullRequests.List(ctx, repo, scm.PullRequestListOptions{})
if err != nil {
helpers.Fail(err)
return
}
for _, pr := range prs {
fmt.Printf("Found PullRequest:\n")
data, err := yaml.Marshal(pr)
if err != nil {
helpers.Fail(errors.Wrap(err, "failed to marshal PR as YAML"))
return
}
fmt.Printf("%s:\n", string(data))
}
return
}
prText := args[2]
number, err := strconv.Atoi(prText)
if err != nil {
helpers.Fail(errors.Wrapf(err, "failed to parse PR number: %s", prText))
return
}
fmt.Printf("Getting PR\n")
pr, _, err := client.PullRequests.Find(ctx, repo, number)
if err != nil {
helpers.Fail(err)
return
}
fmt.Printf("Found PullRequest:\n")
data, err := yaml.Marshal(pr)
if err != nil {
helpers.Fail(errors.Wrap(err, "failed to marshal PR as YAML"))
return
}
fmt.Printf("%s:\n", string(data))
}