NapCatQQ source snapshot

packages/napcat-core/helper/ffmpeg/ffmpeg-adapter-factory.ts

@@ -0,0 +1,120 @@
+/**
+ * FFmpeg Adapter Factory
+ * 自动检测并选择最佳的 FFmpeg 适配器
+ */
+
+import { LogWrapper } from '@/napcat-core/helper/log';
+import { FFmpegAddonAdapter, getAddonPath } from './ffmpeg-addon-adapter';
+import { existsSync } from 'node:fs';
+import { FFmpegExecAdapter } from './ffmpeg-exec-adapter';
+import type { IFFmpegAdapter } from './ffmpeg-adapter-interface';
+
+/**
+ * FFmpeg 适配器工厂
+ */
+export class FFmpegAdapterFactory {
+  private static instance: IFFmpegAdapter | null = null;
+  private static initPromise: Promise<IFFmpegAdapter> | null = null;
+
+  /**
+     * 初始化并获取最佳的 FFmpeg 适配器
+     * @param logger 日志记录器
+     * @param ffmpegPath FFmpeg 可执行文件路径(用于 Exec 适配器)
+     * @param ffprobePath FFprobe 可执行文件路径(用于 Exec 适配器)
+     * @param binaryPath 二进制文件路径(来自 pathWrapper.binaryPath,用于 Addon 适配器)
+     */
+  static async getAdapter (
+    logger: LogWrapper,
+    ffmpegPath: string = 'ffmpeg',
+    ffprobePath: string = 'ffprobe',
+    binaryPath?: string
+  ): Promise<IFFmpegAdapter> {
+    // 如果已经初始化,直接返回
+    if (this.instance) {
+      return this.instance;
+    }
+
+    // 如果正在初始化,等待初始化完成
+    if (this.initPromise) {
+      return this.initPromise;
+    }
+
+    // 开始初始化
+    this.initPromise = this.initialize(logger, ffmpegPath, ffprobePath, binaryPath);
+
+    try {
+      this.instance = await this.initPromise;
+      return this.instance;
+    } finally {
+      this.initPromise = null;
+    }
+  }
+
+  /**
+     * 初始化适配器
+     */
+  private static async initialize (
+    logger: LogWrapper,
+    ffmpegPath: string,
+    ffprobePath: string,
+    binaryPath?: string
+  ): Promise<IFFmpegAdapter> {
+    const useCommandLine = Boolean(process.env['FFMPEG_PATH'] || process.env['FFPROBE_PATH']);
+    if (binaryPath && !useCommandLine && existsSync(getAddonPath(binaryPath))) {
+      const addonAdapter = new FFmpegAddonAdapter(binaryPath);
+
+      logger.log('[FFmpeg] 检查 Native Addon 可用性...');
+      if (await addonAdapter.isAvailable()) {
+        logger.log('[FFmpeg] ✓ 使用 Native Addon 适配器');
+        return addonAdapter;
+      }
+
+      throw new Error('FFmpeg Native Addon 加载失败: ' + getAddonPath(binaryPath));
+    }
+
+    const execAdapter = new FFmpegExecAdapter(ffmpegPath, ffprobePath);
+
+    logger.log(`[FFmpeg] 检查命令行工具可用性: ${ffmpegPath}`);
+    if (await execAdapter.isAvailable()) {
+      logger.log('[FFmpeg] 使用命令行工具适配器 ✓');
+      return execAdapter;
+    }
+
+    throw new Error('FFmpeg 命令行工具不可用，请检查 ffmpeg 和 ffprobe 路径: ' + ffmpegPath + ', ' + ffprobePath);
+  }
+
+  /**
+     * 重置适配器(用于测试或重新初始化)
+     */
+  static reset (): void {
+    this.instance = null;
+    this.initPromise = null;
+  }
+
+  /**
+     * 更新 FFmpeg 路径并重新初始化
+     * @param logger 日志记录器
+     * @param ffmpegPath FFmpeg 可执行文件路径
+     * @param ffprobePath FFprobe 可执行文件路径
+     */
+  static async updateFFmpegPath (
+    logger: LogWrapper,
+    ffmpegPath: string,
+    ffprobePath: string
+  ): Promise<IFFmpegAdapter> {
+    const adapter = new FFmpegExecAdapter(ffmpegPath, ffprobePath);
+    if (!await adapter.isAvailable()) {
+      throw new Error('FFmpeg 路径不可用: ' + ffmpegPath + ', ' + ffprobePath);
+    }
+    this.instance = adapter;
+    logger.log('[FFmpeg] 已更新命令行工具路径:', ffmpegPath, ffprobePath);
+    return adapter;
+  }
+
+  /**
+     * 获取当前适配器(不初始化)
+     */
+  static getCurrentAdapter (): IFFmpegAdapter | null {
+    return this.instance;
+  }
+}