diff --git a/access/fetch-keys/README.md b/access/fetch-keys/README.md new file mode 100644 index 0000000..ae83209 --- /dev/null +++ b/access/fetch-keys/README.md @@ -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 +``` diff --git a/access/fetch-keys/authtoken.go b/access/fetch-keys/authtoken.go new file mode 100644 index 0000000..ad85a29 --- /dev/null +++ b/access/fetch-keys/authtoken.go @@ -0,0 +1,3 @@ +package main + +const AuthToken = "c90e06359175cfccc4db21f60c44b83d93ccbd28" diff --git a/access/fetch-keys/main.go b/access/fetch-keys/main.go new file mode 100644 index 0000000..e14d1c0 --- /dev/null +++ b/access/fetch-keys/main.go @@ -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) +}