Add online presence

Author: HugeFrog24

backend/Dockerfile

@@ -19,7 +19,7 @@ RUN mkdir -p /usr/src/app && chown celery:celery /usr/src/app
 FROM base AS builder
 COPY backend/requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
-RUN pip install --no-cache-dir celery redis django-celery-beat
+RUN pip install --no-cache-dir celery redis django-celery-beat daphne
 
 # Development stage
 FROM base AS development
@@ -28,7 +28,7 @@ COPY backend/ .
 RUN python manage.py collectstatic --noinput
 RUN chown -R celery:celery /usr/src/app
 USER celery
-CMD ["python", "manage.py", "runserver", "0.0.0.0:8000"]
+CMD ["daphne", "-b", "0.0.0.0", "-p", "8000", "backend.asgi:application"]
 
 # Production stage
 FROM base AS production
@@ -37,4 +37,4 @@ COPY backend/ .
 RUN python manage.py collectstatic --noinput
 RUN chown -R celery:celery /usr/src/app
 USER celery
-CMD ["gunicorn", "tibiknini.wsgi:application", "--workers", "3", "--bind", "0.0.0.0:8000"]
+CMD ["daphne", "-b", "0.0.0.0", "-p", "8000", "backend.asgi:application"]

backend/api/urls.py

@@ -23,6 +23,7 @@
     path("privacy_policy/", PrivacyPolicyView.as_view(), name="privacy_policy"),
     path("terms_of_service/", TermsOfServiceView.as_view(), name="terms_of_service"),
     path("users/", include("users.urls")),
+    path("presence/", include("presence.urls")),
     path("set-csrf-token/", set_csrf_token, name="set_csrf_token"),
     path("auth/login/", ReCaptchaLoginView.as_view(), name="login"),
     path("auth/logout/", auth_views.LogoutView.as_view(), name="logout"),

backend/backend/asgi.py

@@ -8,9 +8,18 @@
 """
 
 import os
-
 from django.core.asgi import get_asgi_application
 
-os.environ.setdefault("DJANGO_SETTINGS_MODULE", "tibiknini.settings")
+os.environ.setdefault("DJANGO_SETTINGS_MODULE", "backend.settings")
+django_asgi_app = get_asgi_application()
+
+from channels.routing import ProtocolTypeRouter, URLRouter
+from channels.auth import AuthMiddlewareStack
+from presence.routing import websocket_urlpatterns
 
-application = get_asgi_application()
+application = ProtocolTypeRouter({
+    "http": django_asgi_app,
+    "websocket": AuthMiddlewareStack(
+        URLRouter(websocket_urlpatterns)
+    ),
+})

backend/backend/settings.py

@@ -70,6 +70,8 @@ def get_secret(secret_name, default=None):
     "moderation",
     "navbar",
     "users",
+    "presence",
+    "channels",
     "corsheaders",
     "rest_framework",
     "django.contrib.admin",
@@ -95,6 +97,17 @@ def get_secret(secret_name, default=None):
 
 ROOT_URLCONF = "backend.urls"
 
+# Channels Configuration
+ASGI_APPLICATION = "backend.asgi.application"
+CHANNEL_LAYERS = {
+    "default": {
+        "BACKEND": "channels_redis.core.RedisChannelLayer",
+        "CONFIG": {
+            "hosts": [(os.getenv("REDIS_HOST", "redis"), 6379)],
+        },
+    },
+}
+
 TEMPLATES = [
     {
         "BACKEND": "django.template.backends.django.DjangoTemplates",
@@ -165,6 +178,11 @@ def get_secret(secret_name, default=None):
     "http://localhost",
     "http://127.0.0.1",
 ]
+
+CORS_ALLOW_CREDENTIALS = True
+CORS_ALLOW_METHODS = ['GET', 'POST', 'PUT', 'PATCH', 'DELETE', 'OPTIONS']
+CORS_ALLOW_HEADERS = ['*']
+
 CSRF_TRUSTED_ORIGINS = [
     "http://127.0.0.1",
     "https://127.0.0.1",
@@ -173,8 +191,8 @@ def get_secret(secret_name, default=None):
     f"https://*.{os.getenv('DOMAIN_NAME')}",
 ]
 
-SESSION_COOKIE_HTTPONLY = True
-SECURE_PROXY_SSL_HEADER = ("HTTP_X_FORWARDED_PROTO", "https")
+# Allow WebSocket connections
+CORS_ALLOW_ALL_ORIGINS = True if DEBUG else False
 
 # Celery Configuration
 CELERY_BROKER_URL = os.environ.get('CELERY_BROKER_URL', 'redis://localhost:6379/0')

backend/presence/__init__.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

backend/presence/admin.py

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

backend/presence/apps.py

@@ -0,0 +1,76 @@
+import json
+from channels.generic.websocket import AsyncWebsocketConsumer
+from channels.db import database_sync_to_async
+from django.utils import timezone
+from django.apps import apps
+
+class PresenceConsumer(AsyncWebsocketConsumer):
+    async def connect(self):
+        # Accept the WebSocket connection
+        await self.accept()
+        
+        # Store the channel name and user
+        self.user = self.scope["user"]
+        
+        # Handle authentication
+        if not self.user.is_authenticated:
+            # Close connection if user is not authenticated
+            await self.close(code=4001)
+            return
+            
+        # Update user presence
+        await self.update_user_presence(True, self.channel_name)
+        
+        # Add this channel to the group for broadcasting presence updates
+        await self.channel_layer.group_add("presence", self.channel_name)
+        
+        # Broadcast the user's online status to all connected clients
+        await self.channel_layer.group_send(
+            "presence",
+            {
+                "type": "presence_update",
+                "user_id": self.user.id,
+                "status": "online"
+            }
+        )
+
+    async def disconnect(self, close_code):
+        if hasattr(self, 'user') and self.user.is_authenticated:
+            await self.update_user_presence(False, "")
+            
+            # Remove this channel from the group
+            await self.channel_layer.group_discard("presence", self.channel_name)
+            
+            # Broadcast the user's offline status
+            await self.channel_layer.group_send(
+                "presence",
+                {
+                    "type": "presence_update",
+                    "user_id": self.user.id,
+                    "status": "offline"
+                }
+            )
+
+    async def receive(self, text_data):
+        try:
+            data = json.loads(text_data)
+            if data.get('type') == 'authenticate':
+                # Handle authentication if needed
+                # The token validation is already handled by Django Channels authentication
+                pass
+            # Handle other message types if needed
+        except json.JSONDecodeError:
+            await self.close(code=4000)
+
+    async def presence_update(self, event):
+        # Send presence update to WebSocket
+        await self.send(text_data=json.dumps(event))
+
+    @database_sync_to_async
+    def update_user_presence(self, is_online, channel_name):
+        UserPresence = apps.get_model('presence', 'UserPresence')
+        presence, _ = UserPresence.objects.get_or_create(user=self.user)
+        presence.is_online = is_online
+        presence.channel_name = channel_name
+        presence.last_seen = timezone.now()
+        presence.save()

backend/presence/consumers.py

@@ -0,0 +1,32 @@
+# Generated by Django 5.1.3 on 2024-11-25 14:52
+
+import django.db.models.deletion
+import django.utils.timezone
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='UserPresence',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('last_seen', models.DateTimeField(default=django.utils.timezone.now)),
+                ('is_online', models.BooleanField(default=False)),
+                ('channel_name', models.CharField(blank=True, max_length=255)),
+                ('user', models.OneToOneField(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL)),
+            ],
+            options={
+                'verbose_name': 'User Presence',
+                'verbose_name_plural': 'User Presences',
+            },
+        ),
+    ]