import { create } from "zustand";
import { persist, createJSONStorage } from "zustand/middleware";
import AsyncStorage from "@react-native-async-storage/async-storage";
import {
  buildExplorerCheckoutKey,
  coerceExplorerTabForCheckout,
  isExplorerTab,
  resolveExplorerTabForCheckout,
  type ExplorerTab,
} from "./explorer-tab-memory";
import { isWeb } from "@/constants/platform";
export type { ExplorerTab } from "./explorer-tab-memory";

/**
 * Mobile panel state machine.
 *
 * On mobile, exactly one panel can be visible at a time:
 * - 'agent': Main agent view (no overlay panel)
 * - 'agent-list': Agent list sidebar (left overlay)
 * - 'file-explorer': File explorer sidebar (right overlay)
 *
 * This makes impossible states unrepresentable - you cannot have both
 * sidebars open at the same time on mobile.
 */
type MobilePanelView = "agent" | "agent-list" | "file-explorer";

/**
 * Desktop sidebar state.
 *
 * On desktop, sidebars are independent toggleable panels that don't overlay
 * the main content - they sit alongside it. Both can be open simultaneously.
 */
interface DesktopSidebarState {
  agentListOpen: boolean;
  fileExplorerOpen: boolean;
  focusModeEnabled: boolean;
}

export type SortOption = "name" | "modified" | "size";
export interface ExplorerCheckoutContext {
  serverId: string;
  cwd: string;
  isGit: boolean;
}

export const DEFAULT_SIDEBAR_WIDTH = 320;
export const MIN_SIDEBAR_WIDTH = 200;
export const MAX_SIDEBAR_WIDTH = 600;

export const DEFAULT_EXPLORER_SIDEBAR_WIDTH = 400;
export const MIN_EXPLORER_SIDEBAR_WIDTH = 280;
// Upper bound is intentionally generous; desktop resizing enforces a min-chat-width constraint.
export const MAX_EXPLORER_SIDEBAR_WIDTH = 2000;

export const DEFAULT_EXPLORER_FILES_SPLIT_RATIO = 0.38;
export const MIN_EXPLORER_FILES_SPLIT_RATIO = 0.2;
export const MAX_EXPLORER_FILES_SPLIT_RATIO = 0.8;

interface PanelState {
  // Mobile: which panel is currently shown
  mobileView: MobilePanelView;

  // Desktop: independent sidebar toggles
  desktop: DesktopSidebarState;

  // File explorer settings (shared between mobile/desktop)
  explorerTab: ExplorerTab;
  explorerTabByCheckout: Record<string, ExplorerTab>;
  expandedPathsByWorkspace: Record<string, string[]>;
  diffExpandedPathsByWorkspace: Record<string, string[]>;
  activeExplorerCheckout: ExplorerCheckoutContext | null;
  sidebarWidth: number;
  explorerWidth: number;
  explorerSortOption: SortOption;
  explorerFilesSplitRatio: number;

  // Actions
  openAgentList: () => void;
  openFileExplorer: () => void;
  closeFileExplorer: () => void;
  closeToAgent: () => void;
  toggleAgentList: () => void;
  toggleFileExplorer: () => void;
  toggleBothSidebars: () => void;
  toggleFocusMode: () => void;

  // File explorer settings actions
  setExplorerTab: (tab: ExplorerTab) => void;
  setExplorerTabForCheckout: (params: ExplorerCheckoutContext & { tab: ExplorerTab }) => void;
  setExpandedPathsForWorkspace: (workspaceKey: string, paths: string[]) => void;
  setDiffExpandedPathsForWorkspace: (workspaceKey: string, paths: string[]) => void;
  activateExplorerTabForCheckout: (checkout: ExplorerCheckoutContext) => void;
  setActiveExplorerCheckout: (checkout: ExplorerCheckoutContext | null) => void;
  setSidebarWidth: (width: number) => void;
  setExplorerWidth: (width: number) => void;
  setExplorerSortOption: (option: SortOption) => void;
  setExplorerFilesSplitRatio: (ratio: number) => void;
}

function clampNumber(value: number, min: number, max: number): number {
  if (!Number.isFinite(value)) {
    return min;
  }
  return Math.max(min, Math.min(max, value));
}

function clampSidebarWidth(width: number): number {
  return clampNumber(width, MIN_SIDEBAR_WIDTH, MAX_SIDEBAR_WIDTH);
}

function clampWidth(width: number): number {
  return clampNumber(width, MIN_EXPLORER_SIDEBAR_WIDTH, MAX_EXPLORER_SIDEBAR_WIDTH);
}

function clampExplorerFilesSplitRatio(ratio: number): number {
  return clampNumber(ratio, MIN_EXPLORER_FILES_SPLIT_RATIO, MAX_EXPLORER_FILES_SPLIT_RATIO);
}

function resolveExplorerTabFromActiveCheckout(state: PanelState): ExplorerTab | null {
  if (!state.activeExplorerCheckout) {
    return null;
  }
  return resolveExplorerTabForCheckout({
    serverId: state.activeExplorerCheckout.serverId,
    cwd: state.activeExplorerCheckout.cwd,
    isGit: state.activeExplorerCheckout.isGit,
    explorerTabByCheckout: state.explorerTabByCheckout,
  });
}

const DEFAULT_DESKTOP_OPEN = isWeb;

export const usePanelStore = create<PanelState>()(
  persist(
    (set) => ({
      // Mobile always starts at agent view
      mobileView: "agent",

      // Desktop defaults based on platform
      desktop: {
        agentListOpen: DEFAULT_DESKTOP_OPEN,
        fileExplorerOpen: false,
        focusModeEnabled: false,
      },

      // File explorer defaults
      explorerTab: "changes",
      explorerTabByCheckout: {},
      expandedPathsByWorkspace: {},
      diffExpandedPathsByWorkspace: {},
      activeExplorerCheckout: null,
      sidebarWidth: DEFAULT_SIDEBAR_WIDTH,
      explorerWidth: DEFAULT_EXPLORER_SIDEBAR_WIDTH,
      explorerSortOption: "name",
      explorerFilesSplitRatio: DEFAULT_EXPLORER_FILES_SPLIT_RATIO,

      openAgentList: () =>
        set((state) => ({
          mobileView: "agent-list" as const,
          desktop: { ...state.desktop, agentListOpen: true },
        })),

      openFileExplorer: () =>
        set((state) => {
          const resolvedTab = resolveExplorerTabFromActiveCheckout(state);
          return {
            mobileView: "file-explorer" as MobilePanelView,
            desktop: { ...state.desktop, fileExplorerOpen: true },
            ...(resolvedTab ? { explorerTab: resolvedTab } : {}),
          };
        }),
      closeFileExplorer: () =>
        set((state) => ({
          mobileView: state.mobileView === "file-explorer" ? ("agent" as const) : state.mobileView,
          desktop: {
            ...state.desktop,
            fileExplorerOpen: false,
          },
        })),

      closeToAgent: () =>
        set((state) => ({
          mobileView: "agent" as const,
          // On desktop, closing depends on which panel triggered it
          // This is called when closing via gesture/backdrop, so we close the currently active mobile panel
          desktop: {
            ...state.desktop,
            agentListOpen: state.mobileView === "agent-list" ? false : state.desktop.agentListOpen,
            fileExplorerOpen:
              state.mobileView === "file-explorer" ? false : state.desktop.fileExplorerOpen,
          },
        })),

      toggleAgentList: () =>
        set((state) => ({
          // Mobile: toggle between agent and agent-list
          mobileView: (state.mobileView === "agent-list"
            ? "agent"
            : "agent-list") as MobilePanelView,
          desktop: {
            ...state.desktop,
            agentListOpen: !state.desktop.agentListOpen,
          },
        })),

      toggleFileExplorer: () =>
        set((state) => {
          // Mobile: toggle between agent and file-explorer
          const willOpenMobile = state.mobileView !== "file-explorer";
          const willOpenDesktop = !state.desktop.fileExplorerOpen;
          const nextMobileView: MobilePanelView = willOpenMobile ? "file-explorer" : "agent";
          const nextDesktop = {
            ...state.desktop,
            fileExplorerOpen: willOpenDesktop,
          };
          const nextState: Pick<PanelState, "mobileView" | "desktop"> &
            Partial<Pick<PanelState, "explorerTab">> = {
            mobileView: nextMobileView,
            desktop: nextDesktop,
          };
          if (willOpenMobile || willOpenDesktop) {
            const resolvedTab = resolveExplorerTabFromActiveCheckout(state);
            if (resolvedTab) {
              nextState.explorerTab = resolvedTab;
            }
          }
          return nextState;
        }),

      toggleFocusMode: () =>
        set((state) => ({
          desktop: { ...state.desktop, focusModeEnabled: !state.desktop.focusModeEnabled },
        })),

      toggleBothSidebars: () =>
        set((state) => {
          // If either sidebar is open, close both. Otherwise, open both.
          const anyOpen = state.desktop.agentListOpen || state.desktop.fileExplorerOpen;
          if (anyOpen) {
            return {
              mobileView: "agent" as MobilePanelView,
              desktop: { ...state.desktop, agentListOpen: false, fileExplorerOpen: false },
            };
          }
          const resolvedTab = resolveExplorerTabFromActiveCheckout(state);
          return {
            mobileView: "agent" as MobilePanelView,
            desktop: { ...state.desktop, agentListOpen: true, fileExplorerOpen: true },
            ...(resolvedTab ? { explorerTab: resolvedTab } : {}),
          };
        }),

      setExplorerTab: (tab) => set({ explorerTab: tab }),
      setExplorerTabForCheckout: ({ serverId, cwd, isGit, tab }) =>
        set((state) => {
          const resolvedTab = coerceExplorerTabForCheckout(tab, isGit);
          const key = buildExplorerCheckoutKey(serverId, cwd);
          const nextState: Partial<PanelState> = { explorerTab: resolvedTab };
          if (key) {
            const current = state.explorerTabByCheckout[key];
            if (current !== resolvedTab) {
              nextState.explorerTabByCheckout = {
                ...state.explorerTabByCheckout,
                [key]: resolvedTab,
              };
            }
          }
          return nextState;
        }),
      setExpandedPathsForWorkspace: (workspaceKey, paths) =>
        set((state) => ({
          expandedPathsByWorkspace: { ...state.expandedPathsByWorkspace, [workspaceKey]: paths },
        })),
      setDiffExpandedPathsForWorkspace: (workspaceKey, paths) =>
        set((state) => ({
          diffExpandedPathsByWorkspace: {
            ...state.diffExpandedPathsByWorkspace,
            [workspaceKey]: paths,
          },
        })),
      activateExplorerTabForCheckout: (checkout) =>
        set((state) => ({
          activeExplorerCheckout: checkout,
          explorerTab: resolveExplorerTabForCheckout({
            serverId: checkout.serverId,
            cwd: checkout.cwd,
            isGit: checkout.isGit,
            explorerTabByCheckout: state.explorerTabByCheckout,
          }),
        })),
      setActiveExplorerCheckout: (checkout) =>
        set((state) => {
          const current = state.activeExplorerCheckout;
          if (
            current?.serverId === checkout?.serverId &&
            current?.cwd === checkout?.cwd &&
            current?.isGit === checkout?.isGit
          ) {
            return state;
          }
          return { activeExplorerCheckout: checkout };
        }),
      setSidebarWidth: (width) => set({ sidebarWidth: clampSidebarWidth(width) }),
      setExplorerWidth: (width) => set({ explorerWidth: clampWidth(width) }),
      setExplorerSortOption: (option) => set({ explorerSortOption: option }),
      setExplorerFilesSplitRatio: (ratio) =>
        set({
          explorerFilesSplitRatio: Number.isFinite(ratio)
            ? clampExplorerFilesSplitRatio(ratio)
            : DEFAULT_EXPLORER_FILES_SPLIT_RATIO,
        }),
    }),
    {
      name: "panel-state",
      version: 10,
      storage: createJSONStorage(() => AsyncStorage),
      migrate: (persistedState, version) => {
        const state = persistedState as Partial<PanelState> & Record<string, unknown>;

        if (version < 2) {
          if (isWeb && typeof state.explorerWidth === "number" && state.explorerWidth === 400) {
            state.explorerWidth = DEFAULT_EXPLORER_SIDEBAR_WIDTH;
          }

          if (typeof state.explorerFilesSplitRatio !== "number") {
            state.explorerFilesSplitRatio = DEFAULT_EXPLORER_FILES_SPLIT_RATIO;
          } else {
            state.explorerFilesSplitRatio = clampExplorerFilesSplitRatio(
              state.explorerFilesSplitRatio,
            );
          }
        }

        if (version < 3) {
          if (
            isWeb &&
            typeof state.explorerWidth === "number" &&
            (state.explorerWidth === 400 || state.explorerWidth === 520)
          ) {
            state.explorerWidth = DEFAULT_EXPLORER_SIDEBAR_WIDTH;
          }
        }

        if (!isExplorerTab(state.explorerTab)) {
          state.explorerTab = "changes";
        }

        if (
          version < 4 ||
          typeof state.explorerTabByCheckout !== "object" ||
          !state.explorerTabByCheckout
        ) {
          state.explorerTabByCheckout = {};
        } else {
          const entries = Object.entries(state.explorerTabByCheckout as Record<string, unknown>);
          const next: Record<string, ExplorerTab> = {};
          for (const [key, value] of entries) {
            if (!isExplorerTab(value)) {
              continue;
            }
            next[key] = value;
          }
          state.explorerTabByCheckout = next;
        }

        if (version < 8) {
          const desktop = state.desktop as Record<string, unknown> | undefined;
          if (desktop) {
            if ("zoomed" in desktop) {
              desktop.focusModeEnabled = desktop.zoomed;
              delete desktop.zoomed;
            }
            if ("focused" in desktop) {
              desktop.focusModeEnabled = desktop.focused;
              delete desktop.focused;
            }
            if (typeof desktop.focusModeEnabled !== "boolean") {
              desktop.focusModeEnabled = false;
            }
          }
        }

        if (version < 6 || typeof state.sidebarWidth !== "number") {
          state.sidebarWidth = DEFAULT_SIDEBAR_WIDTH;
        }

        if (
          version < 9 ||
          typeof state.expandedPathsByWorkspace !== "object" ||
          !state.expandedPathsByWorkspace
        ) {
          state.expandedPathsByWorkspace = {};
        }

        if (
          version < 10 ||
          typeof state.diffExpandedPathsByWorkspace !== "object" ||
          !state.diffExpandedPathsByWorkspace
        ) {
          state.diffExpandedPathsByWorkspace = {};
        }

        state.activeExplorerCheckout = null;

        return state as PanelState;
      },
      partialize: (state) => ({
        mobileView: state.mobileView,
        desktop: state.desktop,
        explorerTab: state.explorerTab,
        explorerTabByCheckout: state.explorerTabByCheckout,
        expandedPathsByWorkspace: state.expandedPathsByWorkspace,
        diffExpandedPathsByWorkspace: state.diffExpandedPathsByWorkspace,
        sidebarWidth: state.sidebarWidth,
        explorerWidth: state.explorerWidth,
        explorerSortOption: state.explorerSortOption,
        explorerFilesSplitRatio: state.explorerFilesSplitRatio,
      }),
    },
  ),
);

/**
 * Hook that provides platform-aware panel state.
 *
 * On mobile, uses the state machine (mobileView).
 * On desktop, uses independent booleans (desktop.agentListOpen, desktop.fileExplorerOpen).
 *
 * @param isMobile - Whether the current breakpoint is mobile
 */
export function usePanelState(isMobile: boolean) {
  const store = usePanelStore();

  if (isMobile) {
    return {
      isAgentListOpen: store.mobileView === "agent-list",
      isFileExplorerOpen: store.mobileView === "file-explorer",
      openAgentList: store.openAgentList,
      openFileExplorer: store.openFileExplorer,
      closeAgentList: store.closeToAgent,
      closeFileExplorer: store.closeToAgent,
      toggleAgentList: store.toggleAgentList,
      toggleFileExplorer: store.toggleFileExplorer,
      // Explorer settings
      explorerTab: store.explorerTab,
      explorerTabByCheckout: store.explorerTabByCheckout,
      explorerWidth: store.explorerWidth,
      explorerSortOption: store.explorerSortOption,
      explorerFilesSplitRatio: store.explorerFilesSplitRatio,
      setExplorerTab: store.setExplorerTab,
      setExplorerTabForCheckout: store.setExplorerTabForCheckout,
      activateExplorerTabForCheckout: store.activateExplorerTabForCheckout,
      setActiveExplorerCheckout: store.setActiveExplorerCheckout,
      setExplorerWidth: store.setExplorerWidth,
      setExplorerSortOption: store.setExplorerSortOption,
      setExplorerFilesSplitRatio: store.setExplorerFilesSplitRatio,
    };
  }

  // Desktop: independent toggles
  return {
    isAgentListOpen: store.desktop.agentListOpen,
    isFileExplorerOpen: store.desktop.fileExplorerOpen,
    openAgentList: store.openAgentList,
    openFileExplorer: store.openFileExplorer,
    closeAgentList: () =>
      usePanelStore.setState((state) => ({
        desktop: { ...state.desktop, agentListOpen: false },
      })),
    closeFileExplorer: () =>
      usePanelStore.setState((state) => ({
        desktop: { ...state.desktop, fileExplorerOpen: false },
      })),
    toggleAgentList: store.toggleAgentList,
    toggleFileExplorer: store.toggleFileExplorer,
    // Explorer settings
    explorerTab: store.explorerTab,
    explorerTabByCheckout: store.explorerTabByCheckout,
    explorerWidth: store.explorerWidth,
    explorerSortOption: store.explorerSortOption,
    explorerFilesSplitRatio: store.explorerFilesSplitRatio,
    setExplorerTab: store.setExplorerTab,
    setExplorerTabForCheckout: store.setExplorerTabForCheckout,
    activateExplorerTabForCheckout: store.activateExplorerTabForCheckout,
    setActiveExplorerCheckout: store.setActiveExplorerCheckout,
    setExplorerWidth: store.setExplorerWidth,
    setExplorerSortOption: store.setExplorerSortOption,
    setExplorerFilesSplitRatio: store.setExplorerFilesSplitRatio,
  };
}