Fork project

utils/routers/__init__.py

@@ -0,0 +1,34 @@
+from CTFd.utils import get_config
+
+from .frp import FrpRouter
+from .trp import TrpRouter
+
+_routers = {
+    'frp': FrpRouter,
+    'trp': TrpRouter,
+}
+
+
+def instanciate(cls):
+    return cls()
+
+
+@instanciate
+class Router:
+    _name = ''
+    _router = None
+
+    def __getattr__(self, name: str):
+        router_conftype = get_config("whale:router_type", "frp")
+        if Router._name != router_conftype:
+            Router._router = _routers[router_conftype]()
+            Router._name = router_conftype
+        return getattr(Router._router, name)
+
+    @staticmethod
+    def reset():
+        Router._name = ''
+        Router._router = None
+
+
+__all__ = ["Router"]