[feat] first version

Author: ronething

.github/workflows/today.yaml

@@ -0,0 +1,28 @@
+name: "leetcode-question-today"
+
+on:
+  schedule:
+    - cron: "0 12 * * *"
+  workflow_dispatch:
+
+jobs:
+  today:
+    runs-on: ubuntu-latest
+    steps:
+        - name: Checkout
+          uses: actions/checkout@v2
+          with:
+            fetch-depth: 0
+
+        - name: Set up Go
+          uses: actions/setup-go@v2
+          with:
+            go-version: 1.16
+
+        - name: run cmd
+          run: |
+            go run main.go -slack ${{ secrets.SLACK_URL }}
+
+        - name: echo
+          run: |
+            echo "notify ok" 

.gitignore

@@ -2,4 +2,6 @@
 .vscode/
 
 .DS_Store
-go.sum
+go.sum
+
+leetcode-question-today.exe

README.md

@@ -1,4 +1,4 @@
-<!-- START doctoc generated TOC please keep comment here to allow auto update -->
+<!-- START doctoc generated TOC please keep comment here to allow auto update -->-- START doctoc generated TOC please keep comment here to allow auto update -->
 <!-- DON'T EDIT THIS SECTION, INSTEAD RE-RUN doctoc TO UPDATE -->
 **Table of Contents**  *generated with [DocToc](https://github.com/thlorenz/doctoc)*
 
@@ -8,14 +8,11 @@
 
 <!-- END doctoc generated TOC please keep comment here to allow auto update -->
 
-## go-template
+## leetcode-question-today
 
-### feature
+leetcode 每日一题推送
 
-- go file add license
-- go file format and auto import
-- markdown file add toc
+### acknowledgement
 
-### TODO
-
-- [ ] Makefile
+- graphql
+- leetcode

api/today.go

@@ -0,0 +1,29 @@
+package api
+
+import (
+	"context"
+	"github.com/machinebox/graphql"
+)
+
+func GetTodayQuestion(ctx context.Context) (*QuestionTodayResp, error) {
+	// create a client (safe to share across requests)
+	client := graphql.NewClient("https://leetcode.cn/graphql/")
+
+	// make a request
+	req := graphql.NewRequest(QuestionQuery)
+
+	// set header fields
+	req.Header.Set("Cache-Control", "no-cache")
+	req.Header.Set("content-type", "application/json")
+	req.Header.Set("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/101.0.4951.67 Safari/537.36")
+	req.Header.Set("referer", "https://leetcode.cn/problemset/all/")
+	req.Header.Set("origin", "https://leetcode.cn")
+
+	// run it and capture the response
+	var resp QuestionTodayResp
+	if err := client.Run(ctx, req, &resp); err != nil {
+		return nil, err
+	}
+
+	return &resp, nil
+}

api/types.go

@@ -0,0 +1,82 @@
+package api
+
+const (
+	LeetcodeCn = "https://leetcode.cn"
+	Leetcode   = "https://leetcode.com"
+)
+
+// QuestionQuery graphql query
+const QuestionQuery = `
+	query questionOfToday {
+  todayRecord {
+    date
+    userStatus
+    question {
+      questionId
+      frontendQuestionId: questionFrontendId
+      difficulty
+      title
+      titleCn: translatedTitle
+      titleSlug
+      paidOnly: isPaidOnly
+      freqBar
+      isFavor
+      acRate
+      status
+      solutionNum
+      hasVideoSolution
+      topicTags {
+        name
+        nameTranslated: translatedName
+        id
+      }
+      extra {
+        topCompanyTags {
+          imgUrl
+          slug
+          numSubscribed
+        }
+      }
+    }
+    lastSubmission {
+      id
+    }
+  }
+}
+`
+
+// QuestionTodayResp 注意与官网直接 restful api 请求返回的少一个 data 字段嵌套
+type QuestionTodayResp struct {
+	TodayRecord []struct {
+		Date       string `json:"date"`
+		UserStatus string `json:"userStatus"`
+		Question   struct {
+			QuestionID         string      `json:"questionId"`
+			FrontendQuestionID string      `json:"frontendQuestionId"`
+			Difficulty         string      `json:"difficulty"`
+			Title              string      `json:"title"`
+			TitleCn            string      `json:"titleCn"`
+			TitleSlug          string      `json:"titleSlug"`
+			PaidOnly           bool        `json:"paidOnly"`
+			FreqBar            interface{} `json:"freqBar"`
+			IsFavor            bool        `json:"isFavor"`
+			AcRate             float64     `json:"acRate"`
+			Status             interface{} `json:"status"`
+			SolutionNum        int         `json:"solutionNum"`
+			HasVideoSolution   bool        `json:"hasVideoSolution"`
+			TopicTags          []struct {
+				Name           string `json:"name"`
+				NameTranslated string `json:"nameTranslated"`
+				ID             string `json:"id"`
+			} `json:"topicTags"`
+			Extra struct {
+				TopCompanyTags []struct {
+					ImgURL        string `json:"imgUrl"`
+					Slug          string `json:"slug"`
+					NumSubscribed int    `json:"numSubscribed"`
+				} `json:"topCompanyTags"`
+			} `json:"extra"`
+		} `json:"question"`
+		LastSubmission interface{} `json:"lastSubmission"`
+	} `json:"todayRecord"`
+}

example/graphql_test.go

@@ -0,0 +1,35 @@
+package example_test
+
+import (
+	"context"
+	"github.com/machinebox/graphql"
+	"leetcode-question-today/api"
+	"log"
+	"testing"
+)
+
+func TestGraphql(t *testing.T) {
+	// create a client (safe to share across requests)
+	client := graphql.NewClient("https://leetcode.cn/graphql/")
+
+	// make a request
+	req := graphql.NewRequest(api.QuestionQuery)
+
+	// set header fields
+	req.Header.Set("Cache-Control", "no-cache")
+	req.Header.Set("content-type", "application/json")
+	req.Header.Set("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/101.0.4951.67 Safari/537.36")
+	req.Header.Set("referer", "https://leetcode.cn/problemset/all/")
+	req.Header.Set("origin", "https://leetcode.cn")
+
+	// define a Context for the request
+	ctx := context.Background()
+
+	// run it and capture the response
+	var respData api.QuestionTodayResp
+	if err := client.Run(ctx, req, &respData); err != nil {
+		log.Fatal(err)
+	}
+
+	t.Logf("%+v\n", respData)
+}

go.mod

@@ -0,0 +1,10 @@
+module leetcode-question-today
+
+go 1.16
+
+require (
+	github.com/cloud-org/msgpush v0.0.3 // indirect
+	github.com/machinebox/graphql v0.2.2
+	github.com/matryer/is v1.4.0 // indirect
+	github.com/pkg/errors v0.9.1 // indirect
+)

main.go

@@ -24,9 +24,87 @@
 
 package main
 
-import "fmt"
+import (
+	"context"
+	"flag"
+	"fmt"
+	"github.com/cloud-org/msgpush"
+	"leetcode-question-today/api"
+	"log"
+	"os"
+	"strings"
+)
+
+var (
+	slack string // slack 通知链接
+	wecom string // wecom 通知链接
+	help  bool   // 帮助
+)
+
+func init() {
+	flag.StringVar(&slack, "slack", "", "slack webhook url")
+	flag.StringVar(&wecom, "wecom", "", "wecom webhook token")
+	flag.BoolVar(&help, "h", false, "帮助")
+	flag.Usage = usage
+}
+
+func usage() {
+	fmt.Fprintf(os.Stdout, `leetcode-question-today - leetcode 每日一题推送
+Usage: leetcode-question-today [-h help]
+Options:
+`)
+	flag.PrintDefaults()
+}
 
 func main() {
-	// you can delete this after clone template
-	fmt.Println("hello world")
+	flag.Parse()
+	if help {
+		flag.PrintDefaults()
+		return
+	}
+
+	// 获取每日一题，如果有则推送即可
+	resp, err := api.GetTodayQuestion(context.TODO())
+	if err != nil {
+		log.Printf("获取每日一题发生错误: %v\n", err)
+		return
+	}
+
+	if len(resp.TodayRecord) <= 0 {
+		log.Printf("todayRecord 长度为 0,请检查\n")
+		return
+	}
+
+	today := resp.TodayRecord[0]
+
+	msgTemplate := `每日一题(%s)
+Title: %s
+Tags: %s
+Link: %s
+LinkCN: %s`
+	date := today.Date
+	title := fmt.Sprintf("%s(%s)", today.Question.TitleCn, today.Question.Title)
+	tags := make([]string, 0)
+	for _, tag := range today.Question.TopicTags {
+		tags = append(tags, fmt.Sprintf("%s(%s)", tag.NameTranslated, tag.Name))
+	}
+	tagsValue := strings.Join(tags, "、")
+	link := fmt.Sprintf("%s/%s", api.Leetcode, today.Question.TitleSlug)
+	linkCn := fmt.Sprintf("%s/%s", api.LeetcodeCn, today.Question.TitleSlug)
+
+	content := fmt.Sprintf(msgTemplate, date, title, tagsValue, link, linkCn)
+
+	log.Println(content)
+
+	if slack != "" {
+		s := msgpush.NewSlack(slack)
+		_ = s.Send(content)
+	}
+
+	if wecom != "" {
+		w := msgpush.NewWeCom(wecom)
+		_ = w.SendText(content)
+	}
+
+	return
 }