Add requestMediaKeySytemAccess-specific tests

canalplus · Aug 29, 2024 · 8f3b008 · 8f3b008
1 parent 135a00c
commit 8f3b008
Show file tree

Hide file tree

Showing 10 changed files with 1,625 additions and 218 deletions.
diff --git a/src/compat/eme/eme-api-implementation.ts b/src/compat/eme/eme-api-implementation.ts
@@ -44,7 +44,7 @@ export interface IEmeApiImplementation {
   requestMediaKeySystemAccess: (
     keyType: string,
     config: MediaKeySystemConfiguration[],
-  ) => Promise<IMediaKeySystemAccess | CustomMediaKeySystemAccess>;
+  ) => Promise<IMediaKeySystemAccess>;
 
   /**
    * API allowing to listen for `"encrypted"` events, presumably sent by the

diff --git a/src/experimental/tools/DummyMediaElement/eme.ts b/src/experimental/tools/DummyMediaElement/eme.ts
@@ -22,46 +22,94 @@ import {
   utf8ToStr,
 } from "../../../utils/string_parsing";
 
-const keySystemBlacklist: string[] = [];
-const keySystemWhitelist: string[] = [];
+export interface IRequestMediaKeySystemAccessConfig {
+  /**
+   * For the given arguments of the `navigator.requestMediaKeySystemAccess`
+   * API, returns the resulting `MediaKeySystemConfiguration`, or `null` if
+   * that configuration should be anounced as not supported.
+   *
+   * If not set, the first configuration will be anounced as supported for a
+   * supported key system.
+   *
+   * @param {string} keySystem
+   * @param {Array.<Object>} configs
+   * @returns {Object} config
+   */
+  getMediaKeySystemConfiguration?:
+    | ((
+        keySystem: string,
+        configs: MediaKeySystemConfiguration[],
+      ) => MediaKeySystemConfiguration | null)
+    | undefined;
+
+  /**
+   * For the given key system, returns `true` if it should be anounced as
+   * supported or `false` if it should be anounced as unsupported.
+   *
+   * If not set, all keySystems are supported.
+   * @param {string} keySystem
+   * @returns {boolean}
+   */
+  isKeySystemSupported: ((keySystem: string) => boolean) | undefined;
+}
 
 /**
- * Re-implementation of the EME `navigator.requestMediaKeySystemAccess` API.
- * @param {string} keySystem
- * @param {Array.<Object>} supportedConfigurations
- * @returns {Promise.<Object>}
+ * Return a configured re-implementation of the EME
+ * `navigator.requestMediaKeySystemAccess` API.
+ * @param {Object|undefined} [config]
+ * @returns {Function}
  */
-export function requestMediaKeySystemAccess(
-  keySystem: string,
-  supportedConfigurations: MediaKeySystemConfiguration[],
-): Promise<DummyMediaKeySystemAccess> {
-  if (keySystem === "") {
-    return Promise.reject(
-      new TypeError("`requestMediaKeySystemAccess` error: empty string"),
-    );
-  }
-  if (supportedConfigurations.length === 0) {
-    return Promise.reject(
-      new TypeError("`requestMediaKeySystemAccess` error: no given configuration."),
+export function createRequestMediaKeySystemAccess(
+  config?: IRequestMediaKeySystemAccessConfig | undefined,
+) {
+  /**
+   * Re-implementation of the EME `navigator.requestMediaKeySystemAccess` API.
+   * @param {string} keySystem
+   * @param {Array.<Object>} supportedConfigurations
+   * @returns {Promise.<Object>}
+   */
+  return function requestMediaKeySystemAccess(
+    keySystem: string,
+    supportedConfigurations: MediaKeySystemConfiguration[],
+  ): Promise<DummyMediaKeySystemAccess> {
+    if (keySystem === "") {
+      return Promise.reject(
+        new TypeError("`requestMediaKeySystemAccess` error: empty string"),
+      );
+    }
+    if (supportedConfigurations.length === 0) {
+      return Promise.reject(
+        new TypeError("`requestMediaKeySystemAccess` error: no given configuration."),
+      );
+    }
+    if (
+      typeof config?.isKeySystemSupported === "function" &&
+      !config.isKeySystemSupported(keySystem)
+    ) {
+      const error = new Error(`"${keySystem}" is not a supported keySystem`);
+      error.name = "NotSupportedError";
+      return Promise.reject(error);
+    }
+    let supportedConfiguration: MediaKeySystemConfiguration | null =
+      supportedConfigurations[0] ?? null;
+
+    if (typeof config?.getMediaKeySystemConfiguration === "function") {
+      supportedConfiguration = config.getMediaKeySystemConfiguration(
+        keySystem,
+        supportedConfigurations,
+      );
+    }
+    if (supportedConfiguration === null) {
+      const error = new Error(
+        "`requestMediaKeySystemAccess` error: No configuration supported.",
+      );
+      error.name = "NotSupportedError";
+      return Promise.reject(error);
+    }
+    return Promise.resolve(
+      new DummyMediaKeySystemAccess(keySystem, supportedConfiguration),
     );
-  }
-  if (
-    (keySystemWhitelist.length > 0 && !arrayIncludes(keySystemWhitelist, keySystem)) ||
-    (keySystemBlacklist.length > 0 && arrayIncludes(keySystemBlacklist, keySystem))
-  ) {
-    const error = new Error(`"${keySystem}" is not a supported keySystem`);
-    error.name = "NotSupportedError";
-    return Promise.reject(error);
-  }
-  for (const config of supportedConfigurations) {
-    // TODO configurable configuration validation. Callback?
-    return Promise.resolve(new DummyMediaKeySystemAccess(keySystem, config));
-  }
-  const error = new Error(
-    "`requestMediaKeySystemAccess` error: No configuration supported.",
-  );
-  error.name = "NotSupportedError";
-  return Promise.reject(error);
+  };
 }
 
 /**

diff --git a/src/experimental/tools/DummyMediaElement/html5.ts b/src/experimental/tools/DummyMediaElement/html5.ts
@@ -14,12 +14,20 @@ import {
   keepRangeIntersection,
 } from "../../../utils/ranges";
 import TaskCanceller from "../../../utils/task_canceller";
-import { DummyMediaKeys, requestMediaKeySystemAccess } from "./eme";
+import { DummyMediaKeys, createRequestMediaKeySystemAccess } from "./eme";
+import type { IRequestMediaKeySystemAccessConfig } from "./eme";
 import { DummyMediaSource } from "./mse";
 import TimeRangesWithMetadata, { EventScheduler } from "./utils";
 
 export interface IDummyMediaElementOptions {
   nodeName?: "AUDIO" | "VIDEO" | undefined | null;
+  drmOptions?:
+    | {
+        requestMediaKeySystemAccessConfig?:
+          | IRequestMediaKeySystemAccessConfig
+          | undefined;
+      }
+    | undefined;
 }
 
 export class DummyMediaElement
@@ -152,7 +160,9 @@ export class DummyMediaElement
 
     this.FORCED_MEDIA_SOURCE = DummyMediaSource;
     this.FORCED_EME_API = {
-      requestMediaKeySystemAccess,
+      requestMediaKeySystemAccess: createRequestMediaKeySystemAccess(
+        opts.drmOptions?.requestMediaKeySystemAccessConfig,
+      ),
       onEncrypted: createCompatibleEventListener(["encrypted"]),
       setMediaKeys,
       implementation: "standard",