Add global slice with all styles

Author: bytedream

modules/badge/badge.go

@@ -65,16 +65,20 @@ const (
 	DefaultStyle      = StyleFlat
 )
 
-var StatusColorMap = map[actions_model.Status]string{
-	actions_model.StatusSuccess:   "#4c1",    // Green
-	actions_model.StatusSkipped:   "#dfb317", // Yellow
-	actions_model.StatusUnknown:   "#97ca00", // Light Green
-	actions_model.StatusFailure:   "#e05d44", // Red
-	actions_model.StatusCancelled: "#fe7d37", // Orange
-	actions_model.StatusWaiting:   "#dfb317", // Yellow
-	actions_model.StatusRunning:   "#dfb317", // Yellow
-	actions_model.StatusBlocked:   "#dfb317", // Yellow
-}
+var (
+	StatusColorMap = map[actions_model.Status]string{
+		actions_model.StatusSuccess:   "#4c1",    // Green
+		actions_model.StatusSkipped:   "#dfb317", // Yellow
+		actions_model.StatusUnknown:   "#97ca00", // Light Green
+		actions_model.StatusFailure:   "#e05d44", // Red
+		actions_model.StatusCancelled: "#fe7d37", // Orange
+		actions_model.StatusWaiting:   "#dfb317", // Yellow
+		actions_model.StatusRunning:   "#dfb317", // Yellow
+		actions_model.StatusBlocked:   "#dfb317", // Yellow
+	}
+
+	AllStyles = []Style{StyleFlat, StyleFlatSquare}
+)
 
 // GenerateBadge generates badge with given template
 func GenerateBadge(label, message, color string) Badge {

routers/web/devtest/devtest.go

@@ -128,7 +128,6 @@ func prepareMockDataBadgeCommitSign(ctx *context.Context) {
 
 func prepareMockDataBadgeActionsSvg(ctx *context.Context) {
 	fontFamilyNames := strings.Split(badge.DefaultFontFamily, ",")
-	styles := []badge.Style{badge.StyleFlat, badge.StyleFlatSquare}
 	selectedFontFamilyName := ctx.FormString("font", fontFamilyNames[0])
 	selectedStyle := ctx.FormString("style", badge.DefaultStyle)
 	var badges []badge.Badge
@@ -163,7 +162,7 @@ func prepareMockDataBadgeActionsSvg(ctx *context.Context) {
 	ctx.Data["BadgeSVGs"] = badgeSVGs
 	ctx.Data["BadgeFontFamilyNames"] = fontFamilyNames
 	ctx.Data["SelectedFontFamilyName"] = selectedFontFamilyName
-	ctx.Data["BadgeStyles"] = styles
+	ctx.Data["BadgeStyles"] = badge.AllStyles
 	ctx.Data["SelectedStyle"] = selectedStyle
 }
 

routers/web/repo/actions/badge.go

@@ -27,7 +27,7 @@ func GetWorkflowBadge(ctx *context.Context) {
 	event := ctx.Req.URL.Query().Get("event")
 
 	style := ctx.Req.URL.Query().Get("style")
-	if !slices.Contains([]badge.Style{badge.StyleFlat, badge.StyleFlatSquare}, badge.Style(style)) {
+	if !slices.Contains(badge.AllStyles, badge.Style(style)) {
 		style = badge.DefaultStyle
 	}