AutoAndroidController/src/pages/Processing/Func/chat/ocr-chat.js

/**
 * Func 标签：ocr-chat
 *
 * 约定：src/pages/processing/func/ 目录下每个文件名就是一个"可用标签/能力"。
 * 本文件用于声明该标签存在（供文档/提示词/后续动态加载使用）。
 *
 * 运行时真实执行逻辑由 ActionParser + electronAPI.extractChatHistory + main-js/func/ocr-chat.js 实现。
 */

export const tagName = 'ocr-chat';

export const schema = {
  description: '根据图片识别对话内容，区分好友和自己的对话，输出JSON字符串格式的消息记录。',
  inputs: {
    // 按用户要求：抽象为 avatar1/avatar2，不绑定 friend/me 语义
    avatar1: '参与者1的头像/标识图片路径（相对于工作流目录）',
    avatar2: '参与者2的头像/标识图片路径（相对于工作流目录）',
    variable: '输出变量名（保存JSON字符串格式的消息记录）',
  },
  outputs: {
    variable: 'JSON字符串格式的消息记录（数组，每个元素包含 sender 和 text 字段）',
  },
};

/**
 * 执行 ocr-chat 功能
 * 这个函数会被 ActionParser 调用
 * 
 * @param {Object} params - 参数对象
 * @param {string} params.device - 设备 ID/IP:Port
 * @param {string} params.avatar1 - 参与者1的头像路径
 * @param {string} params.avatar2 - 参与者2的头像路径
 * @param {string} params.folderPath - 工作流文件夹路径
 * @returns {Promise<{success: boolean, messagesJson?: string, messages?: Array, error?: string}>}
 */
export async function executeOcrChat({ device, avatar1, avatar2, folderPath }) {
  try {
    if (!window.electronAPI || !window.electronAPI.extractChatHistory) {
      return { 
        success: false, 
        error: 'extractChatHistory API 不可用' 
      };
    }

    // 调用主进程的 extractChatHistory 函数
    const result = await window.electronAPI.extractChatHistory(
      device,
      avatar1,
      avatar2,
      folderPath
    );

    if (!result.success) {
      return { success: false, error: result.error };
    }

    // 获取消息数组
    let messages = [];
    if (result.messages && Array.isArray(result.messages)) {
      messages = result.messages;
    } else if (result.messagesText) {
      // 如果只有文本，尝试解析（向后兼容）
      const lines = result.messagesText.split('\n').filter(line => line.trim());
      for (const line of lines) {
        const match = line.match(/^(对方|好友|我):\s*(.+)$/);
        if (match) {
          const senderLabel = match[1];
          const sender = (senderLabel === '对方' || senderLabel === '好友') ? 'friend' : 'me';
          const text = match[2].trim();
          messages.push({ sender, text });
        }
      }
    }

    // 将消息数组转换为JSON字符串
    const messagesJson = JSON.stringify(messages);

    return {
      success: true,
      messagesJson: messagesJson,
      messages: messages
    };
  } catch (error) {
    console.error('执行 ocr-chat 失败:', error);
    return { 
      success: false, 
      error: error.message || '提取聊天记录失败' 
    };
  }
}