From 5783e9d616dfb0bfaad1a0e3f48df9988cd68224 Mon Sep 17 00:00:00 2001
From: Niraj Adhikari <nrjadkry@gmail.com>
Date: Mon, 8 Jul 2024 14:20:30 +0545
Subject: [PATCH] update: task events update api for validation process

---
 src/backend/app/tasks/task_crud.py   | 43 ++++++++++++++++++++++++++++
 src/backend/app/tasks/task_routes.py | 37 ++++++++++++++++++++----
 2 files changed, 75 insertions(+), 5 deletions(-)

diff --git a/src/backend/app/tasks/task_crud.py b/src/backend/app/tasks/task_crud.py
index 39948584..6bb574df 100644
--- a/src/backend/app/tasks/task_crud.py
+++ b/src/backend/app/tasks/task_crud.py
@@ -86,6 +86,49 @@ async def map_task(
     return {"project_id": project_id, "task_id": task_id, "comment": comment}
 
 
+async def update_task_state(
+    db: Database,
+    project_id: uuid.UUID,
+    task_id: uuid.UUID,
+    user_id: str,
+    comment: str,
+    initial_state: State,
+    final_state: State,
+):
+    query = """
+                WITH last AS (
+                    SELECT *
+                    FROM task_events
+                    WHERE project_id = :project_id AND task_id = :task_id
+                    ORDER BY event_id DESC
+                    LIMIT 1
+                ),
+                locked AS (
+                    SELECT *
+                    FROM last
+                    WHERE user_id = :user_id AND state = :initial_state
+                )
+                INSERT INTO task_events(event_id, project_id, task_id, user_id, state, comment, created_at)
+                SELECT gen_random_uuid(), project_id, task_id, user_id, :final_state, :comment, now()
+                FROM last
+                WHERE user_id = :user_id
+                RETURNING project_id, task_id, user_id, state;
+        """
+
+    values = {
+        "project_id": str(project_id),
+        "task_id": str(task_id),
+        "user_id": str(user_id),
+        "comment": comment,
+        "initial_state": initial_state.name,
+        "final_state": final_state.name,
+    }
+
+    await db.fetch_one(query, values)
+
+    return {"project_id": project_id, "task_id": task_id, "comment": comment}
+
+
 async def finish(
     db: Database, project_id: uuid.UUID, task_id: uuid.UUID, user_id: str, comment: str
 ):
diff --git a/src/backend/app/tasks/task_routes.py b/src/backend/app/tasks/task_routes.py
index c857d039..92985a37 100644
--- a/src/backend/app/tasks/task_routes.py
+++ b/src/backend/app/tasks/task_routes.py
@@ -1,7 +1,7 @@
 import uuid
 from fastapi import APIRouter, Depends
 from app.config import settings
-from app.models.enums import EventType
+from app.models.enums import EventType, State
 from app.tasks import task_schemas, task_crud
 from app.users.user_deps import login_required
 from app.users.user_schemas import AuthUser
@@ -43,18 +43,45 @@ async def new_event(
                 "Done: locked for mapping",
             )
         case EventType.FINISH:
-            return await task_crud.finish(
+            return await task_crud.update_task_state(
                 db,
                 detail.project_id,
                 detail.task_id,
                 user_id,
                 "Done: unlocked to validate",
+                State.LOCKED_FOR_MAPPING,
+                State.UNLOCKED_TO_VALIDATE,
             )
         case EventType.VALIDATE:
-            pass
+            return await task_crud.update_task_state(
+                db,
+                detail.project_id,
+                detail.task_id,
+                user_id,
+                "Done: locked for validation",
+                State.UNLOCKED_TO_VALIDATE,
+                State.LOCKED_FOR_VALIDATION,
+            )
         case EventType.GOOD:
-            pass
+            return await task_crud.update_task_state(
+                db,
+                detail.project_id,
+                detail.task_id,
+                user_id,
+                "Done: Task is Good",
+                State.LOCKED_FOR_VALIDATION,
+                State.UNLOCKED_DONE,
+            )
+
         case EventType.BAD:
-            pass
+            return await task_crud.update_task_state(
+                db,
+                detail.project_id,
+                detail.task_id,
+                user_id,
+                "Done: needs to redo",
+                State.LOCKED_FOR_VALIDATION,
+                State.UNLOCKED_TO_MAP,
+            )
 
     return True