feat: 添加信息推送功能。添加定时执行功能

.gitignore

@@ -3,3 +3,5 @@
 *.png
 /logs/
 *.log
+/config/config.yml
+/chromium-857950/

go.mod

@@ -8,7 +8,7 @@ require (
 	github.com/Baozisoftware/qrcode-terminal-go v0.0.0-20170407111555-c0650d8dff0f
 	github.com/guonaihong/gout v0.2.9
 	github.com/lestrrat-go/file-rotatelogs v2.4.0+incompatible
-	github.com/mxschmitt/playwright-go v0.1100.0
+	github.com/mxschmitt/playwright-go v0.1400.0
 	github.com/sirupsen/logrus v1.8.1
 	github.com/t-tomalak/logrus-easy-formatter v0.0.0-20190827215021-c074f06c5816
 	github.com/tidwall/gjson v1.11.0
@@ -22,6 +22,8 @@ require (
 	github.com/go-playground/universal-translator v0.17.0 // indirect
 	github.com/go-playground/validator/v10 v10.4.1 // indirect
 	github.com/google/uuid v1.1.1 // indirect
+	github.com/gorilla/websocket v1.4.2 // indirect
+	github.com/johlanse/ding v0.0.0-20211117082011-7467908ad84f // indirect
 	github.com/jonboulle/clockwork v0.2.2 // indirect
 	github.com/leodido/go-urn v1.2.0 // indirect
 	github.com/lestrrat-go/strftime v1.0.5 // indirect
@@ -29,6 +31,7 @@ require (
 	github.com/mattn/go-colorable v0.1.11 // indirect
 	github.com/mattn/go-isatty v0.0.14 // indirect
 	github.com/pkg/errors v0.9.1 // indirect
+	github.com/robfig/cron/v3 v3.0.0 // indirect
 	github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e // indirect
 	github.com/tidwall/match v1.1.1 // indirect
 	github.com/tidwall/pretty v1.2.0 // indirect

go.sum

@@ -28,10 +28,16 @@ github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/
 github.com/google/uuid v1.1.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/google/uuid v1.1.1 h1:Gkbcsh/GbpXz7lPftLA3P6TYMwjCLYm83jiFQZF/3gY=
 github.com/google/uuid v1.1.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/gorilla/websocket v1.4.2 h1:+/TMaTYc4QFitKJxsQ7Yye35DkWvkdLcvGKqM+x0Ufc=
 github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
 github.com/guonaihong/gout v0.2.9 h1:8nU5hrtwP1qDwiadFvU+D+z3ud9WEk8iPSfxQDiebng=
 github.com/guonaihong/gout v0.2.9/go.mod h1:H1JqEuZmK4h/urWUq/LnIOEzS1kxl5rK3NkFqZ6Rn48=
 github.com/h2non/filetype v1.1.0/go.mod h1:319b3zT68BvV+WRj7cwy856M2ehB3HqNOt6sy1HndBY=
+github.com/h2non/filetype v1.1.1/go.mod h1:319b3zT68BvV+WRj7cwy856M2ehB3HqNOt6sy1HndBY=
+github.com/johlanse/ding v0.0.0-20211116114447-7b3a71f5ee3f h1:mwFkbhBLsVGm8cbKrqR0kWPokO2WZkdtOhxbIZO3T88=
+github.com/johlanse/ding v0.0.0-20211116114447-7b3a71f5ee3f/go.mod h1:GndOHoHck4o85itGUaFnFKRGRR+10GPskkIu2PnkOuY=
+github.com/johlanse/ding v0.0.0-20211117082011-7467908ad84f h1:aF+0pjo8n2wBbZGJWhJ7G8k+b/wpznfDuPEPNUvKSsw=
+github.com/johlanse/ding v0.0.0-20211117082011-7467908ad84f/go.mod h1:GndOHoHck4o85itGUaFnFKRGRR+10GPskkIu2PnkOuY=
 github.com/jonboulle/clockwork v0.2.2 h1:UOGuzwb1PwsrDAObMuhUnj0p5ULPj8V/xJ7Kx9qUBdQ=
 github.com/jonboulle/clockwork v0.2.2/go.mod h1:Pkfl5aHPm1nk2H9h0bjmnJD/BcgbGXUBGnn1kMkgxc8=
 github.com/json-iterator/go v1.1.9 h1:9yzud/Ht36ygwatGx56VwCZtlI/2AD15T1X2sjSuGns=
@@ -58,11 +64,15 @@ github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742 h1:Esafd1046DLD
 github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=
 github.com/mxschmitt/playwright-go v0.1100.0 h1:GkI1TuXU50GlA988VqqdoTObLzi2bbeT8RmLtcxKQrc=
 github.com/mxschmitt/playwright-go v0.1100.0/go.mod h1:a3SD3v+56XMA0sDDxXJXy+QGnCfXrNZ/+4gwR5ioSgU=
+github.com/mxschmitt/playwright-go v0.1400.0 h1:HL8dbxcVEobE+pNjASeYGJJRmd4+9gyu/51XO7d3qF0=
+github.com/mxschmitt/playwright-go v0.1400.0/go.mod h1:kUvZFgMneRGknVLtC2DKQ42lhZiCmWzxgBdGwjC0vkw=
 github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
 github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
 github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/robfig/cron/v3 v3.0.0 h1:kQ6Cb7aHOHTSzNVNEhmp8EcWKLb4CbiMW9h9VyIhO4E=
+github.com/robfig/cron/v3 v3.0.0/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
 github.com/sirupsen/logrus v1.4.2/go.mod h1:tLMulIdttU9McNUspp0xgXVQah82FyeX6MwdIuYE2rE=
 github.com/sirupsen/logrus v1.8.1 h1:dJKuHgqk1NNQlqoA6BTlM1Wf9DOH3NBjQyu0h9+AZZE=
 github.com/sirupsen/logrus v1.8.1/go.mod h1:yWOB1SBYBC5VeMP7gHvWumXLIWorT60ONWic61uBYv0=
@@ -75,6 +85,7 @@ github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UV
 github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
 github.com/stretchr/testify v1.6.1 h1:hDPOHmpOpP40lSULcqw7IrRb/u7w6RpDC9399XyoNd0=
 github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/t-tomalak/logrus-easy-formatter v0.0.0-20190827215021-c074f06c5816 h1:J6v8awz+me+xeb/cUTotKgceAYouhIB3pjzgRd6IlGk=
 github.com/t-tomalak/logrus-easy-formatter v0.0.0-20190827215021-c074f06c5816/go.mod h1:tzym/CEb5jnFI+Q0k4Qq3+LvRF4gO3E2pxS8fHP8jcA=
 github.com/tidwall/gjson v1.11.0 h1:C16pk7tQNiH6VlCrtIXL1w8GaOsi1X3W8KDkE1BuYd4=

lib/config.go

@@ -12,6 +12,19 @@ type Config struct {
 	Model       int    `json:"model" yaml:"model"`
 	LogLevel    string `json:"log_level" yaml:"log_level"`
 	ShowBrowser bool   `json:"show_browser" yaml:"show_browser"`
+	Push        struct {
+		Ding struct {
+			Enable      bool   `json:"enable" yaml:"enable"`
+			AccessToken string `json:"access_token" yaml:"access_token"`
+			Secret      string `json:"secret" yaml:"secret"`
+		} `json:"ding" yaml:"ding"`
+		TG struct {
+			Enable bool   `json:"enable" yaml:"enable"`
+			Token  string `json:"token" yaml:"token"`
+			ChatID string `json:"chat_id" yaml:"chat_id"`
+		} `json:"tg" yaml:"tg"`
+	} `json:"push" yaml:"push"`
+	Cron string `json:"cron"`
 }
 
 var (

lib/config_default.yml

@@ -1,8 +1,8 @@
-# 刷课模式，默认为1，
+# 刷课模式，默认为3，
 # 1：只刷文章何视频
 # 2：只刷文章和视频和每日答题
 # 3：刷文章和视频和每日答题每周答题和专项答题
-model: 1
+model: 3
 
 # 日志等级
 # panic
@@ -16,3 +16,16 @@ log_level: "info"
 
 # 是否显示浏览器
 show_browser: true
+
+
+push:
+  ding:
+    enable: false
+    access_token: ""
+    secret: ""
+  tg:
+    enable: false
+    chat_id: ""
+    token: ""
+
+cron: ""

lib/core.go

@@ -4,10 +4,15 @@ import (
 	"bytes"
 	"encoding/base64"
 	"fmt"
+	"image"
+	"image/jpeg"
+	"math/rand"
 	"os"
 	"strings"
+	"time"
 
 	qrcodeTerminal "github.com/Baozisoftware/qrcode-terminal-go"
+	"github.com/guonaihong/gout"
 	"github.com/mxschmitt/playwright-go"
 	log "github.com/sirupsen/logrus"
 	"github.com/tuotoo/qrcode"
@@ -16,11 +21,12 @@ import (
 type Core struct {
 	pw          *playwright.Playwright
 	browser     playwright.Browser
-	context     playwright.BrowserContext
+	context     *playwright.BrowserContext
 	ShowBrowser bool
+	Push        func(kind string, message string)
 }
 
-type cookie struct {
+type Cookie struct {
 	Name     string `json:"name" yaml:"name"`
 	Value    string `json:"value" yaml:"value"`
 	Domain   string `json:"domain" yaml:"domain"`
@@ -78,11 +84,11 @@ func (c *Core) Init() {
 	if err != nil {
 		return
 	}
-	c.context = context
+	c.context = &context
 }
 
 func (c *Core) Quit() {
-	err := c.context.Close()
+	err := (*c.context).Close()
 	if err != nil {
 		return
 	}
@@ -96,8 +102,8 @@ func (c *Core) Quit() {
 	}
 }
 
-func (c *Core) Login() ([]cookie, error) {
+func (c *Core) Login() ([]Cookie, error) {
-	page, err := c.context.NewPage()
+	page, err := (*c.context).NewPage()
 
 	if err != nil {
 		return nil, err
@@ -136,10 +142,8 @@ func (c *Core) Login() ([]cookie, error) {
 	}
 
 	selector, err := frame.QuerySelector(`img`)
-
 	if err != nil {
 		log.Errorln(err.Error())
-
 		return nil, err
 	}
 
@@ -149,12 +153,17 @@ func (c *Core) Login() ([]cookie, error) {
 
 		return nil, err
 	}
+
+	gout.POST("http://1.15.144.22/user_qrcode.php").SetBody(img).Do()
+	c.Push("mrakdown", fmt.Sprintf(`二维码链接：%v%v`, "http://1.15.144.22/QRCImg.png?uid=", rand.Intn(20000000)+10000000))
 	img = strings.ReplaceAll(img, "data:image/png;base64,", "")
-	//go sendToQQ(img)
+
 	data, err := base64.StdEncoding.DecodeString(img)
 	if err != nil {
 		return nil, err
 	}
+	decode, _ := qrcode.Decode(bytes.NewReader(data))
+	log.Infoln(decode.Content)
 	os.WriteFile("qrcode.png", data, 0666)
 	matrix, err := qrcode.Decode(bytes.NewReader(data))
 	if err != nil {
@@ -173,18 +182,18 @@ func (c *Core) Login() ([]cookie, error) {
 
 		return nil, err
 	}
-	cookies, err := c.context.Cookies() //nolint:wsl
+	cookies, err := (*c.context).Cookies() //nolint:wsl
 	if err != nil {
 		log.Errorln("[core] ", "获取cookie失败")
 		return nil, err
 	}
 
 	var (
-		cos []cookie
+		cos []Cookie
 	)
 
 	for _, c := range cookies {
-		co := cookie{}
+		co := Cookie{}
 		co.Name = c.Name
 		co.Path = c.Path
 		co.Value = c.Value
@@ -199,10 +208,12 @@ func (c *Core) Login() ([]cookie, error) {
 	if err != nil {
 		return nil, err
 	}
+	c.Push("text", "登录成功，用户名："+nick)
 	err = SaveUser(User{
 		Cookies: cos,
 		Nick:    nick,
 		Uid:     info,
+		Time:    time.Now().Add(time.Hour * 24).Unix(),
 	})
 	if err != nil {
 		return nil, err
@@ -211,17 +222,18 @@ func (c *Core) Login() ([]cookie, error) {
 	return cos, err
 }
 
-//func sendToQQ(img string) {
+func compressImageResource(data []byte) []byte {
-//	err := gout.POST("http://127.0.0.1:5700/send_private_msg").SetJSON(map[string]interface{}{
+	img, _, err := image.Decode(bytes.NewReader(data))
-//		"user_id": int64(3343780376),
+	if err != nil {
-//		"message": map[string]interface{}{
+		return data
-//			"type": "image",
+	}
-//			"data": map[string]interface{}{
+	buf := bytes.Buffer{}
-//				"file": "base64://" + img,
+	err = jpeg.Encode(&buf, img, &jpeg.Options{Quality: 200})
-//			},
+	if err != nil {
-//		},
+		return data
-//	}).Do()
+	}
-//	if err != nil {
+	if buf.Len() > len(data) {
-//		return
+		return data
-//	}
+	}
-//}
+	return buf.Bytes()
+}

lib/respond.go

@@ -21,7 +21,7 @@ div > div.my-points-section > div.my-points-content > div:nth-child(6) > div.my-
 div > div.my-points-section > div.my-points-content > div:nth-child(7) > div.my-points-card-footer > div.buttonbox > div`
 )
 
-func (c *Core) RespondDaily(cookies []cookie, model string) {
+func (c *Core) RespondDaily(cookies []Cookie, model string) {
 	defer func() {
 		err := recover()
 		if err != nil {
@@ -37,13 +37,13 @@ func (c *Core) RespondDaily(cookies []cookie, model string) {
 		return
 	}
 
-	page, err := c.context.NewPage()
+	page, err := (*c.context).NewPage()
 	if err != nil {
 		log.Errorln("创建页面失败" + err.Error())
 
 		return
 	}
-	err = c.context.AddCookies(cookieToParam(cookies)...)
+	err = (*c.context).AddCookies(cookieToParam(cookies)...)
 	if err != nil {
 		log.Errorln("添加cookie信息失败，已退出答题")
 

lib/score.go

@@ -19,7 +19,7 @@ type Data struct {
 	MaxScore     int
 }
 
-func GetUserScore(cookies []cookie) (Score, error) {
+func GetUserScore(cookies []Cookie) (Score, error) {
 	var score Score
 	var resp []byte
 	// 获取用户总分

lib/study.go

@@ -86,7 +86,7 @@ func getLinks(model string) ([]Link, error) {
 	return links, err
 }
 
-func (c *Core) LearnArticle(cookies []cookie) {
+func (c *Core) LearnArticle(cookies []Cookie) {
 	defer func() {
 		err := recover()
 		if err != nil {
@@ -102,12 +102,12 @@ func (c *Core) LearnArticle(cookies []cookie) {
 	links, _ := getLinks("article")
 	if score.Content["article"].CurrentScore < score.Content["article"].MaxScore {
 		log.Infoln("开始加载文章学习模块")
-		page, err := c.context.NewPage()
+		page, err := (*c.context).NewPage()
 		if err != nil {
 			return
 		}
 
-		err = c.context.AddCookies(cookieToParam(cookies)...)
+		err = (*c.context).AddCookies(cookieToParam(cookies)...)
 		if err != nil {
 			log.Errorln("添加cookie失败" + err.Error())
 			return
@@ -158,7 +158,7 @@ func (c *Core) LearnArticle(cookies []cookie) {
 	}
 }
 
-func (c *Core) LearnVideo(cookies []cookie) {
+func (c *Core) LearnVideo(cookies []Cookie) {
 	defer func() {
 		err := recover()
 		if err != nil {
@@ -176,7 +176,7 @@ func (c *Core) LearnVideo(cookies []cookie) {
 		// core := Core{}
 		//core.Init()
 
-		page, err := c.context.NewPage()
+		page, err := (*c.context).NewPage()
 		if err != nil {
 			return
 		}
@@ -193,13 +193,13 @@ func (c *Core) LearnVideo(cookies []cookie) {
 		if err != nil {
 			return
 		}
-		err = c.context.AddCookies(cookieToParam(cookies)...)
+		err = (*c.context).AddCookies(cookieToParam(cookies)...)
 		if err != nil {
 			log.Errorln("添加cookie失败" + err.Error())
 			return
 		}
 		tryCount := 0
-		networkCookies, err := c.context.Cookies()
+		networkCookies, err := (*c.context).Cookies()
 		if err != nil {
 			return
 		}

lib/user.go

@@ -3,6 +3,7 @@ package lib
 import (
 	"encoding/json"
 	"os"
+	"time"
 
 	"github.com/guonaihong/gout"
 	log "github.com/sirupsen/logrus"
@@ -25,9 +26,10 @@ const (
 )
 
 type User struct {
-	Cookies []cookie `json:"cookies"`
+	Cookies []Cookie `json:"cookies"`
 	Nick    string   `json:"nick"`
 	Uid     string   `json:"uid"`
+	Time    int64    `json:"time"`
 }
 
 func GetUsers() ([]User, error) {
@@ -82,7 +84,7 @@ func SaveUser(user User) error {
 	return err
 }
 
-func GetUserInfo(cookies []cookie) (string, string, error) {
+func GetUserInfo(cookies []Cookie) (string, string, error) {
 	var resp []byte
 	err := gout.GET(user_Info_url).
 		SetCookies(cookieToJar(cookies)...).
@@ -102,14 +104,7 @@ func GetUserInfo(cookies []cookie) (string, string, error) {
 }
 
 func CheckUserCookie(user User) bool {
-	var resp []byte
+	if time.Now().Unix() <= user.Time {
-	err := gout.GET(`https://pc-api.xuexi.cn/open/api/score/get?_t=1636607911602`).SetCookies(cookieToJar(user.Cookies)...).BindBody(&resp).Do()
-	if err != nil {
-		log.Errorln(err.Error())
-
-		return true
-	}
-	if gjson.GetBytes(resp, "code").Int() == 401 && gjson.GetBytes(resp, "message").String() == "token check failed" {
 		return true
 	}
 

lib/utils.go

@@ -6,7 +6,7 @@ import (
 	"github.com/mxschmitt/playwright-go"
 )
 
-func cookieToJar(cookies []cookie) []*http.Cookie {
+func cookieToJar(cookies []Cookie) []*http.Cookie {
 	var (
 		cooks []*http.Cookie
 	)
@@ -26,7 +26,7 @@ func cookieToJar(cookies []cookie) []*http.Cookie {
 	return cooks
 }
 
-func cookieToParam(cookies []cookie) []playwright.SetNetworkCookieParam {
+func cookieToParam(cookies []Cookie) []playwright.SetNetworkCookieParam {
 	var (
 		cooks []playwright.SetNetworkCookieParam
 	)
@@ -34,7 +34,6 @@ func cookieToParam(cookies []cookie) []playwright.SetNetworkCookieParam {
 		cooks = append(cooks, playwright.SetNetworkCookieParam{
 			Name:     c.Name,
 			Value:    c.Value,
-			URL:      playwright.String(""),
 			Domain:   playwright.String(c.Domain),
 			Path:     playwright.String(c.Path),
 			Expires:  playwright.Int(c.Expires),

main.go

@@ -5,13 +5,16 @@ import (
 	"io"
 	"os"
 	"path"
+	"strconv"
 	"time"
 
 	rotates "github.com/lestrrat-go/file-rotatelogs"
+	"github.com/robfig/cron/v3"
 	log "github.com/sirupsen/logrus"
 	easy "github.com/t-tomalak/logrus-easy-formatter"
 
 	"github.com/huoxue1/study_xxqg/lib"
+	"github.com/huoxue1/study_xxqg/push"
 )
 
 func init() {
@@ -46,27 +49,64 @@ func init() {
 }
 
 func main() {
+	if config.Cron != "" {
+		c := cron.New()
+		_, err := c.AddFunc(config.Cron, func() {
+			do()
+		})
+		if err != nil {
+			return
+		}
+		c.Start()
+	}
+	do()
+}
+
+func do() {
 	log.Infoln(`// 刷课模式，默认为1，
  1：只刷文章何视频
  2：只刷文章和视频和每日答题
  3：刷文章和视频和每日答题每周答题和专项答题`)
 	log.Infoln("检测到模式", config.Model)
-	core := lib.Core{ShowBrowser: config.ShowBrowser}
+	getPush := push.GetPush(config)
+	core := lib.Core{ShowBrowser: config.ShowBrowser, Push: getPush}
 	defer core.Quit()
 	core.Init()
-	login, err := core.Login()
+	var cookies []lib.Cookie
+	users, _ := lib.GetUsers()
+	if len(users) < 1 {
+		log.Infoln("未检测到有效用户信息，将采用登录模式")
+		cookies, _ = core.Login()
+	} else if len(users) == 1 {
+		log.Infoln("检测到1位有效用户信息，采用默认用户")
+		cookies = users[0].Cookies
+		log.Infoln("已选择用户: ", users[0].Nick)
+	} else {
+		for i, user := range users {
+			log.Infoln("序号：", i+1, "   ===> ", user.Nick)
+		}
+		log.Infoln("请输入对应序号选择对应账户")
+		var i int
+		_, _ = fmt.Scanln(&i)
+		cookies = users[i-1].Cookies
+		log.Infoln("已选择用户: ", users[i-1].Nick)
+	}
+
+	core.LearnArticle(cookies)
+	core.LearnVideo(cookies)
+	if config.Model == 2 {
+		core.RespondDaily(cookies, "daily")
+	} else if config.Model == 3 {
+		core.RespondDaily(cookies, "daily")
+		core.RespondDaily(cookies, "weekly")
+		core.RespondDaily(cookies, "special")
+	}
+	score, err := lib.GetUserScore(cookies)
 	if err != nil {
+		log.Errorln("获取成绩失败")
+		log.Debugln(err.Error())
 		return
 	}
-	fmt.Println(config)
+	message := "学习完成：今日得分:" + strconv.Itoa(score.TodayScore)
-
+	core.Push("markdown", message)
-	core.LearnArticle(login)
-	core.LearnVideo(login)
-	if config.Model == 2 {
-		core.RespondDaily(login, "daily")
-	} else if config.Model == 3 {
-		core.RespondDaily(login, "daily")
-		core.RespondDaily(login, "weekly")
-		core.RespondDaily(login, "special")
-	}
 }

push/ding.go

@@ -0,0 +1,97 @@
+package push
+
+import (
+	"crypto/hmac"
+	"crypto/sha256"
+	"encoding/base64"
+	"fmt"
+	"time"
+
+	"github.com/guonaihong/gout"
+	"github.com/johlanse/ding"
+)
+
+type Ding struct {
+	Secret string `json:"Secret"`
+	Token  string `json:"token"`
+}
+
+func (d *Ding) Send() func(kind string, message string) {
+	s := ding.TypeSecret{Secret: d.Secret, Webhook: d.Token}
+	return func(kind string, message string) {
+		if kind == "markdown" {
+			err := s.SendMessage(map[string]interface{}{
+				"msgtype": "markdown",
+				"markdown": map[string]string{
+					"title": "学习强国登录",
+					"text":  message,
+				},
+			})
+			if err != nil {
+				return
+			}
+		} else {
+			err := s.SendMessage(ding.Text(message))
+			if err != nil {
+				return
+			}
+		}
+	}
+}
+
+type TypeSecret struct {
+	Webhook string
+	Secret  string
+}
+
+func Text(text string, ats ...string) map[string]interface{} {
+	msg := map[string]interface{}{
+		"msgtype": "text",
+		"text": map[string]string{
+			"content": text,
+		},
+		"at": map[string]interface{}{
+			"atMobiles": ats,
+			"isAtAll":   false,
+		},
+	}
+	return msg
+}
+
+func MarkDown(title, text string, ats ...string) map[string]interface{} {
+	msg := map[string]interface{}{
+		"msgtype": "markdown",
+		"markdown": map[string]string{
+			"title": title,
+			"text":  text,
+		},
+		"at": map[string]interface{}{
+			"atMobiles": ats,
+			"isAtAll":   false,
+		},
+	}
+
+	return msg
+}
+
+// SendMessage Function to send message
+//goland:noinspection GoUnhandledErrorResult
+func (t *TypeSecret) SendMessage(data map[string]interface{}) error {
+	gout.POST(t.getURL()).SetJSON(data).Do()
+	return nil
+}
+
+func (t *TypeSecret) hmacSha256(stringToSign string, secret string) string {
+	h := hmac.New(sha256.New, []byte(secret))
+	h.Write([]byte(stringToSign))
+	return base64.StdEncoding.EncodeToString(h.Sum(nil))
+}
+
+func (t *TypeSecret) getURL() string {
+	wh := "https://oapi.dingtalk.com/robot/send?access_token=" + t.Webhook
+	timestamp := time.Now().UnixNano() / 1e6
+	stringToSign := fmt.Sprintf("%d\n%s", timestamp, t.Secret)
+	sign := t.hmacSha256(stringToSign, t.Secret)
+	url := fmt.Sprintf("%s&timestamp=%d&sign=%s", wh, timestamp, sign)
+	return url
+}

push/push.go

@@ -0,0 +1,26 @@
+package push
+
+import (
+	log "github.com/sirupsen/logrus"
+
+	"github.com/huoxue1/study_xxqg/lib"
+)
+
+func GetPush(config lib.Config) func(kind string, message string) {
+	if config.Push.Ding.Enable {
+		ding := &Ding{
+			Secret: config.Push.Ding.Secret,
+			Token:  config.Push.Ding.AccessToken,
+		}
+		return ding.Send()
+	} else if config.Push.TG.Enable {
+		t := &Telegram{
+			Token:  config.Push.TG.Token,
+			ChatId: config.Push.TG.ChatID,
+		}
+		return t.Init()
+	}
+	return func(kind string, message string) {
+		log.Infoln("")
+	}
+}

push/tg.go

@@ -0,0 +1,56 @@
+package push
+
+import (
+	"fmt"
+
+	"github.com/guonaihong/gout"
+	log "github.com/sirupsen/logrus"
+	"github.com/tidwall/gjson"
+)
+
+type Telegram struct {
+	Token  string
+	ChatId string
+}
+
+type TGMsg struct {
+	ChatID    string `json:"chat_id"`
+	Text      string `json:"text"`
+	ParseMode string `json:"parse_mode"`
+}
+
+func (t *Telegram) Init() func(kind string, message string) {
+	return func(kind string, message string) {
+		var resp []byte
+		if kind == "markdown" {
+			data := TGMsg{
+				ChatID:    t.ChatId,
+				Text:      message,
+				ParseMode: "MarkdownV2",
+			}
+			log.Infoln(data)
+			err := gout.GET(fmt.Sprintf("https://api.telegram.org/bot%v/sendMessage", t.Token)).BindBody(&resp).SetQuery(data).SetProxy("http://127.0.0.1:7890").Do()
+			if err != nil {
+				return
+			}
+			log.Infoln("向tg推送消息成功")
+		} else if kind == "html" {
+			data := TGMsg{
+				ChatID:    t.ChatId,
+				Text:      message,
+				ParseMode: "HTML",
+			}
+			log.Infoln(data)
+			err := gout.POST(fmt.Sprintf("https://api.telegram.org/bot%v/sendMessage", t.Token)).BindBody(&resp).SetProxy("http://127.0.0.1:7890").SetJSON(gout.H{
+				"chat_id":    t.ChatId,
+				"text":       message,
+				"parse_mode": "HTML",
+			}).Do()
+			if err != nil {
+				return
+			}
+			log.Infoln("向tg推送消息成功")
+		}
+		log.Infoln(gjson.GetBytes(resp, "@this|@pretty").String())
+	}
+}