feat: support to configure label color

config/config.go

@@ -91,22 +91,26 @@ type Plan struct {
 // WhenAddOrUpdateOnly is a configuration to notify the plan result contains new or updated in place resources
 type WhenAddOrUpdateOnly struct {
 	Label string `yaml:"label,omitempty"`
+	Color string `yaml:"color,omitempty"`
 }
 
 // WhenDestroy is a configuration to notify the plan result contains destroy operation
 type WhenDestroy struct {
 	Label    string `yaml:"label,omitempty"`
 	Template string `yaml:"template,omitempty"`
+	Color    string `yaml:"color,omitempty"`
 }
 
 // WhenNoChange is a configuration to add a label when the plan result contains no change
 type WhenNoChanges struct {
 	Label string `yaml:"label,omitempty"`
+	Color string `yaml:"color,omitempty"`
 }
 
 // WhenPlanError is a configuration to notify the plan result returns an error
 type WhenPlanError struct {
 	Label string `yaml:"label,omitempty"`
+	Color string `yaml:"color,omitempty"`
 }
 
 // Apply is a terraform apply config

main.go

@@ -122,10 +122,14 @@ func (t *tfnotify) Run(ctx context.Context) error {
 			DestroyWarningTemplate: t.destroyWarningTemplate,
 			WarnDestroy:            t.warnDestroy,
 			ResultLabels: github.ResultLabels{
-				AddOrUpdateLabel: t.config.Terraform.Plan.WhenAddOrUpdateOnly.Label,
-				DestroyLabel:     t.config.Terraform.Plan.WhenDestroy.Label,
-				NoChangesLabel:   t.config.Terraform.Plan.WhenNoChanges.Label,
-				PlanErrorLabel:   t.config.Terraform.Plan.WhenPlanError.Label,
+				AddOrUpdateLabel:      t.config.Terraform.Plan.WhenAddOrUpdateOnly.Label,
+				DestroyLabel:          t.config.Terraform.Plan.WhenDestroy.Label,
+				NoChangesLabel:        t.config.Terraform.Plan.WhenNoChanges.Label,
+				PlanErrorLabel:        t.config.Terraform.Plan.WhenPlanError.Label,
+				AddOrUpdateLabelColor: t.config.Terraform.Plan.WhenAddOrUpdateOnly.Color,
+				DestroyLabelColor:     t.config.Terraform.Plan.WhenDestroy.Color,
+				NoChangesLabelColor:   t.config.Terraform.Plan.WhenNoChanges.Color,
+				PlanErrorLabelColor:   t.config.Terraform.Plan.WhenPlanError.Color,
 			},
 		})
 		if err != nil {

notifier/github/client.go

@@ -120,10 +120,14 @@ func (pr *PullRequest) IsNumber() bool {
 
 // ResultLabels represents the labels to add to the PR depending on the plan result
 type ResultLabels struct {
-	AddOrUpdateLabel string
-	DestroyLabel     string
-	NoChangesLabel   string
-	PlanErrorLabel   string
+	AddOrUpdateLabel      string
+	DestroyLabel          string
+	NoChangesLabel        string
+	PlanErrorLabel        string
+	AddOrUpdateLabelColor string
+	DestroyLabelColor     string
+	NoChangesLabelColor   string
+	PlanErrorLabelColor   string
 }
 
 // HasAnyLabelDefined returns true if any of the internal labels are set

notifier/github/github.go

@@ -14,6 +14,7 @@ type API interface {
 	IssuesListComments(ctx context.Context, number int, opt *github.IssueListCommentsOptions) ([]*github.IssueComment, *github.Response, error)
 	IssuesAddLabels(ctx context.Context, number int, labels []string) ([]*github.Label, *github.Response, error)
 	IssuesRemoveLabel(ctx context.Context, number int, label string) (*github.Response, error)
+	IssuesUpdateLabel(ctx context.Context, label, color string) (*github.Label, *github.Response, error)
 	RepositoriesCreateComment(ctx context.Context, sha string, comment *github.RepositoryComment) (*github.RepositoryComment, *github.Response, error)
 	RepositoriesListCommits(ctx context.Context, opt *github.CommitsListOptions) ([]*github.RepositoryCommit, *github.Response, error)
 	RepositoriesGetCommit(ctx context.Context, sha string) (*github.RepositoryCommit, *github.Response, error)
@@ -55,6 +56,13 @@ func (g *GitHub) IssuesRemoveLabel(ctx context.Context, number int, label string
 	return g.Client.Issues.RemoveLabelForIssue(ctx, g.owner, g.repo, number, label)
 }
 
+// IssuesUpdateLabel is a wrapper of https://pkg.go.dev/github.com/google/go-github/github#IssuesService.EditLabel
+func (g *GitHub) IssuesUpdateLabel(ctx context.Context, label, color string) (*github.Label, *github.Response, error) {
+	return g.Client.Issues.EditLabel(ctx, g.owner, g.repo, label, &github.Label{
+		Color: &color,
+	})
+}
+
 // RepositoriesCreateComment is a wrapper of https://godoc.org/github.com/google/go-github/github#RepositoriesService.CreateComment
 func (g *GitHub) RepositoriesCreateComment(ctx context.Context, sha string, comment *github.RepositoryComment) (*github.RepositoryComment, *github.Response, error) {
 	return g.Client.Repositories.CreateComment(ctx, g.owner, g.repo, sha, comment)

notifier/github/notify.go

@@ -38,27 +38,47 @@ func (g *NotifyService) Notify(ctx context.Context, body string) (exit int, err
 			if err != nil {
 				return result.ExitCode, err
 			}
-			var labelToAdd string
+			var (
+				labelToAdd string
+				labelColor string
+			)
 
 			if result.HasAddOrUpdateOnly {
 				labelToAdd = cfg.ResultLabels.AddOrUpdateLabel
+				labelColor = cfg.ResultLabels.AddOrUpdateLabelColor
 			} else if result.HasDestroy {
 				labelToAdd = cfg.ResultLabels.DestroyLabel
+				labelColor = cfg.ResultLabels.DestroyLabelColor
 			} else if result.HasNoChanges {
 				labelToAdd = cfg.ResultLabels.NoChangesLabel
+				labelColor = cfg.ResultLabels.NoChangesLabelColor
 			} else if result.HasPlanError {
 				labelToAdd = cfg.ResultLabels.PlanErrorLabel
+				labelColor = cfg.ResultLabels.PlanErrorLabelColor
 			}
 
 			if labelToAdd != "" {
-				_, _, err = g.client.API.IssuesAddLabels(
+				labels, _, err := g.client.API.IssuesAddLabels(
 					context.Background(),
 					cfg.PR.Number,
 					[]string{labelToAdd},
 				)
 				if err != nil {
 					return result.ExitCode, err
 				}
+				if labelColor != "" {
+					// set the color of label
+					for _, label := range labels {
+						if labelToAdd == label.GetName() {
+							if label.GetColor() != labelColor {
+								_, _, err := g.client.API.IssuesUpdateLabel(context.Background(), labelToAdd, labelColor)
+								if err != nil {
+									return result.ExitCode, err
+								}
+							}
+						}
+					}
+				}
 			}
 		}
 	}