refactor(auto-reply): split reply pipeline

src/auto-reply/reply/commands-compact.ts

@@ -0,0 +1,119 @@
+import {
+  abortEmbeddedPiRun,
+  compactEmbeddedPiSession,
+  isEmbeddedPiRunActive,
+  waitForEmbeddedPiRunEnd,
+} from "../../agents/pi-embedded.js";
+import type { ClawdbotConfig } from "../../config/config.js";
+import { resolveSessionFilePath } from "../../config/sessions.js";
+import { logVerbose } from "../../globals.js";
+import { enqueueSystemEvent } from "../../infra/system-events.js";
+import { formatContextUsageShort, formatTokenCount } from "../status.js";
+import type { CommandHandler } from "./commands-types.js";
+import { stripMentions, stripStructuralPrefixes } from "./mentions.js";
+import { incrementCompactionCount } from "./session-updates.js";
+
+function extractCompactInstructions(params: {
+  rawBody?: string;
+  ctx: import("../templating.js").MsgContext;
+  cfg: ClawdbotConfig;
+  agentId?: string;
+  isGroup: boolean;
+}): string | undefined {
+  const raw = stripStructuralPrefixes(params.rawBody ?? "");
+  const stripped = params.isGroup
+    ? stripMentions(raw, params.ctx, params.cfg, params.agentId)
+    : raw;
+  const trimmed = stripped.trim();
+  if (!trimmed) return undefined;
+  const lowered = trimmed.toLowerCase();
+  const prefix = lowered.startsWith("/compact") ? "/compact" : null;
+  if (!prefix) return undefined;
+  let rest = trimmed.slice(prefix.length).trimStart();
+  if (rest.startsWith(":")) rest = rest.slice(1).trimStart();
+  return rest.length ? rest : undefined;
+}
+
+export const handleCompactCommand: CommandHandler = async (params) => {
+  const compactRequested =
+    params.command.commandBodyNormalized === "/compact" ||
+    params.command.commandBodyNormalized.startsWith("/compact ");
+  if (!compactRequested) return null;
+  if (!params.command.isAuthorizedSender) {
+    logVerbose(
+      `Ignoring /compact from unauthorized sender: ${params.command.senderId || "<unknown>"}`,
+    );
+    return { shouldContinue: false };
+  }
+  if (!params.sessionEntry?.sessionId) {
+    return {
+      shouldContinue: false,
+      reply: { text: "⚙️ Compaction unavailable (missing session id)." },
+    };
+  }
+  const sessionId = params.sessionEntry.sessionId;
+  if (isEmbeddedPiRunActive(sessionId)) {
+    abortEmbeddedPiRun(sessionId);
+    await waitForEmbeddedPiRunEnd(sessionId, 15_000);
+  }
+  const customInstructions = extractCompactInstructions({
+    rawBody: params.ctx.CommandBody ?? params.ctx.RawBody ?? params.ctx.Body,
+    ctx: params.ctx,
+    cfg: params.cfg,
+    agentId: params.agentId,
+    isGroup: params.isGroup,
+  });
+  const result = await compactEmbeddedPiSession({
+    sessionId,
+    sessionKey: params.sessionKey,
+    messageChannel: params.command.channel,
+    sessionFile: resolveSessionFilePath(sessionId, params.sessionEntry),
+    workspaceDir: params.workspaceDir,
+    config: params.cfg,
+    skillsSnapshot: params.sessionEntry.skillsSnapshot,
+    provider: params.provider,
+    model: params.model,
+    thinkLevel:
+      params.resolvedThinkLevel ?? (await params.resolveDefaultThinkingLevel()),
+    bashElevated: {
+      enabled: false,
+      allowed: false,
+      defaultLevel: "off",
+    },
+    customInstructions,
+    ownerNumbers:
+      params.command.ownerList.length > 0
+        ? params.command.ownerList
+        : undefined,
+  });
+
+  const totalTokens =
+    params.sessionEntry.totalTokens ??
+    (params.sessionEntry.inputTokens ?? 0) +
+      (params.sessionEntry.outputTokens ?? 0);
+  const contextSummary = formatContextUsageShort(
+    totalTokens > 0 ? totalTokens : null,
+    params.contextTokens ?? params.sessionEntry.contextTokens ?? null,
+  );
+  const compactLabel = result.ok
+    ? result.compacted
+      ? result.result?.tokensBefore
+        ? `Compacted (${formatTokenCount(result.result.tokensBefore)} before)`
+        : "Compacted"
+      : "Compaction skipped"
+    : "Compaction failed";
+  if (result.ok && result.compacted) {
+    await incrementCompactionCount({
+      sessionEntry: params.sessionEntry,
+      sessionStore: params.sessionStore,
+      sessionKey: params.sessionKey,
+      storePath: params.storePath,
+    });
+  }
+  const reason = result.reason?.trim();
+  const line = reason
+    ? `${compactLabel}: ${reason} • ${contextSummary}`
+    : `${compactLabel} • ${contextSummary}`;
+  enqueueSystemEvent(line, { sessionKey: params.sessionKey });
+  return { shouldContinue: false, reply: { text: `⚙️ ${line}` } };
+};