koishi-plugin-kkk 1.0.0

package/src/karin/platform/bilibili/genParams.ts

@@ -0,0 +1,30 @@
+import { wbi_sign } from '@ikenxuan/amagi'
+
+import { bilibiliFetcher } from '@/module/utils/amagiClient'
+import { Config } from '@/module/utils/Config'
+
+/**
+ * 计算请求参数
+ * @param apiURL 请求地址
+ * @returns
+ */
+export const genParams = async (apiURL: string): Promise<string> => {
+  if (Config.amagi.cookies.bilibili === '' || Config.amagi.cookies.bilibili === null) return '&platform=html5'
+  const loginInfo = await bilibiliFetcher.fetchLoginStatus()
+  const genSign = await wbi_sign(apiURL, Config.amagi.cookies.bilibili)
+
+  const qn = [6, 16, 32, 64, 74, 80, 112, 116, 120, 125, 126, 127]
+  const isvip = loginInfo.data.data?.vipStatus === 1
+  if (isvip) {
+    return `&fnval=16&fourk=1&${genSign}`
+  } else return `&qn=${qn[3]}&fnval=16`
+}
+
+export const checkCk = async (): Promise<{ Status: 'isLogin' | '!isLogin'; isVIP: boolean }> => {
+  if (Config.amagi.cookies.bilibili === '' || Config.amagi.cookies.bilibili === null) return { Status: '!isLogin', isVIP: false }
+  const loginInfo = await bilibiliFetcher.fetchLoginStatus()
+  const isVIP = loginInfo.data.data?.vipStatus === 1
+  if (isVIP) {
+    return { Status: 'isLogin', isVIP }
+  } else return { Status: 'isLogin', isVIP }
+}

package/src/karin/platform/bilibili/getID.ts

@@ -0,0 +1,163 @@
+import https from 'node:https'
+
+import { logger } from 'node-karin'
+import axios from 'node-karin/axios'
+
+import { bilibiliFetcher } from '@/module/utils/amagiClient'
+import type { BilibiliDataTypes } from '@/types'
+
+export interface BilibiliId {
+  type: BilibiliDataTypes[keyof BilibiliDataTypes]
+  [x: string]: any
+}
+
+/**
+ * return aweme_id
+ * @param {string} url 分享连接
+ * @returns
+ */
+export const getBilibiliID = async (url: string) => {
+  // 如果是专栏链接且带有 opus_fallback 参数，先去掉参数让它自然重定向
+  if (/\/read\/cv\d+/.test(url) && url.includes('opus_fallback')) {
+    url = url.split('?')[0]
+  }
+
+  // 之前这里既没有超时、也没指定 IP 协议族：
+  //   - timeout:0 → 连接被重置/卡住时会**无限等**（表现就是「开始解析」后一直没反应）
+  //   - 没有 family:4 → Node 会先试 IPv6，这台机器上 bilibili 的 IPv6 会 ECONNRESET
+  // 现在：强制 IPv4 + 15 秒超时 + 失败重试一次
+  const request = (target: string) => axios.get(target, {
+    timeout: 15000,
+    maxRedirects: 5,
+    httpsAgent: new https.Agent({ family: 4, keepAlive: true }),
+    headers: {
+      'User-Agent': 'Apifox/1.0.0 (https://apifox.com)'
+    }
+  })
+  let resp
+  try {
+    resp = await request(url)
+  } catch (error: any) {
+    logger.warn('[B站解析] 首次请求失败（%s），用 IPv4 重试一次', error?.code ?? error?.message ?? error)
+    resp = await request(url)
+  }
+  const longLink = resp?.request?.res?.responseUrl ?? resp?.config?.url ?? url
+  let result = {} as BilibiliId
+  let pValue: number | undefined
+  const parsedUrl = new URL(longLink)
+  const pParam = parsedUrl.searchParams.get('p')
+  if (pParam) {
+    pValue = parseInt(pParam, 10)
+    if (isNaN(pValue)) {
+      pValue = undefined
+    }
+  }
+  const pathname = parsedUrl.pathname
+  const hostname = parsedUrl.hostname
+
+  switch (true) {
+    case (hostname === 't.bilibili.com' && /^\/\d+/.test(pathname)) || (hostname === 'www.bilibili.com' && /^\/opus\/\d+/.test(pathname)): {
+      const tMatch = hostname === 't.bilibili.com' ? pathname.match(/^\/(\d+)/) : null
+      const opusMatch = hostname === 'www.bilibili.com' ? pathname.match(/^\/opus\/(\d+)/) : null
+      const dynamic_id = tMatch ?? opusMatch
+      result = {
+        type: 'dynamic_info',
+        dynamic_id: dynamic_id ? dynamic_id[1] : undefined
+      }
+      break
+    }
+    case /\/read\/cv(\d+)/.test(longLink): {
+      const cvMatch = /\/read\/cv(\d+)/.exec(longLink)
+      result = {
+        type: 'dynamic_info',
+        dynamic_id: cvMatch ? cvMatch[1] : undefined
+      }
+      break
+    }
+    case /\/bangumi\/play\/(\w+)/.test(longLink): {
+      const playMatch = /\/bangumi\/play\/(\w+)/.exec(longLink)
+      const id = playMatch ? playMatch[1] : ''
+      let realid = ''
+      let isEpid = false
+      if (id.startsWith('ss')) {
+        realid = id
+      } else if (id.startsWith('ep')) {
+        realid = id
+        isEpid = true
+      }
+      result = {
+        type: 'bangumi_video_info',
+        isEpid,
+        realid
+      }
+      break
+    }
+    case /(video\/|video-)([A-Za-z0-9]+)/.test(longLink): {
+      const bvideoMatch = /video\/([A-Za-z0-9]+)|bvid=([A-Za-z0-9]+)/.exec(longLink)
+      let bvid = bvideoMatch ? bvideoMatch[1] || bvideoMatch[2] : undefined
+
+      if (bvid && bvid.toLowerCase().startsWith('av')) {
+        const avid = parseInt(bvid.replace(/^av/i, ''))
+        // v7 的 avToBv 是纯本地计算端点，data 直接就是 `{ bvid }` ——
+        // 不再是 v6 那层 `{ code, data: { bvid }, message }` 的 API 信封
+        const convertResult = await bilibiliFetcher.convertAvToBv({ avid })
+        bvid = convertResult.data.bvid
+      }
+
+      result = {
+        type: 'one_video',
+        bvid,
+        ...(pValue !== undefined && { p: pValue })
+      }
+      break
+    }
+    case /festival\/([A-Za-z0-9]+)/.test(longLink): {
+      const festivalMatch = /festival\/([A-Za-z0-9]+)\?bvid=([A-Za-z0-9]+)/.exec(longLink)
+      result = {
+        type: 'one_video',
+        id: festivalMatch ? festivalMatch[2] : undefined
+      }
+      break
+    }
+    case /play\/(\S+?)\??/.test(longLink): {
+      const playMatch = /play\/(\w+)/.exec(longLink)
+      const id = playMatch ? playMatch[1] : ''
+      const isEpid = false
+      if (id.startsWith('ss')) {
+        result.realid = 'season_id'
+      } else if (id.startsWith('ep')) {
+        result.realid = 'ep_id'
+      }
+      result = {
+        type: 'bangumi_video_info',
+        isEpid,
+        realid: playMatch ? playMatch[1] : ''
+      }
+      break
+    }
+    case /^https:\/\/t\.bilibili\.com\/(\d+)/.test(longLink) || /^https:\/\/www\.bilibili\.com\/opus\/(\d+)/.test(longLink): {
+      const tMatch = /^https:\/\/t\.bilibili\.com\/(\d+)/.exec(longLink)
+      const opusMatch = /^https:\/\/www\.bilibili\.com\/opus\/(\d+)/.exec(longLink)
+      const dynamic_id = tMatch ?? opusMatch
+      result = {
+        type: 'dynamic_info',
+        dynamic_id: dynamic_id ? dynamic_id[1] : dynamic_id
+      }
+      break
+    }
+    case longLink.includes('live.bilibili.com'): {
+      const match = /https?:\/\/live\.bilibili\.com\/(\d+)/.exec(longLink)
+      result = {
+        type: 'live_room_detail',
+        room_id: match ? match[1] : undefined
+      }
+      break
+    }
+    default:
+      logger.warn('无法获取作品ID')
+      break
+  }
+
+  console.log(result)
+  return result
+}

package/src/karin/platform/bilibili/index.ts

@@ -0,0 +1,8 @@
+export * from './bilibili'
+export * from './comments'
+export * from './danmaku'
+export * from './genParams'
+export * from './getID'
+export * from './login'
+export * from './push'
+export * from './riskControl'

package/src/karin/platform/bilibili/live-status.ts

@@ -0,0 +1,36 @@
+const invalidLiveTimes = new Set(['', '-62170012800', '0000-00-00 00:00:00'])
+
+/**
+ * 将B站返回的东八区开播时间转换为数据库可比较的 ISO 时间。
+ *
+ * @param liveTime B站直播间详情返回的开播时间。
+ * @returns 有效的 UTC ISO 时间；无法解析时返回 `null`。
+ */
+export const parseBilibiliLiveStartedAt = (liveTime: string): string | null => {
+  const normalizedLiveTime = liveTime.trim()
+  if (invalidLiveTimes.has(normalizedLiveTime)) return null
+
+  const zonedLiveTime = /^\d{4}-\d{2}-\d{2} \d{2}:\d{2}:\d{2}$/.test(normalizedLiveTime)
+    ? `${normalizedLiveTime.replace(' ', 'T')}+08:00`
+    : normalizedLiveTime
+  const timestamp = Date.parse(zonedLiveTime)
+  return Number.isNaN(timestamp) ? null : new Date(timestamp).toISOString()
+}
+
+/**
+ * 使用直播间和开播时间构造跨检测源稳定的场次缓存键。
+ *
+ * @remarks
+ * 定时推送可能先从 UID 直播状态接口发现开播，也可能在接口异常时回退到动态列表。
+ * 两条路径必须使用同一个场次键，才能避免同一场直播被重复推送。
+ *
+ * @param hostMid UP 主 UID。
+ * @param roomId 直播间长号。
+ * @param liveTime B站直播间详情返回的本场开播时间。
+ * @returns 可用于 `DynamicCaches` 的场次键；开播时间无效时返回 `null`。
+ */
+export const buildBilibiliLiveSessionId = (hostMid: number, roomId: number, liveTime: string): string | null => {
+  const normalizedLiveTime = liveTime.trim()
+  if (!parseBilibiliLiveStartedAt(normalizedLiveTime)) return null
+  return `bilibili-live:${hostMid}:${roomId}:${normalizedLiveTime}`
+}

package/src/karin/platform/bilibili/login.ts

@@ -0,0 +1,130 @@
+import fs from 'node:fs'
+
+import type { AmagiError, Qrcode } from '@ikenxuan/amagi'
+import { logger, syncUpstreamToKoishi, type Message } from 'node-karin'
+
+import { Common, Render } from '@/module/utils'
+import { getAmagiClient, reloadAmagiConfig } from '@/module/utils/amagiClient'
+import { resolveTriggerAvatarUrl } from '@/module/utils/bot'
+import { Config } from '@/module/utils/Config'
+
+/**
+ * 登录过程中发出的消息统一登记，结束时一起撤回，避免二维码留在群里
+ * @param e - 消息事件
+ * @returns 带 send / recall / recallAll 的登记器
+ */
+const createMessageTracker = (e: Message) => {
+  const messageIds: string[] = []
+
+  const recall = async (id: string) => {
+    try {
+      await e.bot.recallMsg(e.contact, id)
+    } catch (error) {
+      logger.debug('[B站登录] 撤回消息失败:', error)
+    }
+    const index = messageIds.indexOf(id)
+    if (index > -1) messageIds.splice(index, 1)
+  }
+
+  return {
+    /**
+     * 发送并登记一条消息
+     * @param message - 消息内容
+     * @returns 这条消息的 id
+     */
+    async send (message: Parameters<Message['reply']>[0]) {
+      const sent = await e.reply(message, { reply: true })
+      if (sent.messageId) messageIds.push(sent.messageId)
+      return sent.messageId
+    },
+    recall,
+    /** 撤回目前登记的全部消息 */
+    async recallAll () {
+      await Promise.all(messageIds.splice(0, messageIds.length).map(recall))
+    }
+  }
+}
+
+/**
+ * 渲染并发送登录二维码，同时落一份到临时目录方便排查。
+ * @param e - 消息事件
+ * @param qrcode - 会话给出的二维码
+ * @param tracker - 消息登记器
+ * @returns 二维码消息的 id，扫码后要单独撤回它
+ */
+const sendQrcode = async (e: Message, qrcode: Qrcode, tracker: ReturnType<typeof createMessageTracker>): Promise<string | undefined> => {
+  const qrimg = await Render(e, 'bilibili/qrcodeImg', {
+    share_url: qrcode.content,
+    avatarUrl: await resolveTriggerAvatarUrl(e)
+  })
+
+  const base64Data = qrimg[0]?.file
+  if (!base64Data) throw new Error('生成二维码图片失败')
+
+  fs.writeFileSync(`${Common.tempDri.default}BilibiliLoginQrcode.png`, Buffer.from(base64Data.replace(/^base64:\/\//, ''), 'base64'))
+  return tracker.send(qrimg)
+}
+
+/** 把会话的失败原因翻成给用户看的话 */
+const noticeOf = (error: AmagiError): string => {
+  if (error.kind === 'risk') return `登录请求被B站风控拦截：${error.message}`
+  if (error.code === 'COOKIE_EXPIRED') return '二维码已失效'
+  return `登录未完成：${error.message}`
+}
+
+/**
+ * B站扫码登录。
+ *
+ * 轮询、退避、超时都在 amagi 的 v7 登录会话里（`client.bilibili.login.qrcode()`），
+ * 这里只负责渲染二维码与落库。相比 v6 的手写轮询有两处行为变好：
+ *
+ * - **cookie 串是干净的。** 原先把响应的 `set-cookie` 数组直接 `join('; ')`，
+ *   连 `Path=/` / `Domain=` / `Expires=` 这些属性一起当成了 cookie 值；
+ *   v7 的策略用 `mergeSetCookie` 只取 name=value。
+ * - **落库后会重载 Client。** 原先 `Config.Modify` 既没 await 也没重载，
+ *   新 cookie 要等文件监听那一跳才生效。
+ * @param e - 消息事件
+ */
+export const bilibiliLogin = async (e: Message) => {
+  const tracker = createMessageTracker(e)
+  let qrcodeMessageId: string | undefined
+  let scanned = false
+
+  try {
+    const session = getAmagiClient().bilibili.login.qrcode()
+
+    const outcome = await session.watch({
+      onQrcode: async (qrcode) => {
+        qrcodeMessageId = await sendQrcode(e, qrcode, tracker)
+      },
+
+      onScanned: async () => {
+        if (scanned) return
+        scanned = true
+        await tracker.send('二维码已扫码，未确认')
+        if (qrcodeMessageId) await tracker.recall(qrcodeMessageId)
+      },
+
+      onSuccess: async (credential) => {
+        await Config.Modify('amagi', 'cookies.bilibili', credential.cookie)
+        // 回写一份到 koishi.yml：不然控制台表单里那个「B站 Cookie」永远是空的
+        // （插件运行时读的是 config.json，控制台读的是 koishi.yml，两边本来是单向的）
+        const synced = await syncUpstreamToKoishi({ amagi: { cookies: { bilibili: credential.cookie } } })
+        const reloaded = reloadAmagiConfig()
+        logger.mark(`[B站登录] 登录凭证已保存${synced ? '（已同步到控制台配置）' : ''}，Amagi Client ${reloaded ? '已重载' : '配置未变化'}`)
+      }
+    })
+
+    if (outcome.ok) {
+      await e.reply('登录成功！用户登录凭证已保存至配置', { reply: true })
+    } else {
+      logger.warn(`[B站登录] 会话结束于失败态: kind=${outcome.error.kind} code=${outcome.error.code}`)
+      await e.reply(noticeOf(outcome.error), { reply: true })
+    }
+  } catch (error) {
+    logger.error('[B站登录] 登录流程出错:', error)
+    await e.reply('登录过程中发生错误，请重试', { reply: true })
+  } finally {
+    await tracker.recallAll()
+  }
+}