2025-07-26

2025-07-26 14:52:54 -05:00
parent 55c03bcafb
commit c543c98bf3
28 changed files with 2779 additions and 146 deletions
--- a/draft/consumers.py
+++ b/draft/consumers.py
@@ -0,0 +1,151 @@
+from channels.generic.websocket import AsyncJsonWebsocketConsumer
+from channels.db import database_sync_to_async
+from django.core.exceptions import PermissionDenied
+from boxofficefantasy.models import League, Season
+from boxofficefantasy.views import parse_season_slug
+from draft.models import DraftSession, DraftPick, DraftMoviePool, DraftParticipant
+
+import random
+
+class DraftConsumer(AsyncJsonWebsocketConsumer):
+    async def connect(self):
+        draft_session_id_hashed = self.scope["url_route"]["kwargs"].get("draft_session_id_hashed")
+        league_slug = self.scope["url_route"]["kwargs"].get("league_slug")
+        season_slug = self.scope["url_route"]["kwargs"].get("season_slug")
+
+        self.draft_session = await self.get_draft_session(draft_session_id_hashed=draft_session_id_hashed, league_slug=league_slug, season_slug=season_slug)
+
+        self.room_group_name = f"draft_{self.draft_session.season.league.slug}_{self.draft_session.season.slug}"
+
+        # Auth check (optional)
+        user = self.scope["user"]
+        if not user.is_authenticated:
+            await self.close()
+            return
+
+        await self.channel_layer.group_add(self.room_group_name, self.channel_name)
+        await self.accept()
+        await self.send_json({"type": "connection.accepted", "user": user.username})
+
+        # Notify others (optional)
+        await self.channel_layer.group_send(
+            self.room_group_name,
+            {
+                "type": "user.joined",
+                "user": user.username,
+            }
+        )
+
+    async def disconnect(self, close_code):
+        await self.channel_layer.group_discard(self.room_group_name, self.channel_name)
+
+    async def receive_json(self, content):
+        event_type = content.get("type")
+        user = self.scope["user"]
+        
+        if event_type == "start_draft":
+            if user.is_staff:
+                await self.start_draft()
+            else:
+                await self.send_json({
+                    "type": "error",
+                    "message": "insufficient privelleges"
+                })
+        elif event_type == "nominate":
+            await self.nominate(content.get("movie"))
+        elif event_type == "bid":
+            await self.place_bid(content.get("amount"), self.scope["user"].username)
+        elif event_type == "message":
+            await self.channel_layer.group_send(
+                self.room_group_name,
+                {
+                    "type": "chat.message",
+                    "user": self.scope["user"].username,
+                    "message": content.get("message"),
+                },
+            )
+
+    # === Broadcast handlers ===
+
+    async def user_joined(self, event):
+        await self.send_json({
+            "type": "user.joined",
+            "user": event["user"]
+        })
+
+    async def chat_message(self, event):
+        await self.send_json({
+            "type": "chat.message",
+            "user": event["user"],
+            "message": event["message"],
+        })
+
+    async def draft_update(self, event):
+        await self.send_json({
+            "type": "draft.update",
+            "state": event["state"],
+        })
+
+    # === Draft logic (stubbed for now) ===
+
+    async def start_draft(self):
+        # Example: shuffle draft order
+        players = await self.get_draft_users()
+        draft_order = random.sample(players, len(players))
+
+        await self.channel_layer.group_send(
+            self.room_group_name,
+            {
+                "type": "draft.update",
+                "state": {
+                    "status": "started",
+                    "order": [p.user.username for p in draft_order],
+                }
+            }
+        )
+
+    async def nominate(self, movie_title):
+        await self.channel_layer.group_send(
+            self.room_group_name,
+            {
+                "type": "draft.update",
+                "state": {
+                    "status": "nominating",
+                    "movie": movie_title,
+                }
+            }
+        )
+
+    async def place_bid(self, amount, user):
+        await self.channel_layer.group_send(
+            self.room_group_name,
+            {
+                "type": "draft.update",
+                "state": {
+                    "status": "bidding",
+                    "bid": {"amount": amount, "user": user}
+                }
+            }
+        )
+
+    # === Example DB Access ===
+
+    @database_sync_to_async
+    def get_draft_users(self):
+        # Replace this with real queryset to fetch users in draft
+        return list(DraftParticipant.objects.select_related('user').filter(draft=self.draft_session).all())
+        return ["alice", "bob", "carol"]
+    
+    @database_sync_to_async
+    def get_draft_session(self, draft_session_id_hashed, league_slug, season_slug):
+        draft_session_id = DraftSession.decode_id(draft_session_id_hashed)
+        if draft_session_id:
+            draft_session = DraftSession.objects.select_related('season', 'season__league').get(pk=draft_session_id)
+        elif league_slug and season_slug:
+            label, year = parse_season_slug(season_slug)
+            season = Season.objects.filter(label=label, year=year).first()
+            draft_session = DraftSession.objects.select_related('season', 'season__league').filter(season=season).first()
+        else:
+            raise Exception()
+        
+        return draft_session