feat(twitch): Add basic user authentication flow.

.gitmodules

@@ -1,12 +1,12 @@
 [submodule "bot/gui"]
 	path = src/gui
-	url = git@github.com:Intery/CafeHelper-GUI.git
+	url = https://github.com/StudyLions/StudyLion-Plugin-GUI.git
 [submodule "skins"]
 	path = skins
-	url = git@github.com:Intery/CafeHelper-Skins.git
+	url = https://github.com/Intery/pillow-skins.git
 [submodule "src/modules/voicefix"]
 	path = src/modules/voicefix
-	url = git@github.com:Intery/StudyLion-voicefix.git
+	url = https://github.com/Intery/StudyLion-voicefix.git
 [submodule "src/modules/streamalerts"]
 	path = src/modules/streamalerts
 	url = https://github.com/Intery/StudyLion-streamalerts.git

data/schema.sql

@@ -1454,7 +1454,6 @@ CREATE TABLE shoutouts(
 CREATE TABLE counters(
   counterid SERIAL PRIMARY KEY,
   name TEXT NOT NULL,
-  category TEXT,
   created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
 );
 CREATE UNIQUE INDEX counters_name ON counters (name);
@@ -1465,7 +1464,6 @@ CREATE TABLE counter_log(
   userid INTEGER NOT NULL,
   value INTEGER NOT NULL,
   created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
-  details TEXT,
   context_str TEXT
 );
 CREATE INDEX counter_log_counterid ON counter_log (counterid);
@@ -1486,81 +1484,6 @@ CREATE UNIQUE INDEX channel_tags_channelid_name ON channel_tags (channelid, name
 
 -- }}}
 
--- Voice Roles {{{
-CREATE TABLE voice_roles(
-  voice_role_id SERIAL PRIMARY KEY,
-  channelid BIGINT NOT NULL,
-  roleid BIGINT NOT NULL,
-  created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-);
-CREATE INDEX voice_role_channels on voice_roles (channelid);
-
--- }}}
-
--- User and Community Profiles {{{ 
-DROP TABLE IF EXISTS community_members;
-DROP TABLE IF EXISTS communities_twitch;
-DROP TABLE IF EXISTS communities_discord;
-DROP TABLE IF EXISTS communities;
-DROP TABLE IF EXISTS profiles_twitch;
-DROP TABLE IF EXISTS profiles_discord;
-DROP TABLE IF EXISTS user_profiles;
-
-
-CREATE TABLE user_profiles(
-  profileid SERIAL PRIMARY KEY,
-  nickname TEXT,
-  migrated INTEGER REFERENCES user_profiles (profileid) ON DELETE CASCADE ON UPDATE CASCADE,
-  created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-);
-
-CREATE TABLE profiles_discord(
-  linkid SERIAL PRIMARY KEY,
-  profileid INTEGER NOT NULL REFERENCES user_profiles (profileid) ON DELETE CASCADE ON UPDATE CASCADE,
-  userid BIGINT NOT NULL,
-  created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-);
-CREATE INDEX profiles_discord_profileid ON profiles_discord (profileid);
-CREATE UNIQUE INDEX profiles_discord_userid ON profiles_discord (userid);
-
-CREATE TABLE profiles_twitch(
-  linkid SERIAL PRIMARY KEY,
-  profileid INTEGER NOT NULL REFERENCES user_profiles (profileid) ON DELETE CASCADE ON UPDATE CASCADE,
-  userid TEXT NOT NULL,
-  created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-);
-CREATE INDEX profiles_twitch_profileid ON profiles_twitch (profileid);
-CREATE UNIQUE INDEX profiles_twitch_userid ON profiles_twitch (userid);
-
-
-CREATE TABLE communities(
-  communityid SERIAL PRIMARY KEY,
-  migrated INTEGER REFERENCES user_profiles (profileid) ON DELETE CASCADE ON UPDATE CASCADE,
-  created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-);
-
-CREATE TABLE communities_discord(
-  guildid BIGINT PRIMARY KEY,
-  communityid INTEGER NOT NULL REFERENCES communities (communityid) ON DELETE CASCADE ON UPDATE CASCADE,
-  linked_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-);
-CREATE INDEX communities_discord_communityid ON communities_discord (communityid);
-
-CREATE TABLE communities_twitch(
-  channelid TEXT PRIMARY KEY,
-  communityid INTEGER NOT NULL REFERENCES communities (communityid) ON DELETE CASCADE ON UPDATE CASCADE,
-  linked_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-);
-CREATE INDEX communities_twitch_communityid ON communities_twitch (communityid);
-
-CREATE TABLE community_members(
-  memberid SERIAL PRIMARY KEY,
-  communityid INTEGER NOT NULL REFERENCES communities (communityid) ON DELETE CASCADE ON UPDATE CASCADE,
-  profileid INTEGER NOT NULL REFERENCES user_profiles (profileid) ON DELETE CASCADE ON UPDATE CASCADE,
-  created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-);
-CREATE UNIQUE INDEX community_members_communityid_profileid ON community_members (communityid, profileid);
--- }}}
 
 -- Twitch User Auth {{{
 CREATE TABLE twitch_user_auth(
@@ -1571,6 +1494,7 @@ CREATE TABLE twitch_user_auth(
   obtained_at TIMESTAMPTZ
 );
 
+
 CREATE TABLE twitch_user_scopes(
   userid TEXT REFERENCES twitch_user_auth (userid) ON DELETE CASCADE ON UPDATE CASCADE,
   scope TEXT

src/bot.py

@@ -98,7 +98,6 @@ async def main():
                 config=conf,
                 initial_extensions=[
                     'utils', 'core', 'analytics',
-                    'twitch',
                     'modules',
                     'babel',
                     'tracking.voice', 'tracking.text',

src/meta/CrocBot.py

@@ -1,12 +1,9 @@
-from collections import defaultdict
 from typing import TYPE_CHECKING
 
 import logging
 
-import twitchio
 from twitchio.ext import commands
 from twitchio.ext import pubsub
-from twitchio.ext.commands.core import itertools
 
 from data import Database
 
@@ -26,51 +23,5 @@ class CrocBot(commands.Bot):
         self.data = data
         self.pubsub = pubsub.PubSubPool(self)
 
-        self._member_cache = defaultdict(dict)
-
     async def event_ready(self):
         logger.info(f"Logged in as {self.nick}. User id is {self.user_id}")
-
-    async def event_join(self, channel: twitchio.Channel, user: twitchio.User):
-        self._member_cache[channel.name][user.name] = user
-
-    async def event_message(self, message: twitchio.Message):
-        if message.channel and message.author:
-            self._member_cache[message.channel.name][message.author.name] = message.author
-        await self.handle_commands(message)
-
-    async def seek_user(self, userstr: str, matching=True, fuzzy=True):
-        if userstr.startswith('@'):
-            matching = False
-        userstr = userstr.strip('@ ')
-
-        result = None
-        if matching and len(userstr) >= 3:
-            lowered = userstr.lower()
-            full_matches = []
-            for user in itertools.chain(*(cmems.values() for cmems in self._member_cache.values())):
-                matchstr = user.name.lower()
-                print(matchstr)
-                if matchstr.startswith(lowered):
-                    result = user
-                    break
-                if lowered in matchstr:
-                    full_matches.append(user)
-            if result is None and full_matches:
-                result = full_matches[0]
-        print(result)
-
-        if result is None:
-            lookup = userstr
-        elif result.id is None:
-            lookup = result.name
-        else:
-            lookup = None
-
-        if lookup:
-            found = await self.fetch_users(names=[lookup])
-            if found:
-                result = found[0]
-
-        # No matches found
-        return result

src/meta/LionBot.py

@@ -27,7 +27,6 @@ if TYPE_CHECKING:
     from meta.CrocBot import CrocBot
     from core.cog import CoreCog
     from core.config import ConfigCog
-    from twitch.cog import TwitchAuthCog
     from tracking.voice.cog import VoiceTrackerCog
     from tracking.text.cog import TextTrackerCog
     from modules.config.cog import GuildConfigCog
@@ -50,7 +49,6 @@ if TYPE_CHECKING:
     from modules.topgg.cog import TopggCog
     from modules.user_config.cog import UserConfigCog
     from modules.video_channels.cog import VideoCog
-    from modules.profiles.cog import ProfileCog
 
 logger = logging.getLogger(__name__)
 
@@ -144,10 +142,6 @@ class LionBot(Bot):
     # To make the type checker happy about fetching cogs by name
     # TODO: Move this to stubs at some point
 
-    @overload
-    def get_cog(self, name: Literal['ProfileCog']) -> 'ProfileCog':
-        ...
-
     @overload
     def get_cog(self, name: Literal['CoreCog']) -> 'CoreCog':
         ...
@@ -160,10 +154,6 @@ class LionBot(Bot):
     def get_cog(self, name: Literal['VoiceTrackerCog']) -> 'VoiceTrackerCog':
         ...
 
-    @overload
-    def get_cog(self, name: Literal['TwitchAuthCog']) -> 'TwitchAuthCog':
-        ...
-
     @overload
     def get_cog(self, name: Literal['TextTrackerCog']) -> 'TextTrackerCog':
         ...

src/meta/LionCog.py

@@ -22,7 +22,6 @@ class LionCog(Cog):
         cls._placeholder_groups_ = set()
         cls._twitch_cmds_ = {}
         cls._twitch_events_ = {}
-        cls._twitch_events_loaded_ = set()
 
         for base in reversed(cls.__mro__):
             for elem, value in base.__dict__.items():
@@ -48,27 +47,6 @@ class LionCog(Cog):
 
         return await super()._inject(bot, *args, *kwargs)
 
-    def add_twitch_command(self, bot: Bot, command: Command):
-        """
-        Dynamically register a command with the given bot.
-
-        The command will be deregistered on cog unload.
-        """
-        # Remove any conflicting commands
-        if cmd := bot.get_command(command.name):
-            bot.remove_command(cmd.name)
-            self._twitch_cmds_.pop(command.name, None)
-
-        try:
-            self._twitch_cmds_[command.name] = command
-            command._instance = self
-            command.cog = self
-            bot.add_command(command)
-        except Exception:
-            # Ensure the command doesn't die in the internal command cache
-            self._twitch_cmds_.pop(command.name, None)
-            raise
-
     def _load_twitch_methods(self, bot: Bot):
         for name, command in self._twitch_cmds_.items():
             command._instance = self

src/modules/__init__.py

@@ -2,7 +2,6 @@ this_package = 'modules'
 
 active_discord = [
     '.sysadmin',
-    '.profiles',
     '.config',
     '.user_config',
     '.skins',
@@ -31,7 +30,6 @@ active_discord = [
     '.nowdoing',
     '.shoutouts',
     '.tagstrings',
-    '.voiceroles',
 ]
 
 async def setup(bot):

src/modules/counters/cog.py

@@ -25,75 +25,6 @@ class PERIOD(Enum):
     YEAR = ('this year', 'y', 'year', 'yearly')
 
 
-def counter_cmd_factory(
-    counter: str,
-    response: str,
-    default_period: Optional[PERIOD] = PERIOD.STREAM,
-    context: Optional[str] = None
-):
-    context = context or f"cmd: {counter}"
-    async def counter_cmd(cog, ctx: commands.Context, *, args: Optional[str] = None):
-        userid = int(ctx.author.id)
-        channelid = int((await ctx.channel.user()).id)
-        period, start_time = await cog.parse_period(channelid, '', default=default_period)
-
-        args = (args or '').strip(" 󠀀 ")
-        splits = args.split(maxsplit=1)
-        splits = [split.strip() for split in splits if split]
-
-        details = None
-        amount = 1
-        
-        if splits:
-            if splits[0].isdigit() or (splits[0].startswith('-') and splits[0][1:].isdigit()):
-                amount = int(splits[0])
-                splits = splits[1:]
-            if splits:
-                details = ' '.join(splits)
-
-        await cog.add_to_counter(
-            counter, userid, amount,
-            context=context,
-            details=details
-        )
-        lb = await cog.leaderboard(counter, start_time=start_time)
-        user_total = lb.get(userid, 0)
-        total = sum(lb.values())
-        await ctx.reply(
-            response.format(
-                total=total,
-                period=period,
-                period_name=period.value[0],
-                detailsorname=details or counter,
-                user_total=user_total,
-            )
-        )
-
-    async def lb_cmd(cog, ctx: commands.Context, *, args: str = ''):
-        user = await ctx.channel.user()
-        await ctx.reply(await cog.formatted_lb(counter, args, int(user.id)))
-
-    async def undo_cmd(cog, ctx: commands.Context):
-        userid = int(ctx.author.id)
-        channelid = int((await ctx.channel.user()).id)
-        _counter = await cog.fetch_counter(counter)
-        query = cog.data.CounterEntry.fetch_where(
-            counterid=_counter.counterid,
-            userid=userid,
-        )
-        query.order_by('created_at', direction=ORDER.DESC)
-        query.limit(1)
-        results = await query
-        if not results:
-            await ctx.reply("Nothing to delete!")
-        else:
-            row = results[0]
-            await row.delete()
-            await ctx.reply("Undo successful!")
-
-    return (counter_cmd, lb_cmd, undo_cmd)
-
-
 class CounterCog(LionCog):
     def __init__(self, bot: LionBot):
         self.bot = bot
@@ -107,7 +38,6 @@ class CounterCog(LionCog):
 
     async def cog_load(self):
         self._load_twitch_methods(self.crocbot)
-        await self.load_counter_commands()
 
         await self.data.init()
         await self.load_counters()
@@ -116,29 +46,6 @@ class CounterCog(LionCog):
     async def cog_unload(self):
         self._unload_twitch_methods(self.crocbot)
 
-    async def load_counter_commands(self):
-        rows = await self.data.CounterCommand.fetch_where()
-        for row in rows:
-            counter = await self.data.Counter.fetch(row.counterid)
-            counter_cb, lb_cb, undo_cb = counter_cmd_factory(
-                counter.name,
-                row.response
-            )
-            cmds = []
-            main_cmd = commands.command(name=row.name)(counter_cb)
-            cmds.append(main_cmd)
-            if row.lbname:
-                lb_cmd = commands.command(name=row.lbname)(lb_cb)
-                cmds.append(lb_cmd)
-            if row.undoname:
-                undo_cmd = commands.command(name=row.undoname)(undo_cb)
-                cmds.append(undo_cmd)
-
-            for cmd in cmds:
-                self.add_twitch_command(self.crocbot, cmd)
-
-        logger.info(f"(Re)Loaded {len(rows)} counter commands!")
-
     async def cog_check(self, ctx):
         return True
 
@@ -173,19 +80,13 @@ class CounterCog(LionCog):
         if row:
             await self.data.CounterEntry.table.delete_where(counterid=row.counterid)
 
-    async def add_to_counter(
+    async def add_to_counter(self, counter: str, userid: int, value: int, context: Optional[str]=None):
-        self,
-        counter: str, userid: int, value: int,
-        context: Optional[str]=None,
-        details: Optional[str]=None,
-    ):
         row = await self.fetch_counter(counter)
         return await self.data.CounterEntry.create(
             counterid=row.counterid,
             userid=userid,
             value=value,
-            context_str=context,
+            context_str=context
-            details=details
         )
 
     async def leaderboard(self, counter: str, start_time=None):
@@ -254,43 +155,8 @@ class CounterCog(LionCog):
         elif subcmd == 'clear':
             await self.reset_counter(name)
             await ctx.reply(f"'{name}' counter reset.")
-        elif subcmd == 'alias':
-            splits = args.split(maxsplit=3) if args else []
-            counter = await self.fetch_counter(name)
-            rows = await self.data.CounterCommand.fetch_where(counterid=counter.counterid)
-            existing = rows[0] if rows else None
-            if existing and not args:
-                # Show current alias
-                await ctx.reply(
-                    f"Counter '{name}' aliases: '!{existing.name}' to add to counter; "
-                    f"'!{existing.lbname}' to view counter leaderboard; "
-                    f"'!{existing.undoname}' to undo (your) last addition."
-                )
-            elif len(splits) < 4:
-                # Show usage
-                await ctx.reply(
-                    "USAGE: !counter <name> alias <cmdname> <lbname> <undoname> <response> -- "
-                    "Response accepts keywords {total}, {period}, {period_name}, {detailsorname}, {user_total}."
-                )
         else:
-                # Create new alias
+            await ctx.reply(f"Unrecognised subcommand {subcmd}. Supported subcommands: 'show', 'add', 'lb', 'clear'.")
-                cmdname, lbname, undoname, response = splits
-                # Remove any existing alias
-                await self.data.CounterCommand.table.delete_where(name=cmdname)
-
-                alias = await self.data.CounterCommand.create(
-                    name=cmdname,
-                    counterid=counter.counterid,
-                    lbname=lbname, undoname=undoname, response=response
-                )
-                await self.load_counter_commands()
-                await ctx.reply(
-                    f"Alias created for counter '{name}': '!{alias.name}' to add to counter; "
-                    f"'!{alias.lbname}' to view counter leaderboard; "
-                    f"'!{alias.undoname}' to undo (your) last addition."
-                )
-        else:
-            await ctx.reply(f"Unrecognised subcommand {subcmd}. Supported subcommands: 'show', 'add', 'lb', 'clear', 'alias'.")
 
     async def parse_period(self, userid: int, periodstr: str, default=PERIOD.STREAM):
         if periodstr:
@@ -345,3 +211,82 @@ class CounterCog(LionCog):
             return f"{counter} {period.value[-1]} leaderboard --- {lbstr}"
         else:
             return f"{counter} {period.value[-1]} leaderboard is empty!"
+
+    # Misc actual counter commands
+    # TODO: Factor this out to a different module...
+    @commands.command()
+    async def tea(self, ctx: commands.Context, *, args: Optional[str]=None):
+        userid = int(ctx.author.id)
+        channelid = int((await ctx.channel.user()).id)
+        period, start_time = await self.parse_period(channelid, '')
+        counter = 'tea'
+
+        await self.add_to_counter(
+            counter,
+            userid,
+            1,
+            context='cmd: tea'
+        )
+        lb = await self.leaderboard(counter, start_time=start_time)
+        user_total = lb.get(userid, 0)
+        total = sum(lb.values())
+        await ctx.reply(f"Enjoy your tea! We have had {total} cups of tea {period.value[0]}.")
+        
+    @commands.command()
+    async def tealb(self, ctx: commands.Context, *, args: str = ''):
+        user = await ctx.channel.user()
+        await ctx.reply(await self.formatted_lb('tea', args, int(user.id)))
+
+    @commands.command()
+    async def coffee(self, ctx: commands.Context, *, args: Optional[str]=None):
+        userid = int(ctx.author.id)
+        channelid = int((await ctx.channel.user()).id)
+        period, start_time = await self.parse_period(channelid, '')
+        counter = 'coffee'
+
+        await self.add_to_counter(
+            counter,
+            userid,
+            1,
+            context='cmd: coffee'
+        )
+        lb = await self.leaderboard(counter, start_time=start_time)
+        user_total = lb.get(userid, 0)
+        total = sum(lb.values())
+        await ctx.reply(f"Enjoy your coffee! We have had {total} cups of coffee {period.value[0]}.")
+        
+    @commands.command()
+    async def coffeelb(self, ctx: commands.Context, *, args: str = ''):
+        user = await ctx.channel.user()
+        await ctx.reply(await self.formatted_lb('coffee', args, int(user.id)))
+
+    @commands.command()
+    async def water(self, ctx: commands.Context, *, args: Optional[str]=None):
+        userid = int(ctx.author.id)
+        channelid = int((await ctx.channel.user()).id)
+        period, start_time = await self.parse_period(channelid, '')
+        counter = 'water'
+
+        await self.add_to_counter(
+            counter,
+            userid,
+            1,
+            context='cmd: water'
+        )
+        lb = await self.leaderboard(counter, start_time=start_time)
+        user_total = lb.get(userid, 0)
+        total = sum(lb.values())
+        await ctx.reply(f"Good job hydrating! We have had {total} cups of water {period.value[0]}.")
+
+    @commands.command()
+    async def waterlb(self, ctx: commands.Context, *, args: str = ''):
+        user = await ctx.channel.user()
+        await ctx.reply(await self.formatted_lb('water', args, int(user.id)))
+
+    @commands.command()
+    async def stuff(self, ctx: commands.Context, *, args: str = ''):
+        await ctx.reply(f"Stuff {args}")
+
+    @cmds.hybrid_command('water')
+    async def d_water_cmd(self, ctx):
+        await ctx.reply(repr(ctx))

src/modules/counters/data.py

@@ -10,8 +10,7 @@ class CounterData(Registry):
         CREATE TABLE counters(
             counterid SERIAL PRIMARY KEY,
             name TEXT NOT NULL,
-            created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+            created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-            category TEXT
         );
         CREATE UNIQUE INDEX counters_name ON counters (name);
         """
@@ -20,7 +19,6 @@ class CounterData(Registry):
 
         counterid = Integer(primary=True)
         name = String()
-        category = String()
         created_at = Timestamp()
 
     class CounterEntry(RowModel):
@@ -33,8 +31,7 @@ class CounterData(Registry):
             userid INTEGER NOT NULL,
             value INTEGER NOT NULL,
             created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
-            context_str TEXT,
+            context_str TEXT
-            details TEXT
         );
         CREATE INDEX counter_log_counterid ON counter_log (counterid);
         """
@@ -47,28 +44,5 @@ class CounterData(Registry):
         value = Integer()
         created_at = Timestamp()
         context_str = String()
-        details = String()
 
-    class CounterCommand(RowModel):
-        """
-        Schema
-        ------
-        CREATE TABLE counter_commands(
-            name TEXT PRIMARY KEY,
-            counterid INTEGER NOT NULL REFERENCES counters (counterid) ON UPDATE CASCADE ON DELETE CASCADE,
-            lbname TEXT,
-            undoname TEXT,
-            response TEXT NOT NULL
-        );
-        """
-        # NOTE: This table will be replaced by aliases soon anyway
-        # So no need to worry about integrity or future-proofing
-        _tablename_ = 'counter_commands'
-        _cache_ = {}
-
-        name = String(primary=True)
-        counterid = Integer()
-        lbname = String()
-        undoname = String()
-        response = String()
 

src/modules/counters/migration.sql

@@ -1,9 +0,0 @@
-ALTER TABLE counters ADD COLUMN category TEXT;
-ALTER TABLE counter_log ADD COLUMN details TEXT;
-CREATE TABLE counter_commands(
-  name TEXT PRIMARY KEY,
-  counterid INTEGER NOT NULL REFERENCES counters (counterid) ON UPDATE CASCADE ON DELETE CASCADE,
-  lbname TEXT,
-  undoname TEXT,
-  response TEXT NOT NULL
-);

src/modules/pomodoro/cog.py

@@ -47,33 +47,17 @@ class TimerChannel(Channel):
         super().__init__(**kwargs)
         self.cog = cog
 
-        self.channelid = 1261999440160624734
-        self.goal = 12
-
     async def on_connection(self, websocket, event):
         await super().on_connection(websocket, event)
+        timer = self.cog.get_channel_timer(1261999440160624734)
+        if timer is not None:
             await self.send_set(
-            **await self.get_args_for(self.channelid),
+                timer.data.last_started,
-            goal=self.goal,
+                timer.data.focus_length,
+                timer.data.break_length,
                 websocket=websocket,
             )
 
-    async def send_updates(self):
-        await self.send_set(
-            **await self.get_args_for(self.channelid),
-            goal=self.goal,
-        )
-
-    async def get_args_for(self, channelid):
-        timer = self.cog.get_channel_timer(channelid)
-        if timer is None:
-            raise ValueError(f"Timer {channelid} doesn't exist.")
-        return {
-            'start_at': timer.data.last_started,
-            'focus_length': timer.data.focus_length,
-            'break_length': timer.data.break_length,
-        }
-
     async def send_set(self, start_at, focus_length, break_length, goal=12, websocket=None):
         await self.send_event({
             'type': "DO",
@@ -320,6 +304,8 @@ class TimerCog(LionCog):
             return
         if member.bot:
             return
+        if 1148167212901859328 not in [role.id for role in member.roles]:
+            return
 
         # If a member is leaving or joining a running timer, trigger a status update
         if before.channel != after.channel:
@@ -329,7 +315,6 @@ class TimerCog(LionCog):
             tasks = []
             if leaving is not None:
                 tasks.append(asyncio.create_task(leaving.update_status_card()))
-                leaving.last_seen.pop(member.id, None)
             if joining is not None:
                 joining.last_seen[member.id] = utc_now()
                 if not joining.running and joining.auto_restart:
@@ -1074,18 +1059,8 @@ class TimerCog(LionCog):
     @low_management_ward
     async def streamtimer_update_cmd(self, ctx: LionContext,
                                      new_start: Optional[str] = None,
-                                     new_goal: Optional[int] = None,
+                                     new_goal: int = 12):
-                                     new_channel: Optional[discord.VoiceChannel] = None,
+        timer = self.get_channel_timer(1261999440160624734)
-                                     ):
-        if new_channel is not None:
-            channelid = self.channel.channelid = new_channel.id
-        else:
-            channelid = self.channel.channelid
-
-        if new_goal is not None:
-            self.channel.goal = new_goal
-
-        timer = self.get_channel_timer(channelid)
         if timer is None:
             return
         if new_start:
@@ -1093,5 +1068,10 @@ class TimerCog(LionCog):
             start_at = await self.bot.get_cog('Reminders').parse_time_static(new_start, timezone)
             await timer.data.update(last_started=start_at)
 
-        await self.channel.send_updates()
+        await self.channel.send_set(
+            timer.data.last_started,
+            timer.data.focus_length,
+            timer.data.break_length,
+            goal=new_goal,
+        )
         await ctx.reply("Stream Timer Updated")

src/modules/profiles/__init__.py

@@ -1,8 +0,0 @@
-import logging
-
-logger = logging.getLogger(__name__)
-
-from .cog import ProfileCog
-
-async def setup(bot):
-    await bot.add_cog(ProfileCog(bot))

src/modules/profiles/cog.py

@@ -1,404 +0,0 @@
-import asyncio
-from enum import Enum
-from typing import Optional, overload
-from datetime import timedelta
-
-import discord
-from discord import app_commands as appcmds
-from discord.ext import commands as cmds
-from twitchAPI.type import AuthScope
-import twitchio
-from twitchio.ext import commands
-from twitchio import User
-from twitchAPI.object.api import TwitchUser
-
-
-from data.queries import ORDER
-from meta import LionCog, LionBot, CrocBot, LionContext
-from meta.logger import log_wrap
-from utils.lib import utc_now
-from . import logger
-from .data import ProfileData
-from .profile import UserProfile
-from .community import Community
-
-
-class ProfileCog(LionCog):
-    def __init__(self, bot: LionBot):
-        self.bot = bot
-
-        assert bot.crocbot is not None
-        self.crocbot: CrocBot = bot.crocbot
-        self.data = bot.db.load_registry(ProfileData())
-
-        self._profile_migrators = {}
-        self._comm_migrators = {}
-
-    async def cog_load(self):
-        await self.data.init()
-
-    async def cog_check(self, ctx):
-        return True
-
-    # Profile API
-    def add_profile_migrator(self, migrator, name=None):
-        name = name or migrator.__name__
-        self._profile_migrators[name or migrator.__name__] = migrator
-
-        logger.info(
-            f"Added user profile migrator {name}: {migrator}"
-        )
-        return migrator
-
-    def del_profile_migrator(self, name: str):
-        migrator = self._profile_migrators.pop(name, None)
-
-        logger.info(
-            f"Removed user profile migrator {name}: {migrator}"
-        )
-
-    @log_wrap(action="profile migration")
-    async def migrate_profile(self, source_profile, target_profile) -> list[str]:
-        logger.info(
-            f"Beginning user profile migration from {source_profile!r} to {target_profile!r}"
-        )
-        results = []
-        # Wrap this in a transaction so if something goes wrong with migration,
-        # we roll back safely (although this may mess up caches)
-        async with self.bot.db.connection() as conn:
-            self.bot.db.conn = conn
-            async with conn.transaction():
-                for name, migrator in self._profile_migrators.items():
-                    try:
-                        result = await migrator(source_profile, target_profile)
-                        if result:
-                            results.append(result)
-                    except Exception:
-                        logger.exception(
-                            f"Unexpected exception running user profile migrator {name} "
-                            f"migrating {source_profile!r} to {target_profile!r}."
-                        )
-                        raise
-
-                # Move all Discord and Twitch profile references over to the new profile
-                discord_rows = await self.data.DiscordProfileRow.table.update_where(
-                    profileid=source_profile.profileid
-                ).set(profileid=target_profile.profileid)
-                results.append(f"Migrated {len(discord_rows)} attached discord account(s).")
-
-                twitch_rows = await self.data.TwitchProfileRow.table.update_where(
-                    profileid=source_profile.profileid
-                ).set(profileid=target_profile.profileid)
-                results.append(f"Migrated {len(twitch_rows)} attached twitch account(s).")
-
-                # And then mark the old profile as migrated
-                await source_profile.update(migrated=target_profile.profileid)
-                results.append("Marking old profile as migrated.. finished!")
-        return results
-
-    async def fetch_profile_by_id(self, profile_id: int) -> UserProfile:
-        """
-        Fetch a UserProfile by the given id.
-        """
-        return await UserProfile.fetch(self.bot, profile_id=profile_id)
-
-    async def fetch_profile_discord(self, user: discord.Member | discord.User) -> UserProfile:
-        """
-        Fetch or create a UserProfile from the provided discord account.
-        """
-        profile = await UserProfile.fetch_from_discordid(self.bot, user.id)
-        if profile is None:
-            profile = await UserProfile.create_from_discord(self.bot, user)
-        return profile
-
-    async def fetch_profile_twitch(self, user: twitchio.User) -> UserProfile:
-        """
-        Fetch or create a UserProfile from the provided twitch account.
-        """
-        profile = await UserProfile.fetch_from_twitchid(self.bot, user.id)
-        if profile is None:
-            profile = await UserProfile.create_from_twitch(self.bot, user)
-        return profile
-
-    # Community API
-    def add_community_migrator(self, migrator, name=None):
-        name = name or migrator.__name__
-        self._comm_migrators[name or migrator.__name__] = migrator
-
-        logger.info(
-            f"Added community migrator {name}: {migrator}"
-        )
-        return migrator
-
-    def del_community_migrator(self, name: str):
-        migrator = self._comm_migrators.pop(name, None)
-
-        logger.info(
-            f"Removed community migrator {name}: {migrator}"
-        )
-
-    @log_wrap(action="community migration")
-    async def migrate_community(self, source_comm, target_comm) -> list[str]:
-        logger.info(
-            f"Beginning community migration from {source_comm!r} to {target_comm!r}"
-        )
-        results = []
-        # Wrap this in a transaction so if something goes wrong with migration,
-        # we roll back safely (although this may mess up caches)
-        async with self.bot.db.connection() as conn:
-            self.bot.db.conn = conn
-            async with conn.transaction():
-                for name, migrator in self._comm_migrators.items():
-                    try:
-                        result = await migrator(source_comm, target_comm)
-                        if result:
-                            results.append(result)
-                    except Exception:
-                        logger.exception(
-                            f"Unexpected exception running community migrator {name} "
-                            f"migrating {source_comm!r} to {target_comm!r}."
-                        )
-                        raise
-
-                # Move all Discord and Twitch community preferences over to the new profile
-                discord_rows = await self.data.DiscordCommunityRow.table.update_where(
-                    profileid=source_comm.communityid
-                ).set(communityid=target_comm.communityid)
-                results.append(f"Migrated {len(discord_rows)} attached discord guilds.")
-
-                twitch_rows = await self.data.TwitchCommunityRow.table.update_where(
-                    communityid=source_comm.communityid
-                ).set(communityid=target_comm.communityid)
-                results.append(f"Migrated {len(twitch_rows)} attached twitch channel(s).")
-
-                # And then mark the old community as migrated
-                await source_comm.update(migrated=target_comm.communityid)
-                results.append("Marking old community as migrated.. finished!")
-        return results
-
-    async def fetch_community_by_id(self, community_id: int) -> Community:
-        """
-        Fetch a Community by the given id.
-        """
-        return await Community.fetch(self.bot, community_id=community_id)
-
-    async def fetch_community_discord(self, guild: discord.Guild) -> Community:
-        """
-        Fetch or create a Community from the provided discord guild.
-        """
-        comm = await Community.fetch_from_discordid(self.bot, guild.id)
-        if comm is None:
-            comm = await Community.create_from_discord(self.bot, guild)
-        return comm
-
-    async def fetch_community_twitch(self, user: twitchio.User) -> Community:
-        """
-        Fetch or create a Community from the provided twitch account.
-        """
-        community = await Community.fetch_from_twitchid(self.bot, user.id)
-        if community is None:
-            community = await Community.create_from_twitch(self.bot, user)
-        return community
-
-    # ----- Profile Commands -----
-    @cmds.hybrid_group(
-        name='profiles',
-        description="Base comand group for user profiles."
-    )
-    async def profiles_grp(self, ctx: LionContext):
-        ...
-
-    @profiles_grp.group(
-        name='link',
-        description="Base command group for linking profiles"
-    )
-    async def profiles_link_grp(self, ctx: LionContext):
-        ...
-
-    @profiles_link_grp.command(
-        name='twitch',
-        description="Link a twitch account to your current profile."
-    )
-    async def profiles_link_twitch_cmd(self, ctx: LionContext):
-        if not ctx.interaction:
-            return
-
-        await ctx.interaction.response.defer(ephemeral=True)
-
-        # Ask the user to go through auth to get their userid
-        auth_cog = self.bot.get_cog('TwitchAuthCog')
-        flow = await auth_cog.start_auth()
-        message = await ctx.reply(
-            f"Please [click here]({flow.auth.return_auth_url()}) to link your profile "
-            "to Twitch."
-        )
-        authrow = await flow.run()
-        await message.edit(
-            content="Authentication Complete! Beginning profile merge..."
-        )
-
-        results = await self.crocbot.fetch_users(ids=[authrow.userid])
-        if not results:
-            logger.error(
-                f"User {authrow} obtained from Twitch authentication does not exist."
-            )
-            await ctx.error_reply("Sorry, something went wrong. Please try again later!")
-            return
-
-        user = results[0]
-
-        # Retrieve author's profile if it exists
-        author_profile = await UserProfile.fetch_from_discordid(self.bot, ctx.author.id)
-
-        # Check if the twitch-side user has a profile 
-        source_profile = await UserProfile.fetch_from_twitchid(self.bot, user.id)
-
-        if author_profile and source_profile is None:
-            # All we need to do is attach the twitch row
-            await author_profile.attach_twitch(user)
-            await message.edit(
-                content=f"Successfully added Twitch account **{user.name}**! There was no profile data to merge."
-            )
-        elif source_profile and author_profile is None:
-            # Attach the discord row to the profile
-            await source_profile.attach_discord(ctx.author)
-            await message.edit(
-                content=f"Successfully connected to Twitch profile **{user.name}**! There was no profile data to merge."
-            )
-        elif source_profile is None and author_profile is None:
-            profile = await UserProfile.create_from_discord(self.bot, ctx.author)
-            await profile.attach_twitch(user)
-
-            await message.edit(
-                content=f"Opened a new user profile for you and linked Twitch account **{user.name}**."
-            )
-        elif author_profile.profileid == source_profile.profileid:
-            await message.edit(
-                content=f"The Twitch account **{user.name}** is already linked to your profile!"
-            )
-        else:
-            # Migrate the existing profile data to the new profiles
-            try:
-                results = await self.migrate_profile(source_profile, author_profile)
-            except Exception:
-                await ctx.error_reply(
-                    "An issue was encountered while merging your account profiles!\n"
-                    "Migration rolled back, no data has been lost.\n"
-                    "The developer has been notified. Please try again later!"
-                )
-                raise
-
-            content = '\n'.join((
-                "## Connecting Twitch account and merging profiles...",
-                *results,
-                "**Successfully linked account and merged profile data!**"
-            ))
-            await message.edit(content=content)
-
-    # ----- Community Commands -----
-    @cmds.hybrid_group(
-        name='community',
-        description="Base comand group for community profiles."
-    )
-    async def community_grp(self, ctx: LionContext):
-        ...
-
-    @community_grp.group(
-        name='link',
-        description="Base command group for linking communities"
-    )
-    async def community_link_grp(self, ctx: LionContext):
-        ...
-
-    @community_link_grp.command(
-        name='twitch',
-        description="Link a twitch account to this community."
-    )
-    @appcmds.guild_only()
-    @appcmds.default_permissions(manage_guild=True)
-    async def comm_link_twitch_cmd(self, ctx: LionContext):
-        if not ctx.interaction:
-            return
-        assert ctx.guild is not None
-
-        await ctx.interaction.response.defer(ephemeral=True)
-
-        if not ctx.author.guild_permissions.manage_guild:
-            await ctx.error_reply("You need the `MANAGE_GUILD` permission to link this guild to a community.")
-            return
-
-        # Ask the user to go through auth to get their userid
-        auth_cog = self.bot.get_cog('TwitchAuthCog')
-        flow = await auth_cog.start_auth(
-            scopes=[
-                AuthScope.CHAT_EDIT,
-                AuthScope.CHAT_READ,
-                AuthScope.MODERATION_READ,
-                AuthScope.CHANNEL_BOT,
-            ]
-        )
-        message = await ctx.reply(
-            f"Please [click here]({flow.auth.return_auth_url()}) to link your Twitch channel to this server."
-        )
-        authrow = await flow.run()
-        await message.edit(
-            content="Authentication Complete! Beginning community profile merge..."
-        )
-
-        results = await self.crocbot.fetch_users(ids=[authrow.userid])
-        if not results:
-            logger.error(
-                f"User {authrow} obtained from Twitch authentication does not exist."
-            )
-            await ctx.error_reply("Sorry, something went wrong. Please try again later!")
-            return
-
-        user = results[0]
-
-        # Retrieve author's profile if it exists
-        guild_comm = await Community.fetch_from_discordid(self.bot, ctx.guild.id)
-
-        # Check if the twitch-side user has a profile 
-        twitch_comm = await Community.fetch_from_twitchid(self.bot, user.id)
-
-        if guild_comm and twitch_comm is None:
-            # All we need to do is attach the twitch row
-            await guild_comm.attach_twitch(user)
-            await message.edit(
-                content=f"Successfully linked Twitch channel **{user.name}**! There was no community data to merge."
-            )
-        elif twitch_comm and guild_comm is None:
-            # Attach the discord row to the profile
-            await twitch_comm.attach_discord(ctx.guild)
-            await message.edit(
-                content=f"Successfully connected to Twitch channel **{user.name}**!"
-            )
-        elif twitch_comm is None and guild_comm is None:
-            profile = await Community.create_from_discord(self.bot, ctx.guild)
-            await profile.attach_twitch(user)
-
-            await message.edit(
-                content=f"Created a new community for this server and linked Twitch account **{user.name}**."
-            )
-        elif guild_comm.communityid == twitch_comm.communityid:
-            await message.edit(
-                content=f"This server is already linked to the Twitch channel **{user.name}**!"
-            )
-        else:
-            # Migrate the existing profile data to the new profiles
-            try:
-                results = await self.migrate_community(twitch_comm, guild_comm)
-            except Exception:
-                await ctx.error_reply(
-                    "An issue was encountered while merging your community profiles!\n"
-                    "Migration rolled back, no data has been lost.\n"
-                    "The developer has been notified. Please try again later!"
-                )
-                raise
-
-            content = '\n'.join((
-                "## Connecting Twitch account and merging community profiles...",
-                *results,
-                "**Successfully linked account and merged community data!**"
-            ))
-            await message.edit(content=content)

src/modules/profiles/community.py

@@ -1,123 +0,0 @@
-from typing import Optional, Self
-
-import discord
-import twitchio
-
-from meta import LionBot
-from utils.lib import utc_now
-
-from . import logger
-from .data import ProfileData
-
-
-
-class Community:
-    def __init__(self, bot: LionBot, community_row):
-        self.bot = bot
-        self.row: ProfileData.CommunityRow = community_row
-
-    @property
-    def cog(self):
-        return self.bot.get_cog('ProfileCog')
-
-    @property
-    def data(self) -> ProfileData:
-        return self.cog.data
-
-    @property
-    def communityid(self):
-        return self.row.communityid
-
-    def __repr__(self):
-        return f"<Community communityid={self.communityid} row={self.row}>"
-
-    async def attach_discord(self, guild: discord.Guild):
-        """
-        Attach a new discord guild to this community.
-        Assumes the discord guild is not already associated to a community.
-        """
-        discord_row = await self.data.DiscordCommunityRow.create(
-            communityid=self.communityid,
-            guildid=guild.id
-        )
-        logger.info(
-            f"Attached discord guild {guild!r} to community {self!r}"
-        )
-        return discord_row
-
-    async def attach_twitch(self, user: twitchio.User):
-        """
-        Attach a new Twitch user channel to this community.
-        """
-        twitch_row = await self.data.TwitchCommunityRow.create(
-            communityid=self.communityid,
-            channelid=str(user.id)
-        )
-        logger.info(
-            f"Attached twitch channel {user!r} to community {self!r}"
-        )
-        return twitch_row
-
-    async def discord_guilds(self) -> list[ProfileData.DiscordCommunityRow]:
-        """
-        Fetch the Discord guild rows associated to this community.
-        """
-        return await self.data.DiscordCommunityRow.fetch_where(communityid=self.communityid)
-
-    async def twitch_channels(self) -> list[ProfileData.TwitchCommunityRow]:
-        """
-        Fetch the Twitch user rows associated to this profile.
-        """
-        return await self.data.TwitchCommunityRow.fetch_where(communityid=self.communityid)
-
-    @classmethod
-    async def fetch(cls, bot: LionBot, community_id: int) -> Self:
-        community_row = await bot.get_cog('ProfileCog').data.CommunityRow.fetch(community_id)
-        if community_row is None:
-            raise ValueError("Provided community_id does not exist.")
-        return cls(bot, community_row)
-
-    @classmethod
-    async def fetch_from_twitchid(cls, bot: LionBot, channelid: int | str) -> Optional[Self]:
-        data = bot.get_cog('ProfileCog').data
-        rows = await data.TwitchCommunityRow.fetch_where(channelid=str(channelid))
-        if rows:
-            return await cls.fetch(bot, rows[0].communityid)
-
-    @classmethod
-    async def fetch_from_discordid(cls, bot: LionBot, guildid: int) -> Optional[Self]:
-        data = bot.get_cog('ProfileCog').data
-        rows = await data.DiscordCommunityRow.fetch_where(guildid=guildid)
-        if rows:
-            return await cls.fetch(bot, rows[0].communityid)
-
-    @classmethod
-    async def create(cls, bot: LionBot, **kwargs) -> Self:
-        """
-        Create a new empty community with the given initial arguments.
-
-        Communities should usually be created using `create_from_discord` or `create_from_twitch`
-        to correctly setup initial preferences (e.g. name, avatar).
-        """
-        # Create a new community
-        data = bot.get_cog('ProfileCog').data
-        row = await data.CommunityRow.create(created_at=utc_now(), **kwargs)
-        return await cls.fetch(bot, row.communityid)
-
-    @classmethod
-    async def create_from_discord(cls, bot: LionBot, guild: discord.Guild, **kwargs) -> Self:
-        """
-        Create a new community using the given Discord guild as a base.
-        """
-        self = await cls.create(bot, **kwargs)
-        await self.attach_discord(guild)
-        return self
-
-    @classmethod
-    async def create_from_twitch(cls, bot: LionBot, user: twitchio.User, **kwargs) -> Self:
-        """
-        Create a new profile using the given Twitch channel user as a base.
-        """
-        self = await cls.create(bot, **kwargs)
-        await self.attach_twitch(user)
-        return self

src/modules/profiles/data.py

@@ -1,158 +0,0 @@
-from data import Registry, RowModel
-from data.columns import Integer, String, Timestamp
-
-
-class ProfileData(Registry):
-    class UserProfileRow(RowModel):
-        """
-        Schema
-        ------
-        CREATE TABLE user_profiles(
-            profileid SERIAL PRIMARY KEY,
-            nickname TEXT,
-            migrated INTEGER REFERENCES user_profiles (profileid) ON DELETE CASCADE ON UPDATE CASCADE,
-            created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-        );
-        """
-        _tablename_ = 'user_profiles'
-        _cache_ = {}
-
-        profileid = Integer(primary=True)
-        nickname = String()
-        migrated = Integer()
-        created_at = Timestamp()
-
-
-    class DiscordProfileRow(RowModel):
-        """
-        Schema
-        ------
-        CREATE TABLE profiles_discord(
-          linkid SERIAL PRIMARY KEY,
-          profileid INTEGER NOT NULL REFERENCES user_profiles (profileid) ON DELETE CASCADE ON UPDATE CASCADE,
-          userid BIGINT NOT NULL,
-          created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-        );
-        CREATE INDEX profiles_discord_profileid ON profiles_discord (profileid);
-        CREATE UNIQUE INDEX profiles_discord_userid ON profiles_discord (userid);
-        """
-        _tablename_ = 'profiles_discord'
-        _cache_ = {}
-
-        linkid = Integer(primary=True)
-        profileid = Integer()
-        userid = Integer()
-        created_at = Integer()
-
-        @classmethod
-        async def fetch_profile(cls, profileid: int):
-            rows = await cls.fetch_where(profiled=profileid)
-            return next(rows, None)
-
-
-    class TwitchProfileRow(RowModel):
-        """
-        Schema
-        ------
-        CREATE TABLE profiles_twitch(
-          linkid SERIAL PRIMARY KEY,
-          profileid INTEGER NOT NULL REFERENCES user_profiles (profileid) ON DELETE CASCADE ON UPDATE CASCADE,
-          userid TEXT NOT NULL,
-          created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-        );
-        CREATE INDEX profiles_twitch_profileid ON profiles_twitch (profileid);
-        CREATE UNIQUE INDEX profiles_twitch_userid ON profiles_twitch (userid);
-        """
-        _tablename_ = 'profiles_twitch'
-        _cache_ = {}
-
-        linkid = Integer(primary=True)
-        profileid = Integer()
-        userid = String()
-        created_at = Timestamp()
-
-        @classmethod
-        async def fetch_profile(cls, profileid: int):
-            rows = await cls.fetch_where(profiled=profileid)
-            return next(rows, None)
-
-    class CommunityRow(RowModel):
-        """
-        Schema
-        ------
-        CREATE TABLE communities(
-          communityid SERIAL PRIMARY KEY,
-          migrated INTEGER REFERENCES user_profiles (profileid) ON DELETE CASCADE ON UPDATE CASCADE,
-          created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-        );
-        """
-        _tablename_ = 'communities'
-        _cache_ = {}
-
-        communityid = Integer(primary=True)
-        migrated = Integer()
-        created_at = Timestamp()
-
-    class DiscordCommunityRow(RowModel):
-        """
-        Schema
-        ------
-        CREATE TABLE communities_discord(
-          guildid BIGINT PRIMARY KEY,
-          communityid INTEGER NOT NULL REFERENCES communities (communityid) ON DELETE CASCADE ON UPDATE CASCADE,
-          linked_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-        );
-        """
-        _tablename_ = 'communities_discord'
-        _cache_ = {}
-
-        guildid = Integer(primary=True)
-        communityid = Integer()
-        linked_at = Timestamp()
-
-        @classmethod
-        async def fetch_community(cls, communityid: int):
-            rows = await cls.fetch_where(communityd=communityid)
-            return next(rows, None)
-
-    class TwitchCommunityRow(RowModel):
-        """
-        Schema
-        ------
-        CREATE TABLE communities_twitch(
-          channelid TEXT PRIMARY KEY,
-          communityid INTEGER NOT NULL REFERENCES communities (communityid) ON DELETE CASCADE ON UPDATE CASCADE,
-          linked_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-        );
-        """
-        _tablename_ = 'communities_twitch'
-        _cache_ = {}
-
-        channelid = String(primary=True)
-        communityid = Integer()
-        linked_at = Timestamp()
-
-        @classmethod
-        async def fetch_community(cls, communityid: int):
-            rows = await cls.fetch_where(communityd=communityid)
-            return next(rows, None)
-
-    class CommunityMemberRow(RowModel):
-        """
-        Schema
-        ------
-        CREATE TABLE community_members(
-          memberid SERIAL PRIMARY KEY,
-          communityid INTEGER NOT NULL REFERENCES communities (communityid) ON DELETE CASCADE ON UPDATE CASCADE,
-          profileid INTEGER NOT NULL REFERENCES user_profiles (profileid) ON DELETE CASCADE ON UPDATE CASCADE,
-          created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-        );
-        CREATE UNIQUE INDEX community_members_communityid_profileid ON community_members (communityid, profileid);
-        """
-        _tablename_ = 'community_members'
-        _cache_ = {}
-
-        memberid = Integer(primary=True)
-        communityid = Integer()
-        profileid = Integer()
-        created_at = Timestamp()

src/modules/profiles/profile.py

@@ -1,124 +0,0 @@
-from typing import Optional, Self
-
-import discord
-import twitchio
-
-from meta import LionBot
-from utils.lib import utc_now
-
-from . import logger
-from .data import ProfileData
-
-
-
-class UserProfile:
-    def __init__(self, bot: LionBot, profile_row):
-        self.bot = bot
-        self.profile_row: ProfileData.UserProfileRow = profile_row
-
-    @property
-    def cog(self):
-        return self.bot.get_cog('ProfileCog')
-
-    @property
-    def data(self) -> ProfileData:
-        return self.cog.data
-
-    @property
-    def profileid(self):
-        return self.profile_row.profileid
-
-    def __repr__(self):
-        return f"<UserProfile profileid={self.profileid} profile={self.profile_row}>"
-
-    async def attach_discord(self, user: discord.User | discord.Member):
-        """
-        Attach a new discord user to this profile.
-        Assumes the discord user does not itself have a profile.
-        """
-        discord_row = await self.data.DiscordProfileRow.create(
-            profileid=self.profileid, 
-            userid=user.id
-        )
-        logger.info(
-            f"Attached discord user {user!r} to profile {self!r}"
-        )
-        return discord_row
-
-    async def attach_twitch(self, user: twitchio.User):
-        """
-        Attach a new Twitch user to this profile.
-        """
-        twitch_row = await self.data.TwitchProfileRow.create(
-            profileid=self.profileid,
-            userid=str(user.id)
-        )
-        logger.info(
-            f"Attached twitch user {user!r} to profile {self!r}"
-        )
-        return twitch_row
-
-    async def discord_accounts(self) -> list[ProfileData.DiscordProfileRow]:
-        """
-        Fetch the Discord accounts associated to this profile.
-        """
-        return await self.data.DiscordProfileRow.fetch_where(profileid=self.profileid)
-
-    async def twitch_accounts(self) -> list[ProfileData.TwitchProfileRow]:
-        """
-        Fetch the Twitch accounts associated to this profile.
-        """
-        return await self.data.TwitchProfileRow.fetch_where(profileid=self.profileid)
-
-    @classmethod
-    async def fetch(cls, bot: LionBot, profile_id: int) -> Self:
-        profile_row = await bot.get_cog('ProfileCog').data.UserProfileRow.fetch(profile_id)
-        if profile_row is None:
-            raise ValueError("Provided profile_id does not exist.")
-        return cls(bot, profile_row)
-
-    @classmethod
-    async def fetch_from_twitchid(cls, bot: LionBot, userid: int | str) -> Optional[Self]:
-        data = bot.get_cog('ProfileCog').data
-        rows = await data.TwitchProfileRow.fetch_where(userid=str(userid))
-        if rows:
-            return await cls.fetch(bot, rows[0].profileid)
-
-    @classmethod
-    async def fetch_from_discordid(cls, bot: LionBot, userid: int) -> Optional[Self]:
-        data = bot.get_cog('ProfileCog').data
-        rows = await data.DiscordProfileRow.fetch_where(userid=(userid))
-        if rows:
-            return await cls.fetch(bot, rows[0].profileid)
-
-    @classmethod
-    async def create(cls, bot: LionBot, **kwargs) -> Self:
-        """
-        Create a new empty profile with the given initial arguments.
-
-        Profiles should usually be created using `create_from_discord` or `create_from_twitch`
-        to correctly setup initial profile preferences (e.g. name, avatar).
-        """
-        # Create a new profile
-        data = bot.get_cog('ProfileCog').data
-        profile_row = await data.UserProfileRow.create(created_at=utc_now())
-        profile = await cls.fetch(bot, profile_row.profileid)
-        return profile
-
-    @classmethod
-    async def create_from_discord(cls, bot: LionBot, user: discord.Member | discord.User, **kwargs) -> Self:
-        """
-        Create a new profile using the given Discord user as a base.
-        """
-        profile = await cls.create(bot, **kwargs)
-        await profile.attach_discord(user)
-        return profile
-
-    @classmethod
-    async def create_from_twitch(cls, bot: LionBot, user: twitchio.User, **kwargs) -> Self:
-        """
-        Create a new profile using the given Twitch user as a base.
-        """
-        profile = await cls.create(bot, **kwargs)
-        await profile.attach_twitch(user)
-        return profile

src/modules/shoutouts/cog.py

@@ -17,19 +17,9 @@ class ShoutoutCog(LionCog):
     and drop a follow! \
     They {areorwere} streaming {game} at {channel}
     """
-    COWO_SHOUTOUT = """
-    We think that {name} is a great coworker and you should check them out for more productive vibes! \
-    They {areorwere} streaming {game} at {channel}
-    """
-    ART_SHOUTOUT = """
-    We think that {name} is an awesome artist and you should check them out for cool art and cosy vibes! \
-    They {areorwere} streaming {game} at {channel}
-    """
-
     def __init__(self, bot: LionBot):
         self.bot = bot
-        self.crocbot: CrocBot = bot.crocbot
+        self.crocbot = bot.crocbot
-
         self.data = bot.db.load_registry(ShoutoutData())
 
         self.loaded = asyncio.Event()
@@ -69,28 +59,19 @@ class ShoutoutCog(LionCog):
         return replace_multiple(text, mapping)
 
     @commands.command(aliases=['so'])
-    async def shoutout(self, ctx: commands.Context, target: str, typ: Optional[str]=None):
+    async def shoutout(self, ctx: commands.Context, user: twitchio.User):
         # Make sure caller is mod/broadcaster
         # Lookup custom shoutout for this user
         # If it exists use it, otherwise use default shoutout
         if (ctx.author.is_mod or ctx.author.is_broadcaster):
-            user = await self.crocbot.seek_user(target)
-            if user is None:
-                await ctx.reply(f"Couldn't resolve '{target}' to a valid user.")
-            else:
             data = await self.data.CustomShoutout.fetch(int(user.id))
             if data:
                 shoutout = data.content
-                elif typ == 'cowo':
-                    shoutout = self.COWO_SHOUTOUT
-                elif typ == 'art':
-                    shoutout = self.ART_SHOUTOUT
             else:
                 shoutout = self.DEFAULT_SHOUTOUT
             formatted = await self.format_shoutout(shoutout, user)
             await ctx.reply(formatted)
             # TODO: How to /shoutout with lib?
-            # TODO Shoutout queue
 
     @commands.command()
     async def editshoutout(self, ctx: commands.Context, user: twitchio.User, *, text: str):

src/modules/voiceroles/__init__.py

@@ -1,7 +0,0 @@
-import logging
-
-logger = logging.getLogger(__name__)
-
-async def setup(bot):
-    from .cog import VoiceRoleCog
-    await bot.add_cog(VoiceRoleCog(bot))

src/modules/voiceroles/cog.py

@@ -1,166 +0,0 @@
-from collections import defaultdict
-from typing import Optional
-import asyncio
-from cachetools import FIFOCache
-from weakref import WeakValueDictionary
-
-import discord
-from discord.abc import GuildChannel
-from discord.ext import commands as cmds
-from discord import app_commands as appcmds
-
-from meta import LionBot, LionCog, LionContext
-from meta.logger import log_wrap
-from meta.errors import ResponseTimedOut, SafeCancellation, UserInputError
-from utils.ui import Confirm
-
-from . import logger
-from .data import VoiceRoleData
-
-
-class VoiceRoleCog(LionCog):
-    def __init__(self, bot: LionBot):
-        self.bot = bot
-        self.data = bot.db.load_registry(VoiceRoleData())
-
-        self._event_locks: WeakValueDictionary[tuple[int, int], asyncio.Lock] = WeakValueDictionary()
-
-    async def cog_load(self):
-        await self.data.init()
-
-    @LionCog.listener('on_voice_state_update')
-    @log_wrap(action='Voice Role Update')
-    async def voicerole_update(self, member: discord.Member,
-                               before: discord.VoiceState, after: discord.VoiceState):
-        if member.bot:
-            return
-
-        after_channel = after.channel
-        before_channel = before.channel
-        if after_channel == before_channel:
-            return
-
-        task_key = (member.guild.id, member.id)
-        async with self.event_lock(task_key):
-            # Get the roles of the channel they left to remove
-            # Get the roles of the channel they are joining to add
-            # Use a set difference to remove the roles to be added from the ones to remove
-            if before_channel is not None:
-                leaving_roles = await self.get_roles_for(before_channel.id)
-            else:
-                leaving_roles = []
-
-            if after_channel is not None:
-                gaining_roles = await self.get_roles_for(after_channel.id)
-            else:
-                gaining_roles = []
-
-            to_remove = []
-            for role in leaving_roles:
-                if role in member.roles and role not in gaining_roles and role.is_assignable():
-                    to_remove.append(role)
-
-            to_add = []
-            for role in gaining_roles:
-                if role not in member.roles and role.is_assignable():
-                    to_add.append(role)
-
-            if to_remove:
-                await member.remove_roles(*to_remove, reason="Removing voice channel associated roles.")
-            if to_add:
-                await member.add_roles(*to_add, reason="Adding voice channel associated roles.")
-
-            logger.info(
-                    f"Voice roles removed {len(to_remove)} roles "
-                    f"and added {len(to_add)} roles to <uid: {member.id}>"
-            )
-
-    async def get_roles_for(self, channelid: int) -> list[discord.Role]:
-        """
-        Get the voice roles associated to the given channel, as a list.
-
-        Returns an empty list if there are no associated voice roles.
-        """
-        rows = await self.data.VoiceRole.fetch_where(channelid=channelid)
-        channel = self.bot.get_channel(channelid)
-        if not channel:
-            raise ValueError("Provided voice role target channel is not in cache.")
-
-        target_roles = []
-        for row in rows:
-            role = channel.guild.get_role(row.roleid)
-            if role is not None:
-                target_roles.append(role)
-
-        return target_roles
-
-    def event_lock(self, key) -> asyncio.Lock:
-        """
-        Get an asyncio.Lock for the given key.
-
-        Guarantees sequential event handling.
-        """
-        lock = self._event_locks.get(key, None)
-        if lock is None:
-            lock = self._event_locks[key] = asyncio.Lock()
-        logger.debug(f"Getting video event lock {key} (locked: {lock.locked()})")
-        return lock
-
-
-    # -------- Commands --------
-    @cmds.hybrid_group(
-        name='voiceroles',
-        description="Base command group for voice channel -> role associationes."
-    )
-    @appcmds.default_permissions(manage_channels=True)
-    async def voicerole_group(self, ctx: LionContext):
-        ...
-
-    @voicerole_group.command(
-        name="link",
-        description="Link a given voice channel with a given role."
-    )
-    @appcmds.describe(
-        channel="The voice channel to link.",
-        role="The associated role to give to members joining the voice channel."
-    )
-    async def voicerole_link(self, ctx: LionContext,
-                             channel: discord.VoiceChannel,
-                             role: discord.Role):
-        if not ctx.interaction:
-            return
-        if not channel.permissions_for(ctx.author).manage_channels:
-            await ctx.error_reply(f"You don't have the manage channels permission in {channel.mention}")
-            return
-        if not ctx.author.guild_permissions.manage_roles or not (role < ctx.author.top_role):
-            await ctx.error_reply(f"You don't have the permission to manage this role!")
-            return
-
-        await self.data.VoiceRole.table.insert(channelid=channel.id, roleid=role.id)
-        await ctx.reply("Voice role associated!")
-
-    @voicerole_group.command(
-        name="unlink",
-        description="Unlink a given voice channel from a given role."
-    )
-    @appcmds.describe(
-        channel="The voice channel to unlink.",
-        role="The role to remove from this voice channel."
-    )
-    async def voicerole_unlink(self, ctx: LionContext,
-                               channel: discord.VoiceChannel,
-                               role: discord.Role):
-        if not ctx.interaction:
-            return
-        if not channel.permissions_for(ctx.author).manage_channels:
-            await ctx.error_reply(f"You don't have the manage channels permission in {channel.mention}")
-            return
-        if not ctx.author.guild_permissions.manage_roles or not (role < ctx.author.top_role):
-            await ctx.error_reply(f"You don't have the permission to manage this role!")
-            return
-
-        await self.data.VoiceRole.table.delete_where(channelid=channel.id, roleid=role.id)
-        await ctx.reply("Voice role disassociated!")
-
-    # TODO: Display and visual editing of roles.
-

src/modules/voiceroles/data.py

@@ -1,27 +0,0 @@
-from data import Registry, RowModel
-from data.columns import Integer, Timestamp
-
-
-class VoiceRoleData(Registry):
-    class VoiceRole(RowModel):
-        """
-        Schema
-        ------
-        CREATE TABLE voice_roles(
-            voice_role_id SERIAL PRIMARY KEY,
-            channelid BIGINT NOT NULL,
-            roleid BIGINT NOT NULL,
-            created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()
-        );
-        CREATE INDEX voice_role_channels on voice_roles (channelid);
-        """
-        # TODO: Worth associating a guildid to this as well? Denormalises though
-        # Makes more theoretical sense to associated configurable channels to the guilds in a join table.
-        _tablename_ = 'voice_roles'
-        _cache_ = {}
-
-        voice_role_id = Integer(primary=True)
-        channelid = Integer()
-        roleid = Integer()
-
-        created_at = Timestamp()

src/tracking/voice/cog.py

@@ -5,7 +5,7 @@ import datetime as dt
 
 import discord
 from discord.ext import commands as cmds
-from discord import AllowedMentions, app_commands as appcmds
+from discord import app_commands as appcmds
 
 from data import Condition
 from meta import LionBot, LionCog, LionContext
@@ -668,7 +668,7 @@ class VoiceTrackerCog(LionCog):
     @appcmds.describe(
         tag=_p(
             'cmd:now|param:tag|desc',
-            "Describe what you are working!"
+            "Describe what you are working on in 10 characters or less!"
         ),
         user=_p(
             'cmd:now|param:user|desc',
@@ -681,15 +681,17 @@ class VoiceTrackerCog(LionCog):
     )
     @appcmds.guild_only
     async def now_cmd(self, ctx: LionContext,
-                      tag: Optional[str] = None,
+                      tag: Optional[appcmds.Range[str, 0, 10]] = None,
-                      *,
                       user: Optional[discord.Member] = None,
                       clear: Optional[bool] = None
                       ):
         if not ctx.guild:
             return
+        if not ctx.interaction:
+            return
         t = self.bot.translator.t
 
+        await ctx.interaction.response.defer(thinking=True, ephemeral=True)
         is_moderator = await moderator_ctxward(ctx)
         target = user if user is not None else ctx.author
         session = self.get_session(ctx.guild.id, target.id, create=False)
@@ -713,7 +715,7 @@ class VoiceTrackerCog(LionCog):
                         "{mention} has no running session!"
                     )).format(mention=target.mention)
                 )
-            await ctx.reply(embed=error)
+            await ctx.interaction.edit_original_response(embed=error)
             return
 
         if clear:
@@ -721,27 +723,87 @@ class VoiceTrackerCog(LionCog):
             if target == ctx.author:
                 # Clear the author's tag
                 await session.set_tag(None)
-                ack = "Cleared your current task!"
+                ack = discord.Embed(
+                    colour=discord.Colour.brand_green(),
+                    title=t(_p(
+                        'cmd:now|target:self|mode:clear|success|title',
+                        "Session Tag Cleared"
+                    )),
+                    description=t(_p(
+                        'cmd:now|target:self|mode:clear|success|desc',
+                        "Successfully unset your session tag."
+                    ))
+                )
             elif not is_moderator:
                 # Trying to clear someone else's tag without being a moderator
-                ack = "You need to be a moderator to set or clear someone else's task!"
+                ack = discord.Embed(
+                    colour=discord.Colour.brand_red(),
+                    title=t(_p(
+                        'cmd:now|target:other|mode:clear|error:perms|title',
+                        "You can't do that!"
+                    )),
+                    description=t(_p(
+                        'cmd:now|target:other|mode:clear|error:perms|desc',
+                        "You need to be a moderator to set or clear someone else's session tag."
+                    ))
+                )
             else:
                 # Clearing someone else's tag as a moderator
                 await session.set_tag(None)
-                ack = f"Cleared {target}'s current task!"
+                ack = discord.Embed(
+                    colour=discord.Colour.brand_green(),
+                    title=t(_p(
+                        'cmd:now|target:other|mode:clear|success|title',
+                        "Session Tag Cleared!"
+                    )),
+                    description=t(_p(
+                        'cmd:now|target:other|mode:clear|success|desc',
+                        "Cleared {target}'s session tag."
+                    )).format(target=target.mention)
+                )
         elif tag:
             # Tag setting mode
             if target == ctx.author:
                 # Set the author's tag
                 await session.set_tag(tag)
-                ack = f"Set your current task to `{tag}`, good luck! <:goodluck:1266447460146876497>"
+                ack = discord.Embed(
+                    colour=discord.Colour.brand_green(),
+                    title=t(_p(
+                        'cmd:now|target:self|mode:set|success|title',
+                        "Session Tag Set!"
+                    )),
+                    description=t(_p(
+                        'cmd:now|target:self|mode:set|success|desc',
+                        "You are now working on `{new_tag}`. Good luck!"
+                    )).format(new_tag=tag)
+                )
             elif not is_moderator:
                 # Trying the set someone else's tag without being a moderator
-                ack = "You need to be a moderator to set or clear someone else's task!"
+                ack = discord.Embed(
+                    colour=discord.Colour.brand_red(),
+                    title=t(_p(
+                        'cmd:now|target:other|mode:set|error:perms|title',
+                        "You can't do that!"
+                    )),
+                    description=t(_p(
+                        'cmd:now|target:other|mode:set|error:perms|desc',
+                        "You need to be a moderator to set or clear someone else's session tag!"
+                    ))
+                )
             else:
                 # Setting someone else's tag as a moderator
                 await session.set_tag(tag)
-                ack = f"Set {target}'s current task to `{tag}`"
+                ack = discord.Embed(
+                    colour=discord.Colour.brand_green(),
+                    title=t(_p(
+                        'cmd:now|target:other|mode:set|success|title',
+                        "Session Tag Set!"
+                    )),
+                    description=t(_p(
+                        'cmd:now|target:other|mode:set|success|desc',
+                        "Set {target}'s session tag to `{new_tag}`."
+                    )).format(target=target.mention, new_tag=tag)
+                )
         else:
             # Display tag and voice time
             if target == ctx.author:
@@ -753,14 +815,14 @@ class VoiceTrackerCog(LionCog):
                 else:
                     desc = t(_p(
                         'cmd:now|target:self|mode:show_without_tag|desc',
-                        "You have been working in {channel} since {time}! "
+                        "You have been working in {channel} since {time}!\n\n"
                         "Use `/now <tag>` to set what you are working on."
                     ))
             else:
                 if session.tag:
                     desc = t(_p(
                         'cmd:now|target:other|mode:show_with_tag|desc',
-                        "{target} is current working in {channel}! "
+                        "{target} is current working in {channel}!\n"
                         "They have been working on **{tag}** since {time}."
                     ))
                 else:
@@ -768,13 +830,18 @@ class VoiceTrackerCog(LionCog):
                         'cmd:now|target:other|mode:show_without_tag|desc',
                         "{target} has been working in {channel} since {time}!"
                     ))
-            ack = desc.format(
+            desc = desc.format(
                 tag=session.tag,
                 channel=f"<#{session.state.channelid}>",
-                time=discord.utils.format_dt(session.start_time, 'R'),
+                time=discord.utils.format_dt(session.start_time, 't'),
                 target=target.mention,
             )
-        await ctx.reply(ack, allowed_mentions=AllowedMentions.none())
+            ack = discord.Embed(
+                colour=discord.Colour.orange(),
+                description=desc,
+                timestamp=utc_now()
+            )
+        await ctx.interaction.edit_original_response(embed=ack)
 
     # ----- Configuration Commands -----
     @LionCog.placeholder_group

src/twitch/data.py

@@ -76,4 +76,4 @@ class TwitchAuthData(Registry):
     );
     CREATE INDEX twitch_user_scopes_userid ON twitch_user_scopes (userid);
     """
-    user_scopes = Table('twitch_user_scopes')
+    user_scopes = Table('twitch_token_scopes')

src/twitch/userflow.py

@@ -47,7 +47,7 @@ class UserAuthFlow:
                 self._setup_done.set()
                 return await ws.receive_json()
 
-    async def run(self) -> TwitchAuthData.UserAuthRow:
+    async def run(self):
         if not self._setup_done.is_set():
             raise ValueError("Cannot run UserAuthFlow before setup.")
         if self._comm_task is None:

tests/__init__.py

@@ -1,7 +0,0 @@
-# !/bin/python3
-
-import sys
-import os
-
-sys.path.insert(0, os.path.join(os.getcwd()))
-sys.path.insert(0, os.path.join(os.getcwd(), "src"))

tests/gui/cards/goal_sample.py

@@ -1,11 +1,11 @@
 import asyncio
 import datetime as dt
-from gui.cards import WeeklyGoalCard
+from src.cards import WeeklyGoalCard
 
 
 async def get_card():
     card = await WeeklyGoalCard.generate_sample()
-    with open('output/weekly-sample.png', 'wb') as image_file:
+    with open('samples/weekly-sample.png', 'wb') as image_file:
         image_file.write(card.fp.read())
 
 if __name__ == '__main__':

tests/gui/cards/pomo_sample.py

@@ -1,15 +0,0 @@
-import asyncio
-import datetime as dt
-from gui.cards import BreakTimerCard, FocusTimerCard
-
-
-async def get_card():
-    card = await BreakTimerCard.generate_sample()
-    with open('output/break_timer_sample.png', 'wb') as image_file:
-        image_file.write(card.fp.read())
-    card = await FocusTimerCard.generate_sample()
-    with open('output/focus_timer_sample.png', 'wb') as image_file:
-        image_file.write(card.fp.read())
-
-if __name__ == '__main__':
-    asyncio.run(get_card())