package github

import (
	"context"
	"net/http"
	"os"

	ghb "github.com/google/go-github/github"
	"golang.org/x/oauth2"
)

type Client struct {
	apiKey string
}

func NewClient() *Client {
	client := Client{
		apiKey: os.Getenv("WTF_GITHUB_TOKEN"),
	}

	return &client
}

func (client *Client) PullRequests(owner string, repoName string) ([]*ghb.PullRequest, error) {
	oauthClient := client.oauthClient()
	github := ghb.NewClient(oauthClient)

	opts := &ghb.PullRequestListOptions{}

	prs, _, err := github.PullRequests.List(context.Background(), owner, repoName, opts)

	if err != nil {
		return nil, err
	}

	return prs, nil
}

func (client *Client) Repository(owner string, repoName string) (*ghb.Repository, error) {
	oauthClient := client.oauthClient()
	github := ghb.NewClient(oauthClient)

	repo, _, err := github.Repositories.Get(context.Background(), owner, repoName)

	if err != nil {
		return nil, err
	}

	return repo, nil
}

/* -------------------- Unexported Functions -------------------- */

func (client *Client) oauthClient() *http.Client {
	tokenService := oauth2.StaticTokenSource(
		&oauth2.Token{AccessToken: client.apiKey},
	)

	return oauth2.NewClient(context.Background(), tokenService)
}