insta clone

Author: ianshulx

instagram-clone-Django/chat/__init__.py

@@ -0,0 +1,6 @@
+from django.contrib import admin
+
+from chat.models import Room, Message
+
+admin.site.register(Room)
+admin.site.register(Message)

instagram-clone-Django/chat/admin.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class ChatConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name = 'chat'

instagram-clone-Django/chat/apps.py

@@ -0,0 +1,107 @@
+# chat/consumers.py
+import json
+from channels.generic.websocket import AsyncWebsocketConsumer
+from channels.db import database_sync_to_async
+from .models import Room, Message, UserModel
+from posts.models import PostModel  # agar kerak bo'lsa
+
+class ChatConsumer(AsyncWebsocketConsumer):
+    async def connect(self):
+        # room_name from URL, e.g. "private_admin_jamil"
+        self.room_name = self.scope['url_route']['kwargs']['room_name']
+        # group name used for channel_layer
+        self.room_group_name = f"room_{self.room_name}"
+
+        await self.channel_layer.group_add(
+            self.room_group_name,
+            self.channel_name
+        )
+        await self.accept()
+
+    async def disconnect(self, code):
+        # use the group name (not raw room_name)
+        await self.channel_layer.group_discard(
+            self.room_group_name,
+            self.channel_name
+        )
+
+    async def receive(self, text_data):
+        # data expected: {"message": "...", "room_name": "...", "sender": "..."}
+        data_json = json.loads(text_data)
+
+        event = {"type": "send_message", "message": data_json}
+
+        # send to the group (use same group name as added)
+        await self.channel_layer.group_send(self.room_group_name, event)
+
+    async def send_message(self, event):
+        data = event["message"]
+
+        # save to DB (text or post)
+        await self.create_message(data)
+
+        # build response for clients
+        if data.get("post_id"):
+            response = {
+                "sender": data.get("sender"),
+                "post_id": data.get("post_id"),
+                "post_type": data.get("post_type"),
+                "caption": data.get("caption"),
+                "media_url": data.get("media_url"),
+                "username": data.get("username"),
+            }
+        else:
+            response = {
+                "sender": data.get("sender"),
+                "message": data.get("message"),
+            }
+
+        await self.send(text_data=json.dumps({"message": response}))
+
+    # consumers.py - Fix the create_message method
+    # consumers.py - Ensure proper message saving for posts
+    @database_sync_to_async
+    def create_message(self, data):
+        try:
+            room = Room.objects.get(room_name=data.get("room_name") or self.room_name)
+            sender_user = UserModel.objects.get(username=data.get('sender'))
+
+            # If this is a post share
+            if data.get("post_id"):
+                try:
+                    post = PostModel.objects.get(id=data.get("post_id"))
+                    # Create message with post attached - this ensures persistence
+                    message_obj = Message.objects.create(
+                        room=room,
+                        sender=sender_user,
+                        post=post,
+                        message=data.get("message", f"Shared a {post.post_type}")
+                    )
+                    # Return the complete data for WebSocket broadcast
+                    return {
+                        "id": message_obj.id,
+                        "sender": sender_user.username,
+                        "post_id": post.id,
+                        "post_type": post.post_type,
+                        "caption": post.caption,
+                        "media_url": post.contentUrl.url,
+                        "username": post.userID.username,
+                        "message": data.get("message", f"Shared a {post.post_type}")
+                    }
+                except PostModel.DoesNotExist:
+                    # Fallback if post doesn't exist
+                    message_obj = Message.objects.create(
+                        room=room,
+                        sender=sender_user,
+                        message=data.get("message", "Shared content")
+                    )
+                    return {"sender": sender_user.username, "message": data.get("message", "Shared content")}
+            else:
+                # Regular text message
+                text = data.get("message")
+                if text:
+                    message_obj = Message.objects.create(room=room, message=text, sender=sender_user)
+                    return {"sender": sender_user.username, "message": text}
+
+        except (Room.DoesNotExist, UserModel.DoesNotExist):
+            return None

instagram-clone-Django/chat/consumers.py

@@ -0,0 +1,38 @@
+# Generated by Django 5.2.3 on 2025-09-13 20:04
+
+import django.db.models.deletion
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        ('posts', '0001_initial'),
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Room',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('room_name', models.CharField(max_length=100, unique=True)),
+                ('chat_type', models.CharField(choices=[('private', 'Private'), ('public', 'Public')], default='public', max_length=10)),
+                ('members', models.ManyToManyField(related_name='chats', to=settings.AUTH_USER_MODEL)),
+            ],
+        ),
+        migrations.CreateModel(
+            name='Message',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('message', models.CharField(max_length=255)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('post', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, to='posts.postmodel')),
+                ('sender', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
+                ('room', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='messages', to='chat.room')),
+            ],
+        ),
+    ]

instagram-clone-Django/chat/migrations/0001_initial.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.2.3 on 2025-09-14 17:34
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('chat', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='message',
+            name='message',
+            field=models.CharField(blank=True, max_length=255, null=True),
+        ),
+    ]

instagram-clone-Django/chat/migrations/0002_alter_message_message.py

@@ -0,0 +1,35 @@
+from django.db import models
+
+from posts.models import PostModel
+from users.models import UserModel
+
+
+class Room(models.Model):
+    CHAT_TYPE_CHOICES = [
+        ('private', 'Private'),
+        ('public', 'Public'),
+    ]
+    room_name = models.CharField(max_length=100, unique=True)
+    members = models.ManyToManyField(UserModel, related_name='chats')
+    chat_type = models.CharField(max_length=10, choices=CHAT_TYPE_CHOICES, default='public')
+
+    def __str__(self):
+        return self.room_name
+
+    def can_add_member(self):
+        if self.chat_type != 'private':
+            return True
+        if self.chat_type == 'private' and self.members.count() < 2:
+            return True
+        return False
+
+
+class Message(models.Model):
+    room = models.ForeignKey(Room, on_delete=models.CASCADE, related_name='messages')
+    message = models.CharField(max_length=255, blank=True, null=True)
+    sender = models.ForeignKey(UserModel, on_delete=models.CASCADE)
+    post = models.ForeignKey(PostModel, null=True, blank=True, on_delete=models.SET_NULL)
+    created_at = models.DateTimeField(auto_now_add=True)
+
+    def __str__(self):
+        return f'{self.sender} -> {self.room.room_name}'

instagram-clone-Django/chat/migrations/__init__.py

@@ -0,0 +1,6 @@
+from django.urls import path
+from .consumers import ChatConsumer
+
+wsPatterns = [
+    path("ws/messages/<str:room_name>/", ChatConsumer.as_asgi())
+]

instagram-clone-Django/chat/models.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.