factorize code

canalplus · Aug 5, 2024 · 9c2c23f · 9c2c23f
1 parent 42e2dab
commit 9c2c23f
Showing 1 changed file with 30 additions and 44 deletions.
diff --git a/src/compat/eme/eme-api-implementation.ts b/src/compat/eme/eme-api-implementation.ts
@@ -144,29 +144,8 @@ function getEmeApiImplementation(
     let createCustomMediaKeys: (keyType: string) => ICustomMediaKeys;
 
     if (preferredApiType === "webkit" && WebKitMediaKeysConstructor !== undefined) {
-      const compatibleEventListener = createCompatibleEventListener(
-        ["needkey"],
-        undefined /* prefixes */,
-      );
-      onEncrypted = (
-        target: IEventTargetLike,
-        listener: (event?: Event) => void,
-        cancelSignal: CancellationSignal,
-      ) => {
-        compatibleEventListener(
-          target,
-          (event?: Event) => {
-            const patchedEvent = object_assign(
-              { forceSessionRecreation: true },
-              event as MediaEncryptedEvent,
-            );
-            listener(patchedEvent);
-          },
-          cancelSignal,
-        );
-      };
-
       const callbacks = getWebKitMediaKeysCallbacks();
+      onEncrypted = createOnEncryptedForWebkit();
       isTypeSupported = callbacks.isTypeSupported;
       createCustomMediaKeys = callbacks.createCustomMediaKeys;
       setMediaKeys = callbacks.setMediaKeys;
@@ -182,28 +161,7 @@ function getEmeApiImplementation(
         implementation = "older-webkit";
         // This is for WebKit with prefixed EME api
       } else if (WebKitMediaKeysConstructor !== undefined) {
-        const compatibleEventListener = createCompatibleEventListener(
-          ["needkey"],
-          undefined /* prefixes */,
-        );
-        onEncrypted = (
-          target: IEventTargetLike,
-          listener: (event?: Event) => void,
-          cancelSignal: CancellationSignal,
-        ) => {
-          compatibleEventListener(
-            target,
-            (event?: Event) => {
-              const patchedEvent = object_assign(
-                { forceSessionRecreation: true },
-                event as MediaEncryptedEvent,
-              );
-              listener(patchedEvent);
-            },
-            cancelSignal,
-          );
-        };
-
+        onEncrypted = createOnEncryptedForWebkit();
         const callbacks = getWebKitMediaKeysCallbacks();
         isTypeSupported = callbacks.isTypeSupported;
         createCustomMediaKeys = callbacks.createCustomMediaKeys;
@@ -317,6 +275,34 @@ function getEmeApiImplementation(
     implementation,
   };
 }
+/**
+ * Create an event listener for the "webkitneedkey" event
+ * @returns
+ */
+function createOnEncryptedForWebkit(): IEmeApiImplementation["onEncrypted"] {
+  const compatibleEventListener = createCompatibleEventListener(
+    ["needkey"],
+    undefined /* prefixes */,
+  );
+  const onEncrypted = (
+    target: IEventTargetLike,
+    listener: (event?: Event) => void,
+    cancelSignal: CancellationSignal,
+  ) => {
+    compatibleEventListener(
+      target,
+      (event?: Event) => {
+        const patchedEvent = object_assign(
+          { forceSessionRecreation: true },
+          event as MediaEncryptedEvent,
+        );
+        listener(patchedEvent);
+      },
+      cancelSignal,
+    );
+  };
+  return onEncrypted;
+}
 
 /**
  * Set the given MediaKeys on the given HTMLMediaElement.