Add more event types.

src/modules/tracker/component.py

@@ -1,7 +1,7 @@
 from typing import Optional
 import random
 import twitchio
-from twitchio import Scopes, eventsub
+from twitchio import PartialUser, Scopes, eventsub
 from twitchio.ext import commands as cmds
 
 from datamodels import BotChannel, Communities, UserProfile
@@ -333,6 +333,79 @@ class TrackerComponent(cmds.Component):
                 stream_type=payload.type,
             )
 
+    @cmds.Component.listener()
+    async def event_raid(self, payload: twitchio.ChannelRaid):
+        await self._event_raid_out(
+            payload.from_broadcaster,
+            payload.to_broadcaster,
+            payload.viewer_count,
+        )
+        await self._event_raid_in(
+            payload.to_broadcaster,
+            payload.from_broadcaster,
+            payload.viewer_count,
+        )
+
+    async def _event_raid_out(self, broadcaster: PartialUser, to_broadcaster: PartialUser, viewer_count: int):
+        tracked = await TrackingChannel.fetch(broadcaster.id)
+        if tracked and tracked.joined:
+            community = await Communities.fetch_or_create(twitchid=broadcaster.id, name=broadcaster.name)
+            cid = community.communityid
+
+            event_row = await self.data.events.insert(
+                event_type='raidout',
+                communityid=cid,
+                channel_id=broadcaster.id,
+            )
+            detail_row = await self.data.raid_out_events.insert(
+                event_id=event_row['event_id'],
+                target_id=to_broadcaster.id,
+                target_name=to_broadcaster.name,
+                viewer_count=viewer_count
+            )
+
+    async def _event_raid_in(self, broadcaster: PartialUser, from_broadcaster: PartialUser, viewer_count: int):
+        tracked = await TrackingChannel.fetch(broadcaster.id)
+        if tracked and tracked.joined:
+            community = await Communities.fetch_or_create(twitchid=broadcaster.id, name=broadcaster.name)
+            cid = community.communityid
+
+            event_row = await self.data.events.insert(
+                event_type='raidin',
+                communityid=cid,
+                channel_id=broadcaster.id,
+            )
+            detail_row = await self.data.raid_in_events.insert(
+                event_id=event_row['event_id'],
+                source_id=from_broadcaster.id,
+                source_name=from_broadcaster.name,
+                viewer_count=viewer_count
+            )
+
+    async def event_message(self, payload: twitchio.ChatMessage):
+        tracked = await TrackingChannel.fetch(payload.broadcaster.id)
+        if tracked and tracked.joined:
+            community = await Communities.fetch_or_create(twitchid=payload.broadcaster.id, name=payload.broadcaster.name)
+            cid = community.communityid
+            profile = await UserProfile.fetch_or_create(
+                twitchid=payload.chatter.id, name=payload.chatter.name,
+            )
+            pid = profile.profileid
+
+            event_row = await self.data.events.insert(
+                event_type='message',
+                communityid=cid,
+                channel_id=payload.broadcaster.id,
+                profileid=pid,
+                user_id=payload.chatter.id,
+            )
+            detail_row = await self.data.message_events.insert(
+                event_id=event_row['event_id'],
+                message_id=payload.id,
+                message_type=payload.type,
+                content=payload.text,
+                source_channel_id=payload.source_id
+            )
 
     # ----- Commands -----
     @cmds.command(name='starttracking')