'use client'

import { useTranslations } from 'next-intl'
import { useRouter } from 'next/navigation'
import { signOut } from '@/lib/auth-client'
import { LogOut, User, Settings } from 'lucide-react'
import { Avatar, AvatarFallback } from '@/components/ui/avatar'
import { Button } from '@/components/ui/button'
import {
  DropdownMenu,
  DropdownMenuContent,
  DropdownMenuItem,
  DropdownMenuLabel,
  DropdownMenuSeparator,
  DropdownMenuTrigger,
} from '@/components/ui/dropdown-menu'
import { Badge } from '@/components/ui/badge'

interface UserMenuProps {
  user: {
    name: string
    email: string
    role: string
  }
}

const roleColors: Record<string, string> = {
  ADMIN: 'destructive',
  DOCTOR: 'default',
  RECEPTIONIST: 'secondary',
  ACCOUNTANT: 'outline',
  LAB_TECH: 'secondary',
}

export function UserMenu({ user }: UserMenuProps) {
  const t = useTranslations('layout')
  const tRole = useTranslations('staff.role')
  const router = useRouter()
  const initials = user.name
    .split(' ')
    .map((n) => n[0])
    .join('')
    .toUpperCase()
    .slice(0, 2)

  return (
    <DropdownMenu>
      <DropdownMenuTrigger asChild>
        <Button variant="ghost" className="relative h-10 w-10 rounded-full">
          <Avatar className="h-10 w-10">
            <AvatarFallback className="bg-primary text-primary-foreground">
              {initials}
            </AvatarFallback>
          </Avatar>
        </Button>
      </DropdownMenuTrigger>
      <DropdownMenuContent className="w-56" align="end" forceMount>
        <DropdownMenuLabel className="font-normal">
          <div className="flex flex-col space-y-2">
            <p className="text-sm font-medium leading-none">{user.name}</p>
            <p className="text-xs leading-none text-muted-foreground">{user.email}</p>
            <Badge
              variant={roleColors[user.role] as 'default' | 'secondary' | 'destructive' | 'outline'}
              className="w-fit text-xs"
            >
              {tRole.has(user.role) ? tRole(user.role) : user.role}
            </Badge>
          </div>
        </DropdownMenuLabel>
        <DropdownMenuSeparator />
        <DropdownMenuItem onClick={() => router.push('/settings')}>
          <User className="mr-2 h-4 w-4" />
          <span>{t('profile')}</span>
        </DropdownMenuItem>
        <DropdownMenuItem onClick={() => router.push('/settings')}>
          <Settings className="mr-2 h-4 w-4" />
          <span>{t('settings')}</span>
        </DropdownMenuItem>
        <DropdownMenuSeparator />
        <DropdownMenuItem
          className="text-destructive focus:text-destructive"
          onClick={() => signOut({ fetchOptions: { onSuccess: () => router.push('/login') } })}
        >
          <LogOut className="mr-2 h-4 w-4" />
          <span>{t('logOut')}</span>
        </DropdownMenuItem>
      </DropdownMenuContent>
    </DropdownMenu>
  )
}