feat(index): detect outdated token

2 files changed, 23 insertions, 12 deletions

diff --git a/src/due/media.py b/src/due/media.py
index 8c86855..67d9e01 100644
--- a/src/due/media.py
+++ b/src/due/media.py
@@ -2,17 +2,20 @@ import requests
 
 
 def user_id(anilist):
-    return int(
-        requests.post(
-            "https://graphql.anilist.co",
-            json={"query": "{ Viewer { id } }"},
-            headers={
-                "Authorization": anilist["token_type"] + " " + anilist["access_token"],
-                "Content-Type": "application/json",
-                "Accept": "application/json",
-            },
-        ).json()["data"]["Viewer"]["id"]
-    )
+    viewer = requests.post(
+        "https://graphql.anilist.co",
+        json={"query": "{ Viewer { id } }"},
+        headers={
+            "Authorization": anilist["token_type"] + " " + anilist["access_token"],
+            "Content-Type": "application/json",
+            "Accept": "application/json",
+        },
+    ).json()["data"]["Viewer"]
+
+    if viewer is None:
+        return -1
+
+    return int(viewer["id"])
 
 
 def last_activity(id):
diff --git a/src/due/routes/index.py b/src/due/routes/index.py
index ce8d66c..720a462 100644
--- a/src/due/routes/index.py
+++ b/src/due/routes/index.py
@@ -1,6 +1,6 @@
 import time
 from due.html import anime_to_html, manga_to_html, page
-from due.media import create_collection, last_activity, user_name_to_id
+from due.media import create_collection, last_activity, user_id, user_name_to_id
 from flask import make_response, redirect, request, Blueprint
 import json
 import os
@@ -49,6 +49,14 @@ def home():
 
     if request.cookies.get("anilist"):
         anilist = json.loads(request.cookies.get("anilist"))
+
+        if user_id(anilist) == -1:
+            response = redirect("/")
+
+            response.delete_cookie("anilist")
+
+            return response
+
         start = time.time()
         (current_anime, name) = create_collection(
             anilist, "ANIME", request.args.get("username")