"use client"

import { useState } from "react"
import { useRouter } from "next/navigation"
import { MoreHorizontal, UserCheck, UserX } from "lucide-react"
import { toast } from "sonner"

import { Button } from "@/shared/components/ui/button"
import { Avatar, AvatarFallback, AvatarImage } from "@/shared/components/ui/avatar"
import { Card, CardContent, CardFooter, CardHeader } from "@/shared/components/ui/card"
import { cn, getInitials } from "@/shared/lib/utils"
import {
  DropdownMenu,
  DropdownMenuContent,
  DropdownMenuItem,
  DropdownMenuSeparator,
  DropdownMenuTrigger,
} from "@/shared/components/ui/dropdown-menu"
import {
  AlertDialog,
  AlertDialogAction,
  AlertDialogCancel,
  AlertDialogContent,
  AlertDialogDescription,
  AlertDialogFooter,
  AlertDialogHeader,
  AlertDialogTitle,
} from "@/shared/components/ui/alert-dialog"
import type { ClassStudent } from "../types"
import { setStudentEnrollmentStatusAction } from "../actions"

export function StudentsTable({ students }: { students: ClassStudent[] }) {
  const router = useRouter()
  const [workingKey, setWorkingKey] = useState<string | null>(null)
  const [removeTarget, setRemoveTarget] = useState<ClassStudent | null>(null)

  const setStatus = async (student: ClassStudent, status: "active" | "inactive") => {
    const key = `${student.classId}:${student.id}`
    setWorkingKey(key)
    try {
      const res = await setStudentEnrollmentStatusAction(student.classId, student.id, status)
      if (res.success) {
        toast.success(res.message)
        router.refresh()
      } else {
        toast.error(res.message)
      }
    } catch {
      toast.error("Failed to update status")
    } finally {
      setWorkingKey(null)
    }
  }

  return (
    <>
      <div className="grid gap-4 sm:grid-cols-2 lg:grid-cols-3 xl:grid-cols-4">
        {students.map((s) => (
          <Card key={`${s.classId}:${s.id}`} className="overflow-hidden">
            <CardHeader className="flex flex-row items-center gap-4 space-y-0 p-4 pb-2">
              <div className="relative">
                <Avatar className="h-10 w-10 border">
                  <AvatarImage src={s.image || undefined} alt={s.name} />
                  <AvatarFallback>{getInitials(s.name)}</AvatarFallback>
                </Avatar>
                <span className={cn(
                  "absolute bottom-0 right-0 h-3 w-3 rounded-full border-2 border-background",
                  s.status === "active" ? "bg-emerald-500" : "bg-muted-foreground"
                )} />
              </div>
              <div className="flex flex-col flex-1 overflow-hidden">
                <div className="flex items-start justify-between">
                  <div className="flex flex-col overflow-hidden mr-2">
                    <span className="truncate font-semibold text-sm">{s.name}</span>
                    <span className="truncate text-xs text-muted-foreground">{s.email}</span>
                  </div>
                  <div className="flex flex-col items-end gap-0.5 text-xs text-muted-foreground shrink-0">
                    <span className="text-[10px] font-medium text-foreground/80">
                      {s.className}
                    </span>
                    <span className="text-[10px]">
                      {new Date(s.joinedAt).toLocaleDateString("en-GB", {
                        day: "2-digit",
                        month: "2-digit",
                        year: "2-digit"
                      })}
                    </span>
                  </div>
                </div>
              </div>
            </CardHeader>
            <CardContent className="p-4 pt-0">
              {s.subjectScores && Object.keys(s.subjectScores).length > 0 ? (
                <div className="flex flex-col gap-2">
                  <div className="flex flex-wrap gap-1.5">
                    {Object.entries(s.subjectScores).slice(0, 4).map(([subject, score]) => (
                      <div key={subject} className="flex items-center gap-1.5 rounded-md bg-muted/50 px-2 py-1 text-xs border border-muted/50">
                        <span className="font-medium text-muted-foreground/80">{subject}</span>
                        {score !== null ? (
                          <span className={cn(
                            "font-bold",
                            score >= 90 ? "text-emerald-600" :
                            score >= 80 ? "text-primary" :
                            score >= 60 ? "text-yellow-600" : "text-destructive"
                          )}>
                            {score}
                          </span>
                        ) : (
                          <span className="text-muted-foreground">-</span>
                        )}
                      </div>
                    ))}
                    {Object.keys(s.subjectScores).length > 4 && (
                      <div className="flex items-center justify-center rounded-md bg-muted/50 px-2 py-1 text-xs text-muted-foreground font-medium border border-muted/50">
                        +{Object.keys(s.subjectScores).length - 4}
                      </div>
                    )}
                  </div>
                </div>
              ) : (
                <div className="flex items-center justify-center h-[32px] rounded-md bg-muted/20 border border-dashed border-muted">
                  <span className="text-xs text-muted-foreground/50 italic">No recent scores</span>
                </div>
              )}
            </CardContent>
            <CardFooter className="flex items-center justify-between border-t bg-muted/50 p-2">
              <Button variant="ghost" size="sm" className="h-7 text-xs text-muted-foreground" asChild>
                <a href={`mailto:${s.email}`}>Email</a>
              </Button>
              <DropdownMenu>
                <DropdownMenuTrigger asChild>
                  <Button variant="ghost" size="icon" className="h-7 w-7" disabled={workingKey !== null}>
                    <MoreHorizontal className="size-4 text-muted-foreground" />
                  </Button>
                </DropdownMenuTrigger>
                <DropdownMenuContent align="end">
                  {s.status !== "active" ? (
                    <DropdownMenuItem onClick={() => setStatus(s, "active")} disabled={workingKey !== null}>
                      <UserCheck className="mr-2 size-4" />
                      Set active
                    </DropdownMenuItem>
                  ) : (
                    <DropdownMenuItem onClick={() => setStatus(s, "inactive")} disabled={workingKey !== null}>
                      <UserX className="mr-2 size-4" />
                      Set inactive
                    </DropdownMenuItem>
                  )}
                  <DropdownMenuSeparator />
                  <DropdownMenuItem
                    onClick={() => setRemoveTarget(s)}
                    className="text-destructive focus:text-destructive"
                    disabled={s.status === "inactive" || workingKey !== null}
                  >
                    <UserX className="mr-2 size-4" />
                    Remove from class
                  </DropdownMenuItem>
                </DropdownMenuContent>
              </DropdownMenu>
            </CardFooter>
          </Card>
        ))}
      </div>

      <AlertDialog
        open={Boolean(removeTarget)}
        onOpenChange={(open) => {
          if (workingKey !== null) return
          if (!open) setRemoveTarget(null)
        }}
      >
        <AlertDialogContent>
          <AlertDialogHeader>
            <AlertDialogTitle>Remove student from class?</AlertDialogTitle>
            <AlertDialogDescription>
              {removeTarget ? (
                <>
                  This will set <span className="font-medium text-foreground">{removeTarget.name}</span> to inactive in{" "}
                  <span className="font-medium text-foreground">{removeTarget.className}</span>.
                </>
              ) : null}
            </AlertDialogDescription>
          </AlertDialogHeader>
          <AlertDialogFooter>
            <AlertDialogCancel disabled={workingKey !== null}>Cancel</AlertDialogCancel>
            <AlertDialogAction
              className="bg-destructive text-destructive-foreground hover:bg-destructive/90"
              disabled={workingKey !== null}
              onClick={() => {
                if (!removeTarget) return
                setRemoveTarget(null)
                setStatus(removeTarget, "inactive")
              }}
            >
              Remove
            </AlertDialogAction>
          </AlertDialogFooter>
        </AlertDialogContent>
      </AlertDialog>
    </>
  )
}
