feat(twitch): Add basic user authentication flow.

src/twitch/data.py

@@ -1,4 +1,6 @@
-from data import Registry, RowModel
+import datetime as dt
+
+from data import Registry, RowModel, Table
 from data.columns import Integer, String, Timestamp
 
 
@@ -7,22 +9,71 @@ class TwitchAuthData(Registry):
         """
         Schema
         ------
-        CREATE TABLE twitch_tokens(
-          userid BIGINT PRIMARY KEY,
+        CREATE TABLE twitch_user_auth(
+          userid TEXT PRIMARY KEY,
           access_token TEXT NOT NULL,
           expires_at TIMESTAMPTZ NOT NULL,
           refresh_token TEXT NOT NULL,
           obtained_at TIMESTAMPTZ
         );
-
         """
-        _tablename_ = 'twitch_tokens'
+        _tablename_ = 'twitch_user_auth'
         _cache_ = {}
 
         userid = Integer(primary=True)
         access_token = String()
-        expires_at = Timestamp()
         refresh_token = String()
+        expires_at = Timestamp()
         obtained_at = Timestamp()
 
-# TODO: Scopes
+        @classmethod
+        async def update_user_auth(
+            cls, userid: str, token: str, refresh: str,
+            expires_at: dt.datetime, obtained_at: dt.datetime,
+            scopes: list[str]
+            ):
+            if cls._connector is None:
+                raise ValueError("Attempting to use uninitialised Registry.")
+            async with cls._connector.connection() as conn:
+                cls._connector.conn = conn
+                async with conn.transaction():
+                    # Clear row for this userid
+                    await cls.table.delete_where(userid=userid)
+
+                    # Insert new user row
+                    row = await cls.create(
+                        userid=userid,
+                        access_token=token,
+                        refresh_token=refresh,
+                        expires_at=expires_at,
+                        obtained_at=obtained_at
+                    )
+                    # Insert new scope rows
+                    if scopes:
+                        await TwitchAuthData.user_scopes.insert_many(
+                            ('userid', 'scope'),
+                            *((userid, scope) for scope in scopes)
+                        )
+            return row
+
+        @classmethod
+        async def get_scopes_for(cls, userid: str) -> list[str]:
+            """
+            Get a list of scopes stored for the given user.
+            Will return an empty list if the user is not authenticated.
+            """
+            rows = await TwitchAuthData.user_scopes.select_where(userid=userid)
+
+            return [row.scope for row in rows] if rows else []
+
+
+    """
+    Schema
+    ------
+    CREATE TABLE twitch_user_scopes(
+        userid TEXT REFERENCES twitch_user_auth (userid) ON DELETE CASCADE ON UPDATE CASCADE,
+        scope TEXT
+    );
+    CREATE INDEX twitch_user_scopes_userid ON twitch_user_scopes (userid);
+    """
+    user_scopes = Table('twitch_token_scopes')