import useEditor, { type Phase, type Tool } from "@/store/use-editor";
import { useViewer } from "@pascal-app/viewer";
import { ItemTool } from "./item/item-tool";
import { WallTool } from "./wall/wall-tool";
import { ZoneBoundaryEditor } from "./zone/zone-boundary-editor";
import { ZoneTool } from "./zone/zone-tool";

const tools: Record<Phase, Partial<Record<Tool, React.FC>>> = {
  site: {},
  structure: {
    wall: WallTool,
    item: ItemTool,
    zone: ZoneTool,
  },
  furnish: {
    item: ItemTool,
  },
};

export const ToolManager: React.FC = () => {
  const phase = useEditor((state) => state.phase);
  const mode = useEditor((state) => state.mode);
  const tool = useEditor((state) => state.tool);
  const selectedZoneId = useViewer((state) => state.selection.zoneId);

  // Show zone boundary editor when in structure/select mode with a zone selected
  const showZoneBoundaryEditor =
    phase === "structure" && mode === "select" && selectedZoneId !== null;

  // Show build tools when in build mode
  const showBuildTool = mode === "build" && tool !== null;

  const BuildToolComponent = showBuildTool ? tools[phase]?.[tool] : null;

  return (
    <>
      {showZoneBoundaryEditor && <ZoneBoundaryEditor />}
      {BuildToolComponent && <BuildToolComponent />}
    </>
  );
};
