Customisations for nico.

src/bot.py

@@ -1,6 +1,7 @@
 import asyncio
 import logging
 import websockets
+from aiohttp import web
 
 from twitchio.web import AiohttpAdapter
 
@@ -20,6 +21,26 @@ class ProxyAiohttpAdapter(AiohttpAdapter):
     def _find_redirect(self, request):
         return self.redirect_url
 
+    async def oauth_callback(self, request: web.Request) -> web.Response:
+        payload: FetchTokenPayload = await self.fetch_token(request)
+
+        # Change the default success response to no content...
+        print(payload)
+        if 200 <= payload.response.status < 300:
+            return web.Response(
+                status=200,
+                text=(
+                    "Successfully added Croccyfocus to your channel! Good luck with your hyperfocused sessions!\n"
+                    "<p>For further usage and setup information, see the documentation <a href=\"{docs}\">here</a>.</p>"
+                ).format(
+                    docs="https://git.thewisewolf.dev/HoloTech/hyperfocus-twitch-bot/wiki/Focus-docs"
+                ),
+                content_type="text/html",
+            )
+
+        # Return the default error responses...
+        return payload.response
+
 
 async def main():
     db = Database(conf.data['args'])

src/botdata.py

@@ -14,6 +14,7 @@ class UserAuth(RowModel):
       _timestamp TIMESTAMPTZ NOT NULL DEFAULT NOW()
     );
     """
+    _schema_ = 'nico'
     _tablename_ = 'user_auth'
     _cache_ = {}
 
@@ -36,6 +37,7 @@ class BotChannel(RowModel):
       _timestamp TIMESTAMPTZ NOT NULL DEFAULT NOW()
     );
     """
+    _schema_ = 'nico'
     _tablename_ = 'bot_channels'
     _cache_ = {}
 
@@ -73,10 +75,10 @@ class BotData(Registry):
     bot_channels = BotChannel.table
 
     """
-    CREATE TABLE user_auth_scopes(
-      userid TEXT NOT NULL REFERENCES user_auth(userid) ON DELETE CASCADE,
+    CREATE TABLE nico_user_auth_scopes(
+      userid TEXT NOT NULL REFERENCES nico.user_auth(userid) ON DELETE CASCADE,
       scope TEXT NOT NULL
     );
     """
-    user_auth_scopes = Table('user_auth_scopes')
+    user_auth_scopes = Table('user_auth_scopes', schema='nico')