TreeView

"use client";

import { useState } from "react";
import { Badge, TreeView, type PimoluTreeItem } from "@pimolu/react";

const files: PimoluTreeItem[] = [
  {
    id: "src",
    label: "src",
    children: [
      {
        id: "components",
        label: "components",
        children: [
          { id: "button", label: "button.tsx" },
          { id: "card", label: "card.tsx" },
          { id: "tabs", label: "tabs.tsx" },
        ],
      },
      { id: "app", label: "App.tsx" },
      { id: "styles", label: "styles.css" },
    ],
  },
  {
    id: "tests",
    label: "tests",
    children: [
      { id: "button-test", label: "button.test.tsx" },
      { id: "card-test", label: "card.test.tsx" },
    ],
  },
  { id: "readme", label: "README.md" },
  { id: "package", label: "package.json" },
];

function ControlledTree() {
  const [selected, setSelected] = useState("tabs");
  const [expanded, setExpanded] = useState<string[]>([
    "src",
    "components",
  ]);
  return (
    <div className="grid gap-5 min-w-0">
      <div className="flex flex-wrap items-center gap-3 min-w-0">
        <Badge variant="primary">
          {selected ?? "none"}
        </Badge>
        <span className="text-sm text-muted leading-relaxed">
          expanded: {expanded.join(", ") || "none"}
        </span>
      </div>
      <TreeView
        items={files}
        aria-label="Controlled file tree"
        selectedId={selected}
        onSelectedChange={(id) => setSelected(id ?? "")}
        expandedIds={expanded}
        onExpandedChange={setExpanded}
      />
    </div>
  );
}

export function DemoExample() {
  return (<ControlledTree />);
}