/
gh.go
120 lines (97 loc) · 2.56 KB
/
gh.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 gh
import (
"context"
"fmt"
"log"
"os"
"path/filepath"
"gopkg.in/yaml.v2"
"github.com/bgentry/go-netrc/netrc"
"github.com/google/go-github/v50/github"
"golang.org/x/oauth2"
)
var netrcFile = filepath.Join(os.Getenv("HOME"), ".netrc")
var netrcMachines = []string{"api.github.com", "github.com"}
var hubConfigFile = filepath.Join(os.Getenv("HOME"), ".config", "hub")
type githubConnectionInfo struct {
User string `yaml:"user"`
OauthToken string `yaml:"oauth_token"`
Protocol string `yaml:"protocol"`
}
type hubConfig struct {
GitHub []githubConnectionInfo `yaml:"github.com"`
}
type Client struct {
*netrc.Machine
*github.Client
Context context.Context
currentlyAuthedGitHubUser *github.User
}
func loginFromNetrc(rc *netrc.Netrc) (*netrc.Machine, error) {
for _, machineName := range netrcMachines {
machine := rc.FindMachine(machineName)
if machine != nil && !machine.IsDefault() {
return machine, nil
}
}
return nil, fmt.Errorf("no config for any of: %s", netrcMachines)
}
func loginFromNetrcFile() (*netrc.Machine, error) {
rc, err := netrc.ParseFile(netrcFile)
if err != nil {
return nil, err
}
return loginFromNetrc(rc)
}
func loginFromHubConfigFile() (*netrc.Machine, error) {
f, err := os.Open(hubConfigFile)
if err != nil {
return nil, err
}
cfg := &hubConfig{}
yaml.NewDecoder(f).Decode(cfg)
if len(cfg.GitHub) == 0 {
return nil, fmt.Errorf("no config present in: %s", hubConfigFile)
}
rc := &netrc.Netrc{}
for _, siteConf := range cfg.GitHub {
rc.NewMachine("github.com", siteConf.User, siteConf.OauthToken, "")
}
return loginFromNetrc(rc)
}
func getLogin() (*netrc.Machine, error) {
if machine, _ := loginFromNetrcFile(); machine != nil {
return machine, nil
}
if machine, _ := loginFromHubConfigFile(); machine != nil {
return machine, nil
}
return nil, fmt.Errorf("github login missing from %s and %s", netrcFile, hubConfigFile)
}
func NewDefaultClient() (*Client, error) {
machine, err := getLogin()
if err != nil {
return nil, err
}
ts := oauth2.StaticTokenSource(
&oauth2.Token{AccessToken: machine.Password},
)
tc := oauth2.NewClient(oauth2.NoContext, ts)
return &Client{
machine,
github.NewClient(tc),
context.Background(),
nil,
}, nil
}
func (c *Client) CurrentGitHubUser() *github.User {
if c.currentlyAuthedGitHubUser == nil {
currentlyAuthedUser, _, err := c.Users.Get(c.Context, "")
if err != nil {
log.Printf("couldn't fetch currently-auth'd user: %v", err)
return nil
}
c.currentlyAuthedGitHubUser = currentlyAuthedUser
}
return c.currentlyAuthedGitHubUser
}