openclaw-matrix-multiaccounts/src/infra/bonjour-discovery.test.ts

import { describe, expect, it, vi } from "vitest";

import type { runCommandWithTimeout } from "../process/exec.js";
import { discoverGatewayBeacons } from "./bonjour-discovery.js";
import { WIDE_AREA_DISCOVERY_DOMAIN } from "./widearea-dns.js";

describe("bonjour-discovery", () => {
  it("discovers beacons on darwin across local + wide-area domains", async () => {
    const calls: Array<{ argv: string[]; timeoutMs: number }> = [];

    const run = vi.fn(
      async (argv: string[], options: { timeoutMs: number }) => {
        calls.push({ argv, timeoutMs: options.timeoutMs });
        const domain = argv[3] ?? "";

        if (argv[0] === "dns-sd" && argv[1] === "-B") {
          if (domain === "local.") {
            return {
              stdout: [
                "Add 2 3 local. _clawdbot-bridge._tcp. Studio Bridge",
                "Add 2 3 local. _clawdbot-bridge._tcp. Laptop Bridge",
                "",
              ].join("\n"),
              stderr: "",
              code: 0,
              signal: null,
              killed: false,
            };
          }
          if (domain === WIDE_AREA_DISCOVERY_DOMAIN) {
            return {
              stdout: [
                `Add 2 3 ${WIDE_AREA_DISCOVERY_DOMAIN} _clawdbot-bridge._tcp. Tailnet Bridge`,
                "",
              ].join("\n"),
              stderr: "",
              code: 0,
              signal: null,
              killed: false,
            };
          }
        }

        if (argv[0] === "dns-sd" && argv[1] === "-L") {
          const instance = argv[2] ?? "";
          const host =
            instance === "Studio Bridge"
              ? "studio.local"
              : instance === "Laptop Bridge"
                ? "laptop.local"
                : "tailnet.local";
          const tailnetDns =
            instance === "Tailnet Bridge" ? "studio.tailnet.ts.net" : "";
          const txtParts = [
            "txtvers=1",
            `displayName=${instance.replace(" Bridge", "")}`,
            `lanHost=${host}`,
            "gatewayPort=18789",
            "bridgePort=18790",
            "sshPort=22",
            tailnetDns ? `tailnetDns=${tailnetDns}` : null,
          ].filter((v): v is string => Boolean(v));

          return {
            stdout: [
              `${instance}._clawdbot-bridge._tcp. can be reached at ${host}:18790`,
              txtParts.join(" "),
              "",
            ].join("\n"),
            stderr: "",
            code: 0,
            signal: null,
            killed: false,
          };
        }

        throw new Error(`unexpected argv: ${argv.join(" ")}`);
      },
    );

    const beacons = await discoverGatewayBeacons({
      platform: "darwin",
      timeoutMs: 1234,
      run: run as unknown as typeof runCommandWithTimeout,
    });

    expect(beacons).toHaveLength(3);
    expect(beacons.map((b) => b.domain)).toEqual(
      expect.arrayContaining(["local.", WIDE_AREA_DISCOVERY_DOMAIN]),
    );

    const browseCalls = calls.filter(
      (c) => c.argv[0] === "dns-sd" && c.argv[1] === "-B",
    );
    expect(browseCalls.map((c) => c.argv[3])).toEqual(
      expect.arrayContaining(["local.", WIDE_AREA_DISCOVERY_DOMAIN]),
    );
    expect(browseCalls.every((c) => c.timeoutMs === 1234)).toBe(true);
  });

  it("falls back to tailnet DNS probing for wide-area when split DNS is not configured", async () => {
    const calls: Array<{ argv: string[]; timeoutMs: number }> = [];

    const run = vi.fn(
      async (argv: string[], options: { timeoutMs: number }) => {
        calls.push({ argv, timeoutMs: options.timeoutMs });
        const cmd = argv[0];

        if (cmd === "dns-sd" && argv[1] === "-B") {
          return {
            stdout: "",
            stderr: "",
            code: 0,
            signal: null,
            killed: false,
          };
        }

        if (
          cmd === "tailscale" &&
          argv[1] === "status" &&
          argv[2] === "--json"
        ) {
          return {
            stdout: JSON.stringify({
              Self: { TailscaleIPs: ["100.69.232.64"] },
              Peer: {
                "peer-1": { TailscaleIPs: ["100.123.224.76"] },
              },
            }),
            stderr: "",
            code: 0,
            signal: null,
            killed: false,
          };
        }

        if (cmd === "dig") {
          const at = argv.find((a) => a.startsWith("@")) ?? "";
          const server = at.replace(/^@/, "");
          const qname = argv[argv.length - 2] ?? "";
          const qtype = argv[argv.length - 1] ?? "";

          if (
            server === "100.123.224.76" &&
            qtype === "PTR" &&
            qname === "_clawdbot-bridge._tcp.clawdbot.internal"
          ) {
            return {
              stdout: `studio-bridge._clawdbot-bridge._tcp.clawdbot.internal.\n`,
              stderr: "",
              code: 0,
              signal: null,
              killed: false,
            };
          }

          if (
            server === "100.123.224.76" &&
            qtype === "SRV" &&
            qname === "studio-bridge._clawdbot-bridge._tcp.clawdbot.internal"
          ) {
            return {
              stdout: `0 0 18790 studio.clawdbot.internal.\n`,
              stderr: "",
              code: 0,
              signal: null,
              killed: false,
            };
          }

          if (
            server === "100.123.224.76" &&
            qtype === "TXT" &&
            qname === "studio-bridge._clawdbot-bridge._tcp.clawdbot.internal"
          ) {
            return {
              stdout: [
                `"displayName=Studio"`,
                `"transport=bridge"`,
                `"bridgePort=18790"`,
                `"gatewayPort=18789"`,
                `"sshPort=22"`,
                `"tailnetDns=peters-mac-studio-1.sheep-coho.ts.net"`,
                `"cliPath=/opt/homebrew/bin/clawdbot"`,
                "",
              ].join(" "),
              stderr: "",
              code: 0,
              signal: null,
              killed: false,
            };
          }
        }

        throw new Error(`unexpected argv: ${argv.join(" ")}`);
      },
    );

    const beacons = await discoverGatewayBeacons({
      platform: "darwin",
      timeoutMs: 1200,
      domains: [WIDE_AREA_DISCOVERY_DOMAIN],
      run: run as unknown as typeof runCommandWithTimeout,
    });

    expect(beacons).toEqual([
      expect.objectContaining({
        domain: WIDE_AREA_DISCOVERY_DOMAIN,
        instanceName: "studio-bridge",
        displayName: "Studio",
        host: "studio.clawdbot.internal",
        port: 18790,
        tailnetDns: "peters-mac-studio-1.sheep-coho.ts.net",
        gatewayPort: 18789,
        sshPort: 22,
        cliPath: "/opt/homebrew/bin/clawdbot",
      }),
    ]);

    expect(
      calls.some((c) => c.argv[0] === "tailscale" && c.argv[1] === "status"),
    ).toBe(true);
    expect(calls.some((c) => c.argv[0] === "dig")).toBe(true);
  });

  it("normalizes domains and respects domains override", async () => {
    const calls: string[][] = [];
    const run = vi.fn(async (argv: string[]) => {
      calls.push(argv);
      return {
        stdout: "",
        stderr: "",
        code: 0,
        signal: null,
        killed: false,
      };
    });

    await discoverGatewayBeacons({
      platform: "darwin",
      timeoutMs: 1,
      domains: ["local", "clawdbot.internal"],
      run: run as unknown as typeof runCommandWithTimeout,
    });

    expect(calls.filter((c) => c[1] === "-B").map((c) => c[3])).toEqual(
      expect.arrayContaining(["local.", "clawdbot.internal."]),
    );

    calls.length = 0;
    await discoverGatewayBeacons({
      platform: "darwin",
      timeoutMs: 1,
      domains: ["local."],
      run: run as unknown as typeof runCommandWithTimeout,
    });

    expect(calls.filter((c) => c[1] === "-B")).toHaveLength(1);
    expect(calls.filter((c) => c[1] === "-B")[0]?.[3]).toBe("local.");
  });
});