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

import type { runCommandWithTimeout } from "../process/exec.js";

import { discoverGatewayBeacons, resolveGatewayDiscoveryEndpoint } from "./bonjour-discovery.js";

import {

discoverGatewayBeacons,

type GatewayBonjourBeacon,

resolveGatewayDiscoveryEndpoint,

} from "./bonjour-discovery.js";

const WIDE_AREA_DOMAIN = "openclaw.internal.";

type BeaconRecord = {

domain?: string;

instanceName?: string;

displayName?: string;

host?: string;

port?: number;

tailnetDns?: string;

gatewayPort?: number;

sshPort?: number;

cliPath?: string;

txt?: Record<string, unknown>;

};

function collectMatching<T, U>(

items: readonly T[],

predicate: (item: T) => boolean,

return matches;

}

function findBeaconByInstance(beacons: readonly BeaconRecord[], instanceName: string) {

function findBeaconByInstance(

beacons: readonly GatewayBonjourBeacon[],

instanceName: string,

): GatewayBonjourBeacon {

const beacon = beacons.find((item) => item.instanceName === instanceName);

if (!beacon) {

throw new Error(`Expected beacon ${instanceName}`);

}

return beacon;

}

function getOnlyBeacon(beacons: readonly GatewayBonjourBeacon[]): GatewayBonjourBeacon {

expect(beacons).toHaveLength(1);

const beacon = beacons[0];

if (!beacon) {

throw new Error("Expected one beacon");

}

return beacon;

}

describe("bonjour-discovery", () => {

it("discovers beacons on darwin across local + wide-area domains", async () => {

const calls: Array<{ argv: string[]; timeoutMs: number }> = [];

run: run as unknown as typeof runCommandWithTimeout,

});

expect(beacons).toHaveLength(1);

const beacon = beacons[0] as BeaconRecord;

const beacon = getOnlyBeacon(beacons);

expect(beacon.domain).toBe("local.");

expect(beacon.instanceName).toBe("Studio Gateway");

expect(beacon.displayName).toBe("Peter’s Mac Studio");

run: run as unknown as typeof runCommandWithTimeout,

});

expect(beacons).toHaveLength(1);

const beacon = beacons[0] as BeaconRecord;

const beacon = getOnlyBeacon(beacons);

expect(beacon.host).toBe("broken.local");

expect(beacon.port).toBeUndefined();

expect(beacon.gatewayPort).toBeUndefined();

run: run as unknown as typeof runCommandWithTimeout,

});

expect(beacons).toHaveLength(1);

const beacon = beacons[0] as BeaconRecord;

const beacon = getOnlyBeacon(beacons);

expect(beacon.domain).toBe(WIDE_AREA_DOMAIN);

expect(beacon.instanceName).toBe("studio-gateway");

expect(beacon.displayName).toBe("Studio");