client.go

package github

import (
	"context"
	"fmt"
	"net/http"
	"net/url"

	gogithub "github.com/google/go-github/github"
)

//go:generate counterfeiter . Client

type Client interface {
	CurrentUser(*http.Client) (string, error)
	Organizations(*http.Client) ([]string, error)
	Teams(*http.Client) (OrganizationTeams, error)
}

type client struct {
	baseURL string
}

func NewClient(baseURL string) Client {
	return &client{baseURL: baseURL}
}

type OrganizationTeams map[string][]string

func (c *client) CurrentUser(httpClient *http.Client) (string, error) {
	client, err := c.githubClient(httpClient)
	if err != nil {
		return "", err
	}

	currentUser, _, err := client.Users.Get(context.TODO(), "")
	if err != nil {
		return "", err
	}

	return *currentUser.Login, nil
}

func (c *client) Teams(httpClient *http.Client) (OrganizationTeams, error) {
	client, err := c.githubClient(httpClient)
	if err != nil {
		return nil, err
	}

	nextPage := 1
	organizationTeams := OrganizationTeams{}

	for nextPage != 0 {
		teams, resp, err := client.Organizations.ListUserTeams(context.TODO(), &gogithub.ListOptions{Page: nextPage})
		if err != nil {
			return nil, err
		}

		for _, team := range teams {
			organizationName := *team.Organization.Login

			if _, found := organizationTeams[organizationName]; !found {
				organizationTeams[organizationName] = []string{}
			}

			// We add both forms (slug and name) of team
			organizationTeams[organizationName] = append(organizationTeams[organizationName], *team.Name)
			organizationTeams[organizationName] = append(organizationTeams[organizationName], *team.Slug)
		}

		nextPage = resp.NextPage
	}

	return organizationTeams, nil
}

func (c *client) Organizations(httpClient *http.Client) ([]string, error) {
	client, err := c.githubClient(httpClient)
	if err != nil {
		return nil, err
	}

	nextPage := 1
	organizations := []string{}

	for nextPage != 0 {
		orgs, resp, err := client.Organizations.List(context.TODO(), "", &gogithub.ListOptions{Page: nextPage})

		if err != nil {
			return nil, err
		}

		for _, org := range orgs {
			organizations = append(organizations, *org.Login)
		}

		nextPage = resp.NextPage
	}

	return organizations, nil
}

func (c *client) githubClient(httpClient *http.Client) (*gogithub.Client, error) {
	client := gogithub.NewClient(httpClient)
	if c.baseURL != "" {
		var err error
		client.BaseURL, err = url.Parse(c.baseURL)
		if err != nil {
			return nil, fmt.Errorf("invalid github auth API URL '%s'", c.baseURL)
		}
	}

	return client, nil
}