chatgpt-plugin/utils/tools/SendBilibiliTool.js

import fetch from 'node-fetch'

import { formatDate, mkdirs } from '../common.js'
import fs from 'fs'
import { AbstractTool } from './AbstractTool.js'
export class SendVideoTool extends AbstractTool {
  name = 'sendVideo'

  parameters = {
    properties: {
      keyword: {
        type: 'string',
        description: '要发的视频的标题或关键词，用于搜索'
      },
      groupId: {
        type: 'string',
        description: '群号或qq号，发送目标'
      }
    },
    required: ['keyword', 'groupId']
  }

  func = async function (opts) {
    let { keyword, groupId } = opts
    groupId = parseInt(groupId.trim())
    let msg = []
    try {
      let { arcurl, title, pic, description, videoUrl, headers, bvid, author, play, pubdate, like } = await searchBilibili(keyword)
      let group = await Bot.pickGroup(groupId)
      console.log({ arcurl, title, pic, description, videoUrl })
      msg.push(title.replace(/(<([^>]+)>)/ig, '') + '\n')
      msg.push(`UP主：${author} 发布日期：${formatDate(new Date(pubdate * 1000))} 播放量：${play} 点赞：${like}\n`)
      msg.push(arcurl + '\n')
      msg.push(segment.image('https:' + pic))
      msg.push('\n' + description)
      msg.push('\n视频在路上啦！')
      await group.sendMsg(msg)
      const videoResponse = await fetch(videoUrl, { headers })
      const fileType = videoResponse.headers.get('Content-Type').split('/')[1]
      let fileLoc = `data/chatgpt/videos/${bvid}.${fileType}`
      mkdirs('data/chatgpt/videos')
      videoResponse.blob().then(async blob => {
        const arrayBuffer = await blob.arrayBuffer()
        const buffer = Buffer.from(arrayBuffer)
        await fs.writeFileSync(fileLoc, buffer)
        await group.sendMsg(segment.video(fileLoc))
      })
      return `the video ${title.replace(/(<([^>]+)>)/ig, '')} will be shared to ${groupId} after a while, please wait`
    } catch (err) {
      logger.error(err)
      if (msg.length > 0) {
        return `fail to share video, but the video msg is found: ${msg}, you can just tell the information of this video`
      } else {
        return `fail to share video, error: ${err.toString()}`
      }
    }
  }

  description = 'Useful when you want to share a video. The input should be the group number and the keywords that can find the video, connected with a space. If you want to send a specific video, you can give more detailed keywords'
}

export async function searchBilibili (name) {
  let biliRes = await fetch('https://www.bilibili.com',
    {
      // headers: {
      // accept: 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7',
      // Accept: '*/*',
      // 'Accept-Encoding': 'gzip, deflate, br',
      // 'accept-language': 'en-US,en;q=0.9',
      // Connection: 'keep-alive',
      // 'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36'
      // }
    })
  const headers = biliRes.headers.raw()
  const setCookieHeaders = headers['set-cookie']
  if (setCookieHeaders) {
    const cookies = []
    setCookieHeaders.forEach(header => {
      const cookie = header.split(';')[0]
      cookies.push(cookie)
    })
    const cookieHeader = cookies.join('; ')
    let headers = {
      accept: 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7',
      'accept-language': 'en-US,en;q=0.9',
      Referer: 'https://www.bilibili.com',
      'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36',
      cookie: cookieHeader
    }
    let response = await fetch(`https://api.bilibili.com/x/web-interface/search/type?keyword=${name}&search_type=video`,
      {
        headers
      })
    let json = await response.json()
    if (json.data?.numResults > 0) {
      let index = randomIndex()
      let { arcurl, title, pic, description, bvid, author, play, pubdate, like } = json.data.result[Math.min(index, json.data.numResults)]
      let videoInfo = await fetch(`https://api.bilibili.com/x/web-interface/view?bvid=${bvid}`, {
        headers
      })
      videoInfo = await videoInfo.json()
      let cid = videoInfo.data.cid
      let downloadInfo = await fetch(`https://api.bilibili.com/x/player/playurl?bvid=${bvid}&cid=${cid}`, { headers })
      let videoUrl = (await downloadInfo.json()).data.durl[0].url
      return {
        arcurl, title, pic, description, videoUrl, headers, bvid, author, play, pubdate, like
      }
    }
  }

  return {}
}

function randomIndex () {
  // Define weights for each index
  const weights = [5, 4, 3, 2, 1, 1, 1, 1, 1, 1, 1]

  // Compute the total weight
  const totalWeight = weights.reduce((sum, weight) => sum + weight, 0)

  // Generate a random number between 0 and the total weight
  const randomNumber = Math.floor(Math.random() * totalWeight)

  // Choose the index based on the random number and weights
  let weightSum = 0
  for (let i = 0; i < weights.length; i++) {
    weightSum += weights[i]
    if (randomNumber < weightSum) {
      return i
    }
  }
}

console.log('send bilibili')