X Tutup
Skip to content

Commit c2c211d

Browse files
author
Nate Smith
authored
Merge pull request cli#2952 from redreceipt/up-to-date
Adds Branch Up to Date Status
2 parents 4e5aa91 + 57140ad commit c2c211d

File tree

3 files changed

+55
-23
lines changed

3 files changed

+55
-23
lines changed

‎api/queries_pr.go‎

Lines changed: 36 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,11 @@ import (
1616
)
1717

1818
type PullRequestsPayload struct {
19-
ViewerCreated PullRequestAndTotalCount
20-
ReviewRequested PullRequestAndTotalCount
21-
CurrentPR *PullRequest
22-
DefaultBranch string
19+
ViewerCreated PullRequestAndTotalCount
20+
ReviewRequested PullRequestAndTotalCount
21+
CurrentPR *PullRequest
22+
DefaultBranch string
23+
StrictProtection bool
2324
}
2425

2526
type PullRequestAndTotalCount struct {
@@ -28,16 +29,17 @@ type PullRequestAndTotalCount struct {
2829
}
2930

3031
type PullRequest struct {
31-
ID string
32-
Number int
33-
Title string
34-
State string
35-
Closed bool
36-
URL string
37-
BaseRefName string
38-
HeadRefName string
39-
Body string
40-
Mergeable string
32+
ID string
33+
Number int
34+
Title string
35+
State string
36+
Closed bool
37+
URL string
38+
BaseRefName string
39+
HeadRefName string
40+
Body string
41+
Mergeable string
42+
MergeStateStatus string
4143

4244
Author struct {
4345
Login string
@@ -289,7 +291,10 @@ func PullRequests(client *Client, repo ghrepo.Interface, currentPRNumber int, cu
289291
type response struct {
290292
Repository struct {
291293
DefaultBranchRef struct {
292-
Name string
294+
Name string
295+
BranchProtectionRule struct {
296+
RequiresStrictStatusChecks bool
297+
}
293298
}
294299
PullRequests edges
295300
PullRequest *PullRequest
@@ -341,6 +346,7 @@ func PullRequests(client *Client, repo ghrepo.Interface, currentPRNumber int, cu
341346
state
342347
url
343348
headRefName
349+
mergeStateStatus
344350
headRepositoryOwner {
345351
login
346352
}
@@ -357,7 +363,12 @@ func PullRequests(client *Client, repo ghrepo.Interface, currentPRNumber int, cu
357363
queryPrefix := `
358364
query PullRequestStatus($owner: String!, $repo: String!, $headRefName: String!, $viewerQuery: String!, $reviewerQuery: String!, $per_page: Int = 10) {
359365
repository(owner: $owner, name: $repo) {
360-
defaultBranchRef { name }
366+
defaultBranchRef {
367+
name
368+
branchProtectionRule {
369+
requiresStrictStatusChecks
370+
}
371+
}
361372
pullRequests(headRefName: $headRefName, first: $per_page, orderBy: { field: CREATED_AT, direction: DESC }) {
362373
totalCount
363374
edges {
@@ -372,7 +383,12 @@ func PullRequests(client *Client, repo ghrepo.Interface, currentPRNumber int, cu
372383
queryPrefix = `
373384
query PullRequestStatus($owner: String!, $repo: String!, $number: Int!, $viewerQuery: String!, $reviewerQuery: String!, $per_page: Int = 10) {
374385
repository(owner: $owner, name: $repo) {
375-
defaultBranchRef { name }
386+
defaultBranchRef {
387+
name
388+
branchProtectionRule {
389+
requiresStrictStatusChecks
390+
}
391+
}
376392
pullRequest(number: $number) {
377393
...prWithReviews
378394
}
@@ -459,8 +475,9 @@ func PullRequests(client *Client, repo ghrepo.Interface, currentPRNumber int, cu
459475
PullRequests: reviewRequested,
460476
TotalCount: resp.ReviewRequested.TotalCount,
461477
},
462-
CurrentPR: currentPR,
463-
DefaultBranch: resp.Repository.DefaultBranchRef.Name,
478+
CurrentPR: currentPR,
479+
DefaultBranch: resp.Repository.DefaultBranchRef.Name,
480+
StrictProtection: resp.Repository.DefaultBranchRef.BranchProtectionRule.RequiresStrictStatusChecks,
464481
}
465482

466483
return &payload, nil

‎pkg/cmd/factory/http.go‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,8 @@ func NewHTTPClient(io *iostreams.IOStreams, cfg config.Config, appVersion string
8787
api.AddHeaderFunc("Accept", func(req *http.Request) (string, error) {
8888
// antiope-preview: Checks
8989
accept := "application/vnd.github.antiope-preview+json"
90+
// introduced for #2952: pr branch up to date status
91+
accept += ", application/vnd.github.merge-info-preview+json"
9092
if ghinstance.IsEnterprise(req.URL.Hostname()) {
9193
// shadow-cat-preview: Draft pull requests
9294
accept += ", application/vnd.github.shadow-cat-preview"

‎pkg/cmd/pr/status/status.go‎

Lines changed: 17 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ func statusRun(opts *StatusOptions) error {
114114
currentPR = nil
115115
}
116116
if currentPR != nil {
117-
printPrs(opts.IO, 1, *currentPR)
117+
printPrs(opts.IO, 1, prPayload.StrictProtection, *currentPR)
118118
} else if currentPRHeadRef == "" {
119119
shared.PrintMessage(opts.IO, " There is no current branch")
120120
} else {
@@ -124,15 +124,15 @@ func statusRun(opts *StatusOptions) error {
124124

125125
shared.PrintHeader(opts.IO, "Created by you")
126126
if prPayload.ViewerCreated.TotalCount > 0 {
127-
printPrs(opts.IO, prPayload.ViewerCreated.TotalCount, prPayload.ViewerCreated.PullRequests...)
127+
printPrs(opts.IO, prPayload.ViewerCreated.TotalCount, prPayload.StrictProtection, prPayload.ViewerCreated.PullRequests...)
128128
} else {
129129
shared.PrintMessage(opts.IO, " You have no open pull requests")
130130
}
131131
fmt.Fprintln(out)
132132

133133
shared.PrintHeader(opts.IO, "Requesting a code review from you")
134134
if prPayload.ReviewRequested.TotalCount > 0 {
135-
printPrs(opts.IO, prPayload.ReviewRequested.TotalCount, prPayload.ReviewRequested.PullRequests...)
135+
printPrs(opts.IO, prPayload.ReviewRequested.TotalCount, prPayload.StrictProtection, prPayload.ReviewRequested.PullRequests...)
136136
} else {
137137
shared.PrintMessage(opts.IO, " You have no pull requests to review")
138138
}
@@ -178,7 +178,7 @@ func prSelectorForCurrentBranch(baseRepo ghrepo.Interface, prHeadRef string, rem
178178
return
179179
}
180180

181-
func printPrs(io *iostreams.IOStreams, totalCount int, prs ...api.PullRequest) {
181+
func printPrs(io *iostreams.IOStreams, totalCount int, strictProtection bool, prs ...api.PullRequest) {
182182
w := io.Out
183183
cs := io.ColorScheme()
184184

@@ -227,6 +227,19 @@ func printPrs(io *iostreams.IOStreams, totalCount int, prs ...api.PullRequest) {
227227
} else if reviews.Approved {
228228
fmt.Fprint(w, cs.Green("✓ Approved"))
229229
}
230+
231+
// only check if the "up to date" setting is checked in repo settings
232+
if strictProtection {
233+
// add padding between reviews & merge status
234+
fmt.Fprint(w, " ")
235+
236+
if pr.MergeStateStatus == "BEHIND" {
237+
fmt.Fprint(w, cs.Yellow("- Not up to date"))
238+
} else {
239+
fmt.Fprint(w, cs.Green("✓ Up to date"))
240+
}
241+
}
242+
230243
} else {
231244
fmt.Fprintf(w, " - %s", shared.StateTitleWithColor(cs, pr))
232245
}

0 commit comments

Comments
 (0)
X Tutup