mioku-plugin-media 1.0.0 → 2.1.0

package/platforms/types.ts

@@ -3,6 +3,77 @@ export type Platform = "bilibili" | "douyin" | "kuaishou" | "xiaohongshu";
 export interface ParsedMediaUrl {
   platform: Platform;
   id: string;
-  subtype?: "video" | "article" | "note" | "bangumi";
+  subtype?: "video" | "article" | "note" | "bangumi" | "live";
   extra?: Record<string, string>;
 }
+
+export interface AmagiClient {
+  bilibili: {
+    fetcher: {
+      fetchVideoInfo(options: { bvid: string; typeMode?: "strict" | "loose" }): Promise<{
+        success: boolean;
+        data?: any;
+        message?: string;
+      }>;
+      fetchVideoStreamUrl(options: { avid: number; cid: number; typeMode?: "strict" | "loose" }): Promise<{
+        success: boolean;
+        data?: any;
+        message?: string;
+      }>;
+      fetchLiveRoomInfo(options: { room_id: string; typeMode?: "strict" | "loose" }): Promise<{
+        success: boolean;
+        data?: any;
+        message?: string;
+      }>;
+      fetchLiveRoomInitInfo(options: { room_id: string; typeMode?: "strict" | "loose" }): Promise<{
+        success: boolean;
+        data?: any;
+        message?: string;
+      }>;
+      fetchUserCard(options: { host_mid: number; typeMode?: "strict" | "loose" }): Promise<{
+        success: boolean;
+        data?: any;
+        message?: string;
+      }>;
+    };
+  };
+  kuaishou: {
+    fetcher: {
+      fetchVideoWork(options: {
+        photoId: string;
+        typeMode?: "strict" | "loose";
+      }): Promise<{
+        success: boolean;
+        code?: number | string | undefined;
+        data: any;
+        message?: string;
+        error?: any;
+      }>;
+    };
+  };
+  douyin: {
+    fetcher: {
+      parseWork(options: {
+        aweme_id: string;
+        typeMode?: "strict" | "loose";
+      }): Promise<{
+        success: boolean;
+        data?: any;
+        message?: string;
+      }>;
+    };
+  };
+  xiaohongshu: {
+    fetcher: {
+      fetchNoteDetail(options: {
+        note_id: string;
+        xsec_token: string;
+        typeMode?: "strict" | "loose";
+      }): Promise<{
+        success: boolean;
+        data?: any;
+        message?: string;
+      }>;
+    };
+  };
+}

package/platforms/url-parser.ts

@@ -1,4 +1,4 @@
-import type { Platform, ParsedMediaUrl } from "./types";
+import type { ParsedMediaUrl } from "./types";
 
 const BILIBILI_DOMAINS = [
   "bilibili.com",
@@ -20,7 +20,14 @@ const DOUYIN_DOMAINS = [
   "jingxuan.douyin.com",
 ];
 
-const KUAISHOU_DOMAINS = ["kuaishou.com", "www.kuaishou.com", "v.kuaishou.com"];
+const KUAISHOU_DOMAINS = [
+  "kuaishou.com",
+  "www.kuaishou.com",
+  "v.kuaishou.com",
+  "v.m.chenzhongtech.com",
+  "m.chenzhongtech.com",
+  "chenzhongtech.com",
+];
 
 const XIAOHONGSHU_DOMAINS = [
   "xiaohongshu.com",
@@ -31,8 +38,10 @@ const XIAOHONGSHU_DOMAINS = [
 const BV_REGEX = /\b(BV[a-zA-Z0-9]{10,})\b/;
 const AV_REGEX = /\b(av(\d+))\b/i;
 const URL_REGEX = /https?:\/\/[^\s<>"{}|\\^`\[\]]+/gi;
+const BILIBILI_LIVE_REGEX = /\/(\d+)(?:\?|$)/;
 const DOUYIN_ID_REGEX = /\/video\/(\d+)/;
 const KUAISHOU_ID_REGEX = /\/short-video\/([a-zA-Z0-9_-]+)/;
+const KUAISHOU_PHOTO_REGEX = /(?:\/fw)?\/photo\/([a-zA-Z0-9_-]+)/;
 const XHS_NOTE_REGEX = /\/explore\/([a-f0-9]{24})/;
 const XHS_DISCOVERY_REGEX = /\/discovery\/item\/([a-f0-9]{24})/;
 const XHSLINK_NOTE_REGEX = /\/([a-f0-9]{24})/;
@@ -133,6 +142,21 @@ function parseBilibiliUrl(url: string): ParsedMediaUrl | null {
     if (articleMatch) {
       return { platform: "bilibili", id: articleMatch[1], subtype: "article" };
     }
+
+    const liveMatch = url.match(/\/live\/(\d+)/);
+    if (liveMatch) {
+      return { platform: "bilibili", id: liveMatch[1], subtype: "live" };
+    }
+
+    if (
+      hostname === "live.bilibili.com" ||
+      hostname.endsWith(".live.bilibili.com")
+    ) {
+      const roomMatch = url.match(/\/(\d+)(?:\?|$)/);
+      if (roomMatch) {
+        return { platform: "bilibili", id: roomMatch[1], subtype: "live" };
+      }
+    }
   }
 
   return null;
@@ -174,11 +198,18 @@ function parseKuaishouUrl(url: string): ParsedMediaUrl | null {
     return { platform: "kuaishou", id: shortMatch[1], subtype: "video" };
   }
 
-  const photoMatch = url.match(/photo\/(\d+)/);
+  const photoMatch = url.match(KUAISHOU_PHOTO_REGEX);
   if (photoMatch) {
     return { platform: "kuaishou", id: photoMatch[1], subtype: "video" };
   }
 
+  // Handle v.kuaishou.com/xxx short URLs like https://v.kuaishou.com/KfhlEcGV
+  const shortCodeMatch = url.match(/:\/\/[^/]+\/([a-zA-Z0-9_-]+)/);
+  if (shortCodeMatch) {
+    // Store full URL so isShortUrl can detect it and resolveShortUrl can work
+    return { platform: "kuaishou", id: url, subtype: "video" };
+  }
+
   return null;
 }
 
@@ -363,29 +394,29 @@ export function extractMediaUrlFromEvent(event: any): ParsedMediaUrl | null {
 
 export function resolveShortUrl(url: string): Promise<string> {
   return new Promise((resolve) => {
-    try {
-      const controller = new AbortController();
-      const timeout = setTimeout(() => {
-        controller.abort();
-        resolve(url);
-      }, 5000);
-
-      fetch(url, {
-        redirect: "follow",
-        signal: controller.signal,
-        method: "HEAD",
-      })
-        .then((response) => {
-          clearTimeout(timeout);
-          resolve(response.url || url);
-        })
-        .catch(() => {
-          clearTimeout(timeout);
-          resolve(url);
-        });
-    } catch {
+    const controller = new AbortController();
+    const timeout = setTimeout(() => {
+      controller.abort();
       resolve(url);
-    }
+    }, 10000);
+
+    fetch(url, {
+      redirect: "follow",
+      signal: controller.signal,
+    })
+      .then((response) => {
+        clearTimeout(timeout);
+        const finalUrl = response.url;
+        if (finalUrl && finalUrl !== url) {
+          resolve(finalUrl);
+        } else {
+          resolve(url);
+        }
+      })
+      .catch(() => {
+        clearTimeout(timeout);
+        resolve(url);
+      });
   });
 }
 
@@ -399,5 +430,8 @@ export function isShortUrl(parsed: ParsedMediaUrl): boolean {
   if (parsed.platform === "douyin" && parsed.id.startsWith("http")) {
     return true;
   }
+  if (parsed.platform === "kuaishou" && parsed.id.startsWith("http")) {
+    return true;
+  }
   return false;
 }

package/runtime.ts

@@ -3,7 +3,7 @@ import {
   getPluginRuntimeState,
   resetPluginRuntimeState,
   setPluginRuntimeState,
-} from "../../src";
+} from "mioku";
 
 const PLUGIN_NAME = "media";
 

package/skills.ts

@@ -1,7 +1,10 @@
-import type { AISkill, AITool } from "../../src";
-import type { SendNodeContentElement } from "napcat-sdk";
+import type { AISkill, AITool } from "mioku";
 import { getMediaRuntimeState } from "./runtime";
-import { parseMediaUrl, resolveShortUrl, isShortUrl } from "./platforms/url-parser";
+import {
+  parseMediaUrl,
+  resolveShortUrl,
+  isShortUrl,
+} from "./platforms/url-parser";
 import { resolveMedia } from "./platforms/resolvers";
 import { buildInfoMessage, sendMediaResult } from "./utils/message";
 

package/types.ts

@@ -20,9 +20,30 @@ export interface ParsedMediaResult {
   videoUrl: string;
   duration?: number;
   stats?: MediaStats;
+  liveStatus?: string;
+  /** 图集/合辑的图片列表 */
+  images?: string[];
+  /** 图集/合辑的短视频/实况图视频URL列表 */
+  videoUrls?: string[];
+  /** 图集/合辑的背景音乐URL */
+  musicUrl?: string;
+  /** 是否包含实况图 */
+  hasLivePhoto?: boolean;
+  /** 是否为合辑（图集中的多图组合） */
+  isSlides?: boolean;
 }
 
 export interface MediaRuntimeState {
   config: MediaConfig;
-  amagiClient: any;
+  amagiClient: {
+    kuaishou: {
+      fetcher: {
+        fetchVideoWork(options: { photoId: string; typeMode?: "strict" | "loose" }): Promise<{
+          success: boolean;
+          data: any;
+          message?: string;
+        }>;
+      };
+    };
+  };
 }

package/utils/error-handler.ts

@@ -1,5 +1,8 @@
 import type { MiokiContext } from "mioki";
 import type { MediaConfig } from "../types";
+import type { ParsedMediaUrl } from "../platforms/types";
+import type { ParsedMediaResult } from "../types";
+import type { SendNodeElement, SendNodeContentElement } from "napcat-sdk";
 
 function normalizeErrorMessage(error: unknown): string {
   if (error instanceof Error && error.message) {
@@ -22,46 +25,36 @@ export async function handleMediaError(options: {
   platform: string;
   config: MediaConfig;
 }): Promise<void> {
-  const { ctx, event, error, platform, config } = options;
+  const { ctx, error, platform } = options;
   const errorMessage = normalizeErrorMessage(error);
+  ctx.logger.error(`[media] ${platform} 解析失败: ${errorMessage}`);
 
-  ctx.logger.error(`[media] ${platform} 解析失败: ${errorMessage}`, error);
+  // 尝试获取 bot 发送错误信息给用户
+  const selfId = event?.self_id != null ? Number(event.self_id) : undefined;
+  const bot =
+    selfId != null && typeof ctx?.pickBot === "function"
+      ? ctx.pickBot(selfId)
+      : undefined;
 
-  const cause = (error instanceof Error && (error as any).cause) ? (error as any).cause : null;
-  const ckExpired = cause?.error?.errorDescription?.includes("ck可能已经失效");
+  if (!bot) return;
 
-  if (ckExpired) {
-    await event.reply(`${platform} Cookie 已失效，请联系管理员更新`, true);
-    return;
-  }
-
-  if (config.debug) {
-    await event.reply(`${platform} 解析失败: ${errorMessage}`, true);
-    return;
-  }
+  const nickname = String(
+    ctx?.bot?.nickname || event?.sender?.card || event?.sender?.nickname || "媒体解析",
+  );
+  const userId = String(selfId || ctx?.bot?.bot_id || event?.self_id || 0);
 
-  const aiService = ctx.services?.ai as any | undefined;
-  const chatRuntime = aiService?.getChatRuntime?.();
+  // 构建错误提示消息
+  const errorText = `【${platform}】解析失败\n${errorMessage}`;
 
-  if (chatRuntime) {
-    try {
-      await chatRuntime.generateNotice({
-        event,
-        instruction: `媒体解析插件在解析${platform}内容时失败，错误信息: ${errorMessage}。请简短告知用户解析失败，并建议稍后重试或检查链接是否有效。`,
-        send: true,
-        promptInjections: [
-          {
-            title: "Media Plugin Notice",
-            content:
-              "A media link parsing action was triggered. Judge whether the user likely intended this action or triggered it accidentally. If it looks accidental or like a casual mention, weave a natural reply into the conversation without mentioning the plugin, tools, or commands. If the user seems to want this feature, respond helpfully. Keep the response concise.",
-          },
-        ],
-      });
+  try {
+    if (event?.message_type === "group" && event?.group_id != null) {
+      await bot.sendGroupMsg(event.group_id, [ctx.segment.text(errorText)]);
       return;
-    } catch (noticeError) {
-      ctx.logger.error(`[media] AI notice 发送失败: ${normalizeErrorMessage(noticeError)}`, noticeError);
     }
+    if (event?.user_id != null) {
+      await bot.sendPrivateMsg(event.user_id, [ctx.segment.text(errorText)]);
+    }
+  } catch {
+    // ignore send errors
   }
-
-  await event.reply(`解析失败，请稍后重试或检查链接是否有效`, true);
 }

package/utils/message.ts

@@ -1,4 +1,8 @@
-import type { SendNodeElement, SendNodeContentElement, ForwardDisplayOptions } from "napcat-sdk";
+import type {
+  SendNodeElement,
+  SendNodeContentElement,
+  ForwardDisplayOptions,
+} from "napcat-sdk";
 import type { ParsedMediaResult, MediaStats } from "../types";
 import type { ParsedMediaUrl } from "../platforms/types";
 
@@ -37,13 +41,20 @@ function formatCount(count?: number): string {
   return String(count);
 }
 
-export function buildInfoMessage(parsed: ParsedMediaUrl, result: ParsedMediaResult): string {
+export function buildInfoMessage(
+  parsed: ParsedMediaUrl,
+  result: ParsedMediaResult,
+): string {
   const platform = PLATFORM_NAMES[parsed.platform] || parsed.platform;
   const lines: string[] = [];
 
   lines.push(`【${platform}】${result.title}`);
   lines.push(`作者：${result.author}`);
 
+  if (result.liveStatus) {
+    lines.push(`状态：${result.liveStatus}`);
+  }
+
   if (result.duration && result.duration > 0) {
     lines.push(`时长：${formatDuration(result.duration)}`);
   }
@@ -53,22 +64,43 @@ export function buildInfoMessage(parsed: ParsedMediaUrl, result: ParsedMediaResu
     lines.push(`简介：${desc}`);
   }
 
+  // 对于图片内容，显示数量信息
+  if (result.images && result.images.length > 0) {
+    lines.push(`图片：${result.images.length}张`);
+  }
+  if (result.videoUrls && result.videoUrls.length > 0) {
+    lines.push(`视频：${result.videoUrls.length}个`);
+  }
+
   return lines.join("\n");
 }
 
-function buildSummaryText(platform: string, stats?: MediaStats): string {
+function buildSummaryText(
+  platform: string,
+  subtype: string | undefined,
+  stats?: MediaStats,
+): string {
   if (!stats) return "";
 
   const parts: string[] = [];
 
   if (platform === "bilibili") {
-    if (stats.likes != null) parts.push(`赞${formatCount(stats.likes)}`);
-    if (stats.coins != null) parts.push(`币${formatCount(stats.coins)}`);
-    if (stats.favorites != null) parts.push(`藏${formatCount(stats.favorites)}`);
-    if (stats.shares != null) parts.push(`转${formatCount(stats.shares)}`);
+    if (subtype === "live") {
+      if (stats.views != null && stats.views > 0)
+        parts.push(`在线${formatCount(stats.views)}`);
+      if (stats.comments != null && stats.comments > 0)
+        parts.push(`关注${formatCount(stats.comments)}`);
+    } else {
+      if (stats.likes != null) parts.push(`赞${formatCount(stats.likes)}`);
+      if (stats.coins != null) parts.push(`币${formatCount(stats.coins)}`);
+      if (stats.favorites != null)
+        parts.push(`藏${formatCount(stats.favorites)}`);
+      if (stats.shares != null) parts.push(`转${formatCount(stats.shares)}`);
+    }
   } else {
     if (stats.likes != null) parts.push(`赞${formatCount(stats.likes)}`);
-    if (stats.favorites != null) parts.push(`藏${formatCount(stats.favorites)}`);
+    if (stats.favorites != null)
+      parts.push(`藏${formatCount(stats.favorites)}`);
     if (stats.shares != null) parts.push(`转${formatCount(stats.shares)}`);
     if (stats.comments != null) parts.push(`评${formatCount(stats.comments)}`);
   }
@@ -76,16 +108,32 @@ function buildSummaryText(platform: string, stats?: MediaStats): string {
   return parts.join(" ");
 }
 
-function buildForwardDisplay(parsed: ParsedMediaUrl, result: ParsedMediaResult): ForwardDisplayOptions {
+function buildForwardDisplay(
+  parsed: ParsedMediaUrl,
+  result: ParsedMediaResult,
+): ForwardDisplayOptions {
   const displayTitle = PLATFORM_DISPLAY_TITLES[parsed.platform] || "媒体解析";
 
+  let summary = buildSummaryText(parsed.platform, parsed.subtype, result.stats);
+
+  if (result.liveStatus) {
+    summary = summary ? `${result.liveStatus} ${summary}` : result.liveStatus;
+  }
+
+  // 对于图片集/合辑，增加内容数量信息
+  if (result.images && result.images.length > 0) {
+    const imageInfo = `${result.images.length}张图片`;
+    summary = summary ? `${imageInfo} ${summary}` : imageInfo;
+  }
+  if (result.videoUrls && result.videoUrls.length > 0) {
+    const videoInfo = `${result.videoUrls.length}个视频`;
+    summary = summary ? `${videoInfo} ${summary}` : videoInfo;
+  }
+
   return {
     source: displayTitle,
-    news: [
-      { text: truncateText(result.title, 26) },
-      { text: result.author },
-    ],
-    summary: buildSummaryText(parsed.platform, result.stats),
+    news: [{ text: truncateText(result.title, 26) }, { text: result.author }],
+    summary,
   };
 }
 
@@ -116,6 +164,15 @@ function buildForwardNodes(
 ): SendNodeElement[] {
   const nodes: SendNodeElement[] = [];
 
+  // 简介文字优先
+  const infoText = buildInfoMessage(parsed, result);
+  nodes.push({
+    type: "node",
+    user_id: userId,
+    nickname,
+    content: [ctx.segment.text(infoText)],
+  } as SendNodeContentElement);
+
   if (result.coverUrl) {
     nodes.push({
       type: "node",
@@ -125,15 +182,34 @@ function buildForwardNodes(
     } as SendNodeContentElement);
   }
 
-  const infoText = buildInfoMessage(parsed, result);
-  nodes.push({
-    type: "node",
-    user_id: userId,
-    nickname,
-    content: [ctx.segment.text(infoText)],
-  } as SendNodeContentElement);
+  // 处理图片集/合辑
+  if (result.images && result.images.length > 0) {
+    // 如果既有图片又有视频/实况图，发送纯图片
+    // 否则直接发送图片
+    for (const imageUrl of result.images) {
+      nodes.push({
+        type: "node",
+        user_id: userId,
+        nickname,
+        content: [ctx.segment.image(imageUrl)],
+      } as SendNodeContentElement);
+    }
+  }
 
-  if (result.videoUrl) {
+  // 处理短视频/实况图视频
+  if (result.videoUrls && result.videoUrls.length > 0) {
+    for (const videoUrl of result.videoUrls) {
+      nodes.push({
+        type: "node",
+        user_id: userId,
+        nickname,
+        content: [(ctx.segment as any).video(videoUrl)],
+      } as SendNodeContentElement);
+    }
+  }
+
+  // 处理常规视频（单个视频）
+  if (result.videoUrl && (!result.videoUrls || result.videoUrls.length === 0)) {
     nodes.push({
       type: "node",
       user_id: userId,
@@ -194,7 +270,10 @@ export async function sendMediaResult(
   }
 
   const nickname = String(
-    ctx?.bot?.nickname || event?.sender?.card || event?.sender?.nickname || "媒体解析",
+    ctx?.bot?.nickname ||
+      event?.sender?.card ||
+      event?.sender?.nickname ||
+      "媒体解析",
   );
   const userId = String(selfId || ctx?.bot?.bot_id || event?.self_id || 0);
 
@@ -202,43 +281,112 @@ export async function sendMediaResult(
   const forwardPayload = toOneBotForwardFormat(nodes);
   const display = buildForwardDisplay(parsed, result);
 
-  try {
-    if (event?.message_type === "group" && event?.group_id != null) {
-      await bot.api("send_group_forward_msg", {
-        group_id: event.group_id,
-        messages: forwardPayload,
-        source: display.source,
-        news: display.news,
-        summary: display.summary,
-      });
-      return;
-    }
+  if (event?.message_type === "group" && event?.group_id != null) {
+    await bot.api("send_group_forward_msg", {
+      group_id: event.group_id,
+      messages: forwardPayload,
+      source: display.source,
+      news: display.news,
+      summary: display.summary,
+    });
+    return;
+  }
 
-    if (event?.user_id != null) {
-      await bot.api("send_private_forward_msg", {
-        user_id: event.user_id,
-        messages: forwardPayload,
-        source: display.source,
-        news: display.news,
-        summary: display.summary,
-      });
-      return;
-    }
-  } catch (primaryError) {
-    try {
-      if (event?.message_type === "group" && event?.group_id != null) {
-        await bot.sendGroupMsg(event.group_id, nodes);
-        return;
-      }
-      if (event?.user_id != null) {
-        await bot.sendPrivateMsg(event.user_id, nodes);
-        return;
-      }
-    } catch {
-      // fallback to text reply
+  if (event?.user_id != null) {
+    await bot.api("send_private_forward_msg", {
+      user_id: event.user_id,
+      messages: forwardPayload,
+      source: display.source,
+      news: display.news,
+      summary: display.summary,
+    });
+  }
+}
+
+export async function sendDurationLimitResult(
+  ctx: any,
+  event: any,
+  parsed: ParsedMediaUrl,
+  result: ParsedMediaResult,
+  limitMinutes: number,
+): Promise<void> {
+  const selfId = event?.self_id != null ? Number(event.self_id) : undefined;
+  const bot =
+    selfId != null && typeof ctx?.pickBot === "function"
+      ? ctx.pickBot(selfId)
+      : undefined;
+
+  if (!bot) {
+    await event.reply(
+      `【${PLATFORM_NAMES[parsed.platform] || parsed.platform}】视频太大了，发不出来～`,
+    );
+    return;
+  }
+
+  const nickname = String(
+    ctx?.bot?.nickname ||
+      event?.sender?.card ||
+      event?.sender?.nickname ||
+      "媒体解析",
+  );
+  const userId = String(selfId || ctx?.bot?.bot_id || event?.self_id || 0);
+
+  const nodes: SendNodeElement[] = [];
+
+  const platform = PLATFORM_NAMES[parsed.platform] || parsed.platform;
+  const mins = Math.floor((result.duration || 0) / 60);
+  const secs = (result.duration || 0) % 60;
+  const infoText = `【${platform}】${result.title}\n作者：${result.author}\n时长：${mins}分${secs}秒\n\n哎嘿，视频太大了发不出来～请选择更短的视频（不超过 ${limitMinutes} 分钟）`;
+
+  nodes.push({
+    type: "node",
+    user_id: userId,
+    nickname,
+    content: [ctx.segment.text(infoText)],
+  } as SendNodeContentElement);
+
+  if (result.coverUrl) {
+    nodes.push({
+      type: "node",
+      user_id: userId,
+      nickname,
+      content: [ctx.segment.image(result.coverUrl)],
+    } as SendNodeContentElement);
+  }
+
+  // 只发图片，不发视频
+  if (result.images && result.images.length > 0) {
+    for (const imageUrl of result.images) {
+      nodes.push({
+        type: "node",
+        user_id: userId,
+        nickname,
+        content: [ctx.segment.image(imageUrl)],
+      } as SendNodeContentElement);
     }
   }
 
-  const infoText = buildInfoMessage(parsed, result);
-  await event.reply(infoText);
+  const forwardPayload = toOneBotForwardFormat(nodes);
+  const displayTitle = PLATFORM_DISPLAY_TITLES[parsed.platform] || "媒体解析";
+
+  if (event?.message_type === "group" && event?.group_id != null) {
+    await bot.api("send_group_forward_msg", {
+      group_id: event.group_id,
+      messages: forwardPayload,
+      source: displayTitle,
+      news: [{ text: truncateText(result.title, 26) }, { text: result.author }],
+      summary: `视频太长无法发送（${mins}分${secs}秒）`,
+    });
+    return;
+  }
+
+  if (event?.user_id != null) {
+    await bot.api("send_private_forward_msg", {
+      user_id: event.user_id,
+      messages: forwardPayload,
+      source: displayTitle,
+      news: [{ text: truncateText(result.title, 26) }, { text: result.author }],
+      summary: `视频太长无法发送（${mins}分${secs}秒）`,
+    });
+  }
 }