From 0427f63f96f2264de4fd9326cabaf68df22c4cd3 Mon Sep 17 00:00:00 2001
From: shifengbin <shifengbin121@gmail.com>
Date: Thu, 8 Feb 2024 09:26:40 +0100
Subject: [PATCH] feat: TokenFileName

---
 websockify/token_plugins.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/websockify/token_plugins.py b/websockify/token_plugins.py
index d42414e..36a1dbc 100644
--- a/websockify/token_plugins.py
+++ b/websockify/token_plugins.py
@@ -65,6 +65,23 @@ class TokenFile(ReadOnlyTokenFile):
 
         return super().lookup(token)
 
+class TokenFileName(BasePlugin):
+    # source is a directory
+    # token is filename
+    # contents of file is host:port
+    def __init__(self, src):
+        super().__init__(src)
+        if not os.path.isdir(src):
+            raise Exception("TokenFileName plugin requires a directory")
+    
+    def lookup(self, token):
+        token = os.path.basename(token)
+        path = os.path.join(self.source, token)
+        if os.path.exists(path):
+            return open(path).read().strip().split(':')
+        else:
+            return None
+
 
 class BaseTokenAPI(BasePlugin):
     # source is a url with a '%s' in it where the token