import React, { useState, useRef } from "react";
import { ActionIcon, Stack, Tooltip, Divider } from "@mantine/core";
import MenuBookIcon from "@mui/icons-material/MenuBookRounded";
import AppsIcon from "@mui/icons-material/AppsRounded";
import SettingsIcon from "@mui/icons-material/SettingsRounded";
import AutoAwesomeIcon from "@mui/icons-material/AutoAwesomeRounded";
import FolderIcon from "@mui/icons-material/FolderRounded";
import PersonIcon from "@mui/icons-material/PersonRounded";
import NotificationsIcon from "@mui/icons-material/NotificationsRounded";
import { useRainbowThemeContext } from "./RainbowThemeProvider";
import rainbowStyles from '../../styles/rainbow.module.css';
import AppConfigModal from './AppConfigModal';
import { useIsOverflowing } from '../../hooks/useIsOverflowing';
import './QuickAccessBar.css';

interface QuickAccessBarProps {
  onToolsClick: () => void;
  onReaderToggle: () => void;
  selectedToolKey?: string;
  toolRegistry: any;
  leftPanelView: 'toolPicker' | 'toolContent';
  readerMode: boolean;
}

interface ButtonConfig {
  id: string;
  name: string;
  icon: React.ReactNode;
  tooltip: string;
  isRound?: boolean;
  size?: 'sm' | 'md' | 'lg' | 'xl';
  onClick: () => void;
}

function NavHeader({ 
  activeButton, 
  setActiveButton, 
  onReaderToggle, 
  onToolsClick 
}: {
  activeButton: string;
  setActiveButton: (id: string) => void;
  onReaderToggle: () => void;
  onToolsClick: () => void;
}) {
  return (
    <>
      <div className="nav-header">
        <Tooltip label="User Profile" position="right">
          <ActionIcon
            size="md"
            variant="subtle"
            className="action-icon-style"
          >
            <PersonIcon sx={{ fontSize: "1rem" }} />
          </ActionIcon>
        </Tooltip>
        <Tooltip label="Notifications" position="right">
          <ActionIcon
            size="md"
            variant="subtle"
            className="action-icon-style"
          >
            <NotificationsIcon sx={{ fontSize: "1rem" }} />
          </ActionIcon>
        </Tooltip>
      </div>
      {/* Divider after top icons */}
      <Divider 
        size="xs" 
        className="nav-header-divider"
      />
      {/* All Tools button below divider */}
      <Tooltip label="View all available tools" position="right">
        <div className="flex flex-col items-center gap-1 mt-4 mb-2">
          <ActionIcon
            size="lg"
            variant="subtle"
            onClick={() => {
              setActiveButton('tools');
              onReaderToggle();
              onToolsClick();
            }}
            style={{
              backgroundColor: activeButton === 'tools' ? 'var(--icon-tools-bg)' : 'var(--icon-inactive-bg)',
              color: activeButton === 'tools' ? 'var(--icon-tools-color)' : 'var(--icon-inactive-color)',
              border: 'none',
              borderRadius: '8px',
            }}
            className={activeButton === 'tools' ? 'activeIconScale' : ''}
          >
            <span className="iconContainer">
              <AppsIcon sx={{ fontSize: "1.75rem" }} />
            </span>
          </ActionIcon>
          <span className={`all-tools-text ${activeButton === 'tools' ? 'active' : 'inactive'}`}>
            All Tools
          </span>
        </div>
      </Tooltip>
    </>
  );
}

const QuickAccessBar = ({
  onToolsClick,
  onReaderToggle,
  selectedToolKey,
  toolRegistry,
  leftPanelView,
  readerMode,
}: QuickAccessBarProps) => {
  const { isRainbowMode } = useRainbowThemeContext();
  const [configModalOpen, setConfigModalOpen] = useState(false);
  const [activeButton, setActiveButton] = useState<string>('tools');
  const scrollableRef = useRef<HTMLDivElement>(null);
  const isOverflow = useIsOverflowing(scrollableRef);

  const buttonConfigs: ButtonConfig[] = [
    {
      id: 'read',
      name: 'Read',
      icon: <MenuBookIcon sx={{ fontSize: "1.5rem" }} />,
      tooltip: 'Read documents',
      size: 'lg',
      isRound: false,
      onClick: () => {
        setActiveButton('read');
        onReaderToggle();
      }
    },
    {
      id: 'sign',
      name: 'Sign',
      icon: 
      <span className="material-symbols-rounded font-size-20">
        signature
      </span>,
      tooltip: 'Sign your document',
      size: 'lg',
      isRound: false,
      onClick: () => setActiveButton('sign')
    },
    {
      id: 'automate',
      name: 'Automate',
      icon: <AutoAwesomeIcon sx={{ fontSize: "1.5rem" }} />,
      tooltip: 'Automate workflows',
      size: 'lg',
      isRound: false,
      onClick: () => setActiveButton('automate')
    },
    {
      id: 'files',
      name: 'Files',
      icon: <FolderIcon sx={{ fontSize: "1.5rem" }} />,
      tooltip: 'Manage files',
      isRound: true,
      size: 'lg',
      onClick: () => setActiveButton('files')
    },
    {
      id: 'activity',
      name: 'Activity',
      icon: 
      <span className="material-symbols-rounded font-size-20">
      vital_signs
      </span>,
      tooltip: 'View activity and analytics',
      isRound: true,
      size: 'lg',
      onClick: () => setActiveButton('activity')
    },
    {
      id: 'config',
      name: 'Config',
      icon: <SettingsIcon sx={{ fontSize: "1rem" }} />,
      tooltip: 'Configure settings',
      size: 'lg',
      onClick: () => {
        setConfigModalOpen(true);
      }
    }
  ];

  const CIRCULAR_BORDER_RADIUS = '50%';
  const ROUND_BORDER_RADIUS = '8px';

  const getBorderRadius = (config: ButtonConfig): string => {
    return config.isRound ? CIRCULAR_BORDER_RADIUS : ROUND_BORDER_RADIUS;
  };

  const getButtonStyle = (config: ButtonConfig) => {
    const isActive = activeButton === config.id;
    
    if (isActive) {
      return {
        backgroundColor: `var(--icon-${config.id}-bg)`,
        color: `var(--icon-${config.id}-color)`,
        border: 'none',
        borderRadius: getBorderRadius(config),
      };
    }
    
    // Inactive state - use consistent inactive colors
    return {
      backgroundColor: 'var(--icon-inactive-bg)',
      color: 'var(--icon-inactive-color)',
      border: 'none',
      borderRadius: getBorderRadius(config),
    };
  };

  return (
    <div
      className={`h-screen flex flex-col w-20 quick-access-bar-main ${isRainbowMode ? 'rainbow-mode' : ''}`}
    >
      {/* Fixed header outside scrollable area */}
      <div className="quick-access-header">
        <NavHeader 
          activeButton={activeButton} 
          setActiveButton={setActiveButton} 
          onReaderToggle={onReaderToggle} 
          onToolsClick={onToolsClick} 
        />
      </div>

      {/* Conditional divider when overflowing */}
      {isOverflow && (
        <Divider 
          size="xs" 
          className="overflow-divider"
        />
      )}

      {/* Scrollable content area */}
      <div
        ref={scrollableRef}
        className="quick-access-bar flex-1"
        onWheel={(e) => {
          // Prevent the wheel event from bubbling up to parent containers
          e.stopPropagation();
        }}
      >
        <div className="scrollable-content">
          {/* Top section with main buttons */}
          <Stack gap="lg" align="center">
            {buttonConfigs.slice(0, -1).map((config, index) => (
              <React.Fragment key={config.id}>
                <Tooltip label={config.tooltip} position="right">
                  <div className="flex flex-col items-center gap-1" style={{ marginTop: index === 0 ? '0.5rem' : "0rem" }}>
                    <ActionIcon
                      size={config.size || 'xl'}
                      variant="subtle"
                      onClick={config.onClick}
                      style={getButtonStyle(config)}
                      className={activeButton === config.id ? 'activeIconScale' : ''}
                    >
                      <span className="iconContainer">
                        {config.icon}
                      </span>
                    </ActionIcon>
                    <span className={`button-text ${activeButton === config.id ? 'active' : 'inactive'}`}>
                      {config.name}
                    </span>
                  </div>
                </Tooltip>
                
                {/* Add divider after Automate button (index 2) */}
                {index === 2 && (
                    <Divider 
                      size="xs" 
                      className="content-divider"
                    />
                )}
              </React.Fragment>
            ))}
          </Stack>
          
          {/* Spacer to push Config button to bottom */}
          <div className="spacer" />
          
          {/* Config button at the bottom */}
          <Tooltip label="Configure settings" position="right">
            <div className="flex flex-col items-center gap-1">
              <ActionIcon
                size="lg"
                variant="subtle"
                onClick={() => {
                  setConfigModalOpen(true);
                }}
                style={{
                  backgroundColor: 'var(--icon-inactive-bg)',
                  color: 'var(--icon-inactive-color)',
                  border: 'none',
                  borderRadius: '8px',
                }}
              >
                <span className="iconContainer">
                  <SettingsIcon sx={{ fontSize: "1rem" }} />
                </span>
              </ActionIcon>
              <span className="config-button-text">
                Config
              </span>
            </div>
          </Tooltip>
        </div>
      </div>

      <AppConfigModal
        opened={configModalOpen}
        onClose={() => setConfigModalOpen(false)}
      />
    </div>
  );
};

export default QuickAccessBar;