Decompose webSessionSetup into reusable parts (Phase 3)

packages/client/src/internal.ts

@@ -7,12 +7,15 @@ export {
 } from "./utils/location.js";
 export { sourceInfo, setSourceInfo } from "./sourceInfo.js";
 export type { SourceInfo } from "./sourceInfo.js";
-export { setSetupStrategy } from "./platform/VoiceSessionSetup.js";
-export { webSessionSetup } from "./platform/web/VoiceSessionSetup.js";
+export {
+  setSetupStrategy,
+  setupWebRTCSession,
+} from "./platform/VoiceSessionSetup.js";
 export type {
   VoiceSessionSetupStrategy,
   VoiceSessionSetupResult,
 } from "./platform/VoiceSessionSetup.js";
+export { createConnection } from "./utils/ConnectionFactory.js";
 export {
   MIN_VOICE_FREQUENCY,
   MAX_VOICE_FREQUENCY,

packages/client/src/platform/VoiceSessionSetup.test.ts

@@ -0,0 +1,60 @@
+import { describe, it, expect, vi } from "vitest";
+
+vi.mock("livekit-client", () => ({
+  Room: vi.fn(),
+  RoomEvent: {},
+  Track: { Kind: { Audio: "audio" }, Source: { Microphone: "microphone" } },
+  ConnectionState: {},
+  createLocalAudioTrack: vi.fn(),
+}));
+
+import { setupWebRTCSession } from "./VoiceSessionSetup.js";
+import { WebRTCConnection } from "../utils/WebRTCConnection.js";
+import { WebSocketConnection } from "../utils/WebSocketConnection.js";
+
+describe("setupWebRTCSession", () => {
+  it("returns input/output from a WebRTCConnection", () => {
+    const mockInput = { close: vi.fn(), setMuted: vi.fn() };
+    const mockOutput = { close: vi.fn(), setVolume: vi.fn() };
+
+    // Create a minimal object that passes the instanceof check
+    const connection = Object.create(WebRTCConnection.prototype, {
+      input: { value: mockInput },
+      output: { value: mockOutput },
+    });
+
+    const result = setupWebRTCSession(connection);
+
+    expect(result.connection).toBe(connection);
+    expect(result.input).toBe(mockInput);
+    expect(result.output).toBe(mockOutput);
+    expect(result.playbackEventTarget).toBeNull();
+    expect(result.detach).toBeTypeOf("function");
+  });
+
+  it("detach is a no-op", () => {
+    const connection = Object.create(WebRTCConnection.prototype, {
+      input: { value: {} },
+      output: { value: {} },
+    });
+
+    const result = setupWebRTCSession(connection);
+    expect(() => result.detach()).not.toThrow();
+  });
+
+  it("throws when given a WebSocketConnection", () => {
+    const connection = Object.create(WebSocketConnection.prototype);
+
+    expect(() => setupWebRTCSession(connection)).toThrow(
+      "setupWebRTCSession requires a WebRTCConnection"
+    );
+  });
+
+  it("throws when given a plain object", () => {
+    const connection = { input: {}, output: {} } as any;
+
+    expect(() => setupWebRTCSession(connection)).toThrow(
+      "setupWebRTCSession requires a WebRTCConnection"
+    );
+  });
+});

packages/client/src/platform/VoiceSessionSetup.ts

@@ -3,6 +3,7 @@ import type { BaseConnection } from "../utils/BaseConnection.js";
 import type { InputController } from "../InputController.js";
 import type { OutputController } from "../OutputController.js";
 import type { PlaybackEventTarget } from "../OutputController.js";
+import { WebRTCConnection } from "../utils/WebRTCConnection.js";
 
 export type VoiceSessionSetupResult = {
   connection: BaseConnection;
@@ -30,3 +31,26 @@ export let setupStrategy: VoiceSessionSetupStrategy | undefined;
 export function setSetupStrategy(strategy: VoiceSessionSetupStrategy) {
   setupStrategy = strategy;
 }
+
+/**
+ * Sets up a voice session for a WebRTC connection.
+ * Platform-agnostic: extracts the input/output controllers that
+ * WebRTCConnection already provides via LiveKit.
+ */
+export function setupWebRTCSession(
+  connection: BaseConnection
+): VoiceSessionSetupResult {
+  if (!(connection instanceof WebRTCConnection)) {
+    throw new Error(
+      "setupWebRTCSession requires a WebRTCConnection. " +
+        `Received: ${connection.constructor.name}`
+    );
+  }
+  return {
+    connection,
+    input: connection.input,
+    output: connection.output,
+    playbackEventTarget: null,
+    detach: () => {},
+  };
+}

packages/client/src/platform/web/VoiceSessionSetup.ts

@@ -2,6 +2,7 @@ import type { Options } from "../../BaseConversation.js";
 import type { BaseConnection } from "../../utils/BaseConnection.js";
 import {
   setSetupStrategy,
+  setupWebRTCSession,
   type VoiceSessionSetupResult,
 } from "../VoiceSessionSetup.js";
 import { MediaDeviceOutput } from "./output.js";
@@ -13,53 +14,40 @@ import { attachConnectionToOutput } from "../../utils/attachConnectionToOutput.j
 import { createConnection } from "../../utils/ConnectionFactory.js";
 
 /**
- * Sets up input and output controllers for an existing connection.
- * Shared helper used by platform-specific setup strategies.
+ * Sets up WebSocket-specific input and output controllers using
+ * web MediaDevice APIs (AudioContext, AudioWorklet, etc.).
  */
-export async function setupInputOutput(
+async function setupWebSocketIO(
   options: Options,
-  connection: BaseConnection
+  connection: WebSocketConnection
 ): Promise<Omit<VoiceSessionSetupResult, "connection">> {
-  if (connection instanceof WebRTCConnection) {
-    return {
-      input: connection.input,
-      output: connection.output,
-      playbackEventTarget: null,
-      detach: () => {},
-    };
-  } else if (connection instanceof WebSocketConnection) {
-    const [input, output] = await Promise.all([
-      MediaDeviceInput.create({
-        ...connection.inputFormat,
-        preferHeadphonesForIosDevices: options.preferHeadphonesForIosDevices,
-        inputDeviceId: options.inputDeviceId,
-        workletPaths: options.workletPaths,
-        libsampleratePath: options.libsampleratePath,
-      }),
-      MediaDeviceOutput.create({
-        ...connection.outputFormat,
-        outputDeviceId: options.outputDeviceId,
-        workletPaths: options.workletPaths,
-      }),
-    ]);
+  const [input, output] = await Promise.all([
+    MediaDeviceInput.create({
+      ...connection.inputFormat,
+      preferHeadphonesForIosDevices: options.preferHeadphonesForIosDevices,
+      inputDeviceId: options.inputDeviceId,
+      workletPaths: options.workletPaths,
+      libsampleratePath: options.libsampleratePath,
+    }),
+    MediaDeviceOutput.create({
+      ...connection.outputFormat,
+      outputDeviceId: options.outputDeviceId,
+      workletPaths: options.workletPaths,
+    }),
+  ]);
 
-    const detachInput = attachInputToConnection(input, connection);
-    const detachOutput = attachConnectionToOutput(connection, output);
+  const detachInput = attachInputToConnection(input, connection);
+  const detachOutput = attachConnectionToOutput(connection, output);
 
-    return {
-      input,
-      output,
-      playbackEventTarget: output,
-      detach: () => {
-        detachInput();
-        detachOutput();
-      },
-    };
-  } else {
-    throw new Error(
-      `Unsupported connection type: ${connection.constructor.name}`
-    );
-  }
+  return {
+    input,
+    output,
+    playbackEventTarget: output,
+    detach: () => {
+      detachInput();
+      detachOutput();
+    },
+  };
 }
 
 /**
@@ -70,8 +58,14 @@ export async function webSessionSetup(
   options: Options
 ): Promise<VoiceSessionSetupResult> {
   const connection = await createConnection(options);
-  const io = await setupInputOutput(options, connection);
-  return { connection, ...io };
+
+  if (connection instanceof WebSocketConnection) {
+    const io = await setupWebSocketIO(options, connection);
+    return { connection, ...io };
+  }
+
+  // WebRTC — platform-agnostic setup
+  return setupWebRTCSession(connection);
 }
 
 // Register the web strategy as the default

packages/react-native/src/index.react-native.ts

@@ -6,7 +6,8 @@ import {
 import type { Options } from "@elevenlabs/client";
 import {
   setSetupStrategy,
-  webSessionSetup,
+  createConnection,
+  setupWebRTCSession,
   type VoiceSessionSetupResult,
 } from "@elevenlabs/client/internal";
 import { attachNativeVolume } from "./nativeVolume.js";
@@ -36,7 +37,8 @@ async function reactNativeSessionSetup(
   });
   await AudioSession.startAudioSession();
 
-  const result = attachNativeVolume(await webSessionSetup(options));
+  const connection = await createConnection(options);
+  const result = attachNativeVolume(setupWebRTCSession(connection));
 
   const originalDetach = result.detach;
   return {