import { notFound } from "next/navigation"
import { getTranslations } from "next-intl/server"

import { getStudentHomeworkTakeData } from "@/modules/homework/data-access"
import { getCurrentStudentUser } from "@/modules/users/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 getCurrentStudentUser()
  if (!student) return notFound()

  const t = await getTranslations("student")

  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="space-y-8">
        <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>{t("assignment.due", { date: data.assignment.dueAt ? formatDate(data.assignment.dueAt) : "-" })}</span>
          </div>
        </div>

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

  return (
    <div className="space-y-4">
      <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>{t("assignment.due", { date: data.assignment.dueAt ? formatDate(data.assignment.dueAt) : "-" })}</span>
          <span className="mx-2" aria-hidden="true">•</span>
          <span>{t("assignment.maxAttempts", { count: data.assignment.maxAttempts })}</span>
        </div>
      </div>

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