From dfbaf2fd65fe16f5a66b5b16afef83502ea4b237 Mon Sep 17 00:00:00 2001
From: sinavir <sinavir@sinavir.fr>
Date: Sat, 12 Oct 2024 17:37:55 +0200
Subject: [PATCH] feat(frontend/management command): Add a command to craft
 tokens

---
 frontend/frontend/management/__init__.py      |  0
 .../frontend/management/commands/__init__.py  |  0
 .../management/commands/craft_token.py        | 22 ++++++++++++++
 frontend/frontend/utils.py                    | 21 +++++++++++++
 frontend/frontend/views.py                    | 30 +++----------------
 5 files changed, 47 insertions(+), 26 deletions(-)
 create mode 100644 frontend/frontend/management/__init__.py
 create mode 100644 frontend/frontend/management/commands/__init__.py
 create mode 100644 frontend/frontend/management/commands/craft_token.py
 create mode 100644 frontend/frontend/utils.py

diff --git a/frontend/frontend/management/__init__.py b/frontend/frontend/management/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/frontend/frontend/management/commands/__init__.py b/frontend/frontend/management/commands/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/frontend/frontend/management/commands/craft_token.py b/frontend/frontend/management/commands/craft_token.py
new file mode 100644
index 0000000..0edca29
--- /dev/null
+++ b/frontend/frontend/management/commands/craft_token.py
@@ -0,0 +1,22 @@
+import pprint
+
+from django.core.management.base import BaseCommand
+from frontend.utils import craft_token
+
+
+class Command(BaseCommand):
+    help = "Craft a token for the backend"
+
+    def add_arguments(self, parser):
+
+        parser.add_argument(
+            "--is_cof",
+            action="store_true",
+        )
+        parser.add_argument("user", type=str)
+        parser.add_argument("exp_time", type=int)
+
+    def handle(self, *args, **options):
+        token = craft_token(options["user"], options["is_cof"], options["exp_time"])
+
+        self.stdout.write(f"Token:\n{pprint.pformat(token)}")
diff --git a/frontend/frontend/utils.py b/frontend/frontend/utils.py
new file mode 100644
index 0000000..4c71678
--- /dev/null
+++ b/frontend/frontend/utils.py
@@ -0,0 +1,21 @@
+from datetime import datetime, timedelta, timezone
+
+import jwt
+from django.conf import settings
+
+
+def craft_token(username, is_cof, hours=9):
+    claims = {
+        "exp": datetime.now(tz=timezone.utc) + timedelta(hours=hours),
+        "sub": "ragb",
+        "user": username,
+        "is_cof": is_cof,
+        "scope": "modify",
+    }
+    return {
+        "token": jwt.encode(
+            claims,
+            settings.JWT_SECRET,
+        ),
+        "claims": claims,
+    }
diff --git a/frontend/frontend/views.py b/frontend/frontend/views.py
index 47479c8..95c7062 100644
--- a/frontend/frontend/views.py
+++ b/frontend/frontend/views.py
@@ -1,6 +1,3 @@
-from datetime import datetime, timedelta, timezone
-
-import jwt
 from django.conf import settings
 from django.contrib.auth.mixins import LoginRequiredMixin
 from django.core.exceptions import ViewDoesNotExist
@@ -8,6 +5,8 @@ from django.http import Http404, JsonResponse
 from django.views import View
 from django.views.generic.base import TemplateView
 
+from .utils import craft_token
+
 
 def get_context_from_proj(kind, chans):
     print(kind, chans)
@@ -40,20 +39,7 @@ def get_context_from_proj(kind, chans):
 
 class TokenView(LoginRequiredMixin, View):
     def get(self, request, *arg, **kwargs):
-        return JsonResponse(
-            {
-                "token": jwt.encode(
-                    {
-                        "exp": datetime.now(tz=timezone.utc) + timedelta(hours=9),
-                        "sub": "ragb",
-                        "user": self.request.user.username,
-                        "is_cof": self.requests.user.groups.filter(name="cof").exists(),
-                        "scope": "modify",
-                    },
-                    settings.JWT_SECRET,
-                )
-            }
-        )
+        return JsonResponse(craft_token(self.request.user.username, self.request.user.groups.filter(name="cof").exists()))
 
 
 class LightView(TemplateView):
@@ -64,15 +50,7 @@ class LightView(TemplateView):
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)
         if self.request.user.is_authenticated:
-            context["jwt"] = jwt.encode(
-                {
-                    "exp": datetime.now(tz=timezone.utc) + timedelta(hours=9),
-                    "sub": "ragb",
-                    "user": self.request.user.username,
-                    "scope": "modify",
-                },
-                settings.JWT_SECRET,
-            )
+            context["jwt"] = craft_token(self.request.user.username, self.request.user.groups.filter(name="cof").exists())["token"]
         context["websocket_endpoint"] = settings.WEBSOCKET_ENDPOINT
         light = self.kwargs["light"]
         if light not in settings.LIGHTS["lights"]: