support custom labels

Author: odanado

README.md

@@ -50,6 +50,7 @@ jobs:
 
 - `--from`: The base branch name. Required.
 - `--to`: The target branch name. Required.
+- `--labels`: Specify the labels to add to the pull request as a comma-separated list of strings. Optional.
 
 ## Environment Variables
 
@@ -72,7 +73,6 @@ While inspired by git-pr-release, this tool pays homage to its predecessor yet i
 
 # TODO
 - [ ] Support a custom template
-- [ ] Support custom labels
 - [ ] Add more testing
 
 # Release flow

github_client.go

@@ -125,3 +125,11 @@ func (c *GithubClient) UpdatePullRequest(ctx context.Context, prNumber int, titl
 
 	return pr, nil
 }
+
+func (c *GithubClient) AddLabelsToPullRequest(ctx context.Context, prNumber int, labels []string) error {
+	if len(labels) == 0 {
+		return nil
+	}
+	_, _, err := c.client.Issues.AddLabelsToIssue(ctx, c.owner, c.repo, prNumber, labels)
+	return err
+}

github_client_test.go

@@ -178,3 +178,28 @@ func TestCreatePullRequest_alreadyExists(t *testing.T) {
 		t.Errorf("PullRequests.List returned %+v, want %+v", created, false)
 	}
 }
+
+func TestAddLabelsToPullRequest(t *testing.T) {
+	ctx := context.Background()
+
+	mux := http.NewServeMux()
+
+	mux.HandleFunc(
+		"/repos/owner/repo/issues/1/labels",
+		func(w http.ResponseWriter, r *http.Request) {
+			fmt.Fprint(w, `[]`)
+		},
+	)
+
+	ts := httptest.NewServer(mux)
+	defer ts.Close()
+
+	apiUrl, _ := url.Parse(ts.URL)
+	client := NewClient(GithubClientOptions{owner: "owner", repo: "repo", githubToken: "githubToken", apiUrl: apiUrl})
+
+	err := client.AddLabelsToPullRequest(ctx, 1, []string{"label1", "label2"})
+
+	if err != nil {
+		t.Errorf("PullRequests.Get returned error: %v", err)
+	}
+}

main.go

@@ -12,8 +12,9 @@ import (
 
 type Options struct {
 	// from flag
-	from string
-	to   string
+	from   string
+	to     string
+	labels []string
 
 	// from env
 	owner       string
@@ -23,8 +24,9 @@ type Options struct {
 }
 
 func getOptions() (Options, error) {
-	from := flag.String("from", "", "source branch")
-	to := flag.String("to", "", "target branch")
+	from := flag.String("from", "", "The base branch name.")
+	to := flag.String("to", "", "The target branch name.")
+	labelsFlag := flag.String("labels", "", "Specify the labels to add to the pull request as a comma-separated list of strings.")
 	flag.Parse()
 
 	githubToken := os.Getenv("GITHUB_TOKEN")
@@ -35,9 +37,15 @@ func getOptions() (Options, error) {
 
 	apiUrl, _ := url.Parse(rawApiUrl)
 
+	var labels []string
+	if labelsFlag != nil {
+		labels = strings.Split(*labelsFlag, ",")
+	}
+
 	return Options{
 		from:        *from,
 		to:          *to,
+		labels:      labels,
 		owner:       owner,
 		repo:        repo,
 		gitHubToken: githubToken,
@@ -101,6 +109,14 @@ func run(options Options) error {
 		logger.Println("The pull request already exists. The body was updated.", pr.GetNumber())
 	}
 
+	if len(options.labels) > 0 {
+		err := client.AddLabelsToPullRequest(ctx, pr.GetNumber(), options.labels)
+		if err != nil {
+			return err
+		}
+		logger.Println("Added labels to the pull request.", pr.GetNumber())
+	}
+
 	return nil
 }