Support swapping out the E2EEManager for RN (#1345)

* Support swapping out the E2EEManager for RN

* changeset

* formatting

Author: davidliu

.changeset/dirty-carpets-check.md

@@ -0,0 +1,5 @@
+---
+'livekit-client': patch
+---
+
+Support swapping out the E2EEManager for react-native

src/e2ee/E2eeManager.ts

@@ -16,7 +16,7 @@ import type { BaseKeyProvider } from './KeyProvider';
 import { E2EE_FLAG } from './constants';
 import { type E2EEManagerCallbacks, EncryptionEvent, KeyProviderEvent } from './events';
 import type {
-  E2EEOptions,
+  E2EEManagerOptions,
   E2EEWorkerMessage,
   EnableMessage,
   EncodeMessage,
@@ -31,10 +31,21 @@ import type {
 } from './types';
 import { isE2EESupported, isScriptTransformSupported } from './utils';
 
+export interface BaseE2EEManager {
+  setup(room: Room): void;
+  setupEngine(engine: RTCEngine): void;
+  setParticipantCryptorEnabled(enabled: boolean, participantIdentity: string): void;
+  setSifTrailer(trailer: Uint8Array): void;
+  on<E extends keyof E2EEManagerCallbacks>(event: E, listener: E2EEManagerCallbacks[E]): this;
+}
+
 /**
  * @experimental
  */
-export class E2EEManager extends (EventEmitter as new () => TypedEventEmitter<E2EEManagerCallbacks>) {
+export class E2EEManager
+  extends (EventEmitter as new () => TypedEventEmitter<E2EEManagerCallbacks>)
+  implements BaseE2EEManager
+{
   protected worker: Worker;
 
   protected room?: Room;
@@ -43,7 +54,7 @@ export class E2EEManager extends (EventEmitter as new () => TypedEventEmitter<E2
 
   private keyProvider: BaseKeyProvider;
 
-  constructor(options: E2EEOptions) {
+  constructor(options: E2EEManagerOptions) {
     super();
     this.keyProvider = options.keyProvider;
     this.worker = options.worker;

src/e2ee/types.ts

@@ -1,5 +1,6 @@
 import type { LogLevel } from '../logger';
 import type { VideoCodec } from '../room/track/options';
+import type { BaseE2EEManager } from './E2eeManager';
 import type { BaseKeyProvider } from './KeyProvider';
 
 export interface BaseMessage {
@@ -137,10 +138,16 @@ export type KeyInfo = {
   keyIndex?: number;
 };
 
-export type E2EEOptions = {
+export type E2EEManagerOptions = {
   keyProvider: BaseKeyProvider;
   worker: Worker;
 };
+export type E2EEOptions =
+  | E2EEManagerOptions
+  | {
+      /** For react-native usage. */
+      e2eeManager: BaseE2EEManager;
+    };
 
 export type DecodeRatchetOptions = {
   /** attempts  */

src/index.ts

@@ -45,6 +45,7 @@ export { RpcError, type RpcInvocationData, type PerformRpcParams } from './room/
 export * from './connectionHelper/ConnectionCheck';
 export * from './connectionHelper/checks/Checker';
 export * from './e2ee';
+export type { BaseE2EEManager } from './e2ee/E2eeManager';
 export * from './options';
 export * from './room/errors';
 export * from './room/events';

src/room/Room.ts

@@ -33,7 +33,7 @@ import { EventEmitter } from 'events';
 import type TypedEmitter from 'typed-emitter';
 import 'webrtc-adapter';
 import { EncryptionEvent } from '../e2ee';
-import { E2EEManager } from '../e2ee/E2eeManager';
+import { type BaseE2EEManager, E2EEManager } from '../e2ee/E2eeManager';
 import log, { LoggerNames, getLogger } from '../logger';
 import type {
   InternalRoomConnectOptions,
@@ -158,7 +158,7 @@ class Room extends (EventEmitter as new () => TypedEmitter<RoomEventCallbacks>)
 
   private disconnectLock: Mutex;
 
-  private e2eeManager: E2EEManager | undefined;
+  private e2eeManager: BaseE2EEManager | undefined;
 
   private connectionReconcileInterval?: ReturnType<typeof setInterval>;
 
@@ -252,7 +252,11 @@ class Room extends (EventEmitter as new () => TypedEmitter<RoomEventCallbacks>)
 
   private setupE2EE() {
     if (this.options.e2ee) {
-      this.e2eeManager = new E2EEManager(this.options.e2ee);
+      if ('e2eeManager' in this.options.e2ee) {
+        this.e2eeManager = this.options.e2ee.e2eeManager;
+      } else {
+        this.e2eeManager = new E2EEManager(this.options.e2ee);
+      }
       this.e2eeManager.on(
         EncryptionEvent.ParticipantEncryptionStatusChanged,
         (enabled, participant) => {