feat(classes): optimize teacher dashboard ui and implement grade management

src/modules/classes/actions.ts

@@ -1,11 +1,11 @@
 "use server";
 
 import { revalidatePath } from "next/cache"
-import { and, eq, sql } from "drizzle-orm"
+import { and, eq, sql, or, inArray } from "drizzle-orm"
 import { auth } from "@/auth"
 
 import { db } from "@/shared/db"
-import { grades } from "@/shared/db/schema"
+import { grades, classes } from "@/shared/db/schema"
 import type { ActionState } from "@/shared/types/action-state"
 import {
   createAdminClass,
@@ -138,6 +138,201 @@ export async function deleteTeacherClassAction(classId: string): Promise<ActionS
   }
 }
 
+export async function createGradeClassAction(
+  prevState: ActionState<string> | undefined,
+  formData: FormData
+): Promise<ActionState<string>> {
+  const session = await auth()
+  const userId = session?.user?.id
+  if (!userId) return { success: false, message: "Unauthorized" }
+
+  const schoolName = formData.get("schoolName")
+  const schoolId = formData.get("schoolId")
+  const name = formData.get("name")
+  const grade = formData.get("grade")
+  const gradeId = formData.get("gradeId")
+  const teacherId = formData.get("teacherId")
+  const homeroom = formData.get("homeroom")
+  const room = formData.get("room")
+
+  if (typeof name !== "string" || name.trim().length === 0) {
+    return { success: false, message: "Class name is required" }
+  }
+  if (typeof gradeId !== "string" || gradeId.trim().length === 0) {
+    return { success: false, message: "Grade selection is required" }
+  }
+  if (typeof teacherId !== "string" || teacherId.trim().length === 0) {
+    return { success: false, message: "Teacher is required" }
+  }
+
+  // Verify access
+  const [managedGrade] = await db
+    .select({ id: grades.id })
+    .from(grades)
+    .where(and(eq(grades.id, gradeId), or(eq(grades.gradeHeadId, userId), eq(grades.teachingHeadId, userId))))
+    .limit(1)
+
+  if (!managedGrade) {
+    return { success: false, message: "You do not have permission to create classes for this grade" }
+  }
+
+  try {
+    const id = await createAdminClass({
+      schoolName: typeof schoolName === "string" ? schoolName : null,
+      schoolId: typeof schoolId === "string" ? schoolId : null,
+      name,
+      grade: typeof grade === "string" ? grade : "", // Should be passed from UI based on selected grade
+      gradeId,
+      teacherId,
+      homeroom: typeof homeroom === "string" ? homeroom : null,
+      room: typeof room === "string" ? room : null,
+    })
+    revalidatePath("/management/grade/classes")
+    return { success: true, message: "Class created successfully", data: id }
+  } catch (error) {
+    return { success: false, message: error instanceof Error ? error.message : "Failed to create class" }
+  }
+}
+
+export async function updateGradeClassAction(
+  classId: string,
+  prevState: ActionState | undefined,
+  formData: FormData
+): Promise<ActionState> {
+  const session = await auth()
+  const userId = session?.user?.id
+  if (!userId) return { success: false, message: "Unauthorized" }
+
+  const schoolName = formData.get("schoolName")
+  const schoolId = formData.get("schoolId")
+  const name = formData.get("name")
+  const grade = formData.get("grade")
+  const gradeId = formData.get("gradeId")
+  const teacherId = formData.get("teacherId")
+  const homeroom = formData.get("homeroom")
+  const room = formData.get("room")
+  const subjectTeachers = formData.get("subjectTeachers")
+
+  if (typeof classId !== "string" || classId.trim().length === 0) {
+    return { success: false, message: "Missing class id" }
+  }
+
+  // Verify access: Check if the class belongs to a managed grade
+  const [cls] = await db
+    .select({ gradeId: classes.gradeId })
+    .from(classes)
+    .where(eq(classes.id, classId))
+    .limit(1)
+  
+  if (!cls || !cls.gradeId) {
+    return { success: false, message: "Class not found or not linked to a grade" }
+  }
+
+  const [managedGrade] = await db
+    .select({ id: grades.id })
+    .from(grades)
+    .where(and(eq(grades.id, cls.gradeId), or(eq(grades.gradeHeadId, userId), eq(grades.teachingHeadId, userId))))
+    .limit(1)
+
+  if (!managedGrade) {
+    return { success: false, message: "You do not have permission to update this class" }
+  }
+
+  // If changing gradeId, verify target grade too
+  if (typeof gradeId === "string" && gradeId !== cls.gradeId) {
+     const [targetGrade] = await db
+      .select({ id: grades.id })
+      .from(grades)
+      .where(and(eq(grades.id, gradeId), or(eq(grades.gradeHeadId, userId), eq(grades.teachingHeadId, userId))))
+      .limit(1)
+      
+      if (!targetGrade) {
+        return { success: false, message: "You do not have permission to move class to this grade" }
+      }
+  }
+
+  try {
+    await updateAdminClass(classId, {
+      schoolName: typeof schoolName === "string" ? schoolName : undefined,
+      schoolId: typeof schoolId === "string" ? schoolId : undefined,
+      name: typeof name === "string" ? name : undefined,
+      grade: typeof grade === "string" ? grade : undefined,
+      gradeId: typeof gradeId === "string" ? gradeId : undefined,
+      teacherId: typeof teacherId === "string" ? teacherId : undefined,
+      homeroom: typeof homeroom === "string" ? homeroom : undefined,
+      room: typeof room === "string" ? room : undefined,
+    })
+
+    if (typeof subjectTeachers === "string" && subjectTeachers.trim().length > 0) {
+      const parsed = JSON.parse(subjectTeachers) as unknown
+      if (!Array.isArray(parsed)) throw new Error("Invalid subject teachers")
+
+      await setClassSubjectTeachers({
+        classId,
+        assignments: parsed.flatMap((item) => {
+          if (!item || typeof item !== "object") return []
+          const subject = (item as { subject?: unknown }).subject
+          const teacherId = (item as { teacherId?: unknown }).teacherId
+
+          if (typeof subject !== "string" || !isClassSubject(subject)) return []
+
+          if (teacherId === null || typeof teacherId === "undefined") {
+            return [{ subject, teacherId: null }]
+          }
+
+          if (typeof teacherId !== "string") return []
+          const trimmed = teacherId.trim()
+          return [{ subject, teacherId: trimmed.length > 0 ? trimmed : null }]
+        }),
+      })
+    }
+
+    revalidatePath("/management/grade/classes")
+    return { success: true, message: "Class updated successfully" }
+  } catch (error) {
+    return { success: false, message: error instanceof Error ? error.message : "Failed to update class" }
+  }
+}
+
+export async function deleteGradeClassAction(classId: string): Promise<ActionState> {
+  const session = await auth()
+  const userId = session?.user?.id
+  if (!userId) return { success: false, message: "Unauthorized" }
+
+  if (typeof classId !== "string" || classId.trim().length === 0) {
+    return { success: false, message: "Missing class id" }
+  }
+
+  // Verify access
+  const [cls] = await db
+    .select({ gradeId: classes.gradeId })
+    .from(classes)
+    .where(eq(classes.id, classId))
+    .limit(1)
+  
+  if (!cls || !cls.gradeId) {
+    return { success: false, message: "Class not found or not linked to a grade" }
+  }
+
+  const [managedGrade] = await db
+    .select({ id: grades.id })
+    .from(grades)
+    .where(and(eq(grades.id, cls.gradeId), or(eq(grades.gradeHeadId, userId), eq(grades.teachingHeadId, userId))))
+    .limit(1)
+
+  if (!managedGrade) {
+    return { success: false, message: "You do not have permission to delete this class" }
+  }
+
+  try {
+    await deleteAdminClass(classId)
+    revalidatePath("/management/grade/classes")
+    return { success: true, message: "Class deleted successfully" }
+  } catch (error) {
+    return { success: false, message: error instanceof Error ? error.message : "Failed to delete class" }
+  }
+}
+
 export async function enrollStudentByEmailAction(
   classId: string,
   prevState: ActionState | null,
@@ -171,14 +366,19 @@ export async function joinClassByInvitationCodeAction(
   }
 
   const session = await auth()
-  if (!session?.user?.id || String(session.user.role ?? "") !== "student") {
+  const role = String(session?.user?.role ?? "")
+  if (!session?.user?.id || (role !== "student" && role !== "teacher")) {
     return { success: false, message: "Unauthorized" }
   }
 
   try {
     const classId = await enrollStudentByInvitationCode(session.user.id, code)
-    revalidatePath("/student/learning/courses")
-    revalidatePath("/student/schedule")
+    if (role === "student") {
+      revalidatePath("/student/learning/courses")
+      revalidatePath("/student/schedule")
+    } else {
+      revalidatePath("/teacher/classes/my")
+    }
     revalidatePath("/profile")
     return { success: true, message: "Joined class successfully", data: { classId } }
   } catch (error) {