@keyou007/agent-bot 0.1.14-alpha.0 → 0.1.14-alpha.2

package/src/cli/Initializer.ts

@@ -43,6 +43,12 @@ export function shouldConfigureAgentsDuringInitialization(status: Initialization
   return status === "created" || status === "reset";
 }
 
+export function shouldConfigureGroupMessagesDuringInitialization(status: InitializationStatus): boolean {
+  return status === "created" || status === "reset";
+}
+
+export type GroupMessageResponseMode = "all" | "mention-only";
+
 export interface ConfiguredAgentOption {
   name: string;
   title: string;
@@ -238,6 +244,57 @@ export function writeConfiguredAgentSelection(
   return true;
 }
 
+export function readGroupMessageResponseMode(configPath: string): GroupMessageResponseMode {
+  const document = parseConfigurationDocument(fs.readFileSync(configPath, "utf8"), configPath);
+  if (!isMap(document.contents)) {
+    throw new Error(cliText(
+      `Configuration file must contain a YAML mapping: ${configPath}`,
+      `配置文件必须包含 YAML 映射：${configPath}`,
+    ));
+  }
+  const feishu = document.contents.get("feishu", true);
+  if (!isMap(feishu)) {
+    throw new Error(cliText(
+      `Configuration feishu must contain a YAML mapping: ${configPath}`,
+      `配置文件的 feishu 必须是 YAML 映射：${configPath}`,
+    ));
+  }
+  const enabled = document.getIn(["feishu", "respondToAllGroupMessages"]);
+  if (typeof enabled !== "boolean") {
+    throw new Error(cliText(
+      `Configuration feishu.respondToAllGroupMessages must be a boolean: ${configPath}`,
+      `配置项 feishu.respondToAllGroupMessages 必须是布尔值：${configPath}`,
+    ));
+  }
+  return enabled ? "all" : "mention-only";
+}
+
+export function writeGroupMessageResponseMode(
+  configPath: string,
+  mode: GroupMessageResponseMode,
+): boolean {
+  const original = fs.readFileSync(configPath, "utf8");
+  const document = parseConfigurationDocument(original, configPath);
+  if (!isMap(document.contents)) {
+    throw new Error(cliText(
+      `Configuration file must contain a YAML mapping: ${configPath}`,
+      `配置文件必须包含 YAML 映射：${configPath}`,
+    ));
+  }
+  const feishu = document.contents.get("feishu", true);
+  if (!isMap(feishu)) {
+    throw new Error(cliText(
+      `Configuration feishu must contain a YAML mapping: ${configPath}`,
+      `配置文件的 feishu 必须是 YAML 映射：${configPath}`,
+    ));
+  }
+  const enabled = mode === "all";
+  if (document.getIn(["feishu", "respondToAllGroupMessages"]) === enabled) return false;
+  feishu.set("respondToAllGroupMessages", enabled);
+  writeFileAtomically(configPath, document.toString());
+  return true;
+}
+
 export function readFeishuCredentials(envPath: string, env: NodeJS.ProcessEnv = process.env): FeishuCredentialState {
   const fileValues = fs.existsSync(envPath) ? parseDotEnv(fs.readFileSync(envPath, "utf8")) : {};
   const appId = firstNonBlank(env.FEISHU_APP_ID, fileValues.FEISHU_APP_ID);