Rename package to api

api/client.go

@@ -0,0 +1,147 @@
+package api
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"os"
+	"os/user"
+	"regexp"
+)
+
+type graphQLResponse struct {
+	Data   interface{}
+	Errors []struct {
+		Message string
+	}
+}
+
+/*
+graphQL usage
+
+type repoResponse struct {
+	repository struct {
+		createdAt string
+	}
+}
+
+query := `query {
+	repository(owner: "golang", name: "go") {
+		createdAt
+	}
+}`
+
+variables := map[string]string{}
+
+var resp repoResponse
+err := graphql(query, map[string]string{}, &resp)
+if err != nil {
+	panic(err)
+}
+
+fmt.Printf("%+v\n", resp)
+*/
+func graphQL(query string, variables map[string]string, v interface{}) error {
+	url := "https://api.github.com/graphql"
+	reqBody, err := json.Marshal(map[string]interface{}{"query": query, "variables": variables})
+	if err != nil {
+		panic(err)
+	}
+
+	req, err := http.NewRequest("POST", url, bytes.NewBuffer(reqBody))
+	if err != nil {
+		panic(err)
+	}
+	debugRequest(req, string(reqBody))
+
+	req.Header.Set("Authorization", "token "+getToken())
+	req.Header.Set("Content-Type", "application/json; charset=utf-8")
+
+	client := &http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		panic(err)
+	}
+	defer resp.Body.Close()
+
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return err
+	}
+	debugResponse(resp, string(body))
+	return handleResponse(resp, body, v)
+}
+
+func handleResponse(resp *http.Response, body []byte, v interface{}) error {
+	success := resp.StatusCode >= 200 && resp.StatusCode < 300
+
+	if success {
+		gr := &graphQLResponse{Data: v}
+		err := json.Unmarshal(body, &gr)
+		if err != nil {
+			return err
+		}
+		if len(gr.Errors) > 0 {
+			errorMessages := gr.Errors[0].Message
+			for _, e := range gr.Errors[1:] {
+				errorMessages += ", " + e.Message
+			}
+			return fmt.Errorf("graphql error: '%s'", errorMessages)
+		}
+		return nil
+	}
+
+	return handleHTTPError(resp, body)
+}
+
+func handleHTTPError(resp *http.Response, body []byte) error {
+	var message string
+	var parsedBody struct {
+		Message string `json:"message"`
+	}
+	err := json.Unmarshal(body, &parsedBody)
+	if err != nil {
+		message = string(body)
+	} else {
+		message = parsedBody.Message
+	}
+
+	return fmt.Errorf("http error, '%s' failed (%d): '%s'", resp.Request.URL, resp.StatusCode, message)
+}
+
+// TODO: THIS IS NO GOOD. I need to figure out if the GraphQL function has direct
+// access to the token, or if we should pass the token into the GraphQL function. For
+// now I'm asuming that this has direct access so I'm simulating that with this function.
+func getToken() string {
+	usr, err := user.Current()
+	if err != nil {
+		panic(err)
+	}
+
+	content, err := ioutil.ReadFile(usr.HomeDir + "/.config/hub")
+	if err != nil {
+		panic(err)
+	}
+
+	r := regexp.MustCompile(`oauth_token: (\w+)`)
+	token := r.FindStringSubmatch(string(content))
+	return token[1]
+}
+
+func debugRequest(req *http.Request, body string) {
+	if _, ok := os.LookupEnv("DEBUG"); !ok {
+		return
+	}
+
+	fmt.Printf("DEBUG: GraphQL request to %s:\n %s\n\n", req.URL, body)
+}
+
+func debugResponse(resp *http.Response, body string) {
+	if _, ok := os.LookupEnv("DEBUG"); !ok {
+		return
+	}
+
+	fmt.Printf("DEBUG: GraphQL response:\n%+v\n\n%s\n\n", resp, body)
+}

api/queries.go

@@ -0,0 +1,162 @@
+package api
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/github/gh-cli/git"
+	"github.com/github/gh-cli/github"
+)
+
+type PullRequestsPayload struct {
+	ViewerCreated   []PullRequest
+	ReviewRequested []PullRequest
+	CurrentPR       *PullRequest
+}
+
+type PullRequest struct {
+	Number      int
+	Title       string
+	URL         string
+	HeadRefName string
+}
+
+func PullRequests() (PullRequestsPayload, error) {
+	type edges struct {
+		Edges []struct {
+			Node PullRequest
+		}
+		PageInfo struct {
+			HasNextPage bool
+			EndCursor   string
+		}
+	}
+
+	type response struct {
+		Repository struct {
+			PullRequests edges
+		}
+		ViewerCreated   edges
+		ReviewRequested edges
+	}
+
+	query := `
+		fragment pr on PullRequest {
+			number
+			title
+			url
+			headRefName
+		}
+
+		query($owner: String!, $repo: String!, $headRefName: String!, $viewerQuery: String!, $reviewerQuery: String!, $per_page: Int = 10) {
+			repository(owner: $owner, name: $repo) {
+				pullRequests(headRefName: $headRefName, first: 1) {
+					edges {
+						node {
+							...pr
+						}
+					}
+				}
+			}
+			viewerCreated: search(query: $viewerQuery, type: ISSUE, first: $per_page) {
+				edges {
+					node {
+						...pr
+					}
+				}
+				pageInfo {
+					hasNextPage
+				}
+			}
+			reviewRequested: search(query: $reviewerQuery, type: ISSUE, first: $per_page) {
+				edges {
+					node {
+						...pr
+					}
+				}
+				pageInfo {
+					hasNextPage
+				}
+			}
+		}
+	`
+
+	project := project()
+	owner := project.Owner
+	repo := project.Name
+	currentBranch := currentBranch()
+
+	viewerQuery := fmt.Sprintf("repo:%s/%s state:open is:pr author:%s", owner, repo, currentUsername())
+	reviewerQuery := fmt.Sprintf("repo:%s/%s state:open review-requested:%s", owner, repo, currentUsername())
+
+	variables := map[string]string{
+		"viewerQuery":   viewerQuery,
+		"reviewerQuery": reviewerQuery,
+		"owner":         owner,
+		"repo":          repo,
+		"headRefName":   currentBranch,
+	}
+
+	var resp response
+	err := graphQL(query, variables, &resp)
+	if err != nil {
+		return PullRequestsPayload{}, err
+	}
+
+	var viewerCreated []PullRequest
+	for _, edge := range resp.ViewerCreated.Edges {
+		viewerCreated = append(viewerCreated, edge.Node)
+	}
+
+	var reviewRequested []PullRequest
+	for _, edge := range resp.ReviewRequested.Edges {
+		reviewRequested = append(reviewRequested, edge.Node)
+	}
+
+	var currentPR *PullRequest
+	for _, edge := range resp.Repository.PullRequests.Edges {
+		currentPR = &edge.Node
+	}
+
+	payload := PullRequestsPayload{
+		viewerCreated,
+		reviewRequested,
+		currentPR,
+	}
+
+	return payload, nil
+}
+
+// These will be replaced by nate's context stuff
+// 💥💥💥💥💥💥💥💥💥💥💥💥💥💥💥💥💥💥💥💥💥
+func project() github.Project {
+	remotes, error := github.Remotes()
+	if error != nil {
+		panic(error)
+	}
+
+	for _, remote := range remotes {
+		if project, error := remote.Project(); error == nil {
+			return *project
+		}
+	}
+
+	panic("Could not get the project. What is a project? I don't know, it's kind of like a git repository I think?")
+}
+
+func currentBranch() string {
+	currentBranch, err := git.Head()
+	if err != nil {
+		panic(err)
+	}
+
+	return strings.Replace(currentBranch, "refs/heads/", "", 1)
+}
+
+func currentUsername() string {
+	host, err := github.CurrentConfig().DefaultHost()
+	if err != nil {
+		panic(err)
+	}
+	return host.User
+}