137 lines
2.8 KiB
Go
137 lines
2.8 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/github/gh-cli/context"
|
|
)
|
|
|
|
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
|
|
}
|
|
}
|
|
}
|
|
`
|
|
|
|
ghRepo, err := context.Current().BaseRepo()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
currentBranch, err := context.Current().Branch()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
currentUsername, err := context.Current().AuthLogin()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
owner := ghRepo.Owner
|
|
repo := ghRepo.Name
|
|
|
|
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 nil, 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
|
|
}
|