import {useState, useMemo, useRef} from "react";
import {
  Button,
  Box,
  Text,
  Flex,
  Grid,
  Dialog,
  Tabs,
  Badge,
  Menu,
  HStack,
  useRecipe,
  useSlotRecipe
} from "@chakra-ui/react";
import {Settings, Plus, LayoutGrid, Move, Save, X} from 'lucide-react';
import {useDashboardStore} from "../../store/useDashboardStore.jsx";
import {WIDGET_REGISTRY} from "../../constants/widgetRegistry.jsx";

export function SettingsButton() {
  // 1. ☁Store connections
  const isEditing = useDashboardStore(state => state.isEditing);
  const enterEditMode = useDashboardStore(state => state.enterEditMode);
  const saveEdit = useDashboardStore(state => state.saveEdit);
  const cancelEdit = useDashboardStore(state => state.cancelEdit);
  const addWidget = useDashboardStore(state => state.addWidget);

  const recipe = useRecipe({key: "glassTab"});
  const tabStyles = recipe();

  // Pull the glass container styles for the dropdown
  const panelRecipe = useSlotRecipe({key: "glassPanel"});
  const panelStyles = panelRecipe({layout: "dropdown"});

  const [isLibraryOpen, setIsLibraryOpen] = useState(false);

  // Smart Add Handler
  const handleAddWidget = (widgetId, sizeKey) => {
    enterEditMode();
    addWidget(widgetId, sizeKey);
  };

  const categorizedWidgets = useMemo(() => {
    return Object.entries(WIDGET_REGISTRY).reduce((acc, [id, widget]) => {
      const cat = widget.category || 'General';
      if (!acc[cat]) acc[cat] = [];
      acc[cat].push({id, ...widget});
      return acc;
    }, {});
  }, []);

  const categories = Object.keys(categorizedWidgets);

  // use ref to capture html node for placing menu accordingly
  const menuRef = useRef(null);
  const getAnchorRect = () => {
    if (!menuRef.current) return null;
    return menuRef.current.getBoundingClientRect();
  };

  return (
    <>
      {isEditing ? (
        // 📝 EDIT MODE: The 3-Pill Layout
        <HStack gap={2}>
          {/* 1. The Add Widget Button */}
          <Box
            as="button"
            css={tabStyles}
            data-state="open"
            onClick={() => setIsLibraryOpen(true)}
            color={{ base: "blue.500", _dark: "blue.300" }}
            _hover={{ bg: "rgba(66, 153, 225, 0.1)" }}
          >
            <LayoutGrid size={20} style={{ flexShrink: 0 }} />
            <Box as="span">Widget Library</Box>
          </Box>

          {/* 2. The Save Button */}
          <Box
            as="button"
            css={tabStyles}
            data-state="open"
            onClick={saveEdit}
            color={{ base: "green.600", _dark: "green.400" }}
            _hover={{ bg: "rgba(72, 187, 120, 0.1)" }}
          >
            <Save size={20} style={{ flexShrink: 0 }} />
            <Box as="span">Save Layout</Box>
          </Box>

          {/* 3. The Discard Button */}
          <Box
            as="button"
            css={tabStyles}
            data-state="open"
            onClick={cancelEdit}
            color={{ base: "red.500", _dark: "red.400" }}
            _hover={{ bg: "rgba(245, 101, 101, 0.1)" }}
          >
            <X size={20} style={{ flexShrink: 0 }} />
            <Box as="span">Discard</Box>
          </Box>
        </HStack>
      ) : (
        // 👀 VIEW MODE: The Static Dropdown Menu
        <Menu.Root positioning={{ getAnchorRect }}>
          <Menu.Trigger asChild>
            <Box ref={menuRef} as="button" css={tabStyles}>
              {/* size increased to 24 to match tabs. flexShrink: 0 is CRITICAL */}
              <Settings size={24} style={{ flexShrink: 0 }} />
              <Box as="span">Dashboard Settings</Box>
            </Box>
          </Menu.Trigger>

          <Menu.Positioner>
            <Menu.Content css={panelStyles.root} zIndex="dropdown" minW="200px">
              <Menu.Item
                onClick={enterEditMode}
                cursor="pointer"
                _hover={{ WebkitTextStroke: "0.5px currentColor" }}
              >
                <Move size={16} style={{ marginRight: '8px' }} /> Edit Layout
              </Menu.Item>

              <Menu.Separator my={1} borderColor="glassSeperator" />

              <Menu.Item
                onClick={() => setIsLibraryOpen(true)}
                cursor="pointer"
                _hover={{ WebkitTextStroke: "0.5px currentColor" }}
              >
                <LayoutGrid size={16} style={{ marginRight: '8px' }} /> Widget Library
              </Menu.Item>
            </Menu.Content>
          </Menu.Positioner>
        </Menu.Root>
      )}

      {/* THE WIDGET LIBRARY MODAL (Controlled by isLibraryOpen) */}
      <Dialog.Root
        open={isLibraryOpen}
        onOpenChange={(e) => setIsLibraryOpen(e.open)}
        size="xl"
        placement="center"
        closeOnInteractOutside={true}
        closeOnEscapeKeyDown={true}
      >
        <Dialog.Backdrop bg="blackAlpha.600"/>
        <Dialog.Positioner>
          <Dialog.Content bg="gray.50" borderRadius="xl" overflow="hidden" boxShadow="xl" maxW="800px">
            <Dialog.Header bg="gray.800" color="white" py={4}>
              <Dialog.Title fontSize="xl" fontWeight="bold">Add Dashboard Widgets</Dialog.Title>
            </Dialog.Header>

            <Dialog.Body p={0}>
              <Tabs.Root defaultValue={categories[0]} variant="enclosed" colorScheme="blue">
                <Tabs.List bg="white" px={4} pt={4} borderBottom="1px solid" borderColor="gray.200">
                  {categories.map(cat => (
                    <Tabs.Trigger key={cat} value={cat} pb={2}>{cat}</Tabs.Trigger>
                  ))}
                </Tabs.List>

                {categories.map(cat => (
                  <Tabs.Content key={cat} value={cat} p={6}>
                    <Grid templateColumns="repeat(auto-fill, minmax(300px, 1fr))" gap={4}>
                      {categorizedWidgets[cat].map(widget => (
                        <Flex key={widget.id} direction="column" bg="white" p={4} borderRadius="md" border="1px solid"
                              borderColor="gray.200" boxShadow="sm">
                          <Text fontWeight="bold" fontSize="md" mb={1}>{widget.name}</Text>
                          <Text fontSize="sm" color="gray.500" mb={4} flex="1">{widget.description}</Text>

                          <Box borderTop="1px solid" borderColor="gray.100" pt={3}>
                            <Text fontSize="xs" fontWeight="bold" color="gray.400" textTransform="uppercase" mb={2}>Add
                              to Grid:</Text>
                            <HStack gap={2} flexWrap="wrap">
                              {Object.keys(widget.allowedSizes).map(sizeKey => (
                                <Badge
                                  key={sizeKey}
                                  as="button"
                                  onClick={() => handleAddWidget(widget.id, sizeKey)}
                                  colorScheme="gray"
                                  variant="subtle"
                                  px={3}
                                  py={1}
                                  borderRadius="full"
                                  cursor="pointer"
                                  _hover={{bg: "blue.50", color: "blue.600", transform: "scale(1.05)"}}
                                  transition="all 0.2s"
                                >
                                  <Plus size={12} style={{display: 'inline', marginRight: '4px'}}/>
                                  {sizeKey.charAt(0).toUpperCase() + sizeKey.slice(1)}
                                </Badge>
                              ))}
                            </HStack>
                          </Box>
                        </Flex>
                      ))}
                    </Grid>
                  </Tabs.Content>
                ))}
              </Tabs.Root>
            </Dialog.Body>
            <Dialog.CloseTrigger position="absolute" top={3} right={3} color="white"/>
          </Dialog.Content>
        </Dialog.Positioner>
      </Dialog.Root>
    </>
  );
}