@@ -40,12 +40,17 @@
 ### 做题记录
 
 ![](https://leetcode-badge.haozibi.dev/v1cn/chart/submission-calendar/haozibi.svg)
-- LeetCodeCN: `https://leetcode-badge.haozibi.dev/v1cn/chart/submission-calendar/{LeetCode_ID}.svg`
+- LeetCodeCN: `https://leetcode-badge.haozibi.dev/v1cn/chart/submission-calendar/{LeetCode_ID}.svg?type=past-year&color=yellow`
 
 参数
 
-- `?type=last-year`: 过去一年
-- `?`: 默认为 今年
+- type: 范围
+- 默认: 今年
+- past-year: 过去一年
+- color: 颜色
+- green(默认)
+- yellow
+- blue
 
 注意事项:
 

@@ -154,11 +154,14 @@ func (a *APP) getSubCal(name string, r *http.Request) ([]byte, error) {
 err error
 )
 
-var f func(data map[int64]int) ([]byte, error)
+var f func(data map[int64]int, color string) ([]byte, error)
+
 t := r.URL.Query().Get("type")
+color := r.URL.Query().Get("color")
+
 switch t {
-case "last-year":
-f = heatmap.LastYear
+case "past-year":
+f = heatmap.PastYear
 default:
 f = heatmap.CurrYear
 }
@@ -191,7 +194,7 @@ func (a *APP) getSubCal(name string, r *http.Request) ([]byte, error) {
 res[i] = v
 }
 
-body, err = f(res)
+body, err = f(res, color)
 if err != nil {
 return nil, err
 }

@@ -0,0 +1,14 @@
+package card
+
+func Build() ([]byte, error) {
+return nil, nil
+}
+
+type Info struct {
+Easy int
+EasyTotal int
+Media int
+MediaTotal int
+Hard int
+HardTotal int
+}

@@ -0,0 +1,7 @@
+package card
+
+import "testing"
+
+func TestBuild(t *testing.T) {
+Build()
+}

@@ -14,37 +14,73 @@ import (
 ".com/haozibi/leetcode-badge/internal/statics"
 )
 
-// LastYear 过去一年
-func LastYear(data map[int64]int) ([]byte, error) {
+// PastYear 过去一年
+func PastYear(data map[int64]int, color string) ([]byte, error) {
 now := time.Now()
-return Build(now.AddDate(-1, 0, 0).Unix(), now.Unix(), data)
+return Build(now.AddDate(-1, 0, 0).Unix(), now.Unix(), data, color)
 }
 
 // CurrYear 当前年
-func CurrYear(data map[int64]int) ([]byte, error) {
+func CurrYear(data map[int64]int, color string) ([]byte, error) {
 now := time.Now()
 s := time.Date(now.Year(), time.January, 1, 0, 0, 0, 0, time.UTC)
 e := time.Date(now.Year(), time.December, 31, 23, 59, 59, 0, time.UTC)
 
-return Build(s.Unix(), e.Unix(), data)
+return Build(s.Unix(), e.Unix(), data, color)
 }
 
-func Build(start, end int64, data map[int64]int) ([]byte, error) {
+func Build(start, end int64, data map[int64]int, color string) ([]byte, error) {
+
+cfg := &CalendarHeatmapConfig{
+Colors: []string{"#EBEDF0", "#9BE9A8", "#40C463", "#30A14E", "#216E39"},
+BlockSize: 11,
+BlockRoundness: 2,
+BlockMargin: 2,
+MonthLabels: []string{"Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"},
+MonthLabelHeight: 15,
+WeekdayLabels: []string{"", "Mon", "", "Wed", "", "Fri", ""},
+weekLabelWidth: 0,
+MonthSpace: 1,
+}
+
+switch color {
+case "yellow":
+cfg.Colors = Yellow
+case "green":
+cfg.Colors = Green
+case "blue":
+cfg.Colors = Blue
+default:
+cfg.Colors = Green
+}
 
 input := make(map[Date]int, len(data))
 for k, v := range data {
 if k >= start && k <= end {
 t := time.Unix(k, 0)
+vv := 0
+if v >= 1 && v <= 3 {
+vv = 1
+}
+if v > 3 && v <= 6 {
+vv = 2
+}
+if v > 6 && v <= 10 {
+vv = 3
+}
+if v > 10 {
+vv = 4
+}
 input[Date{
 Year: t.Year(),
 Month: t.Month(),
 Day: t.Day(),
-}] = v % 5 // only 5 color
+}] = vv // only 5 color
 }
 }
 
 st, et := time.Unix(start, 0), time.Unix(end, 0)
-h := New(nil)
+h := New(cfg)
 buf := h.Generate(
 Date{
 Year: st.Year(),

@@ -0,0 +1,8 @@
+package heatmap
+
+// 0.15 0.35 0.55 0.75 0.94
+var (
+Yellow = []string{"#FDF3B7", "#F5C670", "#EC7C46", "#C9312B", "#751428"}
+Green = []string{"#EBEDF0", "#9BE9A8", "#40C463", "#30A14E", "#216E39"}
+Blue = []string{"#E9F0F9", "#B7D1E5", "#6CA1CC", "#3568AB", "#152F67"}
+)

@@ -121,10 +121,7 @@ func (c *CalendarHeatmap) Generate(dateFrom, dateTo Date, data map[Date]int) *by
 }
 
 // draw svg
-size := weeks
-for i := 0; i < months; i++ {
-size += months * c.Config.MonthSpace
-}
+size := weeks + months*c.Config.MonthSpace
 canvasPos := c.getPosition(size, days)
 canvas.Start(canvasPos.X, canvasPos.Y)