Merge key fetcher #6

Merged
zeridon merged 15 commits from master into master 2019-07-29 14:19:52 +03:00
3 changed files with 151 additions and 0 deletions
Showing only changes of commit 3516e25029 - Show all commits

View File

@ -0,0 +1,11 @@
Use the Github API to fetch configured SSH keys for a team.
Build by
```
go get
go build
```
Run
```
./fetch-keys --help
```

View File

@ -0,0 +1,3 @@
package main
const AuthToken = "c90e06359175cfccc4db21f60c44b83d93ccbd28"

137
access/fetch-keys/main.go Normal file
View File

@ -0,0 +1,137 @@
package main
import (
"bytes"
"context"
"fmt"
"github.com/akamensky/argparse"
"github.com/google/go-github/github" // with go modules disabled
"golang.org/x/oauth2"
"io/ioutil"
"os"
)
func main() {
parser := argparse.NewParser("sshkeys", "Fetch SSH keys for a Github team members")
ghTeam := parser.String("t", "team", &argparse.Options{Required: false, Help: "Github Team name", Default: "NOC"})
ghOrganization := parser.String("o", "org", &argparse.Options{Required: false, Help: "Github Organization name", Default: "OpenFest"})
authToken := parser.String("a", "authtoken", &argparse.Options{Required: false, Help: "Github Auth token", Default: AuthToken})
// Parse input
err := parser.Parse(os.Args)
if err != nil {
fmt.Print(parser.Usage(err))
os.Exit(1)
}
ctx := context.Background()
ts := oauth2.StaticTokenSource(
&oauth2.Token{AccessToken: *authToken},
)
tc := oauth2.NewClient(ctx, ts)
client := github.NewClient(tc)
var nocTeam *github.Team
var nocMembers []*string
for nextPage := 0; ; {
// list all teams for the specified org
opt := &github.ListOptions{nextPage, 50}
teams, rsp, err := client.Teams.ListTeams(ctx, *ghOrganization, opt)
if err != nil {
fmt.Errorf("client.ListTeams error: %v", err)
os.Exit(-1)
}
if rsp == nil {
fmt.Errorf("client.Repositories.List returned empty response: %v", err)
}
for _, team := range teams {
if *team.Name == *ghTeam {
nocTeam = team
break
}
}
if rsp.NextPage == 0 || nextPage == rsp.NextPage {
break
}
nextPage = rsp.NextPage
}
if nocTeam == nil {
fmt.Errorf("NOC team not found in OpenFest")
os.Exit(2)
}
for nextPage := 0; ; {
// list all members for the given organization's team
opt := &github.TeamListTeamMembersOptions{
Role: "all",
ListOptions: github.ListOptions{nextPage, 50},
}
users, rsp, err := client.Teams.ListTeamMembers(context.Background(), *nocTeam.ID, opt)
if err != nil {
fmt.Errorf("client.Teams.ListTeamMembers %v", err)
os.Exit(-1)
}
if rsp == nil {
fmt.Errorf("client.Teams.ListTeamMembers: %v", err)
}
for _, user := range users {
nocMembers = append(nocMembers, user.Login)
}
if rsp.NextPage == 0 || nextPage == rsp.NextPage {
break
}
nextPage = rsp.NextPage
}
for _, user := range nocMembers {
fmt.Println("Fetching keys for ", *user)
var sshKeys bytes.Buffer
for nextPage := 0; ; {
// list all teams an org for the current user
opt := &github.ListOptions{nextPage, 50}
keys, rsp, err := client.Users.ListKeys(ctx, *user, opt)
if err != nil {
fmt.Errorf("client.Users.ListKeys error: %v", err)
os.Exit(-1)
}
if rsp == nil {
fmt.Errorf("Users.ListKeyss returned empty response: %v", err)
}
for _, key := range keys {
fmt.Println(*key.Key)
sshKeys.WriteString(*key.Key)
sshKeys.WriteString("\n")
}
if rsp.NextPage == 0 || nextPage == rsp.NextPage {
break
}
nextPage = rsp.NextPage
}
err := ioutil.WriteFile(*user+".key", sshKeys.Bytes(), 0444)
if err != nil {
fmt.Errorf(*user+".key error %v", err)
}
}
os.Exit(0)
}