/
repos.go
110 lines (86 loc) · 2.66 KB
/
repos.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
package api
import (
"context"
"net/http"
"strings"
"github.com/Noah-Huppert/gh-gantt/server/config"
"github.com/Noah-Huppert/gh-gantt/server/libgh"
"github.com/Noah-Huppert/gh-gantt/server/req"
"github.com/Noah-Huppert/gh-gantt/server/resp"
"github.com/Noah-Huppert/golog"
"github.com/google/go-github/github"
)
// ReposHandler implements resp.ResponderHandler by returning a list of GitHub repository names for a user
type ReposHandler struct {
// ctx is application context
ctx context.Context
// logger prints debug information
logger golog.Logger
// cfg is application configuration
cfg config.Config
}
// NewReposHandler creates a new ReposHandler
func NewReposHandler(ctx context.Context, logger golog.Logger, cfg config.Config) ReposHandler {
return ReposHandler{
ctx: ctx,
logger: logger,
cfg: cfg,
}
}
// Handle implements resp.ResponderHandler.Handle
func (h ReposHandler) Handle(r *http.Request) resp.Responder {
// Check auth token
authToken, errResp := req.CheckAuthToken(h.logger, h.cfg, r)
if errResp != nil {
return errResp
}
client := libgh.NewUserClient(h.ctx, authToken.GitHubAuthToken)
// Group repositories for response
reposResp := map[string][]string{}
// Get organizations
repoOwners := []string{""}
orgs, _, err := client.Organizations.List(h.ctx, "", nil)
if err != nil {
return resp.NewStrErrorResponder(h.logger, http.StatusInternalServerError,
"error retrieving user's organizations from GitHub API", err.Error())
}
for _, org := range orgs {
repoOwners = append(repoOwners, *(org.Login))
}
// For each organization get repos
for _, repoOwner := range repoOwners {
nextPage := 1
// Get all repositories, requires a paginated API request to GitHub
for nextPage > 0 {
// Setup request
listOpts := &github.RepositoryListOptions{
ListOptions: github.ListOptions{
PerPage: 100,
Page: nextPage,
},
}
repos, apiResp, err := client.Repositories.List(h.ctx, repoOwner, listOpts)
if err != nil {
return resp.NewStrErrorResponder(h.logger, http.StatusInternalServerError,
"error retrieving repositories from GitHub API", err.Error())
}
// Setup next page of API request
nextPage = apiResp.NextPage
// Add repos to response
for _, repo := range repos {
// Check if owner exists in resp
nameParts := strings.Split(*(repo.FullName), "/")
owner := nameParts[0]
name := nameParts[1]
if _, ok := reposResp[owner]; !ok {
reposResp[owner] = []string{}
}
// Add to resp
reposResp[owner] = append(reposResp[owner], name)
}
}
}
return resp.NewJSONResponder(map[string]interface{}{
"repositories": reposResp,
}, http.StatusOK)
}