From 2b9cbf94b603535198290751e6a2d136153b079c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Joakim=20Hells=C3=A9n?= <tlovinator@gmail.com>
Date: Sun, 7 Jul 2024 22:22:01 +0200
Subject: [PATCH] Refactor subscribe buttons

---
 core/templates/index.html | 35 +++++++++++------------
 core/views.py             | 58 +++++++++++++++++++++------------------
 static/css/style.css      |  4 +++
 3 files changed, 54 insertions(+), 43 deletions(-)

diff --git a/core/templates/index.html b/core/templates/index.html
index dab9a4c..ff154ef 100644
--- a/core/templates/index.html
+++ b/core/templates/index.html
@@ -10,7 +10,6 @@
                     <div class="position-sticky d-none d-lg-block toc">
                         <div class="card">
                             <div class="card-body">
-                                <h5 class="card-title">Games</h5>
                                 <div id="toc-list" class="list-group">
                                     {% for game in games %}
                                         <a class="list-group-item list-group-item-action plain-text-item"
@@ -39,22 +38,24 @@
                                             <a href="{{ game.twitch_url }}" class="text-decoration-none">{{ game.display_name }}</a>
                                         </h2>
                                         <div class="mt-auto">
-                                            <div class="form-check form-switch">
-                                                <input class="form-check-input"
-                                                       type="checkbox"
-                                                       role="switch"
-                                                       id="new-{{ game.game_id }}">
-                                                <label class="form-check-label" for="new-{{ game.game_id }}">
-                                                    Notify when new drop is found on <a href="https://www.twitch.tv/drops/campaigns">Twitch</a>
-                                                </label>
-                                            </div>
-                                            <div class="form-check form-switch">
-                                                <input class="form-check-input"
-                                                       type="checkbox"
-                                                       role="switch"
-                                                       id="live-{{ game.game_id }}">
-                                                <label class="form-check-label" for="live-{{ game.game_id }}">Notify when a drop starts</label>
-                                            </div>
+                                            {% for webhook in webhooks %}
+                                                <div>
+                                                    <img src="{{ webhook.avatar }}?size=32"
+                                                         alt="{{ webhook.name }}"
+                                                         class="rounded-circle"
+                                                         height="32"
+                                                         width="32">
+                                                    <a href="{{ webhook.url }}" target="_blank">{{ webhook.name }}</a>
+                                                    <div class="form-check form-switch">
+                                                        <input class="form-check-input" type="checkbox" id="new-drop-switch">
+                                                        <label class="form-check-label" for="new-drop-switch">Enable new drop notifications</label>
+                                                    </div>
+                                                    <div class="form-check form-switch">
+                                                        <input class="form-check-input" type="checkbox" id="live-drop-switch">
+                                                        <label class="form-check-label" for="live-drop-switch">Enable live drop notifications</label>
+                                                    </div>
+                                                </div>
+                                            {% endfor %}
                                         </div>
                                         {% for campaign in game.campaigns %}
                                             {% if not forloop.first %}<br>{% endif %}
diff --git a/core/views.py b/core/views.py
index 626f3f0..32733f6 100644
--- a/core/views.py
+++ b/core/views.py
@@ -83,6 +83,34 @@ class GameContext:
     slug: str | None = None
 
 
+def get_avatar(webhook_response: Response) -> str:
+    """Get the avatar URL from the webhook response."""
+    avatar: str = "https://cdn.discordapp.com/embed/avatars/0.png"
+    if webhook_response.is_success and webhook_response.json().get("id") and webhook_response.json().get("avatar"):
+        avatar = f'https://cdn.discordapp.com/avatars/{webhook_response.json().get("id")}/{webhook_response.json().get("avatar")}.png'
+    return avatar
+
+
+def get_webhook_data(webhook: str) -> WebhookData:
+    """Get the webhook data."""
+    with hishel.CacheClient(storage=storage, controller=controller) as client:
+        webhook_response: Response = client.get(url=webhook, extensions={"cache_metadata": True})
+
+    return WebhookData(
+        name=webhook_response.json().get("name") if webhook_response.is_success else "Unknown",
+        url=webhook,
+        avatar=get_avatar(webhook_response),
+        status="Success" if webhook_response.is_success else "Failed",
+        response=webhook_response.text,
+    )
+
+
+def get_webhooks(request: HttpRequest) -> list[str]:
+    """Get the webhooks from the cookie."""
+    cookie: str = request.COOKIES.get("webhooks", "")
+    return list(filter(None, cookie.split(",")))
+
+
 def fetch_games() -> list[Game]:
     """Fetch all games with necessary fields."""
     return list(Game.objects.all().only("id", "image_url", "display_name", "slug"))
@@ -191,11 +219,12 @@ def index(request: HttpRequest) -> HttpResponse:
     """Render the index page."""
     list_of_games: list[GameContext] = prepare_game_contexts()
     sorted_list_of_games: list[GameContext] = sort_games_by_campaign_start(list_of_games)
+    webhooks: list[WebhookData] = [get_webhook_data(webhook) for webhook in get_webhooks(request)]
 
     return TemplateResponse(
         request=request,
         template="index.html",
-        context={"games": sorted_list_of_games},
+        context={"games": sorted_list_of_games, "webhooks": webhooks},
     )
 
 
@@ -227,33 +256,10 @@ class WebhooksView(FormView):
     def get_context_data(self: WebhooksView, **kwargs: dict[str, WebhooksView] | DiscordSettingForm) -> dict[str, Any]:
         """Get the context data for the view."""
         context: dict[str, DiscordSettingForm | list[WebhookData]] = super().get_context_data(**kwargs)
-        cookie: str = self.request.COOKIES.get("webhooks", "")
-        webhooks: list[str] = cookie.split(",")
-        webhooks = list(filter(None, webhooks))
-
-        webhook_responses: list[WebhookData] = []
-
-        with hishel.CacheClient(storage=storage, controller=controller) as client:
-            for webhook in webhooks:
-                our_webhook = WebhookData(name="Unknown", url=webhook, status="Failed", response="No response")
-                response: Response = client.get(url=webhook, extensions={"cache_metadata": True})
-                if response.is_success:
-                    our_webhook.name = response.json().get("name", "Unknown")
-                    our_webhook.status = "Success"
-                else:
-                    our_webhook.status = "Failed"
-
-                our_webhook.response = response.text
-
-                if response.json().get("id") and response.json().get("avatar"):
-                    avatar_url: str = f'https://cdn.discordapp.com/avatars/{response.json().get("id")}/{response.json().get("avatar")}.png'
-
-                our_webhook.avatar = avatar_url or "https://cdn.discordapp.com/embed/avatars/0.png"
-
-                webhook_responses.append(our_webhook)
+        webhooks: list[str] = get_webhooks(self.request)
 
         context.update({
-            "webhooks": webhook_responses,
+            "webhooks": [get_webhook_data(webhook) for webhook in webhooks],
             "form": DiscordSettingForm(),
         })
         return context
diff --git a/static/css/style.css b/static/css/style.css
index d3d4cdf..3a1718d 100644
--- a/static/css/style.css
+++ b/static/css/style.css
@@ -93,3 +93,7 @@ a:hover {
 .plain-text-item.active {
     background-color: #af1548;
 }
+
+.subscribe-text {
+    color: #af1548;
+}