koishi-plugin-aka-sa-caller 0.0.1 → 0.0.2

package/lib/index.d.ts

@@ -1,6 +1,9 @@
 import { Context, Schema } from 'koishi';
 export declare const name = "aka-sa-caller";
 export interface Config {
+    apiUrl: string;
+    pollInterval: number;
+    maxPollAttempts: number;
 }
 export declare const Config: Schema<Config>;
-export declare function apply(ctx: Context): void;
+export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -27,8 +27,170 @@ __export(src_exports, {
 module.exports = __toCommonJS(src_exports);
 var import_koishi = require("koishi");
 var name = "aka-sa-caller";
-var Config = import_koishi.Schema.object({});
-function apply(ctx) {
+var Config = import_koishi.Schema.object({
+  apiUrl: import_koishi.Schema.string().default("http://localhost:2380").description("Stock Advisor API地址"),
+  pollInterval: import_koishi.Schema.number().default(2e3).description("轮询间隔（毫秒）"),
+  maxPollAttempts: import_koishi.Schema.number().default(60).description("最大轮询次数（约2分钟）")
+});
+function apply(ctx, config) {
+  let stockCache = [];
+  async function fetchStocks() {
+    try {
+      const response = await ctx.http.get(
+        `${config.apiUrl}/api/stocks`
+      );
+      if (response.success) {
+        return response.stocks;
+      }
+      return [];
+    } catch (error) {
+      ctx.logger.error(`获取股票列表失败: ${error}`);
+      return [];
+    }
+  }
+  __name(fetchStocks, "fetchStocks");
+  async function submitAnalysis(symbol) {
+    try {
+      const response = await ctx.http.post(
+        `${config.apiUrl}/api/analysis/submit`,
+        { symbol }
+      );
+      if (response.success) {
+        return response.task_id;
+      }
+      return null;
+    } catch (error) {
+      ctx.logger.error(`提交分析任务失败: ${error}`);
+      return null;
+    }
+  }
+  __name(submitAnalysis, "submitAnalysis");
+  async function pollTaskStatus(taskId) {
+    for (let i = 0; i < config.maxPollAttempts; i++) {
+      try {
+        const response = await ctx.http.get(`${config.apiUrl}/api/analysis/task/${taskId}`);
+        if (response.success) {
+          const task = response.task;
+          if (task.status === "completed" && task.result) {
+            return task.result;
+          } else if (task.status === "failed") {
+            ctx.logger.error(`任务失败: ${task.error}`);
+            return null;
+          }
+        }
+      } catch (error) {
+        ctx.logger.error(`查询任务状态失败: ${error}`);
+      }
+      await new Promise((resolve) => setTimeout(resolve, config.pollInterval));
+    }
+    ctx.logger.warn(`任务 ${taskId} 超时`);
+    return null;
+  }
+  __name(pollTaskStatus, "pollTaskStatus");
+  function formatAnalysis(analysis) {
+    const parts = [];
+    parts.push(`📊 ${analysis.symbol} 分析报告`);
+    parts.push(`⏰ ${new Date(analysis.timestamp).toLocaleString("zh-CN")}`);
+    parts.push("");
+    const sentimentEmoji = {
+      positive: "✅ 积极",
+      negative: "❌ 消极",
+      neutral: "⚪ 中性"
+    };
+    parts.push(`📈 市场情绪: ${sentimentEmoji[analysis.sentiment] || "未知"}`);
+    parts.push("");
+    if (analysis.action) {
+      const actionEmoji = {
+        "加仓": "🟢",
+        "减仓": "🔴",
+        "持有": "🟡",
+        "观望": "⚪"
+      };
+      parts.push(`🎯 操作建议: ${actionEmoji[analysis.action] || ""} ${analysis.action}`);
+      if (analysis.action_reason) {
+        parts.push(`理由: ${analysis.action_reason}`);
+      }
+      parts.push("");
+    }
+    if (analysis.news_summary) {
+      parts.push("📝 新闻摘要:");
+      parts.push(analysis.news_summary);
+      parts.push("");
+    }
+    if (analysis.fundamental_analysis && analysis.fundamental_analysis !== "数据解析失败" && analysis.fundamental_analysis !== "基本面分析数据不足") {
+      parts.push("🏢 基本面分析:");
+      parts.push(analysis.fundamental_analysis);
+      parts.push("");
+    }
+    if (analysis.market_sentiment && analysis.market_sentiment !== "市场情绪分析数据不足") {
+      parts.push("💹 市场情绪:");
+      parts.push(analysis.market_sentiment);
+      parts.push("");
+    }
+    if (analysis.conclusion) {
+      parts.push("📌 投资总结:");
+      parts.push(analysis.conclusion);
+    }
+    return parts.join("\n");
+  }
+  __name(formatAnalysis, "formatAnalysis");
+  ctx.on("ready", async () => {
+    stockCache = await fetchStocks();
+    ctx.logger.info(`已加载 ${stockCache.length} 只股票`);
+    for (const stock of stockCache) {
+      ctx.command(`${stock.symbol}`, `分析 ${stock.name}`).action(async ({ session }) => {
+        await session.send(`正在分析 ${stock.symbol}...`);
+        const taskId = await submitAnalysis(stock.symbol);
+        if (!taskId) {
+          return "❌ 提交分析任务失败，请稍后重试";
+        }
+        const result = await pollTaskStatus(taskId);
+        if (!result) {
+          return "❌ 分析失败或超时，请稍后重试";
+        }
+        return formatAnalysis(result);
+      });
+    }
+  });
+  ctx.command("代码", "查看所有可用的股票代码").action(async () => {
+    if (stockCache.length === 0) {
+      return "⚠️ 暂无股票，请在 Stock Advisor 中添加";
+    }
+    const lines = ["📋 可用股票代码:"];
+    for (const stock of stockCache) {
+      lines.push(`• ${stock.symbol} - ${stock.name}`);
+    }
+    lines.push("");
+    lines.push("💡 直接输入股票代码即可查询分析");
+    return lines.join("\n");
+  });
+  ctx.command("更新股票代码", "从 Stock Advisor 刷新股票列表").action(async ({ session }) => {
+    await session.send("正在更新...");
+    const oldCount = stockCache.length;
+    stockCache = await fetchStocks();
+    const newCount = stockCache.length;
+    for (const stock of stockCache) {
+      ctx.command(`${stock.symbol}`, `分析 ${stock.name}`).action(async ({ session: session2 }) => {
+        await session2.send(`正在分析 ${stock.symbol}...`);
+        const taskId = await submitAnalysis(stock.symbol);
+        if (!taskId) {
+          return "❌ 提交分析任务失败，请稍后重试";
+        }
+        const result = await pollTaskStatus(taskId);
+        if (!result) {
+          return "❌ 分析失败或超时，请稍后重试";
+        }
+        return formatAnalysis(result);
+      });
+    }
+    if (newCount > oldCount) {
+      return `✅ 更新成功！新增 ${newCount - oldCount} 只股票，当前共 ${newCount} 只`;
+    } else if (newCount < oldCount) {
+      return `✅ 更新成功！减少 ${oldCount - newCount} 只股票，当前共 ${newCount} 只`;
+    } else {
+      return `✅ 更新成功！当前共 ${newCount} 只股票`;
+    }
+  });
 }
 __name(apply, "apply");
 // Annotate the CommonJS export names for ESM import in node:

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-aka-sa-caller",
-  "description": "",
-  "version": "0.0.1",
+  "description": "集成 Stock Advisor 的 Koishi 插件，提供股票分析功能",
+  "version": "0.0.2",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [
@@ -36,8 +36,8 @@
   },
   "koishi": {
     "description": {
-      "en": "",
-      "zh": ""
+      "en": "Koishi plugin for Stock Advisor integration - Analyze stocks with AI-powered insights",
+      "zh": "集成 Stock Advisor 的 Koishi 插件 - 使用 AI 分析股票"
     },
     "service": {
       "required": [],