import { notFound } from "next/navigation"

import { getDemoStudentUser, getStudentHomeworkTakeData } from "@/modules/homework/data-access"
import { HomeworkTakeView } from "@/modules/homework/components/homework-take-view"
import { HomeworkReviewView } from "@/modules/homework/components/student-homework-review-view"
import { formatDate } from "@/shared/lib/utils"

export const dynamic = "force-dynamic"

export default async function StudentAssignmentTakePage({
  params,
}: {
  params: Promise<{ assignmentId: string }>
}) {
  const { assignmentId } = await params
  const student = await getDemoStudentUser()
  if (!student) return notFound()

  const data = await getStudentHomeworkTakeData(assignmentId, student.id)
  if (!data) return notFound()

  // If status is graded or submitted, use the review view
  const status = data.submission?.status
  if (status === "graded" || status === "submitted") {
    return (
      <div className="flex h-full flex-col space-y-4 p-6">
        <div className="flex flex-col gap-1">
          <h2 className="text-2xl font-bold tracking-tight">{data.assignment.title}</h2>
          <div className="text-sm text-muted-foreground">
             <span>Due: {data.assignment.dueAt ? formatDate(data.assignment.dueAt) : "-"}</span>
          </div>
        </div>

        <HomeworkReviewView initialData={data} />
      </div>
    )
  }

  return (
    <div className="flex h-full flex-col space-y-4 p-6">
      <div className="flex flex-col gap-1">
        <h2 className="text-2xl font-bold tracking-tight">{data.assignment.title}</h2>
        <div className="text-sm text-muted-foreground">
          <span>Due: {data.assignment.dueAt ? formatDate(data.assignment.dueAt) : "-"}</span>
          <span className="mx-2">•</span>
          <span>Max Attempts: {data.assignment.maxAttempts}</span>
        </div>
      </div>

      <HomeworkTakeView assignmentId={data.assignment.id} initialData={data} />
    </div>
  )
}