openclaw/ui/src/lib/session-display.test.ts

// @vitest-environment node
import { describe, expect, it } from "vitest";
import {
  resolveChannelSessionInfo,
  resolveSessionDisplayName,
  resolveSessionWorkSubtitle,
} from "./session-display.ts";

describe("resolveSessionDisplayName", () => {
  it("uses the same friendly main-thread name for every agent", () => {
    for (const key of ["main", "agent:main:main", "agent:research:main", "agent:ops-team:main"]) {
      expect(resolveSessionDisplayName(key)).toBe("Main Thread");
    }

    expect(resolveSessionDisplayName("agent:research:main", { displayName: "Research desk" })).toBe(
      "Research desk",
    );
    expect(resolveSessionDisplayName("agent:research:dashboard:main")).toBe("New thread");
    expect(resolveSessionDisplayName("agent:research:main:thread")).toBe("main:thread");
  });

  it("prefers label, then displayName", () => {
    expect(
      resolveSessionDisplayName("agent:main:telegram:direct:42", {
        label: "Alice",
        displayName: "openclaw-tui",
      }),
    ).toBe("Alice");
    expect(
      resolveSessionDisplayName("agent:main:telegram:direct:42", { displayName: "Peter" }),
    ).toBe("Peter");
  });

  it("never renders full raw peer ids for unnamed DMs", () => {
    expect(resolveSessionDisplayName("agent:main:telegram:direct:491234567890")).toBe(
      "Telegram · …567890",
    );
    expect(resolveSessionDisplayName("agent:main:imessage:direct:+4912")).toBe("iMessage · +4912");
  });

  it("does not split UTF-16 surrogate pairs when shortening peer ids", () => {
    expect(resolveSessionDisplayName("agent:main:telegram:direct:12345😀67890")).toBe(
      "Telegram · …67890",
    );
  });

  it("falls back to a friendly name for dashboard sessions instead of the uuid key", () => {
    expect(
      resolveSessionDisplayName("agent:main:dashboard:0f9d5c1e-6d0f-4c9a-9d84-1c2f3a4b5c6d"),
    ).toBe("New thread");
  });

  it("names unnamed work sessions after their checkout", () => {
    expect(
      resolveSessionDisplayName("agent:main:dashboard:uuid", {
        worktree: { branch: "openclaw/wt-3f2a", repoRoot: "/Users/dev/Projects/clawdbot" },
      }),
    ).toBe("clawdbot ⎇ wt-3f2a");
  });

  it("uses a gateway-derived title for otherwise unnamed sessions", () => {
    expect(
      resolveSessionDisplayName("agent:main:dashboard:uuid", {
        label: "agent:main:dashboard:uuid",
        displayName: "agent:main:dashboard:uuid",
        derivedTitle: "Quarterly launch plan",
      }),
    ).toBe("Quarterly launch plan");
  });

  it("keeps explicit and worktree names ahead of derived titles", () => {
    expect(
      resolveSessionDisplayName("agent:main:dashboard:uuid", {
        label: "Release room",
        derivedTitle: "Quarterly launch plan",
      }),
    ).toBe("Release room");
    expect(
      resolveSessionDisplayName("agent:main:dashboard:uuid", {
        worktree: { branch: "openclaw/wt-3f2a", repoRoot: "/repo/clawdbot" },
        derivedTitle: "Quarterly launch plan",
      }),
    ).toBe("clawdbot ⎇ wt-3f2a");
  });

  it("names named subsessions after their slug, never the raw agent key", () => {
    expect(resolveSessionDisplayName("agent:main:node-proof-claude")).toBe("node-proof-claude");
    expect(resolveSessionDisplayName("agent:main:explicit:node-mcp-debug")).toBe("node-mcp-debug");
    expect(
      resolveSessionDisplayName(
        "agent:main:explicit:model-run-0f9d5c1e-6d0f-4c9a-9d84-1c2f3a4b5c6d",
      ),
    ).toBe("model-run-…5c6d");
    expect(resolveSessionDisplayName("agent:main:node-fleet-4de003fbff138fcb9239c9378b2e")).toBe(
      "node-fleet-…8b2e",
    );
  });

  it("can omit only the subagent prefix while preserving its untitled fallback", () => {
    const key = "agent:main:subagent:worker";
    expect(resolveSessionDisplayName(key, { label: "Research sources" })).toBe(
      "Subagent: Research sources",
    );
    expect(
      resolveSessionDisplayName(
        key,
        { label: "Subagent: Research sources" },
        {
          includeSubagentPrefix: false,
        },
      ),
    ).toBe("Research sources");
    expect(resolveSessionDisplayName(key, undefined, { includeSubagentPrefix: false })).toBe(
      "Subagent:",
    );
    expect(
      resolveSessionDisplayName(
        "agent:main:cron:daily",
        { label: "Daily" },
        {
          includeSubagentPrefix: false,
        },
      ),
    ).toBe("Cron: Daily");
  });
});

describe("resolveSessionWorkSubtitle", () => {
  it("combines repo, branch, and node host", () => {
    expect(
      resolveSessionWorkSubtitle({
        worktree: { branch: "openclaw/session-ui", repoRoot: "/repo/clawdbot" },
      }),
    ).toBe("clawdbot ⎇ session-ui");
    expect(
      resolveSessionWorkSubtitle({
        worktree: { branch: "feature/x", repoRoot: "/repo/clawdbot" },
        execNode: "macbook",
      }),
    ).toBe("clawdbot ⎇ feature/x · macbook");
    expect(resolveSessionWorkSubtitle({ execNode: "macbook" })).toBe("macbook");
    expect(resolveSessionWorkSubtitle({})).toBeUndefined();
  });

  it("shortens opaque node ids instead of rendering raw hashes", () => {
    expect(
      resolveSessionWorkSubtitle({ execNode: "11c38726acc6fac280357576c87acc6fac280357" }),
    ).toBe("…0357");
    expect(
      resolveSessionWorkSubtitle({
        worktree: { branch: "openclaw/wt-1", repoRoot: "/repo/clawdbot" },
        execNode: "11c38726acc6fac280357576c87acc6fac280357",
      }),
    ).toBe("clawdbot ⎇ wt-1 · …0357");
  });
});

describe("resolveChannelSessionInfo", () => {
  it("classifies channel-shaped keys and keeps main/dashboard out", () => {
    expect(resolveChannelSessionInfo("agent:main:telegram:group:99")).toEqual({
      channel: "telegram",
      channelSession: true,
    });
    expect(resolveChannelSessionInfo("agent:main:slack:acct-1:channel:C1")).toEqual({
      channel: "slack",
      channelSession: true,
    });
    // dmScope per-peer keys have no channel segment; the row channel wins.
    expect(resolveChannelSessionInfo("agent:main:direct:+123", "whatsapp")).toEqual({
      channel: "whatsapp",
      channelSession: true,
    });
    expect(resolveChannelSessionInfo("agent:main:main", "telegram")).toEqual({
      channelSession: false,
    });
    expect(resolveChannelSessionInfo("agent:main:dashboard:uuid")).toEqual({
      channelSession: false,
    });
  });
});