Add macOS iOS Android companion platform client wrappers

2026-02-16 13:55:35 -08:00
parent 542a8cb70f
commit ce9af106ff
7 changed files with 429 additions and 2 deletions
@@ -0,0 +1,109 @@
+import { describe, expect, it, vi } from 'vitest';
+import {
+  AndroidCompanionClient,
+  IOSCompanionClient,
+  MacOSCompanionClient,
+} from './platformClients.js';
+import type { CompanionRuntimeClient } from './runtimeClient.js';
+
+function createRuntimeMock(): {
+  runtime: CompanionRuntimeClient;
+  connect: ReturnType<typeof vi.fn>;
+  disconnect: ReturnType<typeof vi.fn>;
+  registerNode: ReturnType<typeof vi.fn>;
+  getNodeCapabilities: ReturnType<typeof vi.fn>;
+  setNodeStatus: ReturnType<typeof vi.fn>;
+  setNodeLocation: ReturnType<typeof vi.fn>;
+  getNodeLocation: ReturnType<typeof vi.fn>;
+  setNodePushToken: ReturnType<typeof vi.fn>;
+  getSystemCapabilities: ReturnType<typeof vi.fn>;
+  listSystemNodes: ReturnType<typeof vi.fn>;
+} {
+  const connect = vi.fn(async () => undefined);
+  const disconnect = vi.fn(() => undefined);
+  const registerNode = vi.fn(async () => ({ registered: true }));
+  const getNodeCapabilities = vi.fn(async () => ({ node: { id: 'n1', role: 'companion', registeredAt: Date.now() }, protocol: { serverVersion: 1, nodeVersion: 1, negotiatedVersion: 1 }, capabilities: { declared: [], enabled: [], featureGates: {} } }));
+  const setNodeStatus = vi.fn(async () => ({ updated: true }));
+  const setNodeLocation = vi.fn(async () => ({ updated: true }));
+  const getNodeLocation = vi.fn(async () => ({ node: { id: 'n1', role: 'companion' }, location: null }));
+  const setNodePushToken = vi.fn(async () => ({ updated: true }));
+  const getSystemCapabilities = vi.fn(async () => ({ protocol: { version: 1 }, nodes: { enabled: true, locationEnabled: true, pushEnabled: true, allowedRoles: ['companion'], registered: true }, featureGates: {} }));
+  const listSystemNodes = vi.fn(async () => ({ nodes: [], summary: { total: 0 } }));
+
+  const runtime = {
+    connect,
+    disconnect,
+    registerNode,
+    getNodeCapabilities,
+    setNodeStatus,
+    setNodeLocation,
+    getNodeLocation,
+    setNodePushToken,
+    getSystemCapabilities,
+    listSystemNodes,
+  } as unknown as CompanionRuntimeClient;
+
+  return {
+    runtime,
+    connect,
+    disconnect,
+    registerNode,
+    getNodeCapabilities,
+    setNodeStatus,
+    setNodeLocation,
+    getNodeLocation,
+    setNodePushToken,
+    getSystemCapabilities,
+    listSystemNodes,
+  };
+}
+
+describe('platform companion clients', () => {
+  it('macOS client uses macos platform status and APNs push', async () => {
+    const mock = createRuntimeMock();
+    const client = new MacOSCompanionClient({ runtime: mock.runtime, nodeId: 'mac-node' });
+
+    await client.connect();
+    await client.register();
+    await client.setStatus({ appVersion: '1.0.0', powerSource: 'ac' });
+    await client.setLocation({ latitude: 10, longitude: 20, source: 'manual' });
+    await client.registerPushToken({ token: 'a'.repeat(64), topic: 'dev.flynn.macos', environment: 'production' });
+    await client.listNodes();
+    client.disconnect();
+
+    expect(mock.connect).toHaveBeenCalledOnce();
+    expect(mock.registerNode).toHaveBeenCalledWith(expect.objectContaining({ nodeId: 'mac-node', role: 'companion' }));
+    expect(mock.setNodeStatus).toHaveBeenCalledWith(expect.objectContaining({ platform: 'macos' }));
+    expect(mock.setNodePushToken).toHaveBeenCalledWith(expect.objectContaining({ provider: 'apns', topic: 'dev.flynn.macos' }));
+    expect(mock.listSystemNodes).toHaveBeenCalledWith({ platform: 'macos', role: 'companion' });
+    expect(mock.disconnect).toHaveBeenCalledOnce();
+  });
+
+  it('iOS client uses ios platform status and APNs push', async () => {
+    const mock = createRuntimeMock();
+    const client = new IOSCompanionClient({ runtime: mock.runtime, nodeId: 'ios-node' });
+
+    await client.register();
+    await client.setStatus({ statusText: 'foreground', batteryPct: 52, powerSource: 'battery' });
+    await client.registerPushToken({ token: 'b'.repeat(64), topic: 'dev.flynn.ios', environment: 'sandbox' });
+    await client.listNodes();
+
+    expect(mock.setNodeStatus).toHaveBeenCalledWith(expect.objectContaining({ platform: 'ios' }));
+    expect(mock.setNodePushToken).toHaveBeenCalledWith(expect.objectContaining({ provider: 'apns', environment: 'sandbox' }));
+    expect(mock.listSystemNodes).toHaveBeenCalledWith({ platform: 'ios', role: 'companion' });
+  });
+
+  it('android client uses android platform status and FCM push', async () => {
+    const mock = createRuntimeMock();
+    const client = new AndroidCompanionClient({ runtime: mock.runtime, nodeId: 'android-node' });
+
+    await client.register();
+    await client.setStatus({ appVersion: '2.0.0', powerSource: 'battery' });
+    await client.registerPushToken('c'.repeat(64));
+    await client.listNodes();
+
+    expect(mock.setNodeStatus).toHaveBeenCalledWith(expect.objectContaining({ platform: 'android' }));
+    expect(mock.setNodePushToken).toHaveBeenCalledWith({ provider: 'fcm', token: 'c'.repeat(64) });
+    expect(mock.listSystemNodes).toHaveBeenCalledWith({ platform: 'android', role: 'companion' });
+  });
+});