mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-18 20:50:12 +08:00
118 lines
3.2 KiB
Go
118 lines
3.2 KiB
Go
// Package jandan 煎蛋网无聊图
|
|
package jandan
|
|
|
|
import (
|
|
"fmt"
|
|
"hash/crc64"
|
|
"regexp"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/FloatTech/floatbox/binary"
|
|
fcext "github.com/FloatTech/floatbox/ctxext"
|
|
sql "github.com/FloatTech/sqlite"
|
|
ctrl "github.com/FloatTech/zbpctrl"
|
|
"github.com/FloatTech/zbputils/control"
|
|
"github.com/antchfx/htmlquery"
|
|
"github.com/sirupsen/logrus"
|
|
zero "github.com/wdvxdr1123/ZeroBot"
|
|
"github.com/wdvxdr1123/ZeroBot/message"
|
|
)
|
|
|
|
const (
|
|
api = "http://jandan.net/pic"
|
|
)
|
|
|
|
func init() {
|
|
engine := control.AutoRegister(&ctrl.Options[*zero.Ctx]{
|
|
DisableOnDefault: false,
|
|
Brief: "煎蛋网无聊图",
|
|
Help: "- 来份[屌|弔|吊]图\n- 更新[屌|弔|吊]图\n",
|
|
PublicDataFolder: "Jandan",
|
|
})
|
|
|
|
getdb := fcext.DoOnceOnSuccess(func(ctx *zero.Ctx) bool {
|
|
db = sql.New(engine.DataFolder() + "pics.db")
|
|
_, _ = engine.GetLazyData("pics.db", false)
|
|
err := db.Open(time.Hour)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return false
|
|
}
|
|
err = db.Create("picture", &picture{})
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return false
|
|
}
|
|
n, err := db.Count("picture")
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return false
|
|
}
|
|
logrus.Printf("[jandan]读取%d张图片", n)
|
|
return true
|
|
})
|
|
|
|
engine.OnRegex(`来份[屌|弔|吊]图`, getdb).SetBlock(true).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
u, err := getRandomPicture()
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
ctx.SendChain(message.Image(u))
|
|
})
|
|
|
|
engine.OnRegex(`更新[屌|弔|吊]图`, zero.SuperUserPermission, getdb).SetBlock(true).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
ctx.Send("少女更新中...")
|
|
webpageURL := api
|
|
doc, err := htmlquery.LoadURL(webpageURL)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
re := regexp.MustCompile(`\d+`)
|
|
pageTotal, err := strconv.Atoi(re.FindString(htmlquery.FindOne(doc, "//*[@id='comments']/div[2]/div/span[@class='current-comment-page']/text()").Data))
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
LOOP:
|
|
for i := 0; i < pageTotal; i++ {
|
|
logrus.Debugln("[jandan]", fmt.Sprintf("处理第%d/%d页...", i, pageTotal))
|
|
doc, err = htmlquery.LoadURL(webpageURL)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
picList, err := htmlquery.QueryAll(doc, "//*[@class='view_img_link']")
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
if len(picList) != 0 {
|
|
for _, v := range picList {
|
|
u := "https:" + v.Attr[0].Val
|
|
i := crc64.Checksum(binary.StringToBytes(u), crc64.MakeTable(crc64.ISO))
|
|
mu.RLock()
|
|
ok := db.CanFind("picture", "WHERE id = ?", i)
|
|
mu.RUnlock()
|
|
if !ok {
|
|
mu.Lock()
|
|
_ = db.Insert("picture", &picture{ID: i, URL: u})
|
|
mu.Unlock()
|
|
} else {
|
|
// 开始重复，说明之后都是重复
|
|
break LOOP
|
|
}
|
|
}
|
|
}
|
|
if i != pageTotal-1 {
|
|
webpageURL = "https:" + htmlquery.FindOne(doc, "//*[@id='comments']/div[@class='comments']/div[@class='cp-pagenavi']/a[@class='previous-comment-page']").Attr[1].Val
|
|
}
|
|
}
|
|
ctx.Send("更新完成!")
|
|
})
|
|
}
|