feat: allow std hex colors

Signed-off-by: Mateusz Urbanek <mateusz.urbanek.98@gmail.com>
shanduur · Jul 30, 2023 · a4f91d9 · a4f91d9
1 parent 596cf85
commit a4f91d9
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/cmd/labeler/upload/upload_test.go b/cmd/labeler/upload/upload_test.go
@@ -45,6 +45,17 @@ func TestPostLabel(t *testing.T) {
 			Repo:     testRepo,
 			Cassette: govcr.NewCassetteLoader(fixtures),
 		},
+		{
+			Name: "with_prefix",
+			Label: labels.Label{
+				Name:        testLabelName,
+				Color:       "#" + testLabelColor,
+				Description: &testLabelDescription,
+			},
+			Owner:    testOwner,
+			Repo:     testRepo,
+			Cassette: govcr.NewCassetteLoader(fixtures),
+		},
 	} {
 		t.Run(tc.Name, func(t *testing.T) {
 			tc := tc

diff --git a/labels/labels.go b/labels/labels.go
@@ -2,6 +2,7 @@ package labels
 
 import (
 	"errors"
+	"strings"
 
 	"github.com/google/go-github/v53/github"
 )
@@ -25,9 +26,14 @@ func (l *Label) Validate() error {
 }
 
 func (l *Label) ToGitHub() *github.Label {
+	color := l.Color
+	if strings.HasPrefix(l.Color, "#") {
+		color = strings.TrimPrefix(color, "#")
+	}
+
 	return &github.Label{
 		Name:        &l.Name,
-		Color:       &l.Color,
+		Color:       &color,
 		Description: l.Description,
 	}
 }