From 290da707eace858940909ae7713b8d696f32fe08 Mon Sep 17 00:00:00 2001
From: rlaphoenix <rlaphoenix@pm.me>
Date: Mon, 1 Aug 2022 23:07:59 +0100
Subject: [PATCH] serve: Add ability to get all types of keys in /keys

---
 pywidevine/serve.py | 25 ++++++++++++++-----------
 1 file changed, 14 insertions(+), 11 deletions(-)

diff --git a/pywidevine/serve.py b/pywidevine/serve.py
index a358c31..de7ee39 100644
--- a/pywidevine/serve.py
+++ b/pywidevine/serve.py
@@ -150,16 +150,19 @@ async def keys(request: web.Request) -> web.Response:
 
     # get key type
     key_type = request.match_info["key_type"]
-    try:
-        if key_type.isdigit():
-            key_type = License.KeyContainer.KeyType.Name(int(key_type))
-        else:
-            License.KeyContainer.KeyType.Value(key_type)  # only test
-    except ValueError as e:
-        return web.json_response({
-            "status": 400,
-            "message": f"The Key Type value is invalid, {e}"
-        }, status=400)
+    if key_type == "ALL":
+        key_type = None
+    else:
+        try:
+            if key_type.isdigit():
+                key_type = License.KeyContainer.KeyType.Name(int(key_type))
+            else:
+                License.KeyContainer.KeyType.Value(key_type)  # only test
+        except ValueError as e:
+            return web.json_response({
+                "status": 400,
+                "message": f"The Key Type value is invalid, {e}"
+            }, status=400)
 
     # get cdm
     if session_id not in request.app["sessions"]:
@@ -183,7 +186,7 @@ async def keys(request: web.Request) -> web.Response:
             "permissions": key.permissions,
         }
         for key in cdm._sessions[session_id].keys
-        if key.type == key_type
+        if not key_type or key.type == key_type
     ]
 
     return web.json_response({