mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:59:39 +08:00
138 lines
3.5 KiB
Go
138 lines
3.5 KiB
Go
package pixiv
|
||
|
||
import (
|
||
"crypto/tls"
|
||
"errors"
|
||
"fmt"
|
||
"io/ioutil"
|
||
"net"
|
||
"net/http"
|
||
"strings"
|
||
|
||
"github.com/tidwall/gjson"
|
||
)
|
||
|
||
// Illust 插画信息
|
||
type Illust struct {
|
||
Pid int64 `db:"pid"`
|
||
Title string `db:"title"`
|
||
Caption string `db:"caption"`
|
||
Tags string `db:"tags"`
|
||
ImageUrls string `db:"image_urls"`
|
||
AgeLimit string `db:"age_limit"`
|
||
CreatedTime string `db:"created_time"`
|
||
UserId int64 `db:"user_id"`
|
||
UserName string `db:"user_name"`
|
||
}
|
||
|
||
// IllustInfo 根据p站插画id返回插画信息Illust
|
||
func (this *Illust) IllustInfo(id int64) (err error) {
|
||
api := fmt.Sprintf("https://pixiv.net/ajax/illust/%d", id)
|
||
transport := http.Transport{
|
||
DisableKeepAlives: true,
|
||
// 绕过sni审查
|
||
TLSClientConfig: &tls.Config{
|
||
ServerName: "-",
|
||
InsecureSkipVerify: true,
|
||
},
|
||
// 更改dns
|
||
Dial: func(network, addr string) (net.Conn, error) {
|
||
return net.Dial("tcp", "210.140.131.223:443")
|
||
},
|
||
}
|
||
client := &http.Client{
|
||
Transport: &transport,
|
||
}
|
||
|
||
// 网络请求
|
||
req, err := http.NewRequest("GET", api, nil)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
req.Header.Set("Host", "pixiv.net")
|
||
req.Header.Set("Referer", "pixiv.net")
|
||
req.Header.Set("Accept", "Mozilla/5.0 (Windows NT 6.1; WOW64; rv:6.0) Gecko/20100101 Firefox/6.0")
|
||
|
||
resp, err := client.Do(req)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if code := resp.StatusCode; code != 200 {
|
||
return errors.New(fmt.Sprintf("Search illust's info failed, status %d", code))
|
||
}
|
||
defer resp.Body.Close()
|
||
body, err := ioutil.ReadAll(resp.Body)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
json := gjson.ParseBytes(body).Get("body")
|
||
|
||
// 如果有"R-18"tag则判断为R-18（暂时）
|
||
var ageLimit = "all-age"
|
||
for _, tag := range json.Get("tags.tags.#.tag").Array() {
|
||
if tag.Str == "R-18" {
|
||
ageLimit = "r18"
|
||
break
|
||
}
|
||
}
|
||
// 解决json返回带html格式
|
||
var caption = strings.ReplaceAll(json.Get("illustComment").Str, "<br />", "\n")
|
||
if index := strings.Index(caption, "<"); index != -1 {
|
||
caption = caption[:index]
|
||
}
|
||
// 解析返回插画信息
|
||
this.Pid = json.Get("illustId").Int()
|
||
this.Title = json.Get("illustTitle").Str
|
||
this.Caption = caption
|
||
this.Tags = fmt.Sprintln(json.Get("tags.tags.#.tag").Array())
|
||
this.ImageUrls = json.Get("urls.original").Str
|
||
this.AgeLimit = ageLimit
|
||
this.CreatedTime = json.Get("createDate").Str
|
||
this.UserId = json.Get("userId").Int()
|
||
this.UserName = json.Get("userName").Str
|
||
return nil
|
||
}
|
||
|
||
// BigPic 返回一张XML大图CQ码
|
||
func (i *Illust) BigPic(file string) string {
|
||
var hash = PicHash(file)
|
||
return fmt.Sprintf(`[CQ:xml,data=<?xml version='1.0'
|
||
encoding='UTF-8' standalone='yes' ?><msg serviceID="5"
|
||
templateID="12345" action="" brief="不够涩！"
|
||
sourceMsgId="0" url="" flag="0" adverSign="0" multiMsgFlag="0">
|
||
<item layout="0" advertiser_id="0" aid="0"><image uuid="%s.jpg" md5="%s"
|
||
GroupFiledid="2235033681" filesize="81322" local_path="%s.jpg"
|
||
minWidth="200" minHeight="200" maxWidth="500" maxHeight="1000" />
|
||
</item><source name="%s⭐(id:%d author:%s)" icon=""
|
||
action="" appid="-1" /></msg>]`,
|
||
hash,
|
||
hash,
|
||
hash,
|
||
i.Title,
|
||
i.Pid,
|
||
i.UserName,
|
||
)
|
||
}
|
||
|
||
// NormalPic 返回一张普通图CQ码
|
||
func (i *Illust) NormalPic(file string) string {
|
||
return fmt.Sprintf(`[CQ:image,file=file:///%s]`, file)
|
||
}
|
||
|
||
// DetailPic 返回一张带详细信息的图片CQ码
|
||
func (i *Illust) DetailPic(file string) string {
|
||
return fmt.Sprintf(`[SetuTime] %s
|
||
标题：%s
|
||
插画ID：%d
|
||
画师：%s
|
||
画师ID：%d
|
||
直链：https://pixivel.moe/detail?id=%d`,
|
||
i.NormalPic(file),
|
||
i.Title,
|
||
i.Pid,
|
||
i.UserName,
|
||
i.UserId,
|
||
i.Pid,
|
||
)
|
||
}
|