mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:59:39 +08:00
193 lines
7.7 KiB
Go
193 lines
7.7 KiB
Go
// Package novel 铅笔小说搜索插件
|
|
package novel
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/cookiejar"
|
|
"net/url"
|
|
"regexp"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/antchfx/htmlquery"
|
|
log "github.com/sirupsen/logrus"
|
|
zero "github.com/wdvxdr1123/ZeroBot"
|
|
"github.com/wdvxdr1123/ZeroBot/extension/rate"
|
|
"github.com/wdvxdr1123/ZeroBot/message"
|
|
"github.com/wdvxdr1123/ZeroBot/utils/helper"
|
|
|
|
ub "github.com/FloatTech/zbputils/binary"
|
|
control "github.com/FloatTech/zbputils/control"
|
|
"github.com/FloatTech/zbputils/txt2img"
|
|
|
|
"github.com/FloatTech/ZeroBot-Plugin/order"
|
|
)
|
|
|
|
const (
|
|
websiteURL = "https://www.23qb.net"
|
|
websiteTitle = "铅笔小说"
|
|
errorTitle = "出现错误！"
|
|
username = "zerobot"
|
|
password = "123456"
|
|
submit = "%26%23160%3B%B5%C7%26%23160%3B%26%23160%3B%C2%BC%26%23160%3B"
|
|
ua = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36"
|
|
loginURL = websiteURL + "/login.php?do=submit&jumpurl=https%3A%2F%2Fwww.23qb.net%2F"
|
|
searchURL = websiteURL + "/saerch.php"
|
|
idReg = `/(\d+)/`
|
|
)
|
|
|
|
var (
|
|
gCurCookieJar *cookiejar.Jar
|
|
engine = control.Register("novel", order.PrioNovel, &control.Options{
|
|
DisableOnDefault: false,
|
|
Help: "铅笔小说网搜索\n- 小说[xxx]",
|
|
})
|
|
limit = rate.NewManager(time.Minute, 5)
|
|
)
|
|
|
|
func init() {
|
|
engine.OnRegex("^小说([\u4E00-\u9FA5A-Za-z0-9]{1,25})$").SetBlock(true).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
if !limit.Load(ctx.Event.GroupID).Acquire() {
|
|
ctx.SendChain(message.Text("请稍后重试0x0..."))
|
|
return
|
|
}
|
|
ctx.SendChain(message.Text("少女祈祷中......"))
|
|
login(username, password)
|
|
searchKey := ctx.State["regex_matched"].([]string)[1]
|
|
searchHTML := search(searchKey)
|
|
doc, err := htmlquery.Parse(strings.NewReader(searchHTML))
|
|
if err != nil {
|
|
log.Errorln("[novel]", err)
|
|
}
|
|
htmlTitle := htmlquery.InnerText(htmlquery.FindOne(doc, "/html/head/title"))
|
|
switch htmlTitle {
|
|
case websiteTitle:
|
|
list, err := htmlquery.QueryAll(doc, "//dl[@id='nr']")
|
|
if err != nil {
|
|
log.Errorln("[novel]", err)
|
|
}
|
|
if len(list) != 0 {
|
|
text := ""
|
|
for _, v := range list {
|
|
bookName := htmlquery.InnerText(htmlquery.FindOne(v, "/dd[1]/h3/a[1]"))
|
|
category := htmlquery.InnerText(htmlquery.FindOne(v, "/dt/span[1]"))
|
|
author := htmlquery.InnerText(htmlquery.FindOne(v, "/dd[2]/span[1]"))
|
|
status := htmlquery.InnerText(htmlquery.FindOne(v, "/dd[2]/span[2]"))
|
|
wordNumbers := htmlquery.InnerText(htmlquery.FindOne(v, "/dd[2]/span[3]"))
|
|
description := htmlquery.InnerText(htmlquery.FindOne(v, "/dd[3]"))
|
|
updateTime := htmlquery.InnerText(htmlquery.FindOne(v, "/dd[1]/h3/span[1]"))
|
|
latestChapter := htmlquery.InnerText(htmlquery.FindOne(v, "/dd[4]/a[1]"))
|
|
|
|
reg := regexp.MustCompile(idReg)
|
|
id := reg.FindStringSubmatch(htmlquery.SelectAttr(htmlquery.FindOne(v, "/dt/a[1]"), "href"))[1]
|
|
|
|
webpageURL := websiteURL + "/book/" + id + "/"
|
|
downloadURL := websiteURL + "/modules/article/txtarticle.php?id=" + id
|
|
text += fmt.Sprintf("书名:%s\n类型:%s\n作者:%s\n状态:%s\n字数:%s\n简介:%s\n更新时间:%s\n最新章节:%s\n网页链接:%s\n下载地址:%s\n\n", bookName, category, author, status, wordNumbers, description, updateTime, latestChapter, webpageURL, downloadURL)
|
|
}
|
|
data, err := txt2img.RenderToBase64(text, txt2img.FontFile, 400, 20)
|
|
if err != nil {
|
|
log.Println("err:", err)
|
|
}
|
|
if id := ctx.SendChain(message.Image("base64://" + helper.BytesToString(data))); id.ID() == 0 {
|
|
ctx.SendChain(message.Text("ERROR: 可能被风控了"))
|
|
}
|
|
} else {
|
|
text := htmlquery.InnerText(htmlquery.FindOne(doc, "//div[@id='tipss']"))
|
|
text = strings.ReplaceAll(text, " ", "")
|
|
text = strings.ReplaceAll(text, "本站", websiteURL)
|
|
ctx.SendChain(message.Text(text))
|
|
}
|
|
case errorTitle:
|
|
ctx.SendChain(message.Text(errorTitle))
|
|
text := htmlquery.InnerText(htmlquery.FindOne(doc, "//div[@style='text-align: center;padding:10px']"))
|
|
text = strings.ReplaceAll(text, " ", "")
|
|
ctx.SendChain(message.Text(text))
|
|
default:
|
|
bookName := htmlquery.SelectAttr(htmlquery.FindOne(doc, "//meta[@property='og:novel:book_name']"), "content")
|
|
category := htmlquery.SelectAttr(htmlquery.FindOne(doc, "//meta[@property='og:novel:category']"), "content")
|
|
author := htmlquery.SelectAttr(htmlquery.FindOne(doc, "//meta[@property='og:novel:author']"), "content")
|
|
status := htmlquery.SelectAttr(htmlquery.FindOne(doc, "//meta[@property='og:novel:status']"), "content")
|
|
description := htmlquery.InnerText(htmlquery.FindOne(doc, "//div[@id='bookintro']/p"))
|
|
updateTime := htmlquery.SelectAttr(htmlquery.FindOne(doc, "//meta[@property='og:novel:update_time']"), "content")
|
|
latestChapter := htmlquery.SelectAttr(htmlquery.FindOne(doc, "//meta[@property='og:novel:latest_chapter_name']"), "content")
|
|
|
|
reg := regexp.MustCompile(idReg)
|
|
id := reg.FindStringSubmatch(htmlquery.SelectAttr(htmlquery.FindOne(doc, "//meta[@property='og:novel:read_url']"), "content"))[1]
|
|
webpageURL := websiteURL + "/book/" + id + "/"
|
|
downloadURL := websiteURL + "/modules/article/txtarticle.php?id=" + id
|
|
text := fmt.Sprintf("书名:%s\n类型:%s\n作者:%s\n状态:%s\n简介:%s\n更新时间:%s\n最新章节:%s\n网页链接:%s\n下载地址:%s\n", bookName, category, author, status, description, updateTime, latestChapter, webpageURL, downloadURL)
|
|
data, err := txt2img.RenderToBase64(text, txt2img.FontFile, 400, 20)
|
|
if err != nil {
|
|
log.Println("err:", err)
|
|
}
|
|
if id := ctx.SendChain(message.Image("base64://" + helper.BytesToString(data))); id.ID() == 0 {
|
|
ctx.SendChain(message.Text("ERROR: 可能被风控了"))
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
func login(username, password string) {
|
|
gCurCookieJar, _ = cookiejar.New(nil)
|
|
client := &http.Client{
|
|
Jar: gCurCookieJar,
|
|
}
|
|
usernameData, err := ub.UTF82GBK(helper.StringToBytes(username))
|
|
if err != nil {
|
|
log.Errorln("[novel]", err)
|
|
}
|
|
usernameGbk := helper.BytesToString(usernameData)
|
|
passwordData, err := ub.UTF82GBK(helper.StringToBytes(password))
|
|
if err != nil {
|
|
log.Errorln("[novel]", err)
|
|
}
|
|
passwordGbk := helper.BytesToString(passwordData)
|
|
loginReq, err := http.NewRequest("POST", loginURL, strings.NewReader(fmt.Sprintf("username=%s&password=%s&usecookie=315360000&action=login&submit=%s", url.QueryEscape(usernameGbk), url.QueryEscape(passwordGbk), submit)))
|
|
if err != nil {
|
|
log.Errorln("[novel]", err)
|
|
}
|
|
loginReq.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
|
loginReq.Header.Set("User-Agent", ua)
|
|
loginResp, err := client.Do(loginReq)
|
|
if err != nil {
|
|
log.Errorln("[novel]", err)
|
|
}
|
|
defer loginResp.Body.Close()
|
|
}
|
|
|
|
func search(searchKey string) (searchHTML string) {
|
|
searchKeyData, err := ub.UTF82GBK(helper.StringToBytes(searchKey))
|
|
if err != nil {
|
|
log.Errorln("[novel]", err)
|
|
}
|
|
searchKeyGbk := helper.BytesToString(searchKeyData)
|
|
client := &http.Client{
|
|
Jar: gCurCookieJar,
|
|
}
|
|
searchReq, err := http.NewRequest("POST", searchURL, strings.NewReader(fmt.Sprintf("searchkey=%s&searchtype=all", url.QueryEscape(searchKeyGbk))))
|
|
if err != nil {
|
|
log.Errorln("[novel]", err)
|
|
}
|
|
searchReq.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
|
searchReq.Header.Set("User-Agent", ua)
|
|
searchResp, err := client.Do(searchReq)
|
|
if err != nil {
|
|
log.Errorln("[novel]", err)
|
|
}
|
|
defer searchResp.Body.Close()
|
|
searchData, err := ioutil.ReadAll(searchResp.Body)
|
|
if err != nil {
|
|
log.Errorf("[novel] get response for url=%s got error=%s\n", searchURL, err.Error())
|
|
}
|
|
searchData, err = ub.GBK2UTF8(searchData)
|
|
if err != nil {
|
|
log.Errorln("[novel]", err)
|
|
}
|
|
searchHTML = helper.BytesToString(searchData)
|
|
return searchHTML
|
|
}
|