From caf67a69989f1a1e97e400481bb23a1c3aeb9e03 Mon Sep 17 00:00:00 2001
From: Andy <scene@scene.com>
Date: Sat, 31 Jan 2026 16:58:10 -0700
Subject: [PATCH] feat(env): add ML-Worker binary for DRM licensing

---
 unshackle/commands/env.py  | 7 +++++++
 unshackle/core/binaries.py | 2 ++
 2 files changed, 9 insertions(+)

diff --git a/unshackle/commands/env.py b/unshackle/commands/env.py
index 94d9a0b..adc48d4 100644
--- a/unshackle/commands/env.py
+++ b/unshackle/commands/env.py
@@ -52,6 +52,13 @@ def check() -> None:
             "desc": "DRM decryption",
             "cat": "DRM",
         },
+        {
+            "name": "ML-Worker",
+            "binary": binaries.ML_Worker,
+            "required": False,
+            "desc": "DRM licensing",
+            "cat": "DRM",
+        },
         # HDR Processing
         {"name": "dovi_tool", "binary": binaries.DoviTool, "required": False, "desc": "Dolby Vision", "cat": "HDR"},
         {
diff --git a/unshackle/core/binaries.py b/unshackle/core/binaries.py
index 1b56061..d984c0c 100644
--- a/unshackle/core/binaries.py
+++ b/unshackle/core/binaries.py
@@ -53,6 +53,7 @@ DoviTool = find("dovi_tool")
 HDR10PlusTool = find("hdr10plus_tool", "HDR10Plus_tool")
 Mp4decrypt = find("mp4decrypt")
 Docker = find("docker")
+ML_Worker = find("ML-Worker")
 
 
 __all__ = (
@@ -73,5 +74,6 @@ __all__ = (
     "HDR10PlusTool",
     "Mp4decrypt",
     "Docker",
+    "ML_Worker",
     "find",
 )