import { register } from "@rpa/core";
import { RuntimeError } from "@rpa/core";
import { chatCompletion, requireAiConfig } from "@rpa/core";

function inputText(args: Record<string, unknown>, nodeId: string): string {
  for (const key of ["input", "hint_word", "text", "content", "sentence", "prompt", "sourceFilePath", "filePath", "source", "document"]) {
    const v = String(args[key] ?? "").trim();
    if (v) return v;
  }
  throw new RuntimeError("validation_error", "输入文本为空", { nodeId, componentId: nodeId });
}

async function readMaybeFile(value: string): Promise<string> {
  const looksLikeFile = /\.(txt|md|csv|json|log)$/i.test(value) && value.length < 260 && !value.includes("\n");
  if (!looksLikeFile) return value;
  try {
    const fs = await import("node:fs/promises");
    const stat = await fs.stat(value);
    if (stat.isFile()) return await fs.readFile(value, "utf8");
  } catch { /* treat the value as literal text */ }
  return value;
}

async function complete(prompt: string, input: string, temperature = 0.3): Promise<string> {
  requireAiConfig();
  return chatCompletion([
    { role: "system", content: prompt },
    { role: "user", content: input }
  ], { temperature });
}

const TEXT_TASKS: Array<{ ids: string[]; system: string; task: string }> = [
  { ids: ["DataService_AIGenTxt"], system: "你是写作助手，根据用户要求生成高质量中文内容。", task: "AI 生成文本" },
  { ids: ["DataService_LargeModel", "DataService_ZBrain", "DataService_AgentCall", "NLP_XunFei_NLPBaseServer"], system: "你是智能助手，尽力完成用户的任务。", task: "通用模型调用" },
  { ids: ["DataService_ChatBot"], system: "你是对话机器人，用中文自然对话。", task: "对话机器人" },
  { ids: ["DataService_KeyWord", "NLP_KeyWordsAnalysis", "NLP_SplitText"], system: "从文本中提取关键词，输出 JSON 数组（字符串列表），不要输出其他内容。", task: "关键词抽取" },
  { ids: ["DataService_Correct"], system: "修正文本中的错别字与语病，直接输出修正后的全文。", task: "文本纠错" },
  { ids: ["DataService_Review"], system: '审核文本内容是否合规，输出 JSON：{\"pass\":true/false,\"reason\":\"原因\"}。', task: "文本审核" },
  { ids: ["DataService_TextComparison", "Scenario_TextComparison"], system: "比较两段文本差异，输出简明差异摘要。", task: "文档比对" },
  { ids: ["NLP_SplitWords"], system: '对文本分词，输出 JSON 数组（字符串列表）。', task: "分词" },
  { ids: ["NLP_EmotionalAnalysis"], system: '分析文本情感倾向，输出 JSON：{\"label\":\"正面/负面/中性\",\"score\":0.99}。', task: "情感分析" },
  { ids: ["NLP_SemanticSimilarity"], system: '判断两段文本语义相似度，输出 JSON：{\"score\":0.99}（0-1）。输入格式：文本1\n---\n文本2。', task: "语义相似度" },
  { ids: ["NLP_PerspectiveAnalysis"], system: '分析文本中的观点与立场，输出 JSON：{\"perspectives\":[{\"view\":\"...\",\"stance\":\"支持/反对/中立\"}]}。', task: "观点分析" },
  { ids: ["NLP_AnalyseProductReview", "NLP_BatchAnalyseProductReview"], system: '分析商品评论，输出 JSON：{\"summary\":\"...\",\"sentiment\":\"...\",\"issues\":[]}。', task: "商品评论分析" },
];

for (const task of TEXT_TASKS) {
  register(task.ids, async (ctx, node, args) => {
    const rawInput = inputText(args, node.id);
    const input = await readMaybeFile(rawInput);
    const result = await complete(task.system, input);
    ctx.log(task.task + "完成", "info", node.id);
    return result;
  });
}