feat: display today streak counter

cmd/submit.go

@@ -61,6 +61,8 @@ leetgo submit w330/
 					log.Info("added failed case to testcases.txt")
 				}
 			}
+
+			_ = showTodayStreak(c)
 		}
 
 		if hasFailedCase {
@@ -145,3 +147,14 @@ func appendToTestCases(q *leetcode.QuestionData, result *leetcode.SubmitCheckRes
 	err = utils.WriteFile(testCasesFile.GetPath(), content)
 	return true, err
 }
+
+func showTodayStreak(c leetcode.Client) error {
+	streak, err := c.GetStreakCounter()
+	if err != nil {
+		return err
+	}
+	if streak.TodayCompleted {
+		fmt.Println("You have completed today's problem!\nTotal streak:", streak.StreakCount)
+	}
+	return nil
+}

leetcode/client.go

@@ -40,8 +40,8 @@ func (e UnexpectedStatusCode) IsError() bool {
 
 func (e UnexpectedStatusCode) Error() string {
 	body := "<empty body>"
-	if len(e.Body) > 100 {
-		body = e.Body[:100] + "..."
+	if len(e.Body) > 500 {
+		body = e.Body[:500] + "..."
 	}
 	return fmt.Sprintf("[%d %s] %s", e.Code, http.StatusText(e.Code), body)
 }
@@ -81,6 +81,7 @@ type Client interface {
 	GetContestQuestionData(contestSlug string, questionSlug string) (*QuestionData, error)
 	RegisterContest(slug string) error
 	UnregisterContest(slug string) error
+	GetStreakCounter() (StreakCounter, error)
 }
 
 type cnClient struct {
@@ -1012,3 +1013,25 @@ func (c *cnClient) GetQuestionTags() ([]QuestionTag, error) {
 	}
 	return tags, nil
 }
+
+func (c *cnClient) GetStreakCounter() (StreakCounter, error) {
+	query := `
+query getStreakCounter {
+  problemsetStreakCounter {
+    today
+    streakCount
+    daysSkipped
+    todayCompleted
+  }
+}`
+	var resp gjson.Result
+	_, err := c.graphqlPost(
+		graphqlRequest{query: query, authType: requireAuth}, &resp,
+	)
+	if err != nil {
+		return StreakCounter{}, err
+	}
+	var counter StreakCounter
+	err = json.Unmarshal(utils.StringToBytes(resp.Get("data.problemsetStreakCounter").Raw), &counter)
+	return counter, err
+}

leetcode/models.go

@@ -259,3 +259,10 @@ type QuestionTag struct {
 	TypeName       string `json:"typeName"`
 	TypeTransName  string `json:"typeTransName"`
 }
+
+type StreakCounter struct {
+	Today          string `json:"today"`
+	StreakCount    int    `json:"streakCount"`
+	DaysSkipped    int    `json:"daysSkipped"`
+	TodayCompleted bool   `json:"todayCompleted"`
+}