koishi-plugin-kkk 1.0.0

package/src/karin/apps/qrlogin.ts

@@ -0,0 +1,160 @@
+import os from 'node:os'
+
+import karin, { logger } from 'node-karin'
+
+import { resolveTriggerAvatarUrl } from '@/module/utils/bot'
+import { Config } from '@/module/utils/Config'
+import { wrapWithErrorHandler } from '@/module/utils/ErrorHandler'
+import { Render } from '@/module/utils/Render'
+
+/**
+ * 获取本机局域网 IP 地址
+ * 优先返回常见局域网网段的 IP（192.168.x.x, 10.x.x.x, 172.16-31.x.x）
+ */
+function getLocalIP(): string {
+  const interfaces = os.networkInterfaces()
+  const candidates: string[] = []
+
+  for (const name of Object.keys(interfaces)) {
+    const netInterface = interfaces[name]
+    if (!netInterface) continue
+    for (const net of netInterface) {
+      if (net.family === 'IPv4' && !net.internal) {
+        candidates.push(net.address)
+      }
+    }
+  }
+
+  // 优先选择常见局域网网段
+  const preferredIP = candidates.find((ip) => {
+    // 192.168.x.x
+    if (ip.startsWith('192.168.')) return true
+    // 10.x.x.x
+    if (ip.startsWith('10.')) return true
+    // 172.16.x.x - 172.31.x.x
+    if (ip.startsWith('172.')) {
+      const second = parseInt(ip.split('.')[1], 10)
+      if (second >= 16 && second <= 31) return true
+    }
+    return false
+  })
+
+  return preferredIP || candidates[0] || '127.0.0.1'
+}
+
+/**
+ * 获取公网 IP 地址
+ * 通过多个公共 API 尝试获取
+ */
+async function getPublicIP(): Promise<string | null> {
+  const apis = ['4.ipw.cn', 'https://api.ipify.org', 'https://icanhazip.com', 'https://ifconfig.me/ip', 'https://api.ip.sb/ip']
+
+  for (const api of apis) {
+    try {
+      const controller = new AbortController()
+      const timeout = setTimeout(() => controller.abort(), 5000)
+
+      const response = await fetch(api, { signal: controller.signal })
+      clearTimeout(timeout)
+
+      if (response.ok) {
+        const ip = (await response.text()).trim()
+        // 验证是否为有效的 IPv4 地址
+        if (/^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$/.test(ip)) {
+          return ip
+        }
+      }
+    } catch {
+      // 继续尝试下一个 API
+    }
+  }
+
+  return null
+}
+
+/**
+ * 根据配置获取服务器地址
+ */
+async function getHostByConfig(): Promise<string> {
+  const addrType = Config.app.qrLoginAddrType || 'lan'
+
+  if (addrType === 'external') {
+    const externalAddr = Config.app.qrLoginExternalAddr
+    if (externalAddr && externalAddr.trim()) {
+      return externalAddr.trim()
+    }
+    // 如果配置了外部地址但未填写，尝试自动获取公网 IP
+    logger.debug('[APP扫码登录] 未配置外部地址，正在尝试获取公网 IP...')
+    const publicIP = await getPublicIP()
+    if (publicIP) {
+      logger.debug(`[APP扫码登录] 获取到公网 IP: ${publicIP}`)
+      return publicIP
+    }
+    logger.warn('[APP扫码登录] 无法获取公网 IP，回退到局域网 IP')
+    return getLocalIP()
+  }
+
+  return getLocalIP()
+}
+
+/**
+ * 生成登录二维码（仅私发给主人）
+ */
+const handleQrLogin = wrapWithErrorHandler(
+  async (e) => {
+    const bot = karin.getBot(e.selfId)
+    const userId = e.userId
+
+    // 先检查好友关系，避免渲染后无法发送浪费资源
+    if (e.isGroup) {
+      const friendList = await bot?.getFriendList()
+      const isFriend = friendList?.some((f) => f.userId === userId)
+      if (!isFriend) {
+        await e.reply('请先添加 Bot 为好友后再试')
+        return true
+      }
+    }
+
+    const port = process.env.HTTP_PORT || '7777'
+    const authKey = process.env.HTTP_AUTH_KEY || ''
+    const host = await getHostByConfig()
+    const protocol = 'http'
+
+    if (!authKey) {
+      await e.reply('未配置 HTTP_AUTH_KEY 环境变量，无法生成登录二维码')
+      return true
+    }
+
+    // 构建二维码数据（JSON 格式，包含服务器信息）
+    const qrData = JSON.stringify({
+      protocol,
+      host,
+      port,
+      authKey
+    })
+
+    const serverUrl = `${protocol}://${host}:${port}`
+
+    // 使用模板系统渲染二维码图片
+    const images = await Render(e, 'other/qrlogin', {
+      share_url: qrData,
+      serverUrl,
+      avatarUrl: await resolveTriggerAvatarUrl(e)
+    })
+
+    // 私发给触发命令的用户
+    await karin.sendMaster(e.selfId, userId, images)
+
+    // 如果是群聊触发，提示已私发
+    if (e.isGroup) {
+      await e.reply('登录二维码已私聊发送，请查收~')
+    }
+
+    return true
+  },
+  {
+    businessName: 'APP扫码登录'
+  }
+)
+
+export const qrLogin = karin.command(/^#?(kkk)?登录$/i, handleQrLogin, { perm: 'master', name: 'kkk-APP扫码登录' })

package/src/karin/apps/statistics.ts

@@ -0,0 +1,203 @@
+import { format } from 'date-fns'
+import karin, { logger } from 'node-karin'
+
+import { Render } from '@/module'
+import { getStatisticsDB } from '@/module/db'
+import { resolveAvatar } from '@/module/utils/avatar'
+import { wrapWithErrorHandler } from '@/module/utils/ErrorHandler'
+import { aggregateGlobal, aggregateGroup, TREND_DAYS } from '@/module/utils/statisticsAggregate'
+
+/** 向 bot 索取头像时用的尺寸档；接口只接受 0/40/100/140 四档 */
+const AVATAR_FETCH_SIZE = 100
+
+/**
+ * 向 bot 要头像 URL。
+ *
+ * 适配器拿不到头像是常态（QQBot 有概率取不到群头像），失败或返回空串都统一返回 undefined，
+ * 交给 `resolveAvatar` 回落到现场生成的 dither 头像。
+ * @param get 真正发起请求的闭包
+ */
+const getAvatarUrl = async (get: () => Promise<string>): Promise<string | undefined> => {
+  try {
+    return (await get()) || undefined
+  } catch (error) {
+    logger.debug('[统计] 获取头像地址失败，将使用生成头像:', error)
+    return undefined
+  }
+}
+
+/**
+ * 两个统计命令。
+ *
+ * 这里只负责「取数 → 聚合 → 补 bot 侧信息 → 渲染」，
+ * 具体的行 → 图表数据变换全在 `@/module/utils/statisticsAggregate` 里（纯函数，可单测）。
+ */
+
+/**
+ * #kkk解析统计 命令
+ * 获取当前群的解析统计数据
+ */
+const handleGroupStatistics = wrapWithErrorHandler(
+  async (e) => {
+    // 获取群组ID
+    const groupId = e.isGroup ? e.contact?.peer || '' : ''
+
+    if (!groupId) {
+      await e.reply('此命令仅支持在群聊中使用')
+      return true
+    }
+
+    // 获取群组信息
+    let groupName = ''
+    let groupMemberCount: number | undefined
+    let groupAvatarUrl: string | undefined
+    try {
+      const groupInfo = await e.bot.getGroupInfo(groupId)
+      groupName = groupInfo?.groupName || ''
+      groupMemberCount = groupInfo?.memberCount
+      groupAvatarUrl = groupInfo?.avatar
+    } catch (error) {
+      logger.debug('[统计] 获取群组信息失败:', error)
+    }
+    // 群头像：优先问 bot 要（`getGroupAvatarUrl` 比 `groupInfo.avatar` 更可靠，
+    // 部分适配器的群信息里不带头像），拿不到就现场生成一个，保证头部不会空着
+    const groupAvatar = await resolveAvatar(groupId, await getAvatarUrl(() => e.bot.getGroupAvatarUrl(groupId, AVATAR_FETCH_SIZE)) ?? groupAvatarUrl)
+
+    // 获取统计数据库实例
+    const statisticsDB = await getStatisticsDB()
+
+    // 逐条取数（node-sqlite3 单连接，顺序执行最稳）
+    const groupStats = await statisticsDB.getGroupStatistics(groupId)
+    const groupHistory = await statisticsDB.getGroupRecentHistory(groupId, TREND_DAYS)
+    const hourRows = await statisticsDB.getGroupHourStats(groupId)
+    const workTypeRows = await statisticsDB.getGroupWorkTypeStats(groupId)
+    const activeDays = await statisticsDB.getGroupActiveDays(groupId)
+    const metricRows = await statisticsDB.getGroupMetricStats(groupId)
+    const globalHistory = await statisticsDB.getRecentHistory(TREND_DAYS)
+    const groupUniqueUsers = await statisticsDB.getGroupUniqueUsers(groupId)
+    const globalSummary = await statisticsDB.getGlobalSummary()
+
+    const { topUserRows, ...groupData } = aggregateGroup({
+      groupStats,
+      groupHistory,
+      hourRows,
+      workTypeRows,
+      metricRows,
+      globalHistory,
+      activeDays,
+      uniqueUsers: groupUniqueUsers
+    })
+
+    // 活跃用户榜要补昵称：群名片优先，取不到回落昵称，再取不到回落用户ID。
+    // 单次 API 失败不能把整张海报带崩，所以逐个兜住。
+    // 昵称和头像一起并行取：十个人串行等十轮太慢，而且单个人失败不该拖垮整张海报
+    const topUsers = await Promise.all(
+      topUserRows.map(async (row) => {
+        const [memberInfo, avatarUrl] = await Promise.all([
+          e.bot.getGroupMemberInfo(groupId, row.key).catch((error) => {
+            logger.debug(`[统计] 获取群成员 ${row.key} 信息失败:`, error)
+            return undefined
+          }),
+          getAvatarUrl(() => e.bot.getAvatarUrl(row.key, AVATAR_FETCH_SIZE))
+        ])
+        return {
+          userId: row.key,
+          name: memberInfo?.card || memberInfo?.nick || row.key,
+          count: row.count,
+          // segments 是聚合层算好的平台构成，补昵称时原样带上
+          segments: row.segments,
+          avatar: await resolveAvatar(row.key, avatarUrl)
+        }
+      })
+    )
+
+    // 渲染统计图片
+    const img = await Render(e, 'statistics/group', {
+      groupId,
+      groupName,
+      groupMemberCount,
+      groupAvatar,
+      generatedAt: format(new Date(), 'yyyy-MM-dd HH:mm'),
+      ...groupData,
+      topUsers,
+      globalTotalGroups: globalSummary.totalGroups,
+      globalTotalParses: globalSummary.totalParses
+    })
+
+    await e.reply(img)
+    return true
+  },
+  {
+    businessName: '群组解析统计'
+  }
+)
+
+export const groupStatistics = karin.command(/^#?kkk解析统计$/, handleGroupStatistics, { name: 'kkk-解析统计' })
+
+/**
+ * #kkk全局解析统计 命令
+ * 获取整个插件的全局解析统计数据
+ */
+const handleGlobalStatistics = wrapWithErrorHandler(
+  async (e) => {
+    // 获取统计数据库实例
+    const statisticsDB = await getStatisticsDB()
+
+    // 逐条取数（node-sqlite3 单连接，顺序执行最稳）
+    const allStats = await statisticsDB.getAllStatistics()
+    const historyData = await statisticsDB.getRecentHistory(TREND_DAYS)
+    const historyCompleteFrom = await statisticsDB.getHistoryCompleteFrom()
+    const workTypeRows = await statisticsDB.getGlobalWorkTypeStats()
+    const firstSeen = await statisticsDB.getGroupFirstSeen()
+    const metricRows = await statisticsDB.getGlobalMetricStats()
+
+    const { topGroupRows, ...globalData } = aggregateGlobal({
+      allStats,
+      historyData,
+      firstSeen,
+      workTypeRows,
+      metricRows,
+      historyCompleteFrom
+    })
+
+    // 群组排行补群名（同样逐个兜住）
+    const topGroups = await Promise.all(
+      topGroupRows.map(async (row) => {
+        const [groupInfo, avatarUrl] = await Promise.all([
+          e.bot.getGroupInfo(row.groupId).catch((error) => {
+            logger.debug(`[统计] 获取群组 ${row.groupId} 信息失败:`, error)
+            return undefined
+          }),
+          getAvatarUrl(() => e.bot.getGroupAvatarUrl(row.groupId, AVATAR_FETCH_SIZE))
+        ])
+        return {
+          ...row,
+          name: groupInfo?.groupName || row.groupId,
+          avatar: await resolveAvatar(row.groupId, avatarUrl ?? groupInfo?.avatar)
+        }
+      })
+    )
+
+    // 统计起始日：所有群里最早的一次记录
+    const statsSince = firstSeen.length > 0 ? firstSeen.map((row) => row.firstSeen.split('T')[0]).sort()[0] : undefined
+
+    // 渲染统计图片
+    const img = await Render(e, 'statistics/global', {
+      generatedAt: format(new Date(), 'yyyy-MM-dd HH:mm'),
+      ...globalData,
+      statsSince,
+      topGroups
+    })
+
+    await e.reply(img)
+    return true
+  },
+  {
+    businessName: '全局解析统计'
+  }
+)
+
+export const globalStatistics = karin.command(/^#?kkk全局解析统计$/, handleGlobalStatistics, {
+  name: 'kkk-全局解析统计',
+  perm: 'master'
+})

package/src/karin/apps/testPush.ts

@@ -0,0 +1,226 @@
+import type { DouyinUserProfileResponse } from '@ikenxuan/amagi'
+import karin, { type ImageElement, logger } from 'node-karin'
+
+import { douyinFetcher } from '@/module/utils/amagiClient'
+import { Config } from '@/module/utils/Config'
+import { wrapWithErrorHandler } from '@/module/utils/ErrorHandler'
+import { getDouyinID } from '@/platform/douyin/getID'
+import { renderFavoriteImage, renderLiveImage, renderRecommendImage, renderWorkImage } from '@/platform/douyin/push/render'
+import { buildDouyinWorkDetail } from '@/platform/douyin/types'
+import type { DouyinWorkDetailData } from '@/platform/douyin/types'
+import { buildDouyinPlayUrl, douyinProcessVideos, type dyVideo } from '@/platform/douyin/videoQuality'
+import { getWorkTypeInfo } from '@/platform/douyin/workType'
+
+/** 构建与生产推送一致的作品二维码链接。 */
+function buildWorkShareLink(aweme: DouyinWorkDetailData, selectedVideo: dyVideo | null): string {
+  const workTypeInfo = getWorkTypeInfo(aweme)
+  if (workTypeInfo.isArticle) return `https://www.douyin.com/article/${aweme.aweme_id}`
+  if (workTypeInfo.isImage) return `https://www.douyin.com/note/${aweme.aweme_id}`
+  const playAddr = selectedVideo?.play_addr ?? aweme.video?.play_addr
+  return playAddr ? buildDouyinPlayUrl(playAddr) : `https://www.douyin.com/video/${aweme.aweme_id}`
+}
+
+/**
+ * 测试抖音推送命令处理器
+ * 支持四种推送类型的预览渲染，无数据库交互，仅用于调试和验证推送卡片效果
+ *
+ * 命令格式：
+ * - #测试抖音作品推送 <作品链接>
+ * - #测试抖音喜欢列表推送 <用户主页链接>
+ * - #测试抖音推荐列表推送 <用户主页链接>
+ * - #测试抖音直播状态推送 <用户主页链接>
+ */
+const handleTestPush = wrapWithErrorHandler(
+  async (e) => {
+    const match = e.msg.match(/^#测试抖音(作品|喜欢列表|推荐列表|直播状态)推送/)
+    if (!match) {
+      e.reply(
+        '支持的命令（前缀按 Koishi 配置）：\n测试抖音作品推送 <作品链接>\n测试抖音喜欢列表推送 <用户主页链接>\n测试抖音推荐列表推送 <用户主页链接>\n测试抖音直播状态推送 <用户主页链接>'
+      )
+      return true
+    }
+
+    /** 提取推送类型和链接 */
+    const pushType = match[1] as '作品' | '喜欢列表' | '推荐列表' | '直播状态'
+    const restMsg = e.msg.replace(match[0], '').trim()
+
+    const urlMatch = restMsg.match(/(https?:\/\/[^\s]+)/i)
+    if (!urlMatch) {
+      e.reply(`请在命令后提供对应的${pushType === '作品' ? '作品' : '用户主页'}链接`)
+      return true
+    }
+    const url = urlMatch[1]
+    const iddata = await getDouyinID(e, url, false)
+    let images: ImageElement[] = []
+
+    switch (pushType) {
+      /** 作品推送：作品链接 → parseWork → 渲染 */
+      case '作品': {
+        if (iddata.type !== 'one_work' || !iddata.aweme_id) {
+          e.reply('该链接不是作品链接，请提供视频/图集/文章链接')
+          return true
+        }
+        logger.mark(`[测试抖音推送] 开始解析作品: ${iddata.aweme_id}`)
+        const workData = await douyinFetcher.parseWork({ aweme_id: iddata.aweme_id })
+        if (!workData.data.aweme_detail) {
+          e.reply('获取作品详情失败，作品可能已被删除或设为私密')
+          return true
+        }
+        const aweme = workData.data.aweme_detail
+        const userinfo = await douyinFetcher.fetchUserProfile({ sec_uid: aweme.author.sec_uid })
+        const Detail_Data = buildDouyinWorkDetail(aweme, { user_info: userinfo.data })
+        // 与生产推送一致：先按画质配置选档，二维码链接和卡片清晰度都从选中那一路视频源派生
+        const selectedVideo = aweme.video?.bit_rate?.length
+          ? douyinProcessVideos(aweme.video.bit_rate, Config.douyin.videoQuality)[0]
+          : null
+        const shareLink = buildWorkShareLink(Detail_Data, selectedVideo)
+        images = await renderWorkImage({
+          e,
+          Detail_Data,
+          create_time: aweme.create_time,
+          shareLink,
+          videoSource: selectedVideo
+        })
+        if (images.length === 0) {
+          e.reply('未能识别该作品类型，无法渲染推送图片')
+          return true
+        }
+        break
+      }
+
+      /** 喜欢列表：用户主页 → fetchUserFavoriteList[0] → 渲染 */
+      case '喜欢列表': {
+        if (iddata.type !== 'user_dynamic' || !iddata.sec_uid) {
+          e.reply('需要用户主页链接以获取喜欢列表')
+          return true
+        }
+        logger.mark(`[测试抖音推送] 开始获取喜欢列表: sec_uid=${iddata.sec_uid}`)
+        const userinfo = await douyinFetcher.fetchUserProfile({ sec_uid: iddata.sec_uid })
+        const favoriteData = await douyinFetcher.fetchUserFavoriteList({
+          sec_uid: iddata.sec_uid,
+          number: 1
+        })
+        if (!favoriteData.data.aweme_list?.length) {
+          e.reply('该用户的喜欢列表为空或未公开')
+          return true
+        }
+        const aweme = favoriteData.data.aweme_list[0]
+        let authorUserInfo: DouyinUserProfileResponse | undefined
+        try {
+          authorUserInfo = (await douyinFetcher.fetchUserProfile({ sec_uid: aweme.author.sec_uid })).data
+        } catch {
+          /* ignore */
+        }
+        const Detail_Data = buildDouyinWorkDetail(aweme, { user_info: userinfo.data, author_user_info: authorUserInfo })
+        const selectedVideo = aweme.video?.bit_rate?.length
+          ? douyinProcessVideos(aweme.video.bit_rate, Config.douyin.videoQuality)[0]
+          : null
+        const shareLink = buildWorkShareLink(Detail_Data, selectedVideo)
+        images = await renderFavoriteImage({
+          e,
+          Detail_Data,
+          create_time: aweme.create_time,
+          shareLink,
+          remark: userinfo.data.user.nickname
+        })
+        if (!images.length) {
+          e.reply('渲染喜欢列表推送图片失败')
+          return true
+        }
+        break
+      }
+
+      /** 推荐列表：用户主页 → fetchUserRecommendList[0] → 渲染 */
+      case '推荐列表': {
+        if (iddata.type !== 'user_dynamic' || !iddata.sec_uid) {
+          e.reply('需要用户主页链接以获取推荐列表')
+          return true
+        }
+        logger.mark(`[测试抖音推送] 开始获取推荐列表: sec_uid=${iddata.sec_uid}`)
+        const userinfo = await douyinFetcher.fetchUserProfile({ sec_uid: iddata.sec_uid })
+        const recommendData = await douyinFetcher.fetchUserRecommendList({
+          sec_uid: iddata.sec_uid,
+          number: 1
+        })
+        if (!recommendData.data.aweme_list?.length) {
+          e.reply('该用户的推荐列表为空或未公开')
+          return true
+        }
+        const aweme = recommendData.data.aweme_list[0]
+        let authorUserInfo: DouyinUserProfileResponse | undefined
+        try {
+          authorUserInfo = (await douyinFetcher.fetchUserProfile({ sec_uid: aweme.author.sec_uid })).data
+        } catch {
+          /* ignore */
+        }
+        const Detail_Data = buildDouyinWorkDetail(aweme, { user_info: userinfo.data, author_user_info: authorUserInfo })
+        const selectedVideo = aweme.video?.bit_rate?.length
+          ? douyinProcessVideos(aweme.video.bit_rate, Config.douyin.videoQuality)[0]
+          : null
+        const shareLink = buildWorkShareLink(Detail_Data, selectedVideo)
+        images = await renderRecommendImage({
+          e,
+          Detail_Data,
+          create_time: aweme.create_time,
+          shareLink,
+          remark: userinfo.data.user.nickname
+        })
+        if (!images.length) {
+          e.reply('渲染推荐列表推送图片失败')
+          return true
+        }
+        break
+      }
+
+      /** 直播状态：用户主页 → 检查 live_status → fetchLiveRoomInfo → 渲染 */
+      case '直播状态': {
+        if (iddata.type !== 'user_dynamic' || !iddata.sec_uid) {
+          e.reply('需要用户主页链接以检查直播状态')
+          return true
+        }
+        const sec_uid = iddata.sec_uid
+        logger.mark(`[测试抖音推送] 开始检查直播状态: sec_uid=${sec_uid}`)
+        const userinfo = await douyinFetcher.fetchUserProfile({ sec_uid })
+        const user = userinfo.data.user
+        if (user.live_status !== 1) {
+          e.reply(`${user.nickname} 当前未在直播`)
+          return true
+        }
+        if (!user.room_data) {
+          e.reply('未获取到直播间信息')
+          return true
+        }
+        const room_data = JSON.parse(user.room_data)
+        const liveInfo = await douyinFetcher.fetchLiveRoomInfo({
+          room_id: user.room_id_str,
+          web_rid: room_data.owner.web_rid
+        })
+        const Detail_Data = { user_info: userinfo.data, room_data, live_data: liveInfo.data }
+        images = await renderLiveImage({ e, Detail_Data })
+        if (!images.length) {
+          e.reply('渲染直播状态推送图片失败')
+          return true
+        }
+        break
+      }
+    }
+
+    e.reply([
+      ...images
+      // segment.markdown('[跳转](mqqapi://forward/url?version=1&src_type=web&url_prefix=' + encodeURIComponent('https://www.douyin.com'))
+    ])
+    logger.mark(`[测试抖音推送] ${pushType}推送渲染完成`)
+    return true
+  },
+  {
+    businessName: '测试抖音推送'
+  }
+)
+
+/** 注册测试推送命令 */
+export const testPush =
+  Config.douyin.switch &&
+  karin.command(/^#测试抖音(作品|喜欢列表|推荐列表|直播状态)推送/, handleTestPush, {
+    name: 'kkk-测试抖音推送',
+    perm: 'master'
+  })