From 330744daf918c75f700a371c10bd79e6532ab4ce Mon Sep 17 00:00:00 2001
From: Chop <28534054+RChopin@users.noreply.github.com>
Date: Tue, 3 Jun 2025 20:46:37 +0200
Subject: [PATCH] feat: Implement task notes functionality with CRUD operations
 and integrate into project tasks section

---
 src/app/api/task-notes/route.js           |  73 +++++++++
 src/app/projects/[id]/edit/page.js        |   3 +-
 src/app/projects/[id]/page.js             |  16 +-
 src/app/tasks/templates/[id]/edit/page.js |   3 +-
 src/components/ProjectTaskForm.js         |  16 +-
 src/components/ProjectTasksSection.js     | 184 +++++++++++++++++++++-
 src/lib/init-db.js                        |  10 +-
 src/lib/queries/notes.js                  |  17 ++
 src/lib/queries/tasks.js                  |   7 +-
 9 files changed, 309 insertions(+), 20 deletions(-)
 create mode 100644 src/app/api/task-notes/route.js

diff --git a/src/app/api/task-notes/route.js b/src/app/api/task-notes/route.js
new file mode 100644
index 0000000..8631073
--- /dev/null
+++ b/src/app/api/task-notes/route.js
@@ -0,0 +1,73 @@
+import {
+	getNotesByTaskId,
+	addNoteToTask,
+	deleteNote,
+} from "@/lib/queries/notes";
+import { NextResponse } from "next/server";
+
+// GET: Get notes for a specific task
+export async function GET(req) {
+	const { searchParams } = new URL(req.url);
+	const taskId = searchParams.get("task_id");
+
+	if (!taskId) {
+		return NextResponse.json({ error: "task_id is required" }, { status: 400 });
+	}
+	try {
+		const notes = getNotesByTaskId(taskId);
+		return NextResponse.json(notes);
+	} catch (error) {
+		console.error("Error fetching task notes:", error);
+		return NextResponse.json(
+			{ error: "Failed to fetch task notes" },
+			{ status: 500 }
+		);
+	}
+}
+
+// POST: Add a note to a task
+export async function POST(req) {
+	try {
+		const { task_id, note } = await req.json();
+
+		if (!task_id || !note) {
+			return NextResponse.json(
+				{ error: "task_id and note are required" },
+				{ status: 400 }
+			);
+		}
+
+		addNoteToTask(task_id, note);
+		return NextResponse.json({ success: true });
+	} catch (error) {
+		console.error("Error adding task note:", error);
+		return NextResponse.json(
+			{ error: "Failed to add task note" },
+			{ status: 500 }
+		);
+	}
+}
+
+// DELETE: Delete a note
+export async function DELETE(req) {
+	try {
+		const { searchParams } = new URL(req.url);
+		const noteId = searchParams.get("note_id");
+
+		if (!noteId) {
+			return NextResponse.json(
+				{ error: "note_id is required" },
+				{ status: 400 }
+			);
+		}
+
+		deleteNote(noteId);
+		return NextResponse.json({ success: true });
+	} catch (error) {
+		console.error("Error deleting note:", error);
+		return NextResponse.json(
+			{ error: "Failed to delete note" },
+			{ status: 500 }
+		);
+	}
+}
diff --git a/src/app/projects/[id]/edit/page.js b/src/app/projects/[id]/edit/page.js
index d23bc06..adc6680 100644
--- a/src/app/projects/[id]/edit/page.js
+++ b/src/app/projects/[id]/edit/page.js
@@ -1,7 +1,8 @@
 import ProjectForm from "@/components/ProjectForm";
 
 export default async function EditProjectPage({ params }) {
-	const res = await fetch(`http://localhost:3000/api/projects/${params.id}`, {
+	const { id } = await params;
+	const res = await fetch(`http://localhost:3000/api/projects/${id}`, {
 		cache: "no-store",
 	});
 	const project = await res.json();
diff --git a/src/app/projects/[id]/page.js b/src/app/projects/[id]/page.js
index 7674af4..7f3f2fe 100644
--- a/src/app/projects/[id]/page.js
+++ b/src/app/projects/[id]/page.js
@@ -13,8 +13,9 @@ import PageContainer from "@/components/ui/PageContainer";
 import PageHeader from "@/components/ui/PageHeader";
 
 export default async function ProjectViewPage({ params }) {
-	const project = getProjectWithContract(params.id);
-	const notes = getNotesForProject(params.id);
+	const { id } = await params;
+	const project = getProjectWithContract(id);
+	const notes = getNotesForProject(id);
 	const daysRemaining = differenceInCalendarDays(
 		parseISO(project.finish_date),
 		new Date()
@@ -70,8 +71,8 @@ export default async function ProjectViewPage({ params }) {
 								</svg>
 								Back to Projects
 							</Button>
-						</Link>
-						<Link href={`/projects/${params.id}/edit`}>
+						</Link>{" "}
+						<Link href={`/projects/${id}/edit`}>
 							<Button variant="secondary">Edit Project</Button>
 						</Link>
 					</div>
@@ -142,7 +143,6 @@ export default async function ProjectViewPage({ params }) {
 						)}
 					</CardContent>
 				</Card>
-
 				<Card>
 					<CardHeader>
 						<h2 className="text-xl font-semibold text-gray-900">
@@ -175,10 +175,10 @@ export default async function ProjectViewPage({ params }) {
 							<p className="text-gray-900">{project.investor}</p>
 						</div>
 					</CardContent>
-				</Card>
+				</Card>{" "}
 			</div>
 
-			<ProjectTasksSection projectId={params.id} />
+			<ProjectTasksSection projectId={id} />
 
 			<Card>
 				<CardHeader>
@@ -186,7 +186,7 @@ export default async function ProjectViewPage({ params }) {
 				</CardHeader>
 				<CardContent>
 					<div className="mb-6">
-						<NoteForm projectId={params.id} />
+						<NoteForm projectId={id} />
 					</div>
 					{notes.length === 0 ? (
 						<div className="text-center py-8">
diff --git a/src/app/tasks/templates/[id]/edit/page.js b/src/app/tasks/templates/[id]/edit/page.js
index 3fa63e6..d357906 100644
--- a/src/app/tasks/templates/[id]/edit/page.js
+++ b/src/app/tasks/templates/[id]/edit/page.js
@@ -6,10 +6,11 @@ import Button from "@/components/ui/Button";
 import TaskTemplateForm from "@/components/TaskTemplateForm";
 
 export default async function EditTaskTemplatePage({ params }) {
+	const { id } = await params;
 	// Fetch the task template
 	const template = db
 		.prepare("SELECT * FROM tasks WHERE task_id = ? AND is_standard = 1")
-		.get(params.id);
+		.get(id);
 
 	if (!template) {
 		notFound();
diff --git a/src/components/ProjectTaskForm.js b/src/components/ProjectTaskForm.js
index 01fc409..434ea2a 100644
--- a/src/components/ProjectTaskForm.js
+++ b/src/components/ProjectTaskForm.js
@@ -10,6 +10,7 @@ export default function ProjectTaskForm({ projectId, onTaskAdded }) {
 	const [selectedTemplate, setSelectedTemplate] = useState("");
 	const [customTaskName, setCustomTaskName] = useState("");
 	const [customMaxWaitDays, setCustomMaxWaitDays] = useState("");
+	const [customDescription, setCustomDescription] = useState("");
 	const [priority, setPriority] = useState("normal");
 	const [isSubmitting, setIsSubmitting] = useState(false);
 
@@ -40,6 +41,7 @@ export default function ProjectTaskForm({ projectId, onTaskAdded }) {
 			} else {
 				requestData.custom_task_name = customTaskName.trim();
 				requestData.custom_max_wait_days = parseInt(customMaxWaitDays) || 0;
+				requestData.custom_description = customDescription.trim();
 			}
 
 			const res = await fetch("/api/project-tasks", {
@@ -47,12 +49,12 @@ export default function ProjectTaskForm({ projectId, onTaskAdded }) {
 				headers: { "Content-Type": "application/json" },
 				body: JSON.stringify(requestData),
 			});
-
 			if (res.ok) {
 				// Reset form
 				setSelectedTemplate("");
 				setCustomTaskName("");
 				setCustomMaxWaitDays("");
+				setCustomDescription("");
 				setPriority("normal");
 				if (onTaskAdded) onTaskAdded();
 			} else {
@@ -141,6 +143,18 @@ export default function ProjectTaskForm({ projectId, onTaskAdded }) {
 							min="0"
 						/>
 					</div>
+					<div>
+						<label className="block text-sm font-medium text-gray-700 mb-2">
+							Description
+						</label>
+						<textarea
+							value={customDescription}
+							onChange={(e) => setCustomDescription(e.target.value)}
+							className="w-full px-3 py-2 border border-gray-300 rounded-md shadow-sm focus:ring-2 focus:ring-blue-500 focus:border-blue-500"
+							placeholder="Enter task description (optional)..."
+							rows={3}
+						/>
+					</div>
 				</div>
 			)}
 
diff --git a/src/components/ProjectTasksSection.js b/src/components/ProjectTasksSection.js
index f447101..cbd3a85 100644
--- a/src/components/ProjectTasksSection.js
+++ b/src/components/ProjectTasksSection.js
@@ -9,12 +9,34 @@ import Badge from "./ui/Badge";
 export default function ProjectTasksSection({ projectId }) {
 	const [projectTasks, setProjectTasks] = useState([]);
 	const [loading, setLoading] = useState(true);
+	const [taskNotes, setTaskNotes] = useState({});
+	const [newNote, setNewNote] = useState({});
+	const [loadingNotes, setLoadingNotes] = useState({});
 	useEffect(() => {
 		const fetchProjectTasks = async () => {
 			try {
 				const res = await fetch(`/api/project-tasks?project_id=${projectId}`);
 				const tasks = await res.json();
 				setProjectTasks(tasks);
+
+				// Fetch notes for each task
+				const notesPromises = tasks.map(async (task) => {
+					try {
+						const notesRes = await fetch(`/api/task-notes?task_id=${task.id}`);
+						const notes = await notesRes.json();
+						return { taskId: task.id, notes };
+					} catch (error) {
+						console.error(`Failed to fetch notes for task ${task.id}:`, error);
+						return { taskId: task.id, notes: [] };
+					}
+				});
+
+				const notesResults = await Promise.all(notesPromises);
+				const notesMap = {};
+				notesResults.forEach(({ taskId, notes }) => {
+					notesMap[taskId] = notes;
+				});
+				setTaskNotes(notesMap);
 			} catch (error) {
 				console.error("Failed to fetch project tasks:", error);
 			} finally {
@@ -24,12 +46,30 @@ export default function ProjectTasksSection({ projectId }) {
 
 		fetchProjectTasks();
 	}, [projectId]);
-
 	const refetchTasks = async () => {
 		try {
 			const res = await fetch(`/api/project-tasks?project_id=${projectId}`);
 			const tasks = await res.json();
 			setProjectTasks(tasks);
+
+			// Refresh notes for all tasks
+			const notesPromises = tasks.map(async (task) => {
+				try {
+					const notesRes = await fetch(`/api/task-notes?task_id=${task.id}`);
+					const notes = await notesRes.json();
+					return { taskId: task.id, notes };
+				} catch (error) {
+					console.error(`Failed to fetch notes for task ${task.id}:`, error);
+					return { taskId: task.id, notes: [] };
+				}
+			});
+
+			const notesResults = await Promise.all(notesPromises);
+			const notesMap = {};
+			notesResults.forEach(({ taskId, notes }) => {
+				notesMap[taskId] = notes;
+			});
+			setTaskNotes(notesMap);
 		} catch (error) {
 			console.error("Failed to fetch project tasks:", error);
 		}
@@ -55,7 +95,6 @@ export default function ProjectTasksSection({ projectId }) {
 			alert("Error updating task status");
 		}
 	};
-
 	const handleDeleteTask = async (taskId) => {
 		if (!confirm("Are you sure you want to delete this task?")) return;
 		try {
@@ -72,6 +111,59 @@ export default function ProjectTasksSection({ projectId }) {
 			alert("Error deleting task");
 		}
 	};
+
+	const handleAddNote = async (taskId) => {
+		const noteContent = newNote[taskId]?.trim();
+		if (!noteContent) return;
+
+		setLoadingNotes((prev) => ({ ...prev, [taskId]: true }));
+
+		try {
+			const res = await fetch("/api/task-notes", {
+				method: "POST",
+				headers: { "Content-Type": "application/json" },
+				body: JSON.stringify({
+					task_id: taskId,
+					note: noteContent,
+				}),
+			});
+
+			if (res.ok) {
+				// Refresh notes for this task
+				const notesRes = await fetch(`/api/task-notes?task_id=${taskId}`);
+				const notes = await notesRes.json();
+				setTaskNotes((prev) => ({ ...prev, [taskId]: notes }));
+				setNewNote((prev) => ({ ...prev, [taskId]: "" }));
+			} else {
+				alert("Failed to add note");
+			}
+		} catch (error) {
+			alert("Error adding note");
+		} finally {
+			setLoadingNotes((prev) => ({ ...prev, [taskId]: false }));
+		}
+	};
+
+	const handleDeleteNote = async (noteId, taskId) => {
+		if (!confirm("Are you sure you want to delete this note?")) return;
+
+		try {
+			const res = await fetch(`/api/task-notes?note_id=${noteId}`, {
+				method: "DELETE",
+			});
+
+			if (res.ok) {
+				// Refresh notes for this task
+				const notesRes = await fetch(`/api/task-notes?task_id=${taskId}`);
+				const notes = await notesRes.json();
+				setTaskNotes((prev) => ({ ...prev, [taskId]: notes }));
+			} else {
+				alert("Failed to delete note");
+			}
+		} catch (error) {
+			alert("Error deleting note");
+		}
+	};
 	const getPriorityVariant = (priority) => {
 		switch (priority) {
 			case "urgent":
@@ -177,8 +269,15 @@ export default function ProjectTasksSection({ projectId }) {
 												>
 													{task.priority}
 												</Badge>
-											</div>
-
+											</div>{" "}
+											{/* Task Description */}
+											{task.description && (
+												<div className="mb-4">
+													<p className="text-sm text-gray-700 bg-gray-50 p-3 rounded-md">
+														{task.description}
+													</p>
+												</div>
+											)}
 											<div className="grid grid-cols-1 sm:grid-cols-3 gap-3 text-sm text-gray-600 mb-4">
 												<div>
 													<span className="font-medium">Max wait days:</span>{" "}
@@ -193,8 +292,7 @@ export default function ProjectTasksSection({ projectId }) {
 													{new Date(task.date_added).toLocaleDateString()}
 												</div>
 											</div>
-
-											<div className="flex items-center gap-3">
+											<div className="flex items-center gap-3 mb-4">
 												<span className="text-sm font-medium text-gray-700">
 													Status:
 												</span>
@@ -217,6 +315,80 @@ export default function ProjectTasksSection({ projectId }) {
 													{task.status.replace("_", " ")}
 												</Badge>
 											</div>
+											{/* Notes Section */}
+											<div className="border-t pt-4">
+												<h5 className="text-sm font-medium text-gray-900 mb-3">
+													Notes ({taskNotes[task.id]?.length || 0})
+												</h5>
+
+												{/* Existing Notes */}
+												{taskNotes[task.id] &&
+													taskNotes[task.id].length > 0 && (
+														<div className="space-y-2 mb-3">
+															{taskNotes[task.id].map((note) => (
+																<div
+																	key={note.note_id}
+																	className="bg-blue-50 p-3 rounded-md flex justify-between items-start"
+																>
+																	<div className="flex-1">
+																		<p className="text-sm text-gray-800">
+																			{note.note}
+																		</p>
+																		<p className="text-xs text-gray-500 mt-1">
+																			{new Date(
+																				note.note_date
+																			).toLocaleDateString()}{" "}
+																			at{" "}
+																			{new Date(
+																				note.note_date
+																			).toLocaleTimeString()}
+																		</p>
+																	</div>
+																	<button
+																		onClick={() =>
+																			handleDeleteNote(note.note_id, task.id)
+																		}
+																		className="ml-2 text-red-500 hover:text-red-700 text-xs font-bold"
+																		title="Delete note"
+																	>
+																		×
+																	</button>
+																</div>
+															))}
+														</div>
+													)}
+
+												{/* Add New Note */}
+												<div className="flex gap-2">
+													<input
+														type="text"
+														placeholder="Add a note..."
+														value={newNote[task.id] || ""}
+														onChange={(e) =>
+															setNewNote((prev) => ({
+																...prev,
+																[task.id]: e.target.value,
+															}))
+														}
+														className="flex-1 px-3 py-1.5 text-sm border border-gray-300 rounded-md focus:ring-2 focus:ring-blue-500 focus:border-blue-500"
+														onKeyPress={(e) => {
+															if (e.key === "Enter") {
+																handleAddNote(task.id);
+															}
+														}}
+													/>
+													<Button
+														size="sm"
+														variant="primary"
+														onClick={() => handleAddNote(task.id)}
+														disabled={
+															loadingNotes[task.id] || !newNote[task.id]?.trim()
+														}
+													>
+														{loadingNotes[task.id] ? "Adding..." : "Add"}
+													</Button>
+												</div>
+											</div>
 										</div>
 
 										<div className="ml-4 flex-shrink-0">
diff --git a/src/lib/init-db.js b/src/lib/init-db.js
index ab30ca6..ef08fdd 100644
--- a/src/lib/init-db.js
+++ b/src/lib/init-db.js
@@ -81,7 +81,6 @@ export default function initializeDatabase() {
 	} catch (e) {
 		// Column already exists, ignore error
 	}
-
 	// Migration: Add description column to tasks table
 	try {
 		db.exec(`
@@ -90,4 +89,13 @@ export default function initializeDatabase() {
 	} catch (e) {
 		// Column already exists, ignore error
 	}
+
+	// Migration: Add description column to project_tasks table
+	try {
+		db.exec(`
+      ALTER TABLE project_tasks ADD COLUMN custom_description TEXT;
+    `);
+	} catch (e) {
+		// Column already exists, ignore error
+	}
 }
diff --git a/src/lib/queries/notes.js b/src/lib/queries/notes.js
index 0bc5af1..b1befbe 100644
--- a/src/lib/queries/notes.js
+++ b/src/lib/queries/notes.js
@@ -12,3 +12,20 @@ export function addNoteToProject(project_id, note) {
 		note
 	);
 }
+
+export function getNotesByTaskId(task_id) {
+	return db
+		.prepare(`SELECT * FROM notes WHERE task_id = ? ORDER BY note_date DESC`)
+		.all(task_id);
+}
+
+export function addNoteToTask(task_id, note) {
+	db.prepare(`INSERT INTO notes (task_id, note) VALUES (?, ?)`).run(
+		task_id,
+		note
+	);
+}
+
+export function deleteNote(note_id) {
+	db.prepare(`DELETE FROM notes WHERE note_id = ?`).run(note_id);
+}
diff --git a/src/lib/queries/tasks.js b/src/lib/queries/tasks.js
index 3e11c81..6640f6a 100644
--- a/src/lib/queries/tasks.js
+++ b/src/lib/queries/tasks.js
@@ -16,6 +16,7 @@ export function getAllProjectTasks() {
 			pt.*,
 			COALESCE(pt.custom_task_name, t.name) as task_name,
 			COALESCE(pt.custom_max_wait_days, t.max_wait_days) as max_wait_days,
+			COALESCE(pt.custom_description, t.description) as description,
 			CASE 
 				WHEN pt.task_template_id IS NOT NULL THEN 'template'
 				ELSE 'custom'
@@ -42,6 +43,7 @@ export function getProjectTasks(projectId) {
 			pt.*,
 			COALESCE(pt.custom_task_name, t.name) as task_name,
 			COALESCE(pt.custom_max_wait_days, t.max_wait_days) as max_wait_days,
+			COALESCE(pt.custom_description, t.description) as description,
 			CASE 
 				WHEN pt.task_template_id IS NOT NULL THEN 'template'
 				ELSE 'custom'
@@ -72,13 +74,14 @@ export function createProjectTask(data) {
 	} else {
 		// Creating custom task
 		const stmt = db.prepare(`
-			INSERT INTO project_tasks (project_id, custom_task_name, custom_max_wait_days, status, priority)
-			VALUES (?, ?, ?, ?, ?)
+			INSERT INTO project_tasks (project_id, custom_task_name, custom_max_wait_days, custom_description, status, priority)
+			VALUES (?, ?, ?, ?, ?, ?)
 		`);
 		return stmt.run(
 			data.project_id,
 			data.custom_task_name,
 			data.custom_max_wait_days || 0,
+			data.custom_description || "",
 			data.status || "pending",
 			data.priority || "normal"
 		);