openclaw/src/auto-reply/reply.triggers.trigger-handling.filters-usage-summary-current-model-provider.cases.ts

import { mkdir, readFile, writeFile } from "node:fs/promises";
import { join } from "node:path";
import { describe, expect, it } from "vitest";
import { normalizeTestText } from "../../test/helpers/normalize-text.js";
import { resolveSessionKey } from "../config/sessions.js";
import {
  getProviderUsageMocks,
  getRunEmbeddedPiAgentMock,
  makeCfg,
  requireSessionStorePath,
  withTempHome,
} from "./reply.triggers.trigger-handling.test-harness.js";

type GetReplyFromConfig = typeof import("./reply.js").getReplyFromConfig;

const usageMocks = getProviderUsageMocks();

async function readSessionStore(storePath: string): Promise<Record<string, unknown>> {
  const raw = await readFile(storePath, "utf-8");
  return JSON.parse(raw) as Record<string, unknown>;
}

function pickFirstStoreEntry<T>(store: Record<string, unknown>): T | undefined {
  const entries = Object.values(store) as T[];
  return entries[0];
}

function getReplyFromConfigNow(getReplyFromConfig: () => GetReplyFromConfig): GetReplyFromConfig {
  return getReplyFromConfig();
}

export function registerTriggerHandlingUsageSummaryCases(params: {
  getReplyFromConfig: () => GetReplyFromConfig;
}): void {
  describe("usage and status command handling", () => {
    it("handles status, usage cycles, and auth-profile status details", async () => {
      await withTempHome(async (home) => {
        const runEmbeddedPiAgentMock = getRunEmbeddedPiAgentMock();
        const getReplyFromConfig = getReplyFromConfigNow(params.getReplyFromConfig);
        usageMocks.loadProviderUsageSummary.mockClear();
        usageMocks.loadProviderUsageSummary.mockResolvedValue({
          updatedAt: 0,
          providers: [
            {
              provider: "anthropic",
              displayName: "Anthropic",
              windows: [
                {
                  label: "5h",
                  usedPercent: 20,
                },
              ],
            },
          ],
        });

        {
          const res = await getReplyFromConfig(
            {
              Body: "/status",
              From: "+1000",
              To: "+2000",
              Provider: "whatsapp",
              SenderE164: "+1000",
              CommandAuthorized: true,
            },
            {},
            makeCfg(home),
          );

          const text = Array.isArray(res) ? res[0]?.text : res?.text;
          expect(text).toContain("Model:");
          expect(text).toContain("OpenClaw");
          expect(normalizeTestText(text ?? "")).toContain("Usage: Claude 80% left");
          expect(usageMocks.loadProviderUsageSummary).toHaveBeenCalledWith(
            expect.objectContaining({ providers: ["anthropic"] }),
          );
          expect(runEmbeddedPiAgentMock).not.toHaveBeenCalled();
        }

        {
          const cfg = makeCfg(home);
          cfg.session = { ...cfg.session, store: join(home, "usage-cycle.sessions.json") };
          const usageStorePath = requireSessionStorePath(cfg);
          const r0 = await getReplyFromConfig(
            {
              Body: "/usage on",
              From: "+1000",
              To: "+2000",
              Provider: "whatsapp",
              SenderE164: "+1000",
              CommandAuthorized: true,
            },
            undefined,
            cfg,
          );
          expect(String((Array.isArray(r0) ? r0[0]?.text : r0?.text) ?? "")).toContain(
            "Usage footer: tokens",
          );

          const r1 = await getReplyFromConfig(
            {
              Body: "/usage",
              From: "+1000",
              To: "+2000",
              Provider: "whatsapp",
              SenderE164: "+1000",
              CommandAuthorized: true,
            },
            undefined,
            cfg,
          );
          expect(String((Array.isArray(r1) ? r1[0]?.text : r1?.text) ?? "")).toContain(
            "Usage footer: full",
          );

          const r2 = await getReplyFromConfig(
            {
              Body: "/usage",
              From: "+1000",
              To: "+2000",
              Provider: "whatsapp",
              SenderE164: "+1000",
              CommandAuthorized: true,
            },
            undefined,
            cfg,
          );
          expect(String((Array.isArray(r2) ? r2[0]?.text : r2?.text) ?? "")).toContain(
            "Usage footer: off",
          );

          const r3 = await getReplyFromConfig(
            {
              Body: "/usage",
              From: "+1000",
              To: "+2000",
              Provider: "whatsapp",
              SenderE164: "+1000",
              CommandAuthorized: true,
            },
            undefined,
            cfg,
          );
          expect(String((Array.isArray(r3) ? r3[0]?.text : r3?.text) ?? "")).toContain(
            "Usage footer: tokens",
          );
          const finalStore = await readSessionStore(usageStorePath);
          expect(pickFirstStoreEntry<{ responseUsage?: string }>(finalStore)?.responseUsage).toBe(
            "tokens",
          );
          expect(runEmbeddedPiAgentMock).not.toHaveBeenCalled();
        }

        {
          runEmbeddedPiAgentMock.mockClear();
          const cfg = makeCfg(home);
          cfg.session = { ...cfg.session, store: join(home, "auth-profile-status.sessions.json") };
          const agentDir = join(home, ".openclaw", "agents", "main", "agent");
          await mkdir(agentDir, { recursive: true });
          await writeFile(
            join(agentDir, "auth-profiles.json"),
            JSON.stringify(
              {
                version: 1,
                profiles: {
                  "anthropic:work": {
                    type: "api_key",
                    provider: "anthropic",
                    key: "sk-test-1234567890abcdef",
                  },
                },
                lastGood: { anthropic: "anthropic:work" },
              },
              null,
              2,
            ),
          );

          const sessionKey = resolveSessionKey("per-sender", {
            From: "+1002",
            To: "+2000",
            Provider: "whatsapp",
          } as Parameters<typeof resolveSessionKey>[1]);
          await writeFile(
            requireSessionStorePath(cfg),
            JSON.stringify(
              {
                [sessionKey]: {
                  sessionId: "session-auth",
                  updatedAt: Date.now(),
                  authProfileOverride: "anthropic:work",
                },
              },
              null,
              2,
            ),
          );

          const res = await getReplyFromConfig(
            {
              Body: "/status",
              From: "+1002",
              To: "+2000",
              Provider: "whatsapp",
              SenderE164: "+1002",
              CommandAuthorized: true,
            },
            {},
            cfg,
          );
          const text = Array.isArray(res) ? res[0]?.text : res?.text;
          expect(text).toContain("api-key");
          expect(text).not.toContain("sk-test");
          expect(text).not.toContain("abcdef");
          expect(text).not.toContain("1234567890abcdef");
          expect(text).toContain("(anthropic:work)");
          expect(text).not.toContain("mixed");
          expect(runEmbeddedPiAgentMock).not.toHaveBeenCalled();
        }
      });
    });
  });
}