577 lines
16 KiB
Go
577 lines
16 KiB
Go
package pr
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/gitlink-org/gitlink-cli/internal/i18n"
|
|
"github.com/gitlink-org/gitlink-cli/internal/output"
|
|
"github.com/gitlink-org/gitlink-cli/shortcuts/common"
|
|
)
|
|
|
|
func Shortcuts(translators ...*i18n.Translator) []*common.Shortcut {
|
|
tr := shortcutTranslator(translators...)
|
|
return []*common.Shortcut{
|
|
{
|
|
Name: "list",
|
|
Description: tr.T("cmd.pr.list.short"),
|
|
Flags: []common.Flag{
|
|
{Name: "state", Short: "s", Usage: tr.T("flag.pr.state"), Default: "open"},
|
|
{Name: "page", Short: "p", Usage: tr.T("flag.page"), Default: "1"},
|
|
{Name: "limit", Short: "l", Usage: tr.T("flag.limit"), Default: "20"},
|
|
},
|
|
Run: func(ctx *common.RuntimeContext) error {
|
|
if err := ctx.ResolveOwnerRepo(); err != nil {
|
|
return err
|
|
}
|
|
q := url.Values{}
|
|
q.Set("page", ctx.Arg("page"))
|
|
q.Set("limit", ctx.Arg("limit"))
|
|
if s := ctx.Arg("state"); s != "" {
|
|
q.Set("state", s)
|
|
}
|
|
env, err := ctx.CallAPIWithQuery("GET", ctx.RepoPath()+"/pulls", q)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return ctx.Output(env)
|
|
},
|
|
},
|
|
{
|
|
Name: "create",
|
|
Description: tr.T("cmd.pr.create.short"),
|
|
Flags: []common.Flag{
|
|
{Name: "title", Short: "t", Usage: tr.T("flag.pr.title"), Required: true},
|
|
{Name: "body", Short: "b", Usage: tr.T("flag.pr.body")},
|
|
{Name: "head", Usage: tr.T("flag.pr.head"), Required: true},
|
|
{Name: "base", Usage: tr.T("flag.pr.base"), Default: "master"},
|
|
},
|
|
Run: func(ctx *common.RuntimeContext) error {
|
|
if err := ctx.ResolveOwnerRepo(); err != nil {
|
|
return err
|
|
}
|
|
title, _ := ctx.RequireArg("title")
|
|
head, _ := ctx.RequireArg("head")
|
|
base := ctx.Arg("base")
|
|
if base == "" {
|
|
base = "master"
|
|
}
|
|
payload := map[string]interface{}{
|
|
"title": title,
|
|
"head": head,
|
|
"base": base,
|
|
}
|
|
if b := ctx.Arg("body"); b != "" {
|
|
payload["body"] = b
|
|
}
|
|
env, err := ctx.CallAPI("POST", ctx.RepoPath()+"/pulls", payload)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return ctx.Output(env)
|
|
},
|
|
},
|
|
{
|
|
Name: "view",
|
|
Description: tr.T("cmd.pr.view.short"),
|
|
Flags: []common.Flag{
|
|
{Name: "id", Short: "i", Usage: tr.T("flag.pr.id"), Required: true},
|
|
},
|
|
Run: func(ctx *common.RuntimeContext) error {
|
|
if err := ctx.ResolveOwnerRepo(); err != nil {
|
|
return err
|
|
}
|
|
id, _ := ctx.RequireArg("id")
|
|
env, err := ctx.CallAPI("GET", fmt.Sprintf("%s/pulls/%s", ctx.RepoPath(), id), nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := enrichPullRequestClosedAt(ctx, env); err != nil {
|
|
return err
|
|
}
|
|
return ctx.Output(env)
|
|
},
|
|
},
|
|
{
|
|
Name: "merge",
|
|
Description: tr.T("cmd.pr.merge.short"),
|
|
Flags: []common.Flag{
|
|
{Name: "id", Short: "i", Usage: tr.T("flag.pr.id"), Required: true},
|
|
{Name: "method", Short: "m", Usage: tr.T("flag.pr.merge_method"), Default: "merge"},
|
|
},
|
|
Run: func(ctx *common.RuntimeContext) error {
|
|
if err := ctx.ResolveOwnerRepo(); err != nil {
|
|
return err
|
|
}
|
|
id, _ := ctx.RequireArg("id")
|
|
method := ctx.Arg("method")
|
|
if method == "" {
|
|
method = "merge"
|
|
}
|
|
payload := map[string]interface{}{
|
|
"do": method,
|
|
}
|
|
env, err := ctx.CallAPI("POST", fmt.Sprintf("%s/pulls/%s/pr_merge", ctx.RepoPath(), id), payload)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return ctx.Output(env)
|
|
},
|
|
},
|
|
{
|
|
Name: "close",
|
|
Description: tr.T("cmd.pr.close.short"),
|
|
Flags: []common.Flag{
|
|
{Name: "id", Short: "i", Usage: tr.T("flag.pr.id"), Required: true},
|
|
},
|
|
Run: func(ctx *common.RuntimeContext) error {
|
|
if err := ctx.ResolveOwnerRepo(); err != nil {
|
|
return err
|
|
}
|
|
id, _ := ctx.RequireArg("id")
|
|
env, err := ctx.CallAPI("POST", fmt.Sprintf("%s/pulls/%s/refuse_merge", ctx.RepoPath(), id), nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return ctx.Output(env)
|
|
},
|
|
},
|
|
{
|
|
Name: "reopen",
|
|
Description: "Reopen a closed pull request",
|
|
Flags: []common.Flag{
|
|
{Name: "id", Short: "i", Usage: "PR number", Required: true},
|
|
},
|
|
Run: func(ctx *common.RuntimeContext) error {
|
|
if err := ctx.ResolveOwnerRepo(); err != nil {
|
|
return err
|
|
}
|
|
id, err := ctx.RequireArg("id")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
env, err := ctx.CallAPI("POST", prV1Path(ctx, id)+"/reopen", nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return ctx.Output(env)
|
|
},
|
|
},
|
|
{
|
|
Name: "files",
|
|
Description: tr.T("cmd.pr.files.short"),
|
|
Flags: []common.Flag{
|
|
{Name: "id", Short: "i", Usage: tr.T("flag.pr.id"), Required: true},
|
|
},
|
|
Run: func(ctx *common.RuntimeContext) error {
|
|
if err := ctx.ResolveOwnerRepo(); err != nil {
|
|
return err
|
|
}
|
|
id, _ := ctx.RequireArg("id")
|
|
env, err := ctx.CallAPI("GET", fmt.Sprintf("%s/pulls/%s/files", ctx.RepoPath(), id), nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return ctx.Output(env)
|
|
},
|
|
},
|
|
{
|
|
Name: "diff",
|
|
Description: tr.T("cmd.pr.diff.short"),
|
|
Flags: []common.Flag{
|
|
{Name: "id", Short: "i", Usage: tr.T("flag.pr.id"), Required: true},
|
|
},
|
|
Run: func(ctx *common.RuntimeContext) error {
|
|
if err := ctx.ResolveOwnerRepo(); err != nil {
|
|
return err
|
|
}
|
|
id, _ := ctx.RequireArg("id")
|
|
env, err := ctx.CallAPI("GET", fmt.Sprintf("%s/pulls/%s/files", ctx.RepoPath(), id), nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return ctx.Output(env)
|
|
},
|
|
},
|
|
{
|
|
Name: "versions",
|
|
Description: tr.T("cmd.pr.versions.short"),
|
|
Flags: []common.Flag{
|
|
{Name: "id", Short: "i", Usage: tr.T("flag.pr.id"), Required: true},
|
|
},
|
|
Run: func(ctx *common.RuntimeContext) error {
|
|
if err := ctx.ResolveOwnerRepo(); err != nil {
|
|
return err
|
|
}
|
|
id, err := ctx.RequireArg("id")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
env, err := ctx.CallAPI("GET", prV1Path(ctx, id)+"/versions", nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return ctx.Output(env)
|
|
},
|
|
},
|
|
{
|
|
Name: "version-diff",
|
|
Description: tr.T("cmd.pr.version_diff.short"),
|
|
Flags: []common.Flag{
|
|
{Name: "id", Short: "i", Usage: tr.T("flag.pr.id"), Required: true},
|
|
{Name: "version-id", Short: "v", Usage: tr.T("flag.pr.version_id"), Required: true},
|
|
{Name: "file", Short: "f", Usage: tr.T("flag.pr.file")},
|
|
},
|
|
Run: func(ctx *common.RuntimeContext) error {
|
|
if err := ctx.ResolveOwnerRepo(); err != nil {
|
|
return err
|
|
}
|
|
id, err := ctx.RequireArg("id")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
versionID, err := ctx.RequireArg("version-id")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
path := fmt.Sprintf("%s/versions/%s/diff", prV1Path(ctx, id), versionID)
|
|
if file := ctx.Arg("file"); file != "" {
|
|
q := url.Values{}
|
|
q.Set("filepath", file)
|
|
env, err := ctx.CallAPIWithQuery("GET", path, q)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return ctx.Output(env)
|
|
}
|
|
env, err := ctx.CallAPI("GET", path, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return ctx.Output(env)
|
|
},
|
|
},
|
|
{
|
|
Name: "reviews",
|
|
Description: tr.T("cmd.pr.reviews.short"),
|
|
Flags: []common.Flag{
|
|
{Name: "id", Short: "i", Usage: tr.T("flag.pr.id"), Required: true},
|
|
{Name: "status", Short: "s", Usage: tr.T("flag.pr.review_status_filter")},
|
|
},
|
|
Run: func(ctx *common.RuntimeContext) error {
|
|
if err := ctx.ResolveOwnerRepo(); err != nil {
|
|
return err
|
|
}
|
|
id, err := ctx.RequireArg("id")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
q := url.Values{}
|
|
if status := ctx.Arg("status"); status != "" {
|
|
if err := validatePRReviewStatus(status); err != nil {
|
|
return err
|
|
}
|
|
q.Set("status", status)
|
|
}
|
|
env, err := ctx.CallAPIWithQuery("GET", prV1Path(ctx, id)+"/reviews", q)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return ctx.Output(env)
|
|
},
|
|
},
|
|
{
|
|
Name: "review",
|
|
Description: tr.T("cmd.pr.review.short"),
|
|
Flags: []common.Flag{
|
|
{Name: "id", Short: "i", Usage: tr.T("flag.pr.id"), Required: true},
|
|
{Name: "status", Short: "s", Usage: tr.T("flag.pr.review_status"), Default: "common"},
|
|
{Name: "content", Short: "c", Usage: tr.T("flag.pr.review_content"), Required: true},
|
|
{Name: "commit", Short: "m", Usage: tr.T("flag.pr.review_commit")},
|
|
{Name: "dry-run", Usage: tr.T("flag.dry_run"), Bool: true, Default: "false"},
|
|
},
|
|
Run: func(ctx *common.RuntimeContext) error {
|
|
if err := ctx.ResolveOwnerRepo(); err != nil {
|
|
return err
|
|
}
|
|
id, err := ctx.RequireArg("id")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
content, err := ctx.RequireArg("content")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
status := ctx.Arg("status")
|
|
if status == "" {
|
|
status = "common"
|
|
}
|
|
if err := validatePRReviewStatus(status); err != nil {
|
|
return err
|
|
}
|
|
payload := map[string]interface{}{
|
|
"content": content,
|
|
"status": status,
|
|
}
|
|
if commit := ctx.Arg("commit"); commit != "" {
|
|
payload["commit_id"] = commit
|
|
}
|
|
if ctx.Arg("dry-run") == "true" {
|
|
return ctx.OutputData(map[string]interface{}{
|
|
"repository": fmt.Sprintf("%s/%s", ctx.Owner, ctx.Repo),
|
|
"pull_request": id,
|
|
"dry_run": true,
|
|
"action": "create_review",
|
|
"payload": payload,
|
|
})
|
|
}
|
|
env, err := ctx.CallAPI("POST", prV1Path(ctx, id)+"/reviews", payload)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Also post a journal comment so the review is visible in the PR conversation.
|
|
prEnv, journalErr := ctx.CallAPI("GET", fmt.Sprintf("%s/pulls/%s", ctx.RepoPath(), id), nil)
|
|
if journalErr == nil {
|
|
if issueID, extractErr := extractIssueID(prEnv); extractErr == nil {
|
|
statusLabel := map[string]string{
|
|
"approved": "approved", "rejected": "rejected", "common": "commented",
|
|
}[status]
|
|
summary := fmt.Sprintf("## Review: %s\n\n%s", statusLabel, content)
|
|
ctx.CallAPI("POST", fmt.Sprintf("/v1/%s/%s/issues/%d/journals", ctx.Owner, ctx.Repo, issueID),
|
|
map[string]interface{}{"notes": summary})
|
|
}
|
|
}
|
|
|
|
return ctx.Output(env)
|
|
},
|
|
},
|
|
{
|
|
Name: "comment",
|
|
Description: tr.T("cmd.pr.comment.short"),
|
|
Flags: []common.Flag{
|
|
{Name: "id", Short: "i", Usage: tr.T("flag.pr.id"), Required: true},
|
|
{Name: "body", Short: "b", Usage: tr.T("flag.comment.body"), Required: true},
|
|
},
|
|
Run: func(ctx *common.RuntimeContext) error {
|
|
if err := ctx.ResolveOwnerRepo(); err != nil {
|
|
return err
|
|
}
|
|
id, _ := ctx.RequireArg("id")
|
|
body, _ := ctx.RequireArg("body")
|
|
|
|
prEnv, err := ctx.CallAPI("GET", fmt.Sprintf("%s/pulls/%s", ctx.RepoPath(), id), nil)
|
|
if err != nil {
|
|
return fmt.Errorf("fetch PR: %w", err)
|
|
}
|
|
issueID, err := extractIssueID(prEnv)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
payload := map[string]interface{}{
|
|
"notes": body,
|
|
}
|
|
env, err := ctx.CallAPI("POST", fmt.Sprintf("/v1/%s/%s/issues/%d/journals", ctx.Owner, ctx.Repo, issueID), payload)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return ctx.Output(env)
|
|
},
|
|
},
|
|
{
|
|
Name: "commits",
|
|
Description: "List commits in a pull request",
|
|
Flags: []common.Flag{
|
|
{Name: "id", Short: "i", Usage: "PR number", Required: true},
|
|
},
|
|
Run: func(ctx *common.RuntimeContext) error {
|
|
if err := ctx.ResolveOwnerRepo(); err != nil {
|
|
return err
|
|
}
|
|
id, err := ctx.RequireArg("id")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
env, err := ctx.CallAPI("GET", prV1Path(ctx, id)+"/commits", nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return ctx.Output(env)
|
|
},
|
|
},
|
|
{
|
|
Name: "branches",
|
|
Description: "List branches for pull request creation",
|
|
Flags: []common.Flag{},
|
|
Run: func(ctx *common.RuntimeContext) error {
|
|
if err := ctx.ResolveOwnerRepo(); err != nil {
|
|
return err
|
|
}
|
|
env, err := ctx.CallAPI("GET", ctx.RepoPath()+"/pulls/get_branches", nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return ctx.Output(env)
|
|
},
|
|
},
|
|
{
|
|
Name: "check-merge",
|
|
Description: "Check if two branches can be merged",
|
|
Flags: []common.Flag{
|
|
{Name: "head", Usage: "Source branch", Required: true},
|
|
{Name: "base", Usage: "Target branch", Required: true},
|
|
},
|
|
Run: func(ctx *common.RuntimeContext) error {
|
|
if err := ctx.ResolveOwnerRepo(); err != nil {
|
|
return err
|
|
}
|
|
head, err := ctx.RequireArg("head")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
base, err := ctx.RequireArg("base")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
payload := map[string]interface{}{
|
|
"head": head,
|
|
"base": base,
|
|
}
|
|
env, err := ctx.CallAPI("POST", ctx.RepoPath()+"/pulls/check_can_merge", payload)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return ctx.Output(env)
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func shortcutTranslator(translators ...*i18n.Translator) *i18n.Translator {
|
|
if len(translators) > 0 && translators[0] != nil {
|
|
return translators[0]
|
|
}
|
|
return i18n.Default()
|
|
}
|
|
|
|
func prV1Path(ctx *common.RuntimeContext, id string) string {
|
|
return fmt.Sprintf("/v1/%s/%s/pulls/%s", ctx.Owner, ctx.Repo, id)
|
|
}
|
|
|
|
func validatePRReviewStatus(status string) error {
|
|
switch status {
|
|
case "common", "approved", "rejected":
|
|
return nil
|
|
default:
|
|
return fmt.Errorf("invalid --status value %q: use common, approved, or rejected", status)
|
|
}
|
|
}
|
|
|
|
func extractIssueID(env *output.Envelope) (int64, error) {
|
|
data, ok := env.Data.(map[string]interface{})
|
|
if !ok {
|
|
return 0, fmt.Errorf("unexpected PR response format")
|
|
}
|
|
issue, ok := data["issue"].(map[string]interface{})
|
|
if !ok {
|
|
return 0, fmt.Errorf("PR response missing issue field")
|
|
}
|
|
idFloat, ok := issue["id"].(float64)
|
|
if !ok {
|
|
return 0, fmt.Errorf("PR response missing issue.id field")
|
|
}
|
|
return int64(idFloat), nil
|
|
}
|
|
|
|
func enrichPullRequestClosedAt(ctx *common.RuntimeContext, env *output.Envelope) error {
|
|
data, ok := env.Data.(map[string]interface{})
|
|
if !ok {
|
|
return nil
|
|
}
|
|
pr, ok := data["pull_request"].(map[string]interface{})
|
|
if !ok || !isClosedPullRequest(pr) || stringField(pr, "closed_at") != "" {
|
|
return nil
|
|
}
|
|
issue, ok := data["issue"].(map[string]interface{})
|
|
if !ok {
|
|
return nil
|
|
}
|
|
issueID, ok := numberField(issue, "id")
|
|
if !ok {
|
|
return nil
|
|
}
|
|
journalsEnv, err := ctx.CallAPI("GET", fmt.Sprintf("/v1/%s/%s/issues/%d/journals", ctx.Owner, ctx.Repo, int64(issueID)), nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
closedAt := extractPullRequestClosedAt(journalsEnv)
|
|
if closedAt == "" {
|
|
return nil
|
|
}
|
|
pr["closed_at"] = closedAt
|
|
data["closed_at"] = closedAt
|
|
return nil
|
|
}
|
|
|
|
func isClosedPullRequest(pr map[string]interface{}) bool {
|
|
if stringField(pr, "pull_request_staus") == "closed" || stringField(pr, "state") == "closed" {
|
|
return true
|
|
}
|
|
status, ok := numberField(pr, "status")
|
|
return ok && int(status) == 2
|
|
}
|
|
|
|
func extractPullRequestClosedAt(env *output.Envelope) string {
|
|
data, ok := env.Data.(map[string]interface{})
|
|
if !ok {
|
|
return ""
|
|
}
|
|
rawJournals, ok := data["journals"].([]interface{})
|
|
if !ok {
|
|
return ""
|
|
}
|
|
for i := len(rawJournals) - 1; i >= 0; i-- {
|
|
journal, ok := rawJournals[i].(map[string]interface{})
|
|
if !ok || stringField(journal, "operate_category") != "status" {
|
|
continue
|
|
}
|
|
content := stringField(journal, "operate_content")
|
|
if !isPullRequestCloseOperation(content) {
|
|
continue
|
|
}
|
|
if updatedAt := stringField(journal, "updated_at"); updatedAt != "" {
|
|
return updatedAt
|
|
}
|
|
if createdAt := stringField(journal, "created_at"); createdAt != "" {
|
|
return createdAt
|
|
}
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func isPullRequestCloseOperation(content string) bool {
|
|
content = strings.ToLower(content)
|
|
return strings.Contains(content, "合并请求") &&
|
|
(strings.Contains(content, "拒绝") || strings.Contains(content, "关闭") || strings.Contains(content, "closed"))
|
|
}
|
|
|
|
func stringField(m map[string]interface{}, key string) string {
|
|
v, _ := m[key].(string)
|
|
return v
|
|
}
|
|
|
|
func numberField(m map[string]interface{}, key string) (float64, bool) {
|
|
switch v := m[key].(type) {
|
|
case float64:
|
|
return v, true
|
|
case int:
|
|
return float64(v), true
|
|
case int64:
|
|
return float64(v), true
|
|
default:
|
|
return 0, false
|
|
}
|
|
}
|