"use client";

import { useState } from "react";
import {
  Folder,
  FolderOpen,
  Plus,
  MoreHorizontal,
  Pencil,
  Trash2,
  FileText,
} from "lucide-react";
import { cn } from "@/lib/utils";
import {
  Dropdown,
  DropdownTrigger,
  DropdownContent,
  DropdownItem,
} from "@/components/ui/dropdown";
import type { NoteFolder } from "@/types";

interface FolderTreeProps {
  folders: NoteFolder[];
  selectedFolderId: string | null;
  onSelect: (folderId: string | null) => void;
  noteCountByFolder: Record<string, number>;
  totalNoteCount: number;
  onCreate: (name: string) => Promise<void>;
  onRename: (id: string, name: string) => Promise<void>;
  onDelete: (id: string) => Promise<void>;
  canManage?: boolean;
}

export function FolderTree({
  folders,
  selectedFolderId,
  onSelect,
  noteCountByFolder,
  totalNoteCount,
  onCreate,
  onRename,
  onDelete,
  canManage = true,
}: FolderTreeProps) {
  const [creating, setCreating] = useState(false);
  const [newName, setNewName] = useState("");
  const [renamingId, setRenamingId] = useState<string | null>(null);
  const [renameValue, setRenameValue] = useState("");

  async function handleCreate() {
    if (!canManage) return;
    if (!newName.trim()) return;
    await onCreate(newName.trim());
    setNewName("");
    setCreating(false);
  }

  async function handleRename(id: string) {
    if (!canManage) return;
    if (!renameValue.trim()) return;
    await onRename(id, renameValue.trim());
    setRenamingId(null);
  }

  return (
    <div className="space-y-1">
      <div className="flex items-center justify-between px-2 mb-2">
        <span className="text-xs font-semibold text-muted-foreground uppercase tracking-wider">
          Folders
        </span>
        <button
          onClick={() => setCreating(true)}
          disabled={!canManage}
          className="p-1 rounded-md hover:bg-accent text-muted-foreground hover:text-foreground transition-colors cursor-pointer"
        >
          <Plus className="h-3.5 w-3.5" />
        </button>
      </div>

      {/* All notes */}
      <button
        onClick={() => onSelect(null)}
        className={cn(
          "flex items-center gap-2 w-full rounded-md px-2 py-1.5 text-sm transition-colors cursor-pointer",
          selectedFolderId === null
            ? "bg-accent text-accent-foreground font-medium"
            : "text-muted-foreground hover:bg-accent/50"
        )}
      >
        <FileText className="h-4 w-4 shrink-0" />
        <span className="flex-1 text-left truncate">All notes</span>
        <span className="text-xs text-muted-foreground">{totalNoteCount}</span>
      </button>

      {/* Folder items */}
      {folders.map((folder) => (
        <div key={folder.id} className="group relative">
          {renamingId === folder.id && canManage ? (
            <div className="px-2">
              <input
                value={renameValue}
                onChange={(e) => setRenameValue(e.target.value)}
                className="w-full h-7 rounded-md border border-input bg-background px-2 text-sm outline-none focus-visible:ring-2 focus-visible:ring-ring"
                onKeyDown={(e) => {
                  if (e.key === "Enter") handleRename(folder.id);
                  if (e.key === "Escape") setRenamingId(null);
                }}
                autoFocus
              />
            </div>
          ) : (
            <button
              onClick={() => onSelect(folder.id)}
              className={cn(
                "flex items-center gap-2 w-full rounded-md px-2 py-1.5 text-sm transition-colors cursor-pointer",
                selectedFolderId === folder.id
                  ? "bg-accent text-accent-foreground font-medium"
                  : "text-muted-foreground hover:bg-accent/50"
              )}
            >
              {selectedFolderId === folder.id ? (
                <FolderOpen className="h-4 w-4 shrink-0" />
              ) : (
                <Folder className="h-4 w-4 shrink-0" />
              )}
              <span className="flex-1 text-left truncate">{folder.name}</span>
              <span className="text-xs text-muted-foreground">
                {noteCountByFolder[folder.id] || 0}
              </span>
            </button>
          )}

          {/* Folder actions */}
          {renamingId !== folder.id && canManage && (
            <div className="absolute right-1 top-1/2 -translate-y-1/2 opacity-0 group-hover:opacity-100 transition-opacity">
              <Dropdown>
                <DropdownTrigger className="p-1 rounded hover:bg-accent/80 transition-colors">
                  <MoreHorizontal className="h-3.5 w-3.5 text-muted-foreground" />
                </DropdownTrigger>
                <DropdownContent>
                  <DropdownItem
                    onClick={() => {
                      setRenamingId(folder.id);
                      setRenameValue(folder.name);
                    }}
                  >
                    <Pencil className="h-3.5 w-3.5" />
                    Rename
                  </DropdownItem>
                  <DropdownItem onClick={() => onDelete(folder.id)} destructive>
                    <Trash2 className="h-3.5 w-3.5" />
                    Delete
                  </DropdownItem>
                </DropdownContent>
              </Dropdown>
            </div>
          )}
        </div>
      ))}

      {/* Create input */}
      {creating && canManage && (
        <div className="px-2">
          <input
            value={newName}
            onChange={(e) => setNewName(e.target.value)}
            placeholder="Folder name"
            className="w-full h-7 rounded-md border border-input bg-background px-2 text-sm outline-none focus-visible:ring-2 focus-visible:ring-ring"
            onKeyDown={(e) => {
              if (e.key === "Enter") handleCreate();
              if (e.key === "Escape") {
                setCreating(false);
                setNewName("");
              }
            }}
            autoFocus
          />
        </div>
      )}
    </div>
  );
}