qqbot-opencode 1.0.0

package/src/app.ts

@@ -0,0 +1,204 @@
+import {
+  loadConfig,
+  expandEnvVariables,
+  findDefaultConfig,
+  runInitWizard,
+} from "./config.js";
+import type { Config } from "./types.js";
+import {
+  startQQConnection,
+  sendC2CMessage,
+  parseMessage,
+  chunkText,
+  clearTokenCache,
+} from "./qq/index.js";
+import {
+  initOpencodeClient,
+  closeOpencodeClient,
+  createSession,
+} from "./opencode/client.js";
+import { handleMessage } from "./handlers/message.js";
+
+const TEXT_CHUNK_LIMIT = 500;
+
+interface CLIArgs {
+  config?: string;
+  init?: boolean;
+}
+
+function parseArgs(): CLIArgs {
+  const args = process.argv.slice(2);
+  const result: CLIArgs = {};
+
+  for (let i = 0; i < args.length; i++) {
+    if (args[i] === "--config" && i + 1 < args.length) {
+      result.config = args[i + 1];
+      i++;
+    } else if (args[i] === "--init" || args[i] === "init") {
+      result.init = true;
+    } else if (args[i] === "--help" || args[i] === "-h") {
+      printHelp();
+      process.exit(0);
+    }
+  }
+
+  return result;
+}
+
+function printHelp(): void {
+  console.log(`
+QQ Bot with OpenCode AI
+
+Usage:
+  qqbot              启动 Bot（使用默认配置）
+  qqbot --config <path>  使用指定配置文件
+  qqbot init         交互式创建配置文件
+  qqbot --help       显示帮助
+
+配置搜索顺序:
+  1. --config 指定路径
+  2. ./config.yaml（当前工作目录）
+  3. ~/.qqbot/config.yaml（用户家目录）
+
+示例:
+  qqbot
+  qqbot --config ./my-config.yaml
+  qqbot init
+  `);
+}
+
+async function main(): Promise<void> {
+  const args = parseArgs();
+
+  if (args.init) {
+    await runInitWizard();
+    return;
+  }
+
+  let configPath: string | null = args.config ?? null;
+
+  if (configPath === null) {
+    configPath = findDefaultConfig();
+    if (configPath !== null) {
+      console.log(`[App] Using default config: ${configPath}`);
+    }
+  }
+
+  if (configPath === null) {
+    console.error("Error: No config file found.");
+    console.error(
+      "\n请先运行 'qqbot init' 创建配置文件，或使用 --config 指定配置文件路径。",
+    );
+    console.error("\n运行 'qqbot --help' 查看更多信息。");
+    process.exit(1);
+  }
+
+  console.log("[App] Loading configuration...");
+  let config: Config;
+
+  try {
+    config = loadConfig(configPath);
+  } catch (err) {
+    console.error(
+      `[App] Failed to load config: ${err instanceof Error ? err.message : String(err)}`,
+    );
+    process.exit(1);
+  }
+
+  const expandedConfig = expandEnvVariables(config) as Config;
+
+  console.log("[App] Initializing OpenCode client...");
+  try {
+    await initOpencodeClient(expandedConfig.opencode);
+    console.log("[App] OpenCode client initialized");
+  } catch (err) {
+    console.error(
+      `[App] Failed to initialize OpenCode: ${err instanceof Error ? err.message : String(err)}`,
+    );
+    process.exit(1);
+  }
+
+  try {
+    await createSession();
+  } catch (err) {
+    console.error(
+      `[App] Failed to create initial session: ${err instanceof Error ? err.message : String(err)}`,
+    );
+  }
+
+  console.log("[App] Starting QQ connection...");
+  try {
+    await startQQConnection({
+      qq: expandedConfig.qq,
+      onMessage: async (event) => {
+        console.log(
+          `[App] Received message from ${event.author.user_openid}: ${event.content.slice(0, 50)}...`,
+        );
+
+        try {
+          const parsed = parseMessage(event as any);
+          const result = await handleMessage(parsed.content, parsed);
+
+          const chunks = chunkText(result.text, TEXT_CHUNK_LIMIT);
+          for (const chunk of chunks) {
+            await sendC2CMessage(expandedConfig.qq, {
+              toOpenid: event.author.user_openid,
+              content: chunk,
+              messageId: event.id,
+            });
+          }
+
+          console.log(`[App] Sent response to ${event.author.user_openid}`);
+        } catch (err) {
+          console.error(
+            `[App] Error handling message: ${err instanceof Error ? err.message : String(err)}`,
+          );
+
+          const errorText = `处理消息时出错: ${err instanceof Error ? err.message : String(err)}`;
+          try {
+            await sendC2CMessage(expandedConfig.qq, {
+              toOpenid: event.author.user_openid,
+              content: errorText,
+              messageId: event.id,
+            });
+          } catch (sendErr) {
+            console.error(`[App] Failed to send error message: ${sendErr}`);
+          }
+        }
+      },
+      onReady: () => {
+        console.log("[App] QQ Bot is ready!");
+      },
+      onError: (err) => {
+        console.error(`[App] QQ connection error: ${err.message}`);
+      },
+      onDisconnect: () => {
+        console.log("[App] QQ disconnected, will attempt to reconnect...");
+      },
+    });
+  } catch (err) {
+    console.error(
+      `[App] Failed to start QQ connection: ${err instanceof Error ? err.message : String(err)}`,
+    );
+    await closeOpencodeClient();
+    process.exit(1);
+  }
+
+  console.log("[App] Bot is running. Press Ctrl+C to stop.");
+
+  const shutdown = async () => {
+    console.log("\n[App] Shutting down...");
+    clearTokenCache();
+    await closeOpencodeClient();
+    console.log("[App] Goodbye!");
+    process.exit(0);
+  };
+
+  process.on("SIGINT", shutdown);
+  process.on("SIGTERM", shutdown);
+}
+
+main().catch((err) => {
+  console.error(`[App] Fatal error: ${err}`);
+  process.exit(1);
+});

package/src/config.ts

@@ -0,0 +1,200 @@
+import fs from "fs";
+import path from "path";
+import os from "os";
+import yaml from "js-yaml";
+import prompts from "prompts";
+import type { Config } from "./types.js";
+
+const CONFIG_DIR = path.join(os.homedir(), ".qqbot");
+const CONFIG_FILE = "config.yaml";
+const CONFIG_PATH = path.join(CONFIG_DIR, CONFIG_FILE);
+
+export function getConfigDir(): string {
+  return CONFIG_DIR;
+}
+
+export function findDefaultConfig(): string | null {
+  const searchPaths = [path.join(process.cwd(), CONFIG_FILE), CONFIG_PATH];
+
+  for (const configPath of searchPaths) {
+    if (fs.existsSync(configPath)) {
+      return configPath;
+    }
+  }
+  return null;
+}
+
+export function loadConfig(configPath: string): Config {
+  const absolutePath = path.resolve(configPath);
+
+  if (!fs.existsSync(absolutePath)) {
+    throw new Error(`Config file not found: ${absolutePath}`);
+  }
+
+  const rawContent = fs.readFileSync(absolutePath, "utf-8");
+  const config = yaml.load(rawContent) as Config;
+
+  validateConfig(config);
+
+  return config;
+}
+
+function validateConfig(config: Config): void {
+  if (!config.qq?.appId) {
+    throw new Error("Missing required field: qq.appId");
+  }
+  if (!config.qq?.clientSecret) {
+    throw new Error("Missing required field: qq.clientSecret");
+  }
+  if (!config.opencode?.port) {
+    config.opencode.port = 4097;
+  }
+  if (!config.opencode?.hostname) {
+    config.opencode.hostname = "127.0.0.1";
+  }
+  if (!config.app?.workingDir) {
+    config.app.workingDir = process.cwd();
+  }
+}
+
+export function expandEnvVariables(obj: unknown): unknown {
+  if (typeof obj === "string") {
+    const envVarPattern = /\$\{([^}]+)\}/g;
+    return obj.replace(envVarPattern, (_, envName) => {
+      return process.env[envName] || "";
+    });
+  }
+  if (Array.isArray(obj)) {
+    return obj.map(expandEnvVariables);
+  }
+  if (obj && typeof obj === "object") {
+    const result: Record<string, unknown> = {};
+    for (const [key, value] of Object.entries(obj)) {
+      result[key] = expandEnvVariables(value);
+    }
+    return result;
+  }
+  return obj;
+}
+
+export function saveConfig(config: Config, destPath?: string): void {
+  const savePath = destPath || CONFIG_PATH;
+
+  if (!fs.existsSync(CONFIG_DIR)) {
+    fs.mkdirSync(CONFIG_DIR, { recursive: true });
+  }
+
+  const yamlContent = yaml.dump(config, {
+    indent: 2,
+    lineWidth: -1,
+    noRefs: true,
+  });
+
+  fs.writeFileSync(savePath, yamlContent, "utf-8");
+}
+
+export async function runInitWizard(): Promise<void> {
+  console.log("\n欢迎使用 qqbot-opencode 配置向导！\n");
+
+  const answers = await prompts([
+    {
+      type: "text",
+      name: "appId",
+      message: "请输入 QQ Bot AppID:",
+      validate: (value) => value.length > 0 || "AppID 不能为空",
+    },
+    {
+      type: "text",
+      name: "clientSecret",
+      message: "请输入 QQ Bot ClientSecret:",
+      validate: (value) => value.length > 0 || "ClientSecret 不能为空",
+    },
+    {
+      type: "confirm",
+      name: "markdownSupport",
+      message: "是否启用 Markdown 渲染? (y/N):",
+      initial: false,
+    },
+    {
+      type: "number",
+      name: "opencodePort",
+      message: "opencode 服务端口 (默认: 4097):",
+      initial: 4097,
+    },
+    {
+      type: "select",
+      name: "providerType",
+      message: "AI Provider 类型:",
+      choices: [
+        { title: "1. Anthropic 兼容 (MiniMax)", value: "anthropic" },
+        { title: "2. OpenAI 兼容", value: "openai" },
+      ],
+      initial: 0,
+    },
+    {
+      type: "text",
+      name: "providerName",
+      message: "请输入 Provider 名称 (默认: minimax):",
+      initial: "minimax",
+    },
+    {
+      type: "text",
+      name: "modelName",
+      message: "AI 模型名称 (默认: MiniMax-M2.7):",
+      initial: "MiniMax-M2.7",
+    },
+    {
+      type: "text",
+      name: "apiKey",
+      message: "API Key:",
+      validate: (value) => value.length > 0 || "API Key 不能为空",
+    },
+  ]);
+
+  const providerType = answers.providerType;
+  const providerName = answers.providerName || "minimax";
+  const modelName = answers.modelName || "MiniMax-M2.7";
+
+  let baseURL: string;
+  if (providerType === "anthropic") {
+    baseURL = "https://api.minimaxi.com/anthropic/v1";
+  } else {
+    baseURL = "https://api.openai.com/v1";
+  }
+
+  const config: Config = {
+    qq: {
+      appId: answers.appId,
+      clientSecret: answers.clientSecret,
+      markdownSupport: answers.markdownSupport,
+    },
+    opencode: {
+      port: answers.opencodePort || 4097,
+      hostname: "127.0.0.1",
+      config: {
+        model: `${providerName}/${modelName}`,
+        provider: {
+          [providerName]: {
+            options: {
+              baseURL,
+              apiKey: answers.apiKey,
+            },
+            models: {
+              [modelName]: {
+                name: `${providerName}/${modelName}`,
+              },
+            },
+          },
+        },
+      },
+    },
+    app: {
+      workingDir: "./",
+    },
+  };
+
+  saveConfig(config);
+
+  console.log("\n配置已保存到 ~/.qqbot/config.yaml");
+  console.log("运行 'qqbot' 启动 Bot\n");
+}

package/src/handlers/index.ts

@@ -0,0 +1,2 @@
+export * from "./session.js";
+export * from "./message.js";

package/src/handlers/message.ts

@@ -0,0 +1,86 @@
+import { sendPrompt } from "../opencode/client.js";
+import {
+  parseSessionCommand,
+  handleSessionNew,
+  handleSessionSwitch,
+  handleSessionList,
+  handleSessionCurrent,
+} from "./session.js";
+import type { ParsedMessage } from "../qq/index.js";
+
+export interface MessageHandlerResult {
+  text: string;
+  success: boolean;
+  isSessionCommand?: boolean;
+}
+
+export async function handleMessage(
+  content: string,
+  parsedMessage: ParsedMessage,
+): Promise<MessageHandlerResult> {
+  const sessionCmd = parseSessionCommand(content);
+
+  if (sessionCmd) {
+    return handleSessionCommand(sessionCmd.command, sessionCmd.args);
+  }
+
+  return handleAIMessage(content, parsedMessage);
+}
+
+async function handleSessionCommand(
+  command: string,
+  args: string,
+): Promise<MessageHandlerResult> {
+  let result;
+
+  switch (command) {
+    case "new":
+      result = await handleSessionNew();
+      break;
+    case "switch":
+      result = await handleSessionSwitch(args);
+      break;
+    case "list":
+      result = await handleSessionList();
+      break;
+    case "current":
+      result = await handleSessionCurrent();
+      break;
+    default:
+      result = {
+        text: "未知的 session 命令",
+        success: false,
+      };
+  }
+
+  return {
+    ...result,
+    isSessionCommand: true,
+  };
+}
+
+async function handleAIMessage(
+  content: string,
+  parsedMessage: ParsedMessage,
+): Promise<MessageHandlerResult> {
+  try {
+    const result = await sendPrompt(content, parsedMessage.imageUrls);
+
+    if (!result.text || result.text.trim() === "") {
+      return {
+        text: "AI 返回了空响应",
+        success: true,
+      };
+    }
+
+    return {
+      text: result.text,
+      success: true,
+    };
+  } catch (err) {
+    return {
+      text: `处理消息失败: ${err instanceof Error ? err.message : String(err)}`,
+      success: false,
+    };
+  }
+}

package/src/handlers/session.ts

@@ -0,0 +1,130 @@
+import {
+  createSession,
+  switchSession,
+  listSessions,
+  getCurrentSession,
+} from "../opencode/client.js";
+
+export interface SessionCommandResult {
+  text: string;
+  success: boolean;
+}
+
+export async function handleSessionNew(): Promise<SessionCommandResult> {
+  try {
+    const session = await createSession();
+    return {
+      text: `已创建新会话: ${session.id}\n标题: ${session.title || "无"}`,
+      success: true,
+    };
+  } catch (err) {
+    return {
+      text: `创建会话失败: ${err instanceof Error ? err.message : String(err)}`,
+      success: false,
+    };
+  }
+}
+
+export async function handleSessionSwitch(
+  sessionId: string,
+): Promise<SessionCommandResult> {
+  if (!sessionId || sessionId.trim() === "") {
+    return {
+      text: "请提供会话 ID，例如: /session-switch <id>",
+      success: false,
+    };
+  }
+
+  try {
+    const session = await switchSession(sessionId.trim());
+    return {
+      text: `已切换到会话: ${session.id}\n标题: ${session.title || "无"}`,
+      success: true,
+    };
+  } catch (err) {
+    return {
+      text: `切换会话失败: ${err instanceof Error ? err.message : String(err)}`,
+      success: false,
+    };
+  }
+}
+
+export async function handleSessionList(): Promise<SessionCommandResult> {
+  try {
+    const sessions = await listSessions();
+
+    if (sessions.length === 0) {
+      return {
+        text: "暂无会话",
+        success: true,
+      };
+    }
+
+    const currentSession = await getCurrentSession();
+    const currentId = currentSession?.id;
+
+    const lines = sessions.map((s, i) => {
+      const marker = s.id === currentId ? " [当前]" : "";
+      const title = s.title ? ` - ${s.title}` : "";
+      return `${i + 1}. ${s.id}${title}${marker}`;
+    });
+
+    return {
+      text: `会话列表:\n${lines.join("\n")}`,
+      success: true,
+    };
+  } catch (err) {
+    return {
+      text: `获取会话列表失败: ${err instanceof Error ? err.message : String(err)}`,
+      success: false,
+    };
+  }
+}
+
+export async function handleSessionCurrent(): Promise<SessionCommandResult> {
+  try {
+    const session = await getCurrentSession();
+
+    if (!session) {
+      return {
+        text: "当前没有活跃的会话",
+        success: true,
+      };
+    }
+
+    return {
+      text: `当前会话:\nID: ${session.id}\n标题: ${session.title || "无"}`,
+      success: true,
+    };
+  } catch (err) {
+    return {
+      text: `获取当前会话失败: ${err instanceof Error ? err.message : String(err)}`,
+      success: false,
+    };
+  }
+}
+
+export function parseSessionCommand(
+  content: string,
+): { command: string; args: string } | null {
+  const trimmed = content.trim();
+
+  if (trimmed === "/session-new") {
+    return { command: "new", args: "" };
+  }
+
+  if (trimmed.startsWith("/session-switch ")) {
+    const args = trimmed.slice("/session-switch ".length).trim();
+    return { command: "switch", args };
+  }
+
+  if (trimmed === "/session-list") {
+    return { command: "list", args: "" };
+  }
+
+  if (trimmed === "/session-current") {
+    return { command: "current", args: "" };
+  }
+
+  return null;
+}