Forward query parameters to shields.io

handlers/badge.go

@@ -1,9 +1,9 @@
 package handlers
 
 import (
- "fmt"
  "log"
  "net/http"
+ "net/url"
 
  "github.com/dgraph-io/badger/v2"
  "github.com/gojp/goreportcard/check"
@@ -26,26 +26,71 @@ func BadgeHandler(w http.ResponseWriter, r *http.Request, db *badger.DB, repo st
  return
  }
 
- http.Redirect(w, r, badgeURL(resp.Grade, style), http.StatusTemporaryRedirect)
+ http.Redirect(w, r, badgeURL(resp.Grade, r.URL.Query()), http.StatusTemporaryRedirect)
 }
 
-func badgeURL(grade check.Grade, style string) string {
- var color string
- switch grade {
- case check.GradeAPlus:
- color = "brightgreen"
- case check.GradeA:
- color = "green"
- case check.GradeB:
- color = "yellowgreen"
- case check.GradeC:
- color = "yellow"
- case check.GradeD:
- color = "orange"
- case check.GradeE:
- fallthrough
- case check.GradeF:
- color = "red"
- }
- return fmt.Sprintf("https://img.shields.io/badge/go%%20report-%s-%s.svg?style=%s", grade, color, style)
+func badgeURL(grade check.Grade, queryParams url.Values) string {
+ badgeURL, _ := url.ParseRequestURI("https://img.shields.io/static/v1")
+
+ style := queryParams.Get("style")
+ color := queryParams.Get("color")
+ labelColor := queryParams.Get("labelColor")
+ logo := queryParams.Get("logo")
+ logoColor := queryParams.Get("logoColor")
+ logoWidth := queryParams.Get("logoWidth")
+ label := queryParams.Get("label")
+
+ if style == "" {
+ style = "flat"
+ }
+
+ if color == "" {
+ switch grade {
+ case check.GradeAPlus:
+ color = "brightgreen"
+ case check.GradeA:
+ color = "green"
+ case check.GradeB:
+ color = "yellowgreen"
+ case check.GradeC:
+ color = "yellow"
+ case check.GradeD:
+ color = "orange"
+ case check.GradeE:
+ fallthrough
+ case check.GradeF:
+ color = "red"
+ }
+ }
+
+ if label == "" {
+ label = "go report"
+ }
+
+ values := badgeURL.Query()
+ values.Set("color", color)
+ values.Set("style", style)
+ values.Set("label", label)
+ values.Set("message", string(grade))
+
+ // optional parameters
+ if labelColor != "" {
+ values.Set("labelColor", labelColor)
+ }
+
+ if logo != "" {
+ values.Set("logo", logo)
+ }
+
+ if logoColor != "" {
+ values.Set("logoColor", logoColor)
+ }
+
+ if logoWidth != "" {
+ values.Set("logoWidth", logoWidth)
+ }
+
+ badgeURL.RawQuery = values.Encode()
+
+ return badgeURL.String()
 }

handlers/badge_test.go

@@ -1,28 +1,96 @@
 package handlers
 
 import (
+ "net/url"
  "testing"
 
  "github.com/gojp/goreportcard/check"
 )
 
 func TestBadgeURL(t *testing.T) {
- for grade, expectedURL := range map[check.Grade]string{
- check.GradeAPlus: "https://img.shields.io/badge/go%20report-A+-brightgreen.svg?style=for-the-badge",
- check.GradeA: "https://img.shields.io/badge/go%20report-A-green.svg?style=for-the-badge",
- check.GradeB: "https://img.shields.io/badge/go%20report-B-yellowgreen.svg?style=for-the-badge",
- check.GradeC: "https://img.shields.io/badge/go%20report-C-yellow.svg?style=for-the-badge",
- check.GradeD: "https://img.shields.io/badge/go%20report-D-orange.svg?style=for-the-badge",
- check.GradeE: "https://img.shields.io/badge/go%20report-E-red.svg?style=for-the-badge",
- check.GradeF: "https://img.shields.io/badge/go%20report-F-red.svg?style=for-the-badge",
- } {
- grade := grade
- expectedURL := expectedURL
- t.Run(string(grade), func(t *testing.T) {
+ cases := []struct {
+ name string
+ grade check.Grade
+ params url.Values
+ want string
+ }{
+ {
+ check.GradeAPlus,
+ check.GradeAPlus,
+ url.Values{},
+ "https://img.shields.io/static/v1?color=brightgreen&label=go+report&message=A%2B&style=flat",
+ },
+ {
+ check.GradeA,
+ check.GradeA,
+ url.Values{},
+ "https://img.shields.io/static/v1?color=green&label=go+report&message=A&style=flat",
+ },
+ {
+ check.GradeB,
+ check.GradeB,
+ url.Values{},
+ "https://img.shields.io/static/v1?color=yellowgreen&label=go+report&message=B&style=flat",
+ },
+ {
+ check.GradeC,
+ check.GradeC,
+ url.Values{},
+ "https://img.shields.io/static/v1?color=yellow&label=go+report&message=C&style=flat",
+ },
+ {
+ check.GradeD,
+ check.GradeD,
+ url.Values{},
+ "https://img.shields.io/static/v1?color=orange&label=go+report&message=D&style=flat",
+ },
+ {
+ check.GradeE,
+ check.GradeE,
+ url.Values{},
+ "https://img.shields.io/static/v1?color=red&label=go+report&message=E&style=flat",
+ },
+ {
+ check.GradeF,
+ check.GradeF,
+ url.Values{},
+ "https://img.shields.io/static/v1?color=red&label=go+report&message=F&style=flat",
+ },
+ {
+ "override style",
+ check.GradeAPlus,
+ url.Values{"style": []string{"for-the-badge"}},
+ "https://img.shields.io/static/v1?color=brightgreen&label=go+report&message=A%2B&style=for-the-badge",
+ },
+ {
+ "override color",
+ check.GradeAPlus,
+ url.Values{"color": []string{"ff69b4"}},
+ "https://img.shields.io/static/v1?color=ff69b4&label=go+report&message=A%2B&style=flat",
+ },
+ {
+ "override logo params",
+ check.GradeAPlus,
+ url.Values{"logo": []string{"go"}, "logoWidth": []string{"100"}, "logoColor": []string{"ff69b4"}},
+ "https://img.shields.io/static/v1?color=brightgreen&label=go+report&logo=go&logoColor=ff69b4&logoWidth=100&message=A%2B&style=flat",
+ },
+ {
+ "override label params",
+ check.GradeAPlus,
+ url.Values{"label": []string{"code quality"}, "labelColor": []string{"0080ff"}},
+ "https://img.shields.io/static/v1?color=brightgreen&label=code+quality&labelColor=0080ff&message=A%2B&style=flat",
+ },
+ }
+
+ for _, tt := range cases {
+ tt := tt
+
+ t.Run(string(tt.name), func(t *testing.T) {
  t.Parallel()
- got := badgeURL(grade, "for-the-badge")
- if got != expectedURL {
- t.Errorf("expected %s, got %s", expectedURL, got)
+
+ got := badgeURL(tt.grade, tt.params)
+ if got != tt.want {
+ t.Errorf("expected %s, got %s", tt.want, got)
  }
  })
  }