558 lines
18 KiB
JavaScript
558 lines
18 KiB
JavaScript
import _ from 'lodash'
|
||
import moment from 'moment'
|
||
import fetch from 'node-fetch'
|
||
import { Config } from '../components/index.js'
|
||
import request from '../lib/request/request.js'
|
||
import { rankType } from '../constants/pixiv.js'
|
||
import { pixivMsg } from '../constants/msg.js'
|
||
import PixivApi from './Pixiv/api.js'
|
||
/** API请求错误文案 */
|
||
|
||
export default new class Pixiv {
|
||
constructor () {
|
||
this.ranktype = rankType
|
||
this.domain = 'http://api.obfs.dev/api/pixiv'
|
||
this.PixivClient = new PixivApi(Config.pixiv.refresh_token)
|
||
}
|
||
|
||
async loginInfo () {
|
||
if (!this.PixivClient?.auth) {
|
||
await this.PixivClient.login()
|
||
}
|
||
if (!this.PixivClient.auth?.user) throw Error('❎ 未获取到登录信息')
|
||
const { profile_image_urls: { px_170x170 }, id, name, account, mail_address, is_premium, x_restrict } = this.PixivClient.auth.user
|
||
return [
|
||
await this._requestPixivImg(px_170x170),
|
||
`\nid：${id}\n`,
|
||
`name：${name}\n`,
|
||
`account：${account}\n`,
|
||
`mail_address：${mail_address}\n`,
|
||
`is_premium：${is_premium}\n`,
|
||
`x_restrict：${x_restrict}`
|
||
]
|
||
}
|
||
|
||
get headers () {
|
||
if (Config.pixiv.pixivDirectConnection) {
|
||
return {
|
||
'Host': 'i.pximg.net',
|
||
'Referer': 'https://www.pixiv.net/',
|
||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36 Edg/108.0.1462.46'
|
||
}
|
||
} else {
|
||
return undefined
|
||
}
|
||
}
|
||
|
||
get proxy () {
|
||
return Config.pixiv.pixivDirectConnection ? 'i.pximg.net' : Config.pixiv.pixivImageProxy
|
||
}
|
||
|
||
/** 开始执行文案 */
|
||
get startMsg () {
|
||
return _.sample(pixivMsg.start)
|
||
}
|
||
|
||
/**
|
||
* 获取插画信息
|
||
* @param {string} ids 插画ID
|
||
* @param filter
|
||
* @returns {object}
|
||
*/
|
||
async illust (ids, filter = false) {
|
||
const params = { id: ids }
|
||
let res = null
|
||
if (this.PixivClient.auth) {
|
||
res = await this.PixivClient.illust(params)
|
||
} else {
|
||
res = await request.get(`${this.domain}/illust`, { params }).then(res => res.json())
|
||
}
|
||
if (res.error) throw Error(res.error?.user_message || '无法获取数据')
|
||
let illust = this._format(res.illust)
|
||
let { id, title, user, tags, total_bookmarks, total_view, url, create_date, x_restrict, illust_ai_type } = illust
|
||
let msg = [
|
||
`标题：${title}\n`,
|
||
`画师：${user.name}\n`,
|
||
`PID：${id}\n`,
|
||
`UID：${user.id}\n`,
|
||
`点赞：${total_bookmarks}\n`,
|
||
`访问：${total_view}\n`,
|
||
`isAI：${illust_ai_type == 2}\n`,
|
||
`发布：${moment(create_date).format('YYYY-MM-DD HH:mm:ss')}\n`,
|
||
`Tag：${tags.join('，')}\n`,
|
||
`直链：https://pixiv.re/${id}.jpg\n`,
|
||
`传送门：https://www.pixiv.net/artworks/${id}`
|
||
]
|
||
if (filter && x_restrict) {
|
||
let linkmsg = ['该作品不适合所有年龄段，请自行使用链接查看：']
|
||
if (url.length > 1) {
|
||
linkmsg.push(...url.map((item, index) => `https://pixiv.re/${id}-${index + 1}.jpg`))
|
||
} else {
|
||
linkmsg.push(`https://pixiv.re/${id}.jpg`)
|
||
}
|
||
throw Error(linkmsg.join('\n'))
|
||
}
|
||
let img = await Promise.all(url.map(async item => await this._requestPixivImg(item)))
|
||
return { msg, img }
|
||
}
|
||
|
||
/**
|
||
* 获取Pixiv榜单
|
||
* @param {number} page 页数
|
||
* @param {Date} date 时间YYYY-MM-DD
|
||
* @param {string} mode 榜单类型
|
||
* @param {boolean} r18 是否为R18榜单
|
||
* @returns {Array}
|
||
*/
|
||
async Rank (page = 1, date = '', mode = '周', r18 = false) {
|
||
// 转为大写
|
||
mode = _.toUpper(mode)
|
||
// 排行榜类型
|
||
let type = this.ranktype[mode].type
|
||
// 总张数
|
||
let pageSizeAll = this.ranktype[mode].total
|
||
// r18处理
|
||
if (r18) {
|
||
let R18 = this.ranktype[mode].r18
|
||
if (!R18) throw Error('该排行没有不适合所有年龄段的分类哦~')
|
||
type = R18.type
|
||
pageSizeAll = R18.total
|
||
}
|
||
// 总页数
|
||
let pageAll = Math.ceil(pageSizeAll / 30)
|
||
if (page > pageAll) throw Error('哪有那么多图片给你辣(•̀へ •́ ╮ )')
|
||
|
||
if (!date) date = moment().subtract(moment().utcOffset(9).hour() >= 12 ? 1 : 2, 'days').format('YYYY-MM-DD')
|
||
|
||
const params = {
|
||
mode: type,
|
||
page,
|
||
date
|
||
}
|
||
let res = null
|
||
if (this.PixivClient.auth) {
|
||
res = await this.PixivClient.rank(params)
|
||
} else {
|
||
res = await request.get(`${this.domain}/rank`, { params }).then(res => res.json())
|
||
}
|
||
if (res.error) throw Error(res.error.message)
|
||
if (_.isEmpty(res.illusts)) throw Error('暂无数据，请等待榜单更新哦(。-ω-)zzz')
|
||
|
||
let illusts = await Promise.all(res.illusts.map(async (item, index) => {
|
||
let list = this._format(item)
|
||
let { id, title, user, tags, total_bookmarks, image_urls } = list
|
||
return [
|
||
`标题：${title}\n`,
|
||
`画师：${user.name}\n`,
|
||
`PID：${id}\n`,
|
||
`UID：${user.id}\n`,
|
||
`点赞：${total_bookmarks}\n`,
|
||
`排名：${(page - 1) * 30 + (index + 1)}\n`,
|
||
`Tag：${_.truncate(tags)}\n`,
|
||
await this._requestPixivImg(image_urls.large)
|
||
]
|
||
}))
|
||
let formatDate = res.next_url.match(/date=(\d{4}-\d{1,2}-\d{1,2})/)[1]
|
||
formatDate = moment(formatDate, 'YYYY-MM-DD').format('YYYY年MM月DD日')
|
||
if (/周/.test(mode)) {
|
||
formatDate = `${moment(formatDate, 'YYYY年MM月DD日').subtract(6, 'days').format('YYYY年MM月DD日')} ~ ${formatDate}`
|
||
} else if (/月/.test(mode)) {
|
||
formatDate = `${moment(formatDate, 'YYYY年MM月DD日').subtract(29, 'days').format('YYYY年MM月DD日')} ~ ${formatDate}`
|
||
}
|
||
let list = [
|
||
`${formatDate}的${mode}${r18 ? 'R18' : ''}榜`,
|
||
`当前为第${page}页，共${pageAll}页，本页共${illusts.length}张，总共${pageSizeAll}张`
|
||
]
|
||
if (page < pageAll) {
|
||
list.push(`可使用 "#看看${date ? `${formatDate}的` : ''}${mode}${r18 ? 'R18' : ''}榜第${page - 0 + 1}页" 翻页`)
|
||
}
|
||
|
||
list.push(...illusts)
|
||
return list
|
||
}
|
||
|
||
/**
|
||
* 根据关键词搜图
|
||
* @param {string} tag 关键词
|
||
* @param {string} page 页数
|
||
* @returns {Array}
|
||
*/
|
||
async vilipixSearchTags (tag, page = 1) {
|
||
const api = 'https://www.vilipix.com/api/v1/picture/public'
|
||
const params = {
|
||
limit: 30,
|
||
tags: tag,
|
||
sort: 'new',
|
||
offset: (page - 1) * 30
|
||
}
|
||
let res = await request.get(api, { params }).then(res => res.json())
|
||
if (res.data.count == 0) throw Error('呜呜呜，人家没有找到相关的插画(ó﹏ò｡)')
|
||
|
||
let pageall = Math.ceil(res.data.count / 30)
|
||
|
||
if (page > pageall) throw Error('啊啊啊，淫家给不了你那么多辣d(ŐдŐ๑)')
|
||
|
||
let list = [
|
||
`当前为第${page}页，共${pageall}页，本页共${res.data.rows.length}张，总共${res.data.count}张`
|
||
]
|
||
if (page < pageall) {
|
||
list.push(`可使用 "#tag搜图${tag}第${page - 0 + 1}页" 翻页`)
|
||
}
|
||
res.data.rows.sort((a, b) => b.like_total - a.like_total)
|
||
for (let i of res.data.rows) {
|
||
let { picture_id, title, original_url, tags } = i
|
||
list.push([
|
||
`标题：${title}\n`,
|
||
`PID：${picture_id}\n`,
|
||
`Tag：${_.truncate(tags)}\n`,
|
||
segment.image(original_url)
|
||
])
|
||
}
|
||
return list
|
||
}
|
||
|
||
/**
|
||
* tag搜图pro
|
||
* @param {string} tag 关键词
|
||
* @param {string} page 页数
|
||
* @param isfilter
|
||
* @returns {*}
|
||
*/
|
||
async searchTags (tag, page = 1, isfilter = true) {
|
||
const params = {
|
||
word: tag,
|
||
page,
|
||
order: 'popular_desc'
|
||
}
|
||
let res = null
|
||
if (this.PixivClient.auth) {
|
||
res = await this.PixivClient.search(params)
|
||
} else {
|
||
res = await request.get(`${this.domain}/search`, { params }).then(res => res.json())
|
||
}
|
||
if (res.error) throw Error(res.error.message)
|
||
if (_.isEmpty(res.illusts)) throw Error('宝~没有数据了哦(๑＞︶＜)و')
|
||
let sortIllusts = _.orderBy(res.illusts, 'total_bookmarks', 'desc')
|
||
let illusts = []
|
||
let filterNum = 0
|
||
let NowNum = res.illusts.length
|
||
for (let i of sortIllusts) {
|
||
let { id, title, user, tags, total_bookmarks, image_urls, x_restrict } = this._format(i)
|
||
if (isfilter && x_restrict) {
|
||
filterNum++
|
||
continue
|
||
}
|
||
illusts.push([
|
||
`标题：${title}\n`,
|
||
`画师：${user.name}\n`,
|
||
`PID：${id}\n`,
|
||
`UID：${user.id}\n`,
|
||
`点赞：${total_bookmarks}\n`,
|
||
`Tag：${_.truncate(tags)}\n`,
|
||
await this._requestPixivImg(image_urls.large)
|
||
])
|
||
}
|
||
if (_.isEmpty(illusts)) throw Error('该页全为涩涩内容已全部过滤(#／。＼#)')
|
||
|
||
return [
|
||
`本页共${NowNum}张${filterNum ? `，过滤${filterNum}张` : ''}\n可尝试使用 "#tagpro搜图${tag}第${page - 0 + 1}页" 翻页\n无数据则代表无下一页`,
|
||
...illusts
|
||
]
|
||
}
|
||
|
||
/**
|
||
* 获取热门tag
|
||
* @returns {Array}
|
||
*/
|
||
async PopularTags () {
|
||
let res = null
|
||
if (this.PixivClient.auth) {
|
||
res = await this.PixivClient.tags()
|
||
} else {
|
||
res = await fetch(`${this.domain}/tags`).then(res => res.json())
|
||
}
|
||
|
||
if (!res.trend_tags) throw Error('呜呜呜，没有获取到数据(๑ १д१)')
|
||
|
||
let list = []
|
||
for (let i of res.trend_tags) {
|
||
let { tag, translated_name } = i
|
||
let url = i.illust.image_urls.large
|
||
list.push(
|
||
[
|
||
`Tag：${tag}\n`,
|
||
`Translated：${translated_name}\n`,
|
||
`Pid：${i.illust.id}\n`,
|
||
await this._requestPixivImg(url)
|
||
]
|
||
)
|
||
}
|
||
return list
|
||
}
|
||
|
||
/**
|
||
* 搜索用户插画
|
||
* @param {number | string} keyword 用户uid或名称
|
||
* @param {number | string} page 页数
|
||
* @param {boolean} isfilter 是否过滤敏感内容
|
||
* @returns {Array}
|
||
*/
|
||
async userIllust (keyword, page = 1, isfilter = true) {
|
||
// 关键词搜索
|
||
if (!/^\d+$/.test(keyword)) {
|
||
let wordlist = null
|
||
if (this.PixivClient.auth) {
|
||
wordlist = await this.PixivClient.search_user({ word: keyword })
|
||
} else {
|
||
wordlist = await request.get(`${this.domain}/search_user`, {
|
||
params: {
|
||
word: keyword
|
||
}
|
||
}).then(res => res.json())
|
||
}
|
||
if (_.isEmpty(wordlist.user_previews)) throw Error('呜呜呜，人家没有找到这个淫d(ŐдŐ๑)')
|
||
keyword = wordlist.user_previews[0].user.id
|
||
}
|
||
const params = {
|
||
id: keyword,
|
||
page
|
||
}
|
||
let res = null
|
||
if (this.PixivClient.auth) {
|
||
res = await this.PixivClient.member_illust(params)
|
||
} else {
|
||
res = await request.get(`${this.domain}/member_illust`, { params }).then(res => res.json())
|
||
}
|
||
|
||
if (res.error) throw Error(res.error.message)
|
||
// 没有作品直接返回信息
|
||
if (_.isEmpty(res.illusts)) throw Error(page >= 2 ? '这一页没有作品辣（＞人＜；）' : 'Σ(っ °Д °;)っ这个淫居然没有作品')
|
||
|
||
let illusts = []
|
||
let filter = 0
|
||
let NowNum = res.illusts.length
|
||
for (let i of res.illusts) {
|
||
let { id: pid, title, tags, total_bookmarks, total_view, url, x_restrict } = this._format(i)
|
||
if (isfilter && x_restrict) {
|
||
filter++
|
||
continue
|
||
}
|
||
illusts.push([
|
||
`标题：${title}\n`,
|
||
`PID：${pid}\n`,
|
||
`点赞：${total_bookmarks}\n`,
|
||
`访问：${total_view}\n`,
|
||
`Tag：${_.truncate(tags)}\n`,
|
||
await this._requestPixivImg(url[0])
|
||
])
|
||
}
|
||
if (_.isEmpty(illusts)) throw Error('该页全为涩涩内容已全部过滤(#／。＼#)')
|
||
let { id: uid, name, profile_image_urls } = res.user
|
||
return [
|
||
[
|
||
await this._requestPixivImg(profile_image_urls.medium),
|
||
`\nUid：${uid}\n`,
|
||
`画师：${name}`
|
||
],
|
||
`本页共${NowNum}张${filter ? `，过滤${filter}张` : ''}\n可尝试使用 "#uid搜图${keyword}第${page - 0 + 1}页" 翻页\n无数据则代表无下一页`,
|
||
...illusts
|
||
]
|
||
}
|
||
|
||
/**
|
||
* 搜索用户
|
||
* @param {string} word 用户name
|
||
* @param {number} page 页数
|
||
* @param {boolean} isfilter 是否过滤敏感内容
|
||
* @returns {Array} 可直接发送的消息数组
|
||
*/
|
||
async searchUser (word, page = 1, isfilter = true) {
|
||
let params = {
|
||
word,
|
||
page,
|
||
size: 10
|
||
}
|
||
let user = null
|
||
if (this.PixivClient.auth) {
|
||
user = await this.PixivClient.search_user(params)
|
||
} else {
|
||
user = await request.get(`${this.domain}/search_user`, { params }).then(res => res.json())
|
||
}
|
||
if (user.error) throw Error(user.error.message)
|
||
if (_.isEmpty(user.user_previews)) throw Error('呜呜呜，人家没有找到这个淫d(ŐдŐ๑)')
|
||
|
||
let msg = await Promise.all(user.user_previews.slice(0, 10).map(async (item, index) => {
|
||
let { id, name, profile_image_urls } = item.user
|
||
let ret = [
|
||
`${(page - 1) * 10 + index + 1}、`,
|
||
await this._requestPixivImg(profile_image_urls),
|
||
`\nid: ${id}\n`,
|
||
`name: ${name}\n`,
|
||
'作品:\n'
|
||
]
|
||
for (let i of item.illusts) {
|
||
let { image_urls, x_restrict } = this._format(i)
|
||
if (isfilter && x_restrict) continue
|
||
ret.push(await this._requestPixivImg(image_urls.square_medium))
|
||
}
|
||
return ret
|
||
}))
|
||
if (msg.length == 30)msg.unshift(`可尝试使用 "#user搜索${word}第${page + 1}页" 翻页`)
|
||
msg.unshift(`当前为第${page}页，已${isfilter ? '开启' : '关闭'}过滤`)
|
||
return msg
|
||
}
|
||
|
||
/**
|
||
* @param limit
|
||
* vilipix随机图片
|
||
* @returns {Array}
|
||
*/
|
||
async vilipixRandomImg (limit) {
|
||
let api = `https://www.vilipix.com/api/v1/picture/recommand?limit=${limit}&offset=${_.random(1, 700)}`
|
||
let res = await request.get(api).then(res => res.json())
|
||
if (!res.data || !res.data.rows) throw Error('呜呜呜，没拿到瑟瑟的图片(˃ ⌑ ˂ഃ )')
|
||
return res.data.rows.map(item => {
|
||
let { picture_id, title, regular_url, tags, like_total } = item
|
||
return [
|
||
`标题：${title}\n`,
|
||
`点赞: ${like_total}\n`,
|
||
`插画ID：${picture_id}\n`,
|
||
`Tag：${_.truncate(tags)}\n`,
|
||
segment.image(regular_url)
|
||
]
|
||
})
|
||
}
|
||
|
||
/**
|
||
* 相关作品
|
||
* @param {string} pid
|
||
* @param isfilter
|
||
* @returns {*}
|
||
*/
|
||
async relatedIllust (pid, isfilter = true) {
|
||
let params = { id: pid }
|
||
let res = null
|
||
if (this.PixivClient.auth) {
|
||
res = await this.PixivClient.related(params)
|
||
} else {
|
||
res = await request.get(`${this.domain}/related`, { params }).then(res => res.json())
|
||
}
|
||
if (res.error) throw Error(res.error.user_message)
|
||
if (_.isEmpty(res.illusts)) throw Error('呃...没有数据(•ิ_•ิ)')
|
||
|
||
let illusts = []
|
||
let filter = 0
|
||
for (let i of res.illusts) {
|
||
let { id, title, user, tags, total_bookmarks, image_urls, x_restrict } = await this._format(i)
|
||
if (isfilter && x_restrict) {
|
||
filter++
|
||
continue
|
||
}
|
||
illusts.push([
|
||
`标题：${title}\n`,
|
||
`画师：${user.name}\n`,
|
||
`PID：${id}\n`,
|
||
`UID：${user.id}\n`,
|
||
`点赞：${total_bookmarks}\n`,
|
||
`Tag：${_.truncate(tags)}\n`,
|
||
await this._requestPixivImg(image_urls.large)
|
||
])
|
||
}
|
||
if (_.isEmpty(illusts)) throw Error('啊啊啊！！！居然全是瑟瑟哒不给你看(＊／ω＼＊)')
|
||
|
||
return [
|
||
`Pid:${pid}的相关作品，共${res.illusts.length}张${filter ? `，过滤${filter}张` : ''}`,
|
||
...illusts
|
||
]
|
||
}
|
||
|
||
/**
|
||
* p站单图
|
||
* @param pro
|
||
*/
|
||
async pximg (pro) {
|
||
let url = 'https://image.anosu.top/pixiv/json'
|
||
const params = {
|
||
r18: pro ? 1 : 0,
|
||
proxy: this.proxy
|
||
}
|
||
let res = await request.get(url, {
|
||
statusCode: 'json',
|
||
params
|
||
})
|
||
let { pid, uid, title, user, tags, url: urls, r18 } = res[0]
|
||
let msg = [
|
||
`Pid: ${pid}\n`,
|
||
`Uid: ${uid}\n`,
|
||
`R18: ${r18 ?? false}\n`,
|
||
`标题：${title}\n`,
|
||
`画师：${user}\n`,
|
||
`Tag：${tags.join('，')}\n`,
|
||
await this._requestPixivImg(urls)
|
||
]
|
||
return msg
|
||
}
|
||
|
||
/**
|
||
* 推荐作品
|
||
* @param {number} num 数量
|
||
* @returns {Promise}
|
||
*/
|
||
async illustRecommended (num) {
|
||
let list = await this.PixivClient.illustRecommended()
|
||
return Promise.all(_.take(list.illusts, num).map(async (item) => {
|
||
let { id, title, user, tags, total_bookmarks, image_urls } = this._format(item)
|
||
return [
|
||
`标题：${title}\n`,
|
||
`画师：${user.name}\n`,
|
||
`PID：${id}\n`,
|
||
`UID：${user.id}\n`,
|
||
`点赞：${total_bookmarks}\n`,
|
||
`Tag：${_.truncate(tags)}`,
|
||
await this._requestPixivImg(image_urls.large)
|
||
]
|
||
}))
|
||
}
|
||
|
||
/**
|
||
* 请求p站图片
|
||
* @param {string} url
|
||
* @returns {Promise}
|
||
*/
|
||
async _requestPixivImg (url) {
|
||
url = url.replace('i.pximg.net', this.proxy)
|
||
logger.debug(`pixiv getImg URL: ${url}`)
|
||
let headers = /s.pximg.net/.test(url) ? undefined : this.headers
|
||
return request.proxyRequestImg(url, { headers })
|
||
}
|
||
|
||
/**
|
||
* 格式化
|
||
* @param {object} illusts 处format理对象
|
||
* @returns {object}
|
||
* title 标题
|
||
* id pid
|
||
* total_bookmarks 点赞
|
||
* total_view 访问量
|
||
* tags 标签
|
||
* url 图片链接
|
||
* user 作者信息
|
||
* image_urls 单张图片
|
||
* x_restrict 是否为全年龄
|
||
* create_date 发布时间
|
||
* illust_ai_type 是否为AI作品
|
||
* visible 是否为可见作品
|
||
*/
|
||
_format (illusts) {
|
||
let url = []
|
||
let { tags, meta_single_page, meta_pages } = illusts
|
||
tags = _.uniq(_.compact(_.flattenDeep(tags?.map(item => Object.values(item)))))
|
||
if (!_.isEmpty(meta_single_page)) {
|
||
url.push(meta_single_page.original_image_url)
|
||
} else {
|
||
url = meta_pages.map(item => item.image_urls.original)
|
||
}
|
||
return { ...illusts, tags, url }
|
||
}
|
||
}()
|