import Button from '@/components/ui/Button'
import { SiteInfo, webuiPrefix } from '@/lib/constants'
import AppSettings from '@/components/AppSettings'
import { TabsList, TabsTrigger } from '@/components/ui/Tabs'
import { useSettingsStore } from '@/stores/settings'
import { useAuthStore } from '@/stores/state'
import { cn } from '@/lib/utils'
import { useTranslation } from 'react-i18next'
import { navigationService } from '@/services/navigation'
import { ZapIcon, GithubIcon, LogOutIcon } from 'lucide-react'
import { Tooltip, TooltipContent, TooltipProvider, TooltipTrigger } from '@/components/ui/Tooltip'
import { WorkspaceSelector } from '@/components/WorkspaceSelector'

interface NavigationTabProps {
  value: string
  currentTab: string
  children: React.ReactNode
}

function NavigationTab({ value, currentTab, children }: NavigationTabProps) {
  return (
    <TabsTrigger
      value={value}
      className={cn(
        'cursor-pointer px-2 py-1 transition-all',
        currentTab === value ? '!bg-emerald-400 !text-zinc-50' : 'hover:bg-background/60'
      )}
    >
      {children}
    </TabsTrigger>
  )
}

function TabsNavigation() {
  const currentTab = useSettingsStore.use.currentTab()
  const { t } = useTranslation()

  return (
    <div className="flex h-8 self-center">
      <TabsList className="h-full gap-2">
        <NavigationTab value="documents" currentTab={currentTab}>
          {t('header.documents')}
        </NavigationTab>
        <NavigationTab value="knowledge-graph" currentTab={currentTab}>
          {t('header.knowledgeGraph')}
        </NavigationTab>
        <NavigationTab value="retrieval" currentTab={currentTab}>
          {t('header.retrieval')}
        </NavigationTab>
        <NavigationTab value="api" currentTab={currentTab}>
          {t('header.api')}
        </NavigationTab>
      </TabsList>
    </div>
  )
}

export default function SiteHeader() {
  const { t } = useTranslation()
  const { isGuestMode, coreVersion, apiVersion, username, webuiTitle, webuiDescription } = useAuthStore()

  const versionDisplay = (coreVersion && apiVersion)
    ? `${coreVersion}/${apiVersion}`
    : null;

  const handleLogout = () => {
    navigationService.navigateToLogin();
  }

  return (
    <header className="border-border/40 bg-background/95 supports-[backdrop-filter]:bg-background/60 sticky top-0 z-50 flex flex-col w-full border-b px-4 backdrop-blur">
      {/* First row: Title and other header items */}
      <div className="flex h-10 items-center justify-between">
        <div className="min-w-[200px] w-auto flex items-center">
          <a href={webuiPrefix} className="flex items-center gap-2">
            <ZapIcon className="size-4 text-emerald-400" aria-hidden="true" />
            <span className="font-bold md:inline-block">{SiteInfo.name}</span>
          </a>
          {webuiTitle && (
            <div className="flex items-center">
              <span className="mx-1 text-xs text-gray-500 dark:text-gray-400">|</span>
              <TooltipProvider>
                <Tooltip>
                  <TooltipTrigger asChild>
                    <span className="font-medium text-sm cursor-default">
                      {webuiTitle}
                    </span>
                  </TooltipTrigger>
                  {webuiDescription && (
                    <TooltipContent side="bottom">
                      {webuiDescription}
                    </TooltipContent>
                  )}
                </Tooltip>
              </TooltipProvider>
            </div>
          )}
        </div>

        <div className="flex h-10 flex-1 items-center justify-center">
          <TabsNavigation />
        </div>

        <nav className="w-[200px] flex items-center justify-end">
          <div className="flex items-center gap-2">
            {versionDisplay && (
              <span className="text-xs text-gray-500 dark:text-gray-400 mr-1">
                v{versionDisplay}
              </span>
            )}
            <AppSettings />
            <Button
              variant="ghost"
              size="icon"
              side="bottom"
              tooltip={isGuestMode ? t('header.login') : `${t('header.logout')} (${username})`}
              onClick={handleLogout}
            >
              <LogOutIcon className="size-4" aria-hidden="true" />
            </Button>
          </div>
        </nav>
      </div>

      {/* Second row: Workspace selector */}
      <div className="flex items-center gap-2 py-1 border-t border-border/40">
        <WorkspaceSelector />
      </div>
    </header>
  )
}
