Labels (#248)

Adds label support for PRs for GitHub and GitLab

Author: hooksie1

digestabotctl/cmd/files.go

@@ -100,6 +100,7 @@ func handlePRForPlatform(platform string, checkout versioncontrol.CheckoutRespon
 		Diff:        commit,
 		Base:        viper.GetString("base"),
 		Head:        viper.GetString("branch"),
+		Labels:      viper.GetStringSlice("labels"),
 		RepoData: platforms.RepoData{
 			Repo:  viper.GetString("repo"),
 			Owner: viper.GetString("owner"),

digestabotctl/cmd/flags.go

@@ -57,6 +57,7 @@ func bindPRFlags(cmd *cobra.Command) {
 	viper.BindPFlag("signing-token", cmd.Flags().Lookup("signing-token"))
 	viper.BindPFlag("name", cmd.Flags().Lookup("name"))
 	viper.BindPFlag("email", cmd.Flags().Lookup("email"))
+	viper.BindPFlag("labels", cmd.Flags().Lookup("labels"))
 }
 
 // prFlags adds the pr flags to the passed in command
@@ -74,4 +75,5 @@ func prFlags(cmd *cobra.Command) {
 	cmd.PersistentFlags().String("signing-token", "", "OIDC token for signing commit")
 	cmd.PersistentFlags().String("name", "digestabotctl", "Name for commit")
 	cmd.PersistentFlags().String("email", "", "Email for commit")
+	cmd.PersistentFlags().StringSlice("labels", []string{"automated pr", "kind/cleanup", "release-note-none"}, "Labels to apply to the PR")
 }

digestabotctl/platforms/github.go

@@ -17,10 +17,21 @@ type GitHub struct {
 }
 
 type GitHubPR struct {
-	Title string `json:"title"`
-	Body  string `json:"body"`
-	Head  string `json:"head"`
-	Base  string `json:"base"`
+	Title  string `json:"title"`
+	Body   string `json:"body"`
+	Head   string `json:"head"`
+	Base   string `json:"base"`
+	Labels []string
+}
+
+type GitHubPRResponse struct {
+	URL    string `json:"url"`
+	ID     int    `json:"id"`
+	Number int    `json:"number"`
+}
+
+type GitHubPRLabelRequest struct {
+	Labels []string `json:"labels"`
 }
 
 func (g GitHub) CreatePR(logger *slog.Logger) error {
@@ -39,6 +50,43 @@ func (g GitHub) CreatePR(logger *slog.Logger) error {
 	req.Header.Add("X-GitHub-Api-Version", "2022-11-28")
 	req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", g.Token))
 
+	resp, err := sendRequest(req)
+	if err != nil {
+		return err
+	}
+	logger.Debug(string(resp))
+	var gr GitHubPRResponse
+	if err := json.Unmarshal(resp, &gr); err != nil {
+		return err
+	}
+
+	if len(g.GitHubPR.Labels) == 0 {
+		return nil
+	}
+
+	return g.CreateLabels(logger, gr.Number)
+}
+
+func (g GitHub) CreateLabels(logger *slog.Logger, pr int) error {
+	url := fmt.Sprintf("https://api.github.com/repos/%s/%s/issues/%d/labels", g.Owner, g.Repo, pr)
+	labelRequest := GitHubPRLabelRequest{
+		Labels: g.GitHubPR.Labels,
+	}
+
+	body, err := json.Marshal(labelRequest)
+	if err != nil {
+		return err
+	}
+	logger.Debug(string(body))
+
+	req, err := http.NewRequest(http.MethodPost, url, bytes.NewReader(body))
+	if err != nil {
+		return err
+	}
+	req.Header.Add("Accept", "application/vnd.github+json")
+	req.Header.Add("X-GitHub-Api-Version", "2022-11-28")
+	req.Header.Add("Authorization", fmt.Sprintf("Bearer %s", g.Token))
+
 	resp, err := sendRequest(req)
 	if err != nil {
 		return err
@@ -60,10 +108,11 @@ func NewGitHub(pr PullRequest) (PRCreator, error) {
 		Repo:  pr.Repo,
 		Token: pr.Token,
 		GitHubPR: GitHubPR{
-			Title: pr.Title,
-			Body:  buf.String(),
-			Head:  pr.Head,
-			Base:  pr.Base,
+			Title:  pr.Title,
+			Body:   buf.String(),
+			Head:   pr.Head,
+			Base:   pr.Base,
+			Labels: pr.Labels,
 		},
 	}, nil
 }

digestabotctl/platforms/gitlab.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"log/slog"
 	"net/http"
+	"strings"
 	"text/template"
 )
 
@@ -21,6 +22,7 @@ type GitLabMR struct {
 	Description string `json:"description"`
 	Source      string `json:"source_branch"`
 	Target      string `json:"target_branch"`
+	Labels      string `json:"labels"`
 }
 
 func (g GitLab) CreatePR(logger *slog.Logger) error {
@@ -64,6 +66,7 @@ func NewGitLab(pr PullRequest) (PRCreator, error) {
 			Description: buf.String(),
 			Source:      pr.Head,
 			Target:      pr.Base,
+			Labels:      strings.Join(pr.Labels, ","),
 		},
 	}, nil
 }

digestabotctl/platforms/platforms.go

@@ -54,6 +54,7 @@ type PullRequest struct {
 	Diff        string
 	Base        string
 	Head        string
+	Labels      []string
 	RepoData
 }