From 7cb638c60b75919572e3f924dbad18aeaf923ca4 Mon Sep 17 00:00:00 2001
From: thec0sm0s <deepakrajko14@gmail.com>
Date: Sun, 10 May 2020 16:55:02 +0530
Subject: [PATCH] Somewhat better index

---
 tests/test_app.py | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/tests/test_app.py b/tests/test_app.py
index 1edeba1..25b4976 100644
--- a/tests/test_app.py
+++ b/tests/test_app.py
@@ -16,15 +16,30 @@ app.config["DISCORD_REDIRECT_URI"] = "http://127.0.0.1:5000/callback"
 discord = DiscordOAuth2Session(app)
 
 
+HYPERLINK = '<a href="{}">{}</a>'
+
+
 @app.route("/")
 def index():
+    if not discord.authorized:
+        return HYPERLINK.format(url_for(".login"), "Login")
+    return f"""
+    {HYPERLINK.format(url_for(".me"), "@ME")}<br />
+    {HYPERLINK.format(url_for(".logout"), "Logout")}<br />
+    {HYPERLINK.format(url_for(".user_guilds"), "My Servers")}<br />
+    {HYPERLINK.format(url_for(".add_to_guild", guild_id=475549041741135881), "Add bot to 475549041741135881.")}    
+    """
+
+
+@app.route("/login/")
+def login():
     return discord.create_session()
 
 
 @app.route("/callback/")
 def callback():
     discord.callback()
-    return redirect(url_for(".me"))
+    return redirect(url_for(".index"))
 
 
 @app.route("/me/")