423 lines
12 KiB
Go
423 lines
12 KiB
Go
package lib
|
||
|
||
import (
|
||
"encoding/json"
|
||
"errors"
|
||
"fmt"
|
||
"math/rand"
|
||
"strconv"
|
||
"time"
|
||
|
||
"github.com/playwright-community/playwright-go"
|
||
log "github.com/sirupsen/logrus"
|
||
|
||
"github.com/sjkhsl/study_xxqg/model"
|
||
"github.com/sjkhsl/study_xxqg/utils"
|
||
)
|
||
|
||
var (
|
||
article_url_list = []string{
|
||
"https://www.xuexi.cn/lgdata/35il6fpn0ohq.json",
|
||
"https://www.xuexi.cn/lgdata/45a3hac2bf1j.json",
|
||
"https://www.xuexi.cn/lgdata/1ajhkle8l72.json",
|
||
"https://www.xuexi.cn/lgdata/1ahjpjgb4n3.json",
|
||
"https://www.xuexi.cn/lgdata/1je1objnh73.json",
|
||
"https://www.xuexi.cn/lgdata/1kvrj9vvv73.json",
|
||
"https://www.xuexi.cn/lgdata/17qonfb74n3.json",
|
||
"https://www.xuexi.cn/lgdata/1i30sdhg0n3.json"}
|
||
|
||
video_url_list = []string{
|
||
"https://www.xuexi.cn/lgdata/3j2u3cttsii9.json",
|
||
"https://www.xuexi.cn/lgdata/1novbsbi47k.json",
|
||
"https://www.xuexi.cn/lgdata/31c9ca1tgfqb.json",
|
||
"https://www.xuexi.cn/lgdata/1oajo2vt47l.json",
|
||
"https://www.xuexi.cn/lgdata/18rkaul9h7l.json",
|
||
"https://www.xuexi.cn/lgdata/2qfjjjrprmdh.json",
|
||
"https://www.xuexi.cn/lgdata/3o3ufqgl8rsn.json",
|
||
"https://www.xuexi.cn/lgdata/525pi8vcj24p.json",
|
||
"https://www.xuexi.cn/lgdata/1742g60067k.json"}
|
||
|
||
yp_url_list = []string{
|
||
"https://www.xuexi.cn/lgdata/1ode6kjlu7m.json",
|
||
"https://www.xuexi.cn/lgdata/1ggb81u8f7m.json",
|
||
"https://www.xuexi.cn/lgdata/139993ri8nm.json",
|
||
"https://www.xuexi.cn/lgdata/u07dubuq7m.json",
|
||
"https://www.xuexi.cn/lgdata/spisr390nm.json",
|
||
"https://www.xuexi.cn/lgdata/1elt18mm57m.json"}
|
||
)
|
||
|
||
type Link struct {
|
||
Editor string `json:"editor"`
|
||
PublishTime string `json:"publishTime"`
|
||
ItemType string `json:"itemType"`
|
||
Author string `json:"author"`
|
||
CrossTime int `json:"crossTime"`
|
||
Source string `json:"source"`
|
||
NameB string `json:"nameB"`
|
||
Title string `json:"title"`
|
||
Type string `json:"type"`
|
||
Url string `json:"url"`
|
||
ShowSource string `json:"showSource"`
|
||
ItemId string `json:"itemId"`
|
||
ThumbImage string `json:"thumbImage"`
|
||
AuditTime string `json:"auditTime"`
|
||
ChannelNames []string `json:"channelNames"`
|
||
Producer string `json:"producer"`
|
||
ChannelIds []string `json:"channelIds"`
|
||
DataValid bool `json:"dataValid"`
|
||
}
|
||
|
||
// 获取学习链接列表
|
||
func getLinks(model string) ([]Link, error) {
|
||
UID := rand.Intn(20000000) + 10000000
|
||
learnUrl := ""
|
||
if model == "article" {
|
||
learnUrl = article_url_list[rand.Intn(7)]
|
||
} else if model == "video" {
|
||
learnUrl = video_url_list[rand.Intn(7)]
|
||
} else if model == "yp" {
|
||
learnUrl = yp_url_list[rand.Intn(7)]
|
||
} else {
|
||
return nil, errors.New("model选择出现错误")
|
||
}
|
||
var (
|
||
resp []byte
|
||
)
|
||
|
||
response, err := utils.GetClient().R().SetQueryParam("_st", strconv.Itoa(UID)).Get(learnUrl)
|
||
if err != nil {
|
||
log.Errorln("请求链接列表出现错误！" + err.Error())
|
||
return nil, err
|
||
}
|
||
resp = response.Bytes()
|
||
|
||
var links []Link
|
||
err = json.Unmarshal(resp, &links)
|
||
if err != nil {
|
||
log.Errorln("解析列表出现错误" + err.Error())
|
||
return nil, err
|
||
}
|
||
return links, err
|
||
}
|
||
|
||
// 文章学习
|
||
func (c *Core) LearnArticle(user *model.User) {
|
||
defer func() {
|
||
err := recover()
|
||
if err != nil {
|
||
log.Errorln("文章学习模块异常结束")
|
||
log.Errorln(err)
|
||
}
|
||
}()
|
||
if c.IsQuit() {
|
||
return
|
||
}
|
||
|
||
score, err := GetUserScore(user.ToCookies())
|
||
if err != nil {
|
||
log.Errorln(err.Error())
|
||
return
|
||
}
|
||
links, _ := getLinks("article")
|
||
if score.Content["article"].CurrentScore < score.Content["article"].MaxScore {
|
||
log.Infoln("开始加载文章学习模块")
|
||
|
||
context, err := c.browser.NewContext(playwright.BrowserNewContextOptions{
|
||
Viewport: &playwright.BrowserNewContextOptionsViewport{
|
||
Width: playwright.Int(1920),
|
||
Height: playwright.Int(1080),
|
||
}})
|
||
_ = context.AddInitScript(playwright.BrowserContextAddInitScriptOptions{
|
||
Script: playwright.String("Object.defineProperties(navigator, {webdriver:{get:()=>undefined}});")})
|
||
if err != nil {
|
||
log.Errorln("创建实例对象错误" + err.Error())
|
||
return
|
||
}
|
||
|
||
defer func(context playwright.BrowserContext) {
|
||
err := context.Close()
|
||
if err != nil {
|
||
log.Errorln("错误的关闭了实例对象" + err.Error())
|
||
}
|
||
}(context)
|
||
|
||
page, err := context.NewPage()
|
||
if err != nil {
|
||
return
|
||
}
|
||
defer func() {
|
||
err := page.Close()
|
||
if err != nil {
|
||
log.Errorln("关闭页面失败")
|
||
return
|
||
}
|
||
}()
|
||
|
||
err = context.AddCookies(user.ToBrowserCookies()...)
|
||
if err != nil {
|
||
log.Errorln("添加cookie失败" + err.Error())
|
||
return
|
||
}
|
||
|
||
tryCount := 0
|
||
|
||
for {
|
||
if tryCount < 20 {
|
||
PrintScore(score)
|
||
n := rand.Intn(len(links))
|
||
_, err := page.Goto(links[n].Url, playwright.PageGotoOptions{
|
||
Referer: playwright.String(links[rand.Intn(len(links))].Url),
|
||
Timeout: playwright.Float(10000),
|
||
WaitUntil: playwright.WaitUntilStateDomcontentloaded,
|
||
})
|
||
if err != nil {
|
||
log.Errorln("页面跳转失败")
|
||
}
|
||
log.Infoln("正在学习文章：" + links[n].Title)
|
||
c.Push(user.PushId, "text", "正在学习文章："+links[n].Title)
|
||
log.Infoln("文章发布时间：" + links[n].PublishTime)
|
||
log.Infoln("文章学习链接：" + links[n].Url)
|
||
learnTime := 60 + rand.Intn(15) + 3
|
||
for i := 0; i < learnTime; i++ {
|
||
if c.IsQuit() {
|
||
return
|
||
}
|
||
fmt.Printf("\r[%v] [INFO]: 正在进行阅读学习中，剩余%d篇，本篇剩余时间%d秒", time.Now().Format("2006-01-02 15:04:05"), score.Content["article"].MaxScore-score.Content["article"].CurrentScore, learnTime-i)
|
||
|
||
if rand.Float32() > 0.5 {
|
||
go func() {
|
||
_, err = page.Evaluate(fmt.Sprintf(`let h = document.body.scrollHeight/120*%d;document.documentElement.scrollTop=h;`, i))
|
||
if err != nil {
|
||
log.Errorln("文章滑动失败")
|
||
}
|
||
}()
|
||
}
|
||
time.Sleep(1 * time.Second)
|
||
}
|
||
fmt.Println()
|
||
score, _ = GetUserScore(user.ToCookies())
|
||
if score.Content["article"].CurrentScore >= score.Content["article"].MaxScore {
|
||
log.Infoln("检测到本次阅读学习分数已满，退出学习")
|
||
break
|
||
}
|
||
|
||
tryCount++
|
||
} else {
|
||
log.Errorln("阅读学习出现异常，稍后可重新学习")
|
||
return
|
||
}
|
||
}
|
||
} else {
|
||
log.Infoln("检测到文章学习已经完成")
|
||
}
|
||
}
|
||
|
||
// 视频学习
|
||
func (c *Core) LearnVideo(user *model.User) {
|
||
defer func() {
|
||
err := recover()
|
||
if err != nil {
|
||
log.Errorln("视频学习模块异常结束")
|
||
log.Errorln(err)
|
||
}
|
||
}()
|
||
if c.IsQuit() {
|
||
return
|
||
}
|
||
score, err := GetUserScore(user.ToCookies())
|
||
if err != nil {
|
||
log.Errorln(err.Error())
|
||
return
|
||
}
|
||
links, _ := getLinks("video")
|
||
if !(score.Content["video"].CurrentScore >= score.Content["video"].MaxScore && score.Content["video_time"].CurrentScore >= score.Content["video_time"].MaxScore) {
|
||
log.Infoln("开始加载视频学习模块")
|
||
// core := Core{}
|
||
// core.Init()
|
||
|
||
context, err := c.browser.NewContext(playwright.BrowserNewContextOptions{
|
||
Viewport: &playwright.BrowserNewContextOptionsViewport{
|
||
Width: playwright.Int(1920),
|
||
Height: playwright.Int(1080),
|
||
}})
|
||
_ = context.AddInitScript(playwright.BrowserContextAddInitScriptOptions{
|
||
Script: playwright.String("Object.defineProperties(navigator, {webdriver:{get:()=>undefined}});")})
|
||
if err != nil {
|
||
log.Errorln("创建实例对象错误" + err.Error())
|
||
return
|
||
}
|
||
defer func(context playwright.BrowserContext) {
|
||
err := context.Close()
|
||
if err != nil {
|
||
log.Errorln("错误的关闭了实例对象" + err.Error())
|
||
}
|
||
}(context)
|
||
|
||
page, err := context.NewPage()
|
||
if err != nil {
|
||
return
|
||
}
|
||
defer func() {
|
||
page.Close()
|
||
}()
|
||
|
||
err = context.AddCookies(user.ToBrowserCookies()...)
|
||
if err != nil {
|
||
log.Errorln("添加cookie失败" + err.Error())
|
||
return
|
||
}
|
||
tryCount := 0
|
||
for {
|
||
if tryCount < 20 {
|
||
PrintScore(score)
|
||
n := rand.Intn(len(links))
|
||
_, err := page.Goto(links[n].Url, playwright.PageGotoOptions{
|
||
Referer: playwright.String(links[rand.Intn(len(links))].Url),
|
||
Timeout: playwright.Float(10000),
|
||
WaitUntil: playwright.WaitUntilStateDomcontentloaded,
|
||
})
|
||
if err != nil {
|
||
log.Errorln("页面跳转失败")
|
||
}
|
||
log.Infoln("正在观看视频：" + links[n].Title)
|
||
c.Push(user.PushId, "text", "正在观看视频："+links[n].Title)
|
||
log.Infoln("视频发布时间：" + links[n].PublishTime)
|
||
log.Infoln("视频学习链接：" + links[n].Url)
|
||
learnTime := 60 + rand.Intn(10)
|
||
for i := 0; i < learnTime; i++ {
|
||
if c.IsQuit() {
|
||
return
|
||
}
|
||
fmt.Printf("\r[%v] [INFO]: 正在进行视频学习中，剩余%d个，当前剩余时间%d秒", time.Now().Format("2006-01-02 15:04:05"), score.Content["video"].MaxScore-score.Content["video"].CurrentScore, learnTime-i)
|
||
|
||
if rand.Float32() > 0.5 {
|
||
go func() {
|
||
_, err := page.Evaluate(fmt.Sprintf(`let h = document.body.scrollHeight/120*%d;document.documentElement.scrollTop=h;`, i))
|
||
if err != nil {
|
||
log.Errorln("视频滑动失败")
|
||
}
|
||
}()
|
||
}
|
||
time.Sleep(1 * time.Second)
|
||
}
|
||
fmt.Println()
|
||
score, _ = GetUserScore(user.ToCookies())
|
||
if score.Content["video"].CurrentScore >= score.Content["video"].MaxScore && score.Content["video_time"].CurrentScore >= score.Content["video_time"].MaxScore {
|
||
log.Infoln("检测到本次视频学习分数已满，退出学习")
|
||
break
|
||
}
|
||
|
||
tryCount++
|
||
} else {
|
||
log.Errorln("视频学习出现异常，稍后可重新学习")
|
||
return
|
||
}
|
||
}
|
||
} else {
|
||
log.Infoln("检测到视频学习已经完成")
|
||
}
|
||
}
|
||
|
||
// 音频学习
|
||
func (c *Core) RadioStation(user *model.User) {
|
||
defer func() {
|
||
err := recover()
|
||
if err != nil {
|
||
log.Errorln("电台学习模块异常结束")
|
||
log.Errorln(err)
|
||
}
|
||
}()
|
||
if c.IsQuit() {
|
||
return
|
||
}
|
||
score, err := GetUserScore(user.ToCookies())
|
||
if err != nil {
|
||
log.Errorln(err.Error())
|
||
return
|
||
}
|
||
links, _ := getLinks("yp")
|
||
if !(score.Content["video"].CurrentScore >= score.Content["video"].MaxScore && score.Content["video_time"].CurrentScore >= score.Content["video_time"].MaxScore) {
|
||
log.Infoln("开始加载音频学习模块")
|
||
context, err := c.browser.NewContext(playwright.BrowserNewContextOptions{
|
||
Viewport: &playwright.BrowserNewContextOptionsViewport{
|
||
Width: playwright.Int(1920),
|
||
Height: playwright.Int(1080),
|
||
}})
|
||
_ = context.AddInitScript(playwright.BrowserContextAddInitScriptOptions{
|
||
Script: playwright.String("Object.defineProperties(navigator, {webdriver:{get:()=>undefined}});")})
|
||
if err != nil {
|
||
log.Errorln("创建实例对象错误" + err.Error())
|
||
return
|
||
}
|
||
defer func(context playwright.BrowserContext) {
|
||
err := context.Close()
|
||
if err != nil {
|
||
log.Errorln("错误的关闭了实例对象" + err.Error())
|
||
}
|
||
}(context)
|
||
|
||
page, err := context.NewPage()
|
||
if err != nil {
|
||
return
|
||
}
|
||
defer func() {
|
||
page.Close()
|
||
}()
|
||
|
||
err = context.AddCookies(user.ToBrowserCookies()...)
|
||
if err != nil {
|
||
log.Errorln("添加cookie失败" + err.Error())
|
||
return
|
||
}
|
||
tryCount := 0
|
||
for {
|
||
if tryCount < 20 {
|
||
PrintScore(score)
|
||
n := rand.Intn(len(links))
|
||
_, err := page.Goto(links[n].Url, playwright.PageGotoOptions{
|
||
Referer: playwright.String(links[rand.Intn(len(links))].Url),
|
||
Timeout: playwright.Float(10000),
|
||
WaitUntil: playwright.WaitUntilStateDomcontentloaded,
|
||
})
|
||
if err != nil {
|
||
log.Errorln("页面跳转失败")
|
||
}
|
||
log.Infoln("正在收听：" + links[n].Title)
|
||
c.Push(user.PushId, "text", "正在收听："+links[n].Title)
|
||
log.Infoln("音频发布时间：" + links[n].PublishTime)
|
||
log.Infoln("音频学习链接：" + links[n].Url)
|
||
learnTime := 60 + rand.Intn(10)
|
||
for i := 0; i < learnTime; i++ {
|
||
if c.IsQuit() {
|
||
return
|
||
}
|
||
fmt.Printf("\r[%v] [INFO]: 正在进行音频学习中，剩余%d个，当前剩余时间%d秒", time.Now().Format("2006-01-02 15:04:05"), score.Content["video"].MaxScore-score.Content["video"].CurrentScore, learnTime-i)
|
||
|
||
if rand.Float32() > 0.5 {
|
||
go func() {
|
||
_, err := page.Evaluate(fmt.Sprintf(`let h = document.body.scrollHeight/120*%d;document.documentElement.scrollTop=h;`, i))
|
||
if err != nil {
|
||
log.Errorln("视频滑动失败")
|
||
}
|
||
}()
|
||
}
|
||
time.Sleep(1 * time.Second)
|
||
}
|
||
fmt.Println()
|
||
score, _ = GetUserScore(user.ToCookies())
|
||
if score.Content["video"].CurrentScore >= score.Content["video"].MaxScore && score.Content["video_time"].CurrentScore >= score.Content["video_time"].MaxScore {
|
||
log.Infoln("检测到本次音频学习分数已满，退出学习")
|
||
break
|
||
}
|
||
|
||
tryCount++
|
||
} else {
|
||
log.Errorln("音频学习出现异常，稍后可重新学习")
|
||
return
|
||
}
|
||
}
|
||
} else {
|
||
log.Infoln("检测到音频学习已经完成")
|
||
}
|
||
}
|