Add notification logic for task assignment and draft publication

Author: japint

apps/codebility/app/home/kanban/[projectId]/[id]/actions.ts

@@ -3,6 +3,7 @@
 import { revalidatePath } from "next/cache";
 import { Task, TaskDraft } from "@/types/home/codev";
 import { createClientServerComponent } from "@/utils/supabase/server";
+import { createNotificationAction } from "@/lib/actions/notification.actions";
 
 interface CodevMember {
   id: string;
@@ -247,8 +248,10 @@ export const updateTask = async (
     const rawDeadline = formData.get("deadline")?.toString();
     const deadline = rawDeadline === "" || rawDeadline === "null" ? null : rawDeadline;
 
+    const title = formData.get("title")?.toString();
+
     const updateData = {
-      title: formData.get("title")?.toString(),
+      title: title,
       description: formData.get("description")?.toString(),
       priority: formData.get("priority")?.toString(),
       difficulty: formData.get("difficulty")?.toString(),
@@ -264,9 +267,10 @@ export const updateTask = async (
       updated_at: new Date().toISOString(),
     };
 
+    // 1. Fetch existing task to check current assignee
     const { data: existingTask, error: fetchError } = await supabase
       .from("tasks")
-      .select("*")
+      .select("codev_id, title, kanban_column_id")
       .eq("id", taskId)
       .single();
 
@@ -275,6 +279,7 @@ export const updateTask = async (
       return { success: false, error: "Could not find task to update" };
     }
 
+    // 2. Perform the update
     const { error: updateError } = await supabase
       .from("tasks")
       .update({
@@ -288,6 +293,32 @@ export const updateTask = async (
       return { success: false, error: updateError.message };
     }
 
+    // 3. NOTIFICATION LOGIC: Trigger if assignee changed and isn't null
+    if (codev_id && codev_id !== existingTask.codev_id) {
+      // Get current user to avoid self-notification
+      const { data: { user } } = await supabase.auth.getUser();
+      
+      if (user && user.id !== codev_id) {
+        // Fetch project info to build the action URL
+        const { data: colData } = await supabase
+          .from("kanban_columns")
+          .select("board_id, kanban_boards(project_id)")
+          .eq("id", existingTask.kanban_column_id)
+          .single();
+
+        await createNotificationAction({
+          recipientId: codev_id,
+          title: "Task Assigned",
+          message: `You've been assigned to: ${title || existingTask.title}`,
+          type: "task",
+          priority: "normal",
+          projectId: (colData?.kanban_boards as any)?.project_id,
+          actionUrl: `/home/kanban/${(colData?.kanban_boards as any)?.project_id}/${colData?.board_id}?taskId=${taskId}`,
+          metadata: { taskId }
+        });
+      }
+    }
+
     return { success: true };
   } catch (error) {
     console.error("Update task error:", error);
@@ -989,6 +1020,25 @@ export const promoteDraft = async (
       return { success: false, error: createError.message };
     }
 
+    // Step 3.5: Trigger Notification if assignee exists in the draft
+    if (draft.codev_id) {
+      // Get current user to avoid self-notification
+      const { data: { user } } = await supabase.auth.getUser();
+      
+      if (user && user.id !== draft.codev_id) {
+        await createNotificationAction({
+          recipientId: draft.codev_id,
+          title: "Task Assigned (from Draft)",
+          message: `The draft "${draft.title}" has been published and assigned to you.`,
+          type: "task",
+          priority: "normal",
+          projectId: draft.project_id,
+          actionUrl: `/home/kanban/${draft.project_id}/${draft.intended_column_id}?taskId=${newTask.id}`,
+          metadata: { taskId: newTask.id }
+        });
+      }
+    }
+
     // Step 4: Delete the draft
     const { error: deleteError } = await supabase
       .from("task_drafts")

apps/codebility/lib/actions/notification.ts

@@ -0,0 +1,52 @@
+import { createClientServerComponent } from "@/utils/supabase/server";
+import { createNotificationAction } from "./notification.actions";
+import { revalidatePath } from "next/cache";
+
+interface AssignTaskParams {
+    taskId: string;
+    assigneeId: string;
+    taskTitle: string;
+    projectId: string;
+}
+
+export async function assignTaskAction({
+    taskId,
+    assigneeId,
+    taskTitle,
+    projectId
+}: AssignTaskParams) {
+    const supabase = await createClientServerComponent();
+
+    // 1. Update the database
+    const { data: task, error: updateError } = await supabase.from("tasks").update({
+        assignee_id: assigneeId,
+        updated_at: new Date().toISOString()
+    }).eq("id", taskId).select().single();
+
+    if (updateError) return { data: null, error: updateError.message };
+
+    // 2. Trigger the notification using your existing action
+    // We don't "await" this if we want the UI to respond faster,
+    // but awaiting is safer for error handling.
+
+    const { data: { user } } = await supabase.auth.getUser();
+    if (user && user.id !== assigneeId) {
+        const { error: notifyError } = await createNotificationAction({
+            recipientId: assigneeId,
+            title: "New Task Assigned",
+            message: `You've been assigned to: ${taskTitle}`,
+            type: "task", // matches your Notification type
+            priority: "normal",
+            projectId: projectId,
+            actionUrl: `/dashboard/projects/${projectId}/kanban`, // Direct link to the task
+            metadata: { taskId }
+        });
+
+        if (notifyError) {
+            console.error("Notification failed:", notifyError);
+        }
+    }
+
+    revalidatePath(`/dashboard/projects/${projectId}/kanban`);
+    return { data: task, error: null };
+}