koishi-plugin-kkk 1.0.0

package/src/karin/app.ts

@@ -0,0 +1 @@
+import('node-karin/start')

package/src/karin/apps/__ready.ts

@@ -0,0 +1,14 @@
+import { logger } from 'node-karin'
+
+import { Root } from '@/module'
+
+const start = globalThis.__kkkLoadStart
+const elapsedMs = typeof start === 'bigint' ? Number(process.hrtime.bigint() - start) / 1_000_000 : 0
+
+const timeText = elapsedMs >= 1000 ? `${String(parseFloat((elapsedMs / 1000).toFixed(2)))}s` : `${Math.round(elapsedMs)}ms`
+
+logger.info(
+  `${logger.violet(`[插件:${Root.pluginName}]`)} ${logger.green(`v${Root.pluginVersion}`)} 初始化完成 ~ 耗时 ${logger.green(timeText)}`
+)
+
+delete (globalThis as any).__kkkLoadStart

package/src/karin/apps/admin.ts

@@ -0,0 +1,127 @@
+import fs from 'node:fs'
+import path from 'node:path'
+
+import karin, { logger } from 'node-karin'
+
+import { Common } from '@/module'
+import { Config } from '@/module/utils/Config'
+import { wrapWithErrorHandler } from '@/module/utils/ErrorHandler'
+import { bilibiliLogin } from '@/platform'
+import { douyinLogin } from '@/platform/douyin/login'
+
+// 包装缓存清理任务
+const handleCacheCleanup = wrapWithErrorHandler(
+  async () => {
+    const twoHoursAgo = Date.now() - 2 * 60 * 60 * 1000 // 2小时前的时间戳
+
+    // 清理视频缓存
+    const videoDeleted = removeOldFiles(Common.tempDri.video, twoHoursAgo)
+    logger.debug(`${Common.tempDri.video} 目录下已删除 ${videoDeleted} 个文件`)
+
+    // 需要本地落盘的图片模式也清理图片缓存目录
+    if (Config.app.imageSendMode !== 'url') {
+      const imageDeleted = removeOldFiles(Common.tempDri.images, twoHoursAgo)
+      logger.debug(`${Common.tempDri.images} 目录下已删除 ${imageDeleted} 个文件`)
+    }
+  },
+  {
+    businessName: '缓存自动删除'
+  }
+)
+
+export const task = Config.app.removeCache && karin.task('[kkk-缓存自动删除]', '*/30 * * * *', handleCacheCleanup, { log: false })
+
+// 包装B站登录命令
+const handleBilibiliLogin = wrapWithErrorHandler(
+  async (e) => {
+    // await e.reply('暂时不可用')
+    // return true
+    await bilibiliLogin(e)
+    return true
+  },
+  {
+    businessName: 'B站登录'
+  }
+)
+
+// 包装抖音登录命令
+const handleDouyinLogin = wrapWithErrorHandler(
+  async (e) => {
+    // await e.reply('暂时不可用')
+    // return true
+    await douyinLogin(e)
+    return true
+  },
+  {
+    businessName: '抖音登录'
+  }
+)
+
+// 包装B站登录命令
+const handlTestWrapWithErrorHandler = wrapWithErrorHandler(
+  async (_e) => {
+    logger.trace('测试跟踪日志')
+    logger.debug('测试调试日志')
+    logger.info('测试信息日志')
+    logger.mark('测试标记日志')
+    logger.warn('测试警告日志')
+    logger.error('测试错误日志')
+    logger.fatal('测试致命日志')
+    throw Error('测试异常')
+  },
+  {
+    businessName: '测试命令-错误捕获'
+  }
+)
+
+export const biLogin = karin.command(/^#?(kkk)?\s*B站\s*(扫码)?\s*登录$/i, handleBilibiliLogin, {
+  perm: Config.bilibili.loginPerm,
+  name: 'kkk-ck管理'
+})
+
+export const dylogin = karin.command(/^#?(kkk)?抖音(扫码)?登录$/, handleDouyinLogin, {
+  perm: Config.douyin.loginPerm,
+  name: 'kkk-ck管理'
+})
+
+export const testWrapWithErrorHandler =
+  process.env.NODE_ENV === 'development'
+    ? karin.command(/^#?(kkk)?\s*测试\s*错误捕获$/i, handlTestWrapWithErrorHandler, {
+        perm: Config.bilibili.loginPerm,
+        name: 'kkk-测试命令-错误捕获'
+      })
+    : null
+
+// 删除指定时间之前的文件
+export const removeOldFiles = (dir: string, beforeTimestamp: number): number => {
+  let deletedCount = 0
+
+  const files = fs.readdirSync(dir)
+
+  for (const file of files) {
+    const filePath = path.join(dir, file)
+    const stats = fs.statSync(filePath)
+
+    if (stats.isDirectory()) {
+      // 递归处理子目录
+      deletedCount += removeOldFiles(filePath, beforeTimestamp)
+
+      // 检查目录是否为空，如果为空则删除
+      const remainingFiles = fs.readdirSync(filePath)
+      if (remainingFiles.length === 0) {
+        fs.rmdirSync(filePath)
+      }
+    } else {
+      // 检查文件的创建时间
+      const fileCreatedTime = stats.birthtimeMs
+
+      // 如果文件创建时间早于指定时间戳，则删除
+      if (fileCreatedTime < beforeTimestamp) {
+        fs.unlinkSync(filePath)
+        deletedCount++
+      }
+    }
+  }
+
+  return deletedCount
+}

package/src/karin/apps/help.ts

@@ -0,0 +1,228 @@
+import karin, { config } from 'node-karin'
+
+import { Render } from '@/module'
+import { Config } from '@/module/utils/Config'
+import { wrapWithErrorHandler } from '@/module/utils/ErrorHandler'
+import { collectRuntimeReport, getLocalChangelog } from '@/module/utils/runtime-report'
+
+type Role = 'master' | 'member'
+type RoleItem = { title: string; description: string; icon?: string | { name: string; color?: string }; roles?: Role[] }
+type RoleMenuGroup = {
+  title: string
+  items: RoleItem[]
+  subGroups?: { title: string; items: RoleItem[] }[]
+}
+
+const HELP_MENU_CONFIG: RoleMenuGroup[] = [
+  {
+    title: '常用功能',
+    items: [
+      {
+        title: '自动识别分享链接进行解析',
+        description: (() => {
+          const platforms = []
+          if (Config.douyin?.switch) platforms.push('抖音')
+          if (Config.bilibili?.switch) platforms.push('哔哩哔哩')
+          if (Config.kuaishou?.switch) platforms.push('快手')
+          if (Config.xiaohongshu?.switch) platforms.push('小红书')
+          return platforms.length > 0 ? `支持「${platforms.join('」「')}」` : '暂无可用平台'
+        })(),
+        icon: 'ph:link-fill',
+        roles: ['member', 'master']
+      },
+      {
+        title: '「解析」',
+        description: '直接发链接或引用消息后发送（指令前缀按 Koishi 配置，本插件不带 # 前缀）；发链接会自动识别平台并弹画质面板',
+        icon: 'ph:magic-wand-fill',
+        roles: ['member', 'master']
+      },
+      {
+        title: 'kkk解析统计',
+        description: '查看当前群组的解析统计数据，包括各平台解析次数、使用用户数等',
+        icon: 'ph:chart-bar-fill',
+        roles: ['member', 'master']
+      },
+      {
+        title: 'kkk全局解析统计',
+        description: '查看全局解析统计数据，包括所有群组的解析情况、趋势分析和群组排行',
+        icon: 'ph:chart-line-up-fill',
+        roles: ['master']
+      }
+    ]
+  },
+  {
+    title: '推送相关',
+    items: [
+      {
+        title: '抖音/B站推送列表',
+        description: '查看当前群的订阅推送列表',
+        icon: 'ph:list-checks-fill',
+        roles: ['master']
+      },
+      {
+        title: '抖音/B站强制推送',
+        description: '全部强制推送：手动模拟一次定时任务；\n强制推送：只在触发群模拟一次定时任务；\n已推送过的不会再推送',
+        icon: 'ph:paper-plane-right-fill',
+        roles: ['master']
+      },
+      {
+        title: 'kkk推送全局忽略 + 链接',
+        description: '对抖音作品或B站动态进行全局忽略，所有群组的推送标记为已处理',
+        icon: 'ph:arrows-clockwise-fill',
+        roles: ['master']
+      }
+    ],
+    subGroups: [
+      {
+        title: '在群聊中再发送一次即可取消订阅',
+        items: [
+          {
+            title: '设置抖音推送 + 抖音号',
+            description: '在群聊中发送以对该群订阅该抖音博主的作品更新',
+            icon: 'ph:bell-fill',
+            roles: Config.douyin.push.permission === 'all' ? ['member', 'master'] : ['master']
+          },
+          {
+            title: '设置B站推送 + UP主UID',
+            description: '在群聊中发送以对该群订阅该B站UP主的稿件/动态更新',
+            icon: 'ph:bell-fill',
+            roles: Config.bilibili.push.permission === 'all' ? ['member', 'master'] : ['master']
+          }
+        ]
+      }
+    ]
+  },
+  {
+    title: '设置相关',
+    items: [
+      {
+        title: 'kkk设置推送机器人 + Bot ID',
+        description: '一键更换推送机器人',
+        icon: 'ph:robot-fill',
+        roles: ['master']
+      },
+      {
+        title: '抖音登录',
+        description: '使用抖音APP扫码登录获取 Cookies',
+        icon: 'logos:tiktok-icon',
+        roles: ['master']
+      },
+      {
+        title: 'B站登录',
+        description: '使用哔哩哔哩APP扫码登录获取 Cookies',
+        icon: {
+          name: 'streamline-ultimate:bilibili-logo-bold',
+          color: '#7fe1fa'
+        },
+        roles: ['master']
+      }
+    ]
+  },
+  {
+    title: '其他',
+    items: [
+      {
+        title: 'kkk版本',
+        description: '查看插件、Karin、Node.js、适配器与系统资源等运行环境诊断信息',
+        icon: 'ph:monitor-fill',
+        roles: ['member', 'master']
+      },
+      {
+        title: '「kkk更新日志」「kkk更新」',
+        description: '查看更新日志或执行插件更新',
+        icon: 'ph:arrows-clockwise-fill',
+        roles: ['master']
+      }
+    ]
+  }
+]
+
+const buildMenuForRole = (role: Role) => {
+  const filterItems = (items: RoleItem[] = []) =>
+    items.filter((i) => !i.roles || i.roles.includes(role)).map(({ title, description, icon }) => ({ title, description, icon }))
+
+  return HELP_MENU_CONFIG.map((group) => {
+    const items = filterItems(group.items)
+    const subGroups = group.subGroups?.map((sg) => ({ title: sg.title, items: filterItems(sg.items) })).filter((s) => s.items.length > 0)
+
+    return { title: group.title, items, subGroups }
+  }).filter((g) => g.items.length > 0 || (g.subGroups && g.subGroups.length > 0))
+}
+
+// 包装帮助命令
+const handleHelp = wrapWithErrorHandler(
+  async (e) => {
+    const masters = config.master().filter((id) => id !== 'console')
+    const isMaster = !!e.sender && masters.includes(e.sender.userId)
+    const role: Role = isMaster ? 'master' : 'member'
+    const menu = buildMenuForRole(role)
+
+    // 将 menu 转换为 list 供前端渲染
+    const list = menu.flatMap((group) => {
+      const groupItems = group.items.map((item) => ({
+        title: item.title,
+        description: item.description
+      }))
+      const subItems =
+        group.subGroups?.flatMap((sg) =>
+          sg.items.map((item) => ({
+            title: item.title,
+            description: item.description
+          }))
+        ) || []
+      return [...groupItems, ...subItems]
+    })
+
+    const img = await Render(e, 'other/help', {
+      title: 'KKK插件帮助页面',
+      menu,
+      list,
+      role
+    })
+    await e.reply(img)
+    return true
+  },
+  {
+    businessName: 'KKK帮助'
+  }
+)
+
+// 包装版本命令
+const handleVersion = wrapWithErrorHandler(
+  async (e) => {
+    const img = await Render(e, 'other/runtime', collectRuntimeReport(e))
+    await e.reply(img)
+    return true
+  },
+  {
+    businessName: 'KKK版本'
+  }
+)
+
+// 包装更新日志命令
+const handleChangelog = wrapWithErrorHandler(
+  async (e) => {
+    const forwardLogs = getLocalChangelog(10)
+    if (!forwardLogs) {
+      throw new Error('当前构建未携带可用的 CHANGELOG.md')
+    }
+
+    const img = await Render(e, 'other/changelog', {
+      markdown: forwardLogs,
+      Tip: false,
+      localVersion: '',
+      remoteVersion: ''
+    })
+    await e.reply(img)
+    return true
+  },
+  {
+    businessName: 'KKK更新日志'
+  }
+)
+
+export const help = karin.command(/^#?kkk帮助$/, handleHelp, { name: 'kkk-帮助' })
+
+export const version = karin.command(/^#?kkk版本$/, handleVersion, { name: 'kkk-版本' })
+
+export const changelog = karin.command(/^#?kkk更新日志$/, handleChangelog, { name: 'kkk-更新日志' })

package/src/karin/apps/push.ts

@@ -0,0 +1,327 @@
+import karin, { logger } from 'node-karin'
+
+import { bilibiliDB, douyinDB } from '@/module/db'
+import { bilibiliFetcher, douyinFetcher } from '@/module/utils/amagiClient'
+import { Config } from '@/module/utils/Config'
+import { wrapWithErrorHandler } from '@/module/utils/ErrorHandler'
+import { Bilibilipush, DouYinpush, getBilibiliID, getDouyinID } from '@/platform'
+
+// 包装抖音推送任务
+const handleDouyinPush = wrapWithErrorHandler(
+  async () => {
+    await new DouYinpush().action()
+    return true
+  },
+  {
+    businessName: '抖音推送任务'
+  }
+)
+
+// 包装B站推送任务
+const handleBilibiliPush = wrapWithErrorHandler(
+  async () => {
+    await new Bilibilipush().action()
+    return true
+  },
+  {
+    businessName: 'B站推送任务'
+  }
+)
+
+// 包装强制推送命令
+const handleForcePush = wrapWithErrorHandler(
+  async (e) => {
+    if (e.msg.includes('抖音')) {
+      await new DouYinpush().action()
+      return true
+    } else if (e.msg.includes('B站')) {
+      await new Bilibilipush().action()
+      return true
+    }
+    return true
+  },
+  {
+    businessName: '强制推送'
+  }
+)
+
+// 包装设置抖音推送命令
+const handleSetDouyinPush = wrapWithErrorHandler(
+  async (e) => {
+    const query = e.msg.replace(/^#设置抖音推送/, '').trim()
+
+    // 检查是否是开启/关闭命令
+    if (query === '开启' || query === '关闭') {
+      const enable = query === '开启'
+      Config.Modify('douyin', 'push.switch', enable)
+      await e.reply(`抖音推送已${enable ? '开启' : '关闭'}，${enable ? '需要重启后生效' : '将在下次重启后停止推送'}`)
+      logger.info(`抖音推送已${enable ? '开启' : '关闭'}`)
+      return true
+    }
+
+    // 原有的订阅逻辑
+    const data = await douyinFetcher.searchContent({
+      query,
+      type: 'user'
+    })
+    await new DouYinpush(e).setting(data.data)
+    return true
+  },
+  {
+    businessName: '设置抖音推送'
+  }
+)
+
+// 包装设置B站推送命令
+const handleSetBilibiliPush = wrapWithErrorHandler(
+  async (e) => {
+    const query = e.msg
+      .replace(/^#设置[bB]站推送/, '')
+      .replace(/^(?:[Uu][Ii][Dd]:)?/, '')
+      .trim()
+
+    // 检查是否是开启/关闭命令
+    if (query === '开启' || query === '关闭') {
+      const enable = query === '开启'
+      Config.Modify('bilibili', 'push.switch', enable)
+      await e.reply(`B站推送已${enable ? '开启' : '关闭'}，${enable ? '需要重启后生效' : '将在下次重启后停止推送'}`)
+      logger.info(`B站推送已${enable ? '开启' : '关闭'}`)
+      return true
+    }
+
+    // 原有的订阅逻辑
+    if (!Config.amagi.cookies.bilibili) {
+      await e.reply('\n请先配置B站Cookie', { at: true })
+      return true
+    }
+    const match = /^(\d+)$/.exec(query)
+    if (match && match[1]) {
+      const data = await bilibiliFetcher.fetchUserCard({
+        host_mid: Number(match[1])
+      })
+      await new Bilibilipush(e).setting(data.data)
+    }
+    return true
+  },
+  {
+    businessName: '设置B站推送'
+  }
+)
+
+// 包装推送列表命令
+const handleBilibiliPushList = wrapWithErrorHandler(
+  async (e) => {
+    await new Bilibilipush(e).renderPushList()
+  },
+  {
+    businessName: 'B站推送列表'
+  }
+)
+
+const handleDouyinPushList = wrapWithErrorHandler(
+  async (e) => {
+    await new DouYinpush(e).renderPushList()
+  },
+  {
+    businessName: '抖音推送列表'
+  }
+)
+
+// 包装设置机器人ID命令
+const handleChangeBotID = wrapWithErrorHandler(
+  async (e) => {
+    const newBotId = e.msg.replace(/^#kkk设置推送机器人/, '')
+
+    // 更新抖音配置和数据库
+    const newDouyinlist = Config.pushlist.douyin.map((item) => {
+      const modifiedGroupIds = item.group_id.map((groupId) => {
+        const [group_id, oldBotId] = groupId.split(':')
+        // 更新数据库中的botId
+        if (oldBotId && oldBotId !== newBotId) {
+          douyinDB.updateGroupBotId(group_id, oldBotId, newBotId).catch((err) => {
+            logger.error(`Failed to update douyin group ${group_id}:`, err)
+          })
+        }
+        return `${group_id}:${newBotId}`
+      })
+      return {
+        ...item,
+        group_id: modifiedGroupIds
+      }
+    })
+
+    // 更新B站配置和数据库
+    const newBilibililist = Config.pushlist.bilibili.map((item) => {
+      const modifiedGroupIds = item.group_id.map((groupId) => {
+        const [group_id, oldBotId] = groupId.split(':')
+        // 更新数据库中的botId
+        if (oldBotId && oldBotId !== newBotId) {
+          bilibiliDB.updateGroupBotId(group_id, oldBotId, newBotId).catch((err) => {
+            logger.error(`Failed to update bilibili group ${group_id}:`, err)
+          })
+        }
+        return `${group_id}:${newBotId}`
+      })
+      return {
+        ...item,
+        group_id: modifiedGroupIds
+      }
+    })
+
+    Config.Modify('pushlist', 'douyin', newDouyinlist)
+    Config.Modify('pushlist', 'bilibili', newBilibililist)
+    await e.reply('推送机器人已修改为' + newBotId)
+    return true
+  },
+  {
+    businessName: '设置推送机器人'
+  }
+)
+
+// 包装全局忽略命令
+const handleGlobalIgnore = wrapWithErrorHandler(
+  async (e) => {
+    const url = e.msg.replace(/^#kkk推送全局忽略/, '').trim()
+
+    if (!url) {
+      await e.reply('请提供链接')
+      return true
+    }
+
+    // 判断平台
+    const isDouyin = /(douyin|iesdouyin)\.com/.test(url)
+    const isBilibili = /bilibili\.com/.test(url)
+
+    if (!isDouyin && !isBilibili) {
+      await e.reply('暂不支持该平台链接')
+      return true
+    }
+
+    if (isDouyin) {
+      const idData = await getDouyinID(e, url, false)
+      if (!idData.aweme_id) {
+        await e.reply('无法解析该抖音链接')
+        return true
+      }
+
+      // 获取作品详情以获取 sec_uid
+      const workInfo = await douyinFetcher.parseWork({ aweme_id: idData.aweme_id })
+      const sec_uid = workInfo.data?.aweme_detail?.author?.sec_uid
+
+      if (!sec_uid) {
+        await e.reply('无法获取该作品作者信息')
+        return true
+      }
+
+      // 检查该作者是否在订阅配置中
+      const subscribedItem = Config.pushlist.douyin?.find((item: any) => item.sec_uid === sec_uid)
+      if (!subscribedItem) {
+        await e.reply('该作品对应的博主未在推送订阅中，跳过')
+        return true
+      }
+
+      // 获取所有订阅该作者的群组
+      const groupIds = subscribedItem.group_id.map((g: string) => g.split(':')[0])
+
+      // 将作品缓存写入所有订阅群组
+      for (const groupId of groupIds) {
+        await douyinDB.addAwemeCache(idData.aweme_id, sec_uid, groupId, 'post')
+      }
+
+      await e.reply(`已忽略抖音作品 ${idData.aweme_id}，共 ${groupIds.length} 个群组的推送标记为已处理`)
+      logger.info(`全局忽略抖音作品 ${idData.aweme_id}，博主 sec_uid: ${sec_uid}`)
+      return true
+    }
+
+    if (isBilibili) {
+      const idData = await getBilibiliID(url)
+      if (!idData.dynamic_id) {
+        await e.reply('无法解析该B站链接或该链接不是动态链接')
+        return true
+      }
+
+      // 获取动态详情以获取 host_mid
+      const dynamicInfo = await bilibiliFetcher.fetchDynamicDetail({
+        dynamic_id: idData.dynamic_id
+      })
+      const host_mid = dynamicInfo.data.data.item.modules.module_author.mid
+
+      if (!host_mid) {
+        await e.reply('无法获取该动态作者信息')
+        return true
+      }
+
+      // 检查该UP主是否在订阅配置中
+      const subscribedItem = Config.pushlist.bilibili?.find((item: any) => item.host_mid === Number(host_mid))
+      if (!subscribedItem) {
+        await e.reply('该动态对应的UP主未在推送订阅中，跳过')
+        return true
+      }
+
+      // 获取所有订阅该UP主的群组
+      const groupIds = subscribedItem.group_id.map((g: string) => g.split(':')[0])
+      const dynamicType = dynamicInfo.data?.data?.card?.desc?.type ?? 'DYNAMIC_TYPE_WORD'
+
+      // 将动态缓存写入所有订阅群组
+      for (const groupId of groupIds) {
+        await bilibiliDB.addDynamicCache(idData.dynamic_id, Number(host_mid), groupId, String(dynamicType))
+      }
+
+      await e.reply(`已忽略B站动态 ${idData.dynamic_id}，共 ${groupIds.length} 个群组的推送标记为已处理`)
+      logger.info(`全局忽略B站动态 ${idData.dynamic_id}，UP主 host_mid: ${host_mid}`)
+      return true
+    }
+
+    return true
+  },
+  {
+    businessName: '推送全局忽略'
+  }
+)
+
+// 注册任务和命令
+export const douyinPush =
+  Config.douyin.push.switch && karin.task('抖音推送', Config.douyin.push.cron, handleDouyinPush, { log: true, type: 'skip' })
+
+export const bilibiliPush =
+  Config.bilibili.push.switch && karin.task('B站推送', Config.bilibili.push.cron, handleBilibiliPush, { log: true, type: 'skip' })
+
+export const forcePush = karin.command(/#(抖音|B站)(全部)?强制推送/, handleForcePush, {
+  name: '𝑪𝒊𝒂𝒍𝒍𝒐～(∠・ω< )⌒★',
+  perm: 'master',
+  event: 'message.group'
+})
+
+export const setdyPush = karin.command(/^#设置抖音推送/, handleSetDouyinPush, {
+  name: 'kkk-推送功能-设置',
+  event: 'message.group',
+  perm: Config.douyin.push.permission
+})
+
+export const setbiliPush = karin.command(/^#设置[bB]站推送/, handleSetBilibiliPush, {
+  name: 'kkk-推送功能-设置',
+  event: 'message.group',
+  perm: Config.bilibili.push.permission
+})
+
+export const bilibiliPushList = karin.command(/^#?[bB]站推送列表$/, handleBilibiliPushList, {
+  name: 'kkk-推送功能-列表',
+  event: 'message.group'
+})
+
+export const douyinPushList = karin.command(/^#?抖音推送列表$/, handleDouyinPushList, {
+  name: 'kkk-推送功能-列表',
+  event: 'message.group'
+})
+
+export const changeBotID = karin.command(/^#kkk设置推送机器人/, handleChangeBotID, {
+  name: 'kkk-推送功能-设置',
+  perm: 'master'
+})
+
+export const globalIgnore = karin.command(/^#kkk推送全局忽略/, handleGlobalIgnore, {
+  name: 'kkk-推送功能-全局忽略',
+  perm: 'master',
+  event: 'message.group',
+  priority: 1
+})