mplement Time and Activity Logging Endpoints

blt/urls.py

@@ -18,6 +18,7 @@
 from blt import settings
 from company.views import ShowBughuntView
 from website.api.views import (
+    ActivityLogViewSet,
     AuthApiViewset,
     BugHuntApiViewset,
     BugHuntApiViewsetV2,
@@ -31,6 +32,7 @@
     ProjectViewSet,
     StatsApiViewset,
     TagApiViewset,
+    TimeLogViewSet,
     UrlCheckApiViewset,
     UserIssueViewSet,
     UserProfileViewSet,
@@ -105,6 +107,8 @@
 router.register(r"userissues", UserIssueViewSet, basename="userissues")
 router.register(r"profile", UserProfileViewSet, basename="profile")
 router.register(r"domain", DomainViewSet, basename="domain")
+router.register(r"timelogs", TimeLogViewSet, basename="timelogs")
+router.register(r"activitylogs", ActivityLogViewSet, basename="activitylogs")
 
 from allauth.socialaccount.providers.facebook import views as facebook_views
 from allauth.socialaccount.providers.github import views as github_views

website/api/views.py

@@ -3,19 +3,24 @@
 
 from django.conf import settings
 from django.contrib.sites.shortcuts import get_current_site
+from django.core.exceptions import ObjectDoesNotExist, ValidationError
 from django.core.files.storage import default_storage
 from django.core.mail import send_mail
 from django.db.models import Count, Q, Sum
 from django.template.loader import render_to_string
+from django.utils import timezone
 from django.utils.text import slugify
 from rest_framework import filters, status, viewsets
 from rest_framework.authentication import TokenAuthentication
+from rest_framework.decorators import action
+from rest_framework.exceptions import NotFound, ParseError
 from rest_framework.pagination import PageNumberPagination
 from rest_framework.permissions import AllowAny, IsAuthenticated, IsAuthenticatedOrReadOnly
 from rest_framework.response import Response
 from rest_framework.views import APIView
 
 from website.models import (
+    ActivityLog,
     Company,
     Contributor,
     Domain,
@@ -27,11 +32,13 @@
     Points,
     Project,
     Tag,
+    TimeLog,
     Token,
     User,
     UserProfile,
 )
 from website.serializers import (
+    ActivityLogSerializer,
     BugHuntPrizeSerializer,
     BugHuntSerializer,
     CompanySerializer,
@@ -40,6 +47,7 @@
     IssueSerializer,
     ProjectSerializer,
     TagSerializer,
+    TimeLogSerializer,
     UserProfileSerializer,
 )
 from website.views import LeaderboardBase, image_validator
@@ -735,3 +743,73 @@
 class TagApiViewset(viewsets.ModelViewSet):
     queryset = Tag.objects.all()
     serializer_class = TagSerializer
+
+
+class TimeLogViewSet(viewsets.ModelViewSet):
+    queryset = TimeLog.objects.all()
+    serializer_class = TimeLogSerializer
+    permission_classes = [IsAuthenticated]
+
+    def perform_create(self, serializer):
+        try:
+            serializer.save(user=self.request.user)
+        except ValidationError as e:
+            raise ParseError(detail=str(e))
+        except Exception as e:
+            raise ParseError(detail="An unexpected error occurred while creating the time log.")
+
+    @action(detail=False, methods=["post"])
+    def start(self, request):
+        """Starts a new time log"""
+        data = request.data
+        data["start_time"] = timezone.now()  # Set start time to current tim
+
+        serializer = self.get_serializer(data=data)
+        try:
+            serializer.is_valid(raise_exception=True)
+            self.perform_create(serializer)
+            return Response(serializer.data, status=status.HTTP_201_CREATED)
+        except ValidationError as e:
+            return Response({"detail": str(e)}, status=status.HTTP_400_BAD_REQUEST)
+        except Exception as e:
+            return Response(
+                {"detail": "An unexpected error occurred while starting the time log."},
+                status=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            )
+
+    @action(detail=True, methods=["post"])
+    def stop(self, request, pk=None):
+        """Stops the time log and calculates duration"""
+        try:
+            timelog = self.get_object()
+        except ObjectDoesNotExist:
+            raise NotFound(detail="Time log not found.")
+
+        timelog.end_time = timezone.now()
+        if timelog.start_time:
+            timelog.duration = timelog.end_time - timelog.start_time
+
+        try:
+            timelog.save()
+            return Response(TimeLogSerializer(timelog).data, status=status.HTTP_200_OK)
+        except ValidationError as e:
+            return Response({"detail": str(e)}, status=status.HTTP_400_BAD_REQUEST)
+        except Exception as e:
+            return Response(
+                {"detail": "An unexpected error occurred while stopping the time log."},
+                status=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            )
+
+
+class ActivityLogViewSet(viewsets.ModelViewSet):
+    queryset = ActivityLog.objects.all()
+    serializer_class = ActivityLogSerializer
+    permission_classes = [IsAuthenticated]
+
+    def perform_create(self, serializer):
+        try:
+            serializer.save(user=self.request.user, recorded_at=timezone.now())
+        except ValidationError as e:
+            raise ParseError(detail=str(e))
+        except Exception as e:
+            raise ParseError(detail="An unexpected error occurred while creating the activity log.")

website/serializers.py

@@ -2,6 +2,7 @@
 from rest_framework import serializers
 
 from website.models import (
+    ActivityLog,
     Company,
     Contributor,
     Domain,
@@ -11,6 +12,7 @@
     Points,
     Project,
     Tag,
+    TimeLog,
     User,
     UserProfile,
 )
@@ -125,3 +127,23 @@ class ContributorSerializer(serializers.ModelSerializer):
     class Meta:
         model = Contributor
         fields = "__all__"
+
+
+class TimeLogSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = TimeLog
+        fields = ["id", "user", "start_time", "end_time", "duration", "github_issue_url", "created"]
+        read_only_fields = [
+            "id",
+            "user",
+            "end_time",
+            "duration",
+            "created",
+        ]  # These fields will be managed automatically
+
+
+class ActivityLogSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = ActivityLog
+        fields = ["id", "user", "window_title", "url", "recorded_at", "created"]
+        read_only_fields = ["id", "user", "recorded_at", "created"]  # Auto-filled fields