From bbb8ff51bcd4ff3c181a4c62cf22bff8578b0fee Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Joakim=20Hells=C3=A9n?= <tlovinator@gmail.com>
Date: Tue, 20 Dec 2022 03:28:14 +0100
Subject: [PATCH] You can now pause and unpause feeds

---
 discord_rss_bot/main.py             | 20 ++++++++++++++++++++
 discord_rss_bot/templates/feed.html |  9 +++++++++
 2 files changed, 29 insertions(+)

diff --git a/discord_rss_bot/main.py b/discord_rss_bot/main.py
index 93e8e09..de211fe 100644
--- a/discord_rss_bot/main.py
+++ b/discord_rss_bot/main.py
@@ -193,6 +193,26 @@ async def create_feed(feed_url: str = Form(), webhook_dropdown: str = Form()) ->
     return RedirectResponse(url=f"/feed/?feed_url={feed_url}", status_code=303)
 
 
+@app.post("/pause")
+async def pause_feed(feed_url: str = Form()) -> dict[str, str] | RedirectResponse:
+    clean_url: str = urllib.parse.quote(feed_url)
+
+    # Disable/pause the feed.
+    reader.disable_feed_updates(feed_url)
+
+    return RedirectResponse(url=f"/feed/?feed_url={clean_url}", status_code=303)
+
+
+@app.post("/unpause")
+async def unpause_feed(feed_url: str = Form()) -> dict[str, str] | RedirectResponse:
+    clean_url: str = urllib.parse.quote(feed_url)
+
+    # Enable/unpause the feed.
+    reader.enable_feed_updates(feed_url)
+
+    return RedirectResponse(url=f"/feed/?feed_url={clean_url}", status_code=303)
+
+
 @app.get("/add", response_class=HTMLResponse)
 def get_add(request: Request) -> _TemplateResponse:
     """
diff --git a/discord_rss_bot/templates/feed.html b/discord_rss_bot/templates/feed.html
index 96f5a0b..f2cd78f 100644
--- a/discord_rss_bot/templates/feed.html
+++ b/discord_rss_bot/templates/feed.html
@@ -39,6 +39,15 @@
             </form>
         </div>
 
+        {% if not feed.updates_enabled %}
+            <form action="/unpause" method="post">
+                <button class="btn btn-dark btn-sm" name="feed_url" value="{{ feed.url }}">Unpause</button>
+            </form>
+        {% else %}
+            <form action="/pause" method="post">
+                <button class="btn btn-danger btn-sm" name="feed_url" value="{{ feed.url }}">Pause</button>
+            </form>
+        {% endif %}
 
     </div>
     {% for entry in entries %}