chat qo'shildi

core/apps/chat/consumers/chat.py

@@ -0,0 +1,105 @@
+import json
+
+from channels.db import database_sync_to_async
+from channels.generic.websocket import AsyncWebsocketConsumer
+from django.contrib.auth.models import AnonymousUser
+
+
+class ChatConsumer(AsyncWebsocketConsumer):
+    """
+    Xona asosidagi chat consumer.
+    Ulanish: ws://host/ws/chat/room/{room_id}/?token=<jwt>
+
+    Matn xabari yuborish:
+        { "message_type": "text", "text": "Salom" }
+
+    Fayl xabari (oldin REST orqali yuklab, keyin URL yuborish):
+        { "message_type": "image", "file_url": "https://...", "text": "caption (ixtiyoriy)" }
+    """
+
+    async def connect(self):
+        user = self.scope.get("user")
+        if not user or isinstance(user, AnonymousUser):
+            await self.close(code=4001)
+            return
+
+        self.room_id = self.scope["url_route"]["kwargs"]["room_id"]
+        self.room_group = f"chat_room_{self.room_id}"
+
+        await self.channel_layer.group_add(self.room_group, self.channel_name)
+        await self.accept()
+
+    async def disconnect(self, close_code):
+        if hasattr(self, "room_group"):
+            await self.channel_layer.group_discard(self.room_group, self.channel_name)
+
+    async def receive(self, text_data):
+        user = self.scope.get("user")
+        if not user or isinstance(user, AnonymousUser):
+            await self.close(code=4001)
+            return
+
+        try:
+            data = json.loads(text_data)
+        except (json.JSONDecodeError, ValueError):
+            await self.send(text_data=json.dumps({"error": "Noto'g'ri format."}))
+            return
+
+        message_type = data.get("message_type", "text")
+        text = (data.get("text") or "").strip()
+
+        # Matn xabari uchun text majburiy
+        if message_type == "text" and not text:
+            await self.send(text_data=json.dumps({"error": "Matn bo'sh bo'lishi mumkin emas."}))
+            return
+
+        # WS orqali faqat matn + caption saqlanadi.
+        # Fayl yuklash uchun REST /chat/messages/ POST ishlatiladi.
+        if message_type != "text":
+            await self.send(
+                text_data=json.dumps(
+                    {"error": "Fayl xabarlarni yuklash uchun REST API dan foydalaning."}
+                )
+            )
+            return
+
+        # DB ga saqlash — post_save signal WS ga broadcast qiladi
+        await self._save_message(user, text)
+
+    async def chat_message(self, event):
+        await self.send(
+            text_data=json.dumps(
+                {
+                    "id": event["id"],
+                    "message_type": event["message_type"],
+                    "text": event["text"],
+                    "file_url": event["file_url"],
+                    "sender": event["sender"],
+                    "created_at": event["created_at"],
+                }
+            )
+        )
+
+    @database_sync_to_async
+    def _save_message(self, user, text):
+        from core.apps.chat.models import ChatmessageModel
+
+        msg = ChatmessageModel.objects.create(
+            room_id=self.room_id,
+            sender=user,
+            message_type="text",
+            text=text,
+        )
+        full_name = user.get_full_name().strip() or str(user.phone)
+        return {
+            "id": msg.id,
+            "message_type": msg.message_type,
+            "text": msg.text,
+            "file_url": None,
+            "sender": {
+                "id": user.id,
+                "full_name": full_name,
+                "role": user.role,
+            },
+            "created_at": msg.created_at.isoformat(),
+        }