Add support for data URIs from HLS playlists in ClearKey

devine/core/drm/clearkey.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import base64
 import shutil
 from pathlib import Path
 from typing import Optional, Union
@@ -58,8 +59,16 @@ class ClearKey:
         if not m3u_key.uri:
             raise ValueError("No URI in M3U Key, unable to get Key.")
 
+        if m3u_key.uri.startswith("data:"):
+            media_types, data = m3u_key.uri[5:].split(",")
+            media_types = media_types.split(";")
+            if "base64" in media_types:
+                data = base64.b64decode(data)
+            key = data
+        else:
+            url = urljoin(m3u_key.base_uri, m3u_key.uri)
             res = requests.get(
-            url=urljoin(m3u_key.base_uri, m3u_key.uri),
+                url=url,
                 headers={
                     "User-Agent": "smartexoplayer/1.1.0 (Linux;Android 8.0.0) ExoPlayerLib/2.13.3"
                 },
@@ -70,11 +79,12 @@ class ClearKey:
                 raise EOFError("Unexpected Empty Response by M3U Key URI.")
             if len(res.content) < 16:
                 raise EOFError(f"Unexpected Length of Key ({len(res.content)} bytes) in M3U Key.")
-
             key = res.content
-        iv = None
+
         if m3u_key.iv:
             iv = bytes.fromhex(m3u_key.iv.replace("0x", ""))
+        else:
+            iv = None
 
         return cls(key=key, iv=iv)