cli · vilmibm · May 8, 2020 · Apr 15, 2020 · Apr 17, 2020 · Apr 17, 2020
diff --git a/api/queries_org.go b/api/queries_org.go
@@ -1,6 +1,11 @@
 package api
 
-import "fmt"
+import (
+	"context"
+	"fmt"
+
+	"github.com/shurcooL/githubv4"
+)
 
 // using API v3 here because the equivalent in GraphQL needs `read:org` scope
 func resolveOrganization(client *Client, orgName string) (string, error) {
@@ -22,3 +27,84 @@ func resolveOrganizationTeam(client *Client, orgName, teamSlug string) (string,
 	err := client.REST("GET", fmt.Sprintf("orgs/%s/teams/%s", orgName, teamSlug), nil, &response)
 	return response.Organization.NodeID, response.NodeID, err
 }
+
+// OrganizationProjects fetches all open projects for an organization
+func OrganizationProjects(client *Client, owner string) ([]RepoProject, error) {
+	var query struct {
+		Organization struct {
+			Projects struct {
+				Nodes    []RepoProject
+				PageInfo struct {
+					HasNextPage bool
+					EndCursor   string
+				}
+			} `graphql:"projects(states: [OPEN], first: 100, orderBy: {field: NAME, direction: ASC}, after: $endCursor)"`
+		} `graphql:"organization(login: $owner)"`
+	}
+
+	variables := map[string]interface{}{
+		"owner":     githubv4.String(owner),
+		"endCursor": (*githubv4.String)(nil),
+	}
+
+	v4 := githubv4.NewClient(client.http)
+
+	var projects []RepoProject
+	for {
+		err := v4.Query(context.Background(), &query, variables)
+		if err != nil {
+			return nil, err
+		}
+
+		projects = append(projects, query.Organization.Projects.Nodes...)
+		if !query.Organization.Projects.PageInfo.HasNextPage {
+			break
+		}
+		variables["endCursor"] = githubv4.String(query.Organization.Projects.PageInfo.EndCursor)
+	}
+
+	return projects, nil
+}
+
+type OrgTeam struct {
+	ID   string
+	Slug string
+}
+
+// OrganizationTeams fetches all the teams in an organization
+func OrganizationTeams(client *Client, owner string) ([]OrgTeam, error) {
+	var query struct {
+		Organization struct {
+			Teams struct {
+				Nodes    []OrgTeam
+				PageInfo struct {
+					HasNextPage bool
+					EndCursor   string
+				}
+			} `graphql:"teams(first: 100, orderBy: {field: NAME, direction: ASC}, after: $endCursor)"`
+		} `graphql:"organization(login: $owner)"`
+	}
+
+	variables := map[string]interface{}{
+		"owner":     githubv4.String(owner),
+		"endCursor": (*githubv4.String)(nil),
+	}
+
+	v4 := githubv4.NewClient(client.http)
+
+	var teams []OrgTeam
+	for {
+		err := v4.Query(context.Background(), &query, variables)
+		if err != nil {
+			return nil, err
+		}
+
+		teams = append(teams, query.Organization.Teams.Nodes...)
+		if !query.Organization.Teams.PageInfo.HasNextPage {
+			break
+		}
+		variables["endCursor"] = githubv4.String(query.Organization.Teams.PageInfo.EndCursor)
+	}
+
+	return teams, nil
+}
diff --git a/api/queries_pr.go b/api/queries_pr.go
@@ -579,6 +579,7 @@ func CreatePullRequest(client *Client, repo *Repository, params map[string]inter
 		mutation CreatePullRequest($input: CreatePullRequestInput!) {
 			createPullRequest(input: $input) {
 				pullRequest {
+					id
 					url
 				}
 			}
@@ -588,7 +589,10 @@ func CreatePullRequest(client *Client, repo *Repository, params map[string]inter
 		"repositoryId": repo.ID,
 	}
 	for key, val := range params {
-		inputParams[key] = val
+		switch key {
+		case "title", "body", "draft", "baseRefName", "headRefName":
+			inputParams[key] = val
+		}
 	}
 	variables := map[string]interface{}{
 		"input": inputParams,
@@ -604,8 +608,70 @@ func CreatePullRequest(client *Client, repo *Repository, params map[string]inter
 	if err != nil {
 		return nil, err
 	}
+	pr := &result.CreatePullRequest.PullRequest
+
+	// metadata parameters aren't currently available in `createPullRequest`,
+	// but they are in `updatePullRequest`
+	updateParams := make(map[string]interface{})
+	for key, val := range params {
+		switch key {
+		case "assigneeIds", "labelIds", "projectIds", "milestoneId":
+			if !isBlank(val) {
+				updateParams[key] = val
+			}
+		}
+	}
+	if len(updateParams) > 0 {
+		updateQuery := `
+		mutation UpdatePullRequest($input: UpdatePullRequestInput!) {
+			updatePullRequest(input: $input) { clientMutationId }
+		}`
+		updateParams["pullRequestId"] = pr.ID
+		variables := map[string]interface{}{
+			"input": updateParams,
+		}
+		err := client.GraphQL(updateQuery, variables, &result)
+		if err != nil {
+			return nil, err
+		}
+	}
 
-	return &result.CreatePullRequest.PullRequest, nil
+	// reviewers are requested in yet another additional mutation
+	reviewParams := make(map[string]interface{})
+	if ids, ok := params["userReviewerIds"]; ok && !isBlank(ids) {
+		reviewParams["userIds"] = ids
+	}
+	if ids, ok := params["teamReviewerIds"]; ok && !isBlank(ids) {
+		reviewParams["teamIds"] = ids
+	}
+
+	if len(reviewParams) > 0 {
+		reviewQuery := `
+		mutation RequestReviews($input: RequestReviewsInput!) {
+			requestReviews(input: $input) { clientMutationId }
+		}`
+		reviewParams["pullRequestId"] = pr.ID
+		variables := map[string]interface{}{
+			"input": reviewParams,
+		}
+		err := client.GraphQL(reviewQuery, variables, &result)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	return pr, nil
+}
+
+func isBlank(v interface{}) bool {
+	switch vv := v.(type) {
+	case string:
+		return vv == ""
+	case []string:
+		return len(vv) == 0
+	default:
+		return true
+	}
 }
 
 func AddReview(client *Client, pr *PullRequest, input *PullRequestReviewInput) error {