import { type AnyRoute, useMatchRoute, useNavigate } from "@tanstack/react-router";
import { useTranslation } from "react-i18next";
import { AnimatePresence } from "motion/react";

import GlassSurface from "@components/surface/GlassSurface";

import classes from "./Navigation.module.css";
import ShopRoute from "@/routes/shop";
import ApiaryRoute from "@/routes/apiary";
import GameRoute from "@/routes/game";
import CashdeskRoute from "@/routes/cashdesk";
import RouletteRoute from "@/routes/roulette";
import TasksRoute from "@/routes/tasks";
import EarningsRoute from "@/routes/earnings";
import { useCallback, useEffect, useRef, useState } from "react";

import shopIcon from "./assets/shop.svg";
import apiaryIcon from "./assets/apiary.svg";
import gameIcon from "./assets/game.svg";
import cashdeskIcon from "./assets/cashdesk.svg";
import menuIcon from "./assets/menu.svg";
import rouletteIcon from "./assets/roulette.svg";
import tasksIcon from "./assets/tasks.svg";
import earningsIcon from "./assets/earnings.svg";
import tg from "@/tg";

const ANIMATION_DURATION = 0.2;
const SPRING_ANIMATION = {
  type: "spring",
  stiffness: 400,
  damping: 20,
  duration: ANIMATION_DURATION,
} as const;

const BAR_HEIGHT = 64;
const ACTIVE_BAR_HEIGHT = 74;
const OFFSCREEN_BAR_OFFSET = 20;

const NAV_ITEMS = [
  { key: "nav.shop", route: ShopRoute, icon: shopIcon },
  { key: "nav.apiary", route: ApiaryRoute, icon: apiaryIcon },
  { key: "nav.game", route: GameRoute, icon: gameIcon },
  { key: "nav.cashdesk", route: CashdeskRoute, icon: cashdeskIcon },
  { key: "nav.menu", isMenu: true, icon: menuIcon },
];
const HORIZONTAL_ENTRANCE_DELAYS = [0.2, 0.1, 0, 0.1, 0.2];

const MENU_ITEMS = [
  { key: "nav.roulette", route: RouletteRoute, icon: rouletteIcon, delay: 0.1 },
  { key: "nav.tasks", route: TasksRoute, icon: tasksIcon, delay: 0.05 },
  { key: "nav.earnings", route: EarningsRoute, icon: earningsIcon, delay: 0 },
];

type BarProps = {
  labelKey: string;
  icon: string;
  active: boolean;
  entranceDelay: number;
  onClick: () => void;
};

function NavBar({ labelKey, icon, active, entranceDelay, onClick }: BarProps) {
  const { t } = useTranslation();

  const isInitial = useRef(true);
  useEffect(() => {
    isInitial.current = false;
  }, []);

  const height = active ? ACTIVE_BAR_HEIGHT : BAR_HEIGHT;
  return (
    <GlassSurface
      className={[classes.bar, active && classes.active]}
      variants={{
        hidden: {
          translateY: ACTIVE_BAR_HEIGHT + OFFSCREEN_BAR_OFFSET,
          height,
        },
        visible: {
          translateY: OFFSCREEN_BAR_OFFSET,
          height: height + OFFSCREEN_BAR_OFFSET,
          transition: { ...SPRING_ANIMATION, delay: entranceDelay },
        },
        ready: {
          translateY: OFFSCREEN_BAR_OFFSET,
          height: height + OFFSCREEN_BAR_OFFSET,
          transition: SPRING_ANIMATION,
        },
      }}
      initial="hidden"
      animate={isInitial.current ? "visible" : "ready"}
      whileTap={{ scale: 0.95 }}
      onClick={onClick}
    >
      <div className={classes.safeContent}>
        <img src={icon} className={classes.icon} alt="" />
        <span className={classes.label}>{t(labelKey)}</span>
      </div>
    </GlassSurface>
  );
}

type MenuBarProps = {
  labelKey: string;
  icon: string;
  active: boolean;
  delay: number;
  onClick: () => void;
};

const MENU_BAR_WIDTH = 94;
const ACTIVE_MENU_BAR_WIDTH = 104;
const OFFSCREEN_MENU_BAR_OFFSET = 20;

function MenuBar({ labelKey, icon, delay, active, onClick }: MenuBarProps) {
  const { t } = useTranslation();

  const [isReady, setIsReady] = useState(false);

  const width = active ? ACTIVE_MENU_BAR_WIDTH : MENU_BAR_WIDTH;
  return (
    <GlassSurface
      className={[classes.menuBar, active && classes.active]}
      variants={{
        hidden: {
          translateX: width + OFFSCREEN_MENU_BAR_OFFSET,
          width,
        },
        visible: {
          translateX: OFFSCREEN_MENU_BAR_OFFSET,
          transition: { ...SPRING_ANIMATION, delay: delay },
          width,
        },
        ready: {
          translateX: OFFSCREEN_MENU_BAR_OFFSET,
          transition: SPRING_ANIMATION,
          width,
        },
        exit: {
          translateX: width + OFFSCREEN_MENU_BAR_OFFSET + 10,
          transition: { ...SPRING_ANIMATION, delay: delay },
          width,
        },
      }}
      onAnimationComplete={(variant) => variant === "visible" && setIsReady(true)}
      initial="hidden"
      animate={isReady ? "ready" : "visible"}
      exit="exit"
      whileTap={{ scale: 0.95 }}
      onClick={onClick}
    >
      <div className={classes.menuBarContent}>
        <img src={icon} className={classes.icon} alt="" />
        <span className={classes.label}>{t(labelKey)}</span>
      </div>
    </GlassSurface>
  );
}

export default function Navigation() {
  const matchRoute = useMatchRoute();
  const navigate = useNavigate();
  const [menuOpen, setMenuOpen] = useState<number>(0);
  const navRef = useRef<HTMLDivElement>(null);

  const handleOutsideClick = useCallback((e: MouseEvent) => {
    if (navRef.current && !navRef.current.contains(e.target as Node)) {
      setMenuOpen(0);
    }
  }, []);

  useEffect(() => {
    if (menuOpen) {
      document.addEventListener("click", handleOutsideClick);
      return () => document.removeEventListener("click", handleOutsideClick);
    }
  }, [menuOpen, handleOutsideClick]);

  const navigateRoute = async (route: AnyRoute, wait = false) => {
    tg.hapticFeedback.click();
    const redirection = navigate({ to: route.to });
    if (wait) {
      await new Promise((resolve) => setTimeout(resolve, 1000 * ANIMATION_DURATION - 100));
      await redirection;
    }
    setMenuOpen(0);
  };

  return (
    <div ref={navRef}>
      <div className={classes.menuOverlay}>
        <AnimatePresence>
          {menuOpen &&
            MENU_ITEMS.map((item) => (
              <MenuBar
                key={`${item.key} ${menuOpen}`}
                active={Boolean(matchRoute({ to: item.route.to }))}
                labelKey={item.key}
                icon={item.icon}
                delay={item.delay}
                onClick={() => navigateRoute(item.route, true)}
              />
            ))}
        </AnimatePresence>
      </div>
      <nav className={classes.navigation}>
        {NAV_ITEMS.map((item, index) => {
          const active = Boolean(item.isMenu ? menuOpen : matchRoute({ to: item.route!.to }));
          return (
            <NavBar
              key={item.key}
              labelKey={item.key}
              icon={item.icon}
              active={active}
              entranceDelay={HORIZONTAL_ENTRANCE_DELAYS[index]}
              onClick={
                item.isMenu
                  ? () => setMenuOpen((v) => (v ? 0 : Math.random()))
                  : active
                    ? () => {}
                    : () => navigateRoute(item.route!)
              }
            />
          );
        })}
      </nav>
    </div>
  );
}