/**
 * @license
 * Copyright 2025 Vybestack LLC
 * SPDX-License-Identifier: Apache-2.0
 */

import { useEffect, useMemo } from 'react';
import { useLogger } from '../../../hooks/useLogger.js';
import { useGitBranchName } from '../../../hooks/useGitBranchName.js';
import { useHookDisplayState } from '../../../hooks/useHookDisplayState.js';
import { getAllLlxprtMdFilenames } from '@vybestack/llxprt-code-core';
import { useKeybindings } from './useKeybindings.js';
import { useLayoutMeasurement } from './useLayoutMeasurement.js';
import { useFlickerDetector } from '../../../hooks/useFlickerDetector.js';
import { useSelectionDebugLogger } from './useSelectionDebugLogger.js';
import { useClearScreenAction } from './useClearScreenAction.js';
import { useConfirmationSelection } from './useConfirmationSelection.js';
import { useInputHistoryBootstrap } from './useInputHistoryBootstrap.js';
import { useInitialPromptSubmit } from './useInitialPromptSubmit.js';
import { usePowerShellPlaceholder } from './usePowerShellPlaceholder.js';
import { calculateMainAreaWidth } from '../../../utils/ui-sizing.js';
import type { HistoryItem } from '../../../types.js';
import type { AppBootstrapResult } from './useAppBootstrap.js';
import type { AppDialogsResult } from './useAppDialogs.js';
import type { AppInputResult } from './useAppInput.js';
import type { UiRuntime } from '../../../cliUiRuntime.js';

export interface AppLayoutParams {
  // From bootstrap
  uiRuntime: UiRuntime;
  settings: AppBootstrapResult['settings'];
  runtimeMessageBus: AppBootstrapResult['runtimeMessageBus'];
  consoleMessages: AppBootstrapResult['consoleMessages'];
  clearConsoleMessagesState: AppBootstrapResult['clearConsoleMessagesState'];
  addItem: (item: Omit<HistoryItem, 'id'>, baseTimestamp?: number) => number;
  clearItems: AppBootstrapResult['clearItems'];
  history: AppBootstrapResult['history'];

  // From dialogs
  constrainHeight: AppDialogsResult['constrainHeight'];
  setConstrainHeight: AppDialogsResult['setConstrainHeight'];
  refreshStatic: AppDialogsResult['refreshStatic'];
  showErrorDetails: AppDialogsResult['showErrorDetails'];
  setShowErrorDetails: AppDialogsResult['setShowErrorDetails'];
  showToolDescriptions: AppDialogsResult['showToolDescriptions'];
  setShowToolDescriptions: AppDialogsResult['setShowToolDescriptions'];
  renderMarkdown: AppDialogsResult['renderMarkdown'];
  setRenderMarkdown: AppDialogsResult['setRenderMarkdown'];
  isTodoPanelCollapsed: AppDialogsResult['isTodoPanelCollapsed'];
  setIsTodoPanelCollapsed: AppDialogsResult['setIsTodoPanelCollapsed'];
  isQueuedMessagesPanelCollapsed: AppDialogsResult['isQueuedMessagesPanelCollapsed'];
  setIsQueuedMessagesPanelCollapsed: AppDialogsResult['setIsQueuedMessagesPanelCollapsed'];
  setFooterHeight: AppDialogsResult['setFooterHeight'];
  footerHeight: AppDialogsResult['footerHeight'];
  copyModeEnabled: AppDialogsResult['copyModeEnabled'];
  setCopyModeEnabled: AppDialogsResult['setCopyModeEnabled'];
  useAlternateBuffer: AppDialogsResult['useAlternateBuffer'];
  ideContextState: AppDialogsResult['ideContextState'];
  setDebugMessage: AppDialogsResult['setDebugMessage'];
  isAuthDialogOpen: AppDialogsResult['isAuthDialogOpen'];
  isThemeDialogOpen: AppDialogsResult['isThemeDialogOpen'];
  isEditorDialogOpen: AppDialogsResult['isEditorDialogOpen'];
  isProviderDialogOpen: AppDialogsResult['isProviderDialogOpen'];
  isToolsDialogOpen: AppDialogsResult['isToolsDialogOpen'];
  isCreateProfileDialogOpen: AppDialogsResult['isCreateProfileDialogOpen'];
  showPrivacyNotice: AppDialogsResult['showPrivacyNotice'];
  isWelcomeDialogOpen: AppDialogsResult['isWelcomeDialogOpen'];
  isFolderTrustDialogOpen: AppDialogsResult['isFolderTrustDialogOpen'];
  embeddedShellFocused: AppDialogsResult['embeddedShellFocused'];
  setEmbeddedShellFocused: AppDialogsResult['setEmbeddedShellFocused'];
  startupGuardsInitialized: AppDialogsResult['startupGuardsInitialized'];

  // From input
  streamingState: AppInputResult['streamingState'];
  pendingHistoryItems: AppInputResult['pendingHistoryItems'];
  confirmationRequest: AppInputResult['confirmationRequest'];
  cancelOngoingRequest: AppInputResult['cancelOngoingRequest'];
  activeShellPtyId: AppInputResult['activeShellPtyId'];
  ctrlCPressedOnce: AppInputResult['ctrlCPressedOnce'];
  requestCtrlCExit: AppInputResult['requestCtrlCExit'];
  requestCtrlDExit: AppInputResult['requestCtrlDExit'];
  handleSlashCommand: AppInputResult['handleSlashCommand'];
  inputHistoryStore: AppInputResult['inputHistoryStore'];
  handleUserInputSubmit: AppInputResult['handleUserInputSubmit'];
  handleSteer: AppInputResult['handleSteer'];
  interactiveRuntimeReady: AppInputResult['interactiveRuntimeReady'];
  vimModeEnabled: AppInputResult['vimModeEnabled'];
  terminalHeight: AppInputResult['terminalHeight'];
  terminalWidth: AppInputResult['terminalWidth'];
  buffer: AppInputResult['buffer'];
}
function pickCopyModeKeybindingState(p: AppLayoutParams) {
  return {
    copyModeEnabled: p.copyModeEnabled,
    setCopyModeEnabled: p.setCopyModeEnabled,
    useAlternateBuffer: p.useAlternateBuffer,
  };
}

function buildKeybindingsConfig(
  p: AppLayoutParams,
  exitState: {
    requestCtrlCExit: AppLayoutParams['requestCtrlCExit'];
    requestCtrlDExit: AppLayoutParams['requestCtrlDExit'];
    ctrlCPressedOnce: AppLayoutParams['ctrlCPressedOnce'];
    cancelOngoingRequest: AppLayoutParams['cancelOngoingRequest'];
    bufferTextLength: number;
  },
  displayState: {
    showErrorDetails: AppLayoutParams['showErrorDetails'];
    setShowErrorDetails: AppLayoutParams['setShowErrorDetails'];
    showToolDescriptions: AppLayoutParams['showToolDescriptions'];
    setShowToolDescriptions: AppLayoutParams['setShowToolDescriptions'];
    renderMarkdown: AppLayoutParams['renderMarkdown'];
    setRenderMarkdown: AppLayoutParams['setRenderMarkdown'];
    isTodoPanelCollapsed: AppLayoutParams['isTodoPanelCollapsed'];
    setIsTodoPanelCollapsed: AppLayoutParams['setIsTodoPanelCollapsed'];
    isQueuedMessagesPanelCollapsed: AppLayoutParams['isQueuedMessagesPanelCollapsed'];
    setIsQueuedMessagesPanelCollapsed: AppLayoutParams['setIsQueuedMessagesPanelCollapsed'];
    constrainHeight: AppLayoutParams['constrainHeight'];
    setConstrainHeight: AppLayoutParams['setConstrainHeight'];
    refreshStatic: AppLayoutParams['refreshStatic'];
    addItem: AppLayoutParams['addItem'];
    handleSlashCommand: AppLayoutParams['handleSlashCommand'];
  },
  shellState: {
    activeShellPtyId: AppLayoutParams['activeShellPtyId'];
    setEmbeddedShellFocused: AppLayoutParams['setEmbeddedShellFocused'];
    ideContextState: AppLayoutParams['ideContextState'];
  },
) {
  const { uiRuntime } = p;
  return {
    exit: exitState,
    display: displayState,
    shell: {
      activeShellPtyId: shellState.activeShellPtyId,
      setEmbeddedShellFocused: shellState.setEmbeddedShellFocused,
      getEnableInteractiveShell: () =>
        uiRuntime.shell.getEnableInteractiveShell(),
    },
    copyMode: pickCopyModeKeybindingState(p),
    ideContext: {
      getIdeMode: () => uiRuntime.ide.getIdeMode(),
      ideContextState: shellState.ideContextState,
    },
    mcp: {
      getMcpServers: () => uiRuntime.mcp.getMcpServers(),
    },
  };
}

function useLayoutKeybindingsAndHistory(p: AppLayoutParams) {
  const {
    uiRuntime,
    clearItems,
    clearConsoleMessagesState,
    constrainHeight,
    setConstrainHeight,
    refreshStatic,
    showErrorDetails,
    setShowErrorDetails,
    showToolDescriptions,
    setShowToolDescriptions,
    renderMarkdown,
    setRenderMarkdown,
    isTodoPanelCollapsed,
    setIsTodoPanelCollapsed,
    isQueuedMessagesPanelCollapsed,
    setIsQueuedMessagesPanelCollapsed,
    ideContextState,
    activeShellPtyId,
    setEmbeddedShellFocused,
    ctrlCPressedOnce,
    cancelOngoingRequest,
    requestCtrlCExit,
    requestCtrlDExit,
    handleSlashCommand,
    addItem,
    inputHistoryStore,
    buffer,
    useAlternateBuffer,
  } = p;
  useKeybindings(
    buildKeybindingsConfig(
      p,
      {
        requestCtrlCExit,
        requestCtrlDExit,
        ctrlCPressedOnce,
        cancelOngoingRequest,
        bufferTextLength: buffer.text.length,
      },
      {
        showErrorDetails,
        setShowErrorDetails,
        showToolDescriptions,
        setShowToolDescriptions,
        renderMarkdown,
        setRenderMarkdown,
        isTodoPanelCollapsed,
        setIsTodoPanelCollapsed,
        isQueuedMessagesPanelCollapsed,
        setIsQueuedMessagesPanelCollapsed,
        constrainHeight,
        setConstrainHeight,
        refreshStatic,
        addItem,
        handleSlashCommand,
      },
      { activeShellPtyId, setEmbeddedShellFocused, ideContextState },
    ),
  );
  const logger = useLogger(uiRuntime.storage);
  useInputHistoryBootstrap({ inputHistoryStore, logger });
  const handleClearScreen = useClearScreenAction({
    clearItems,
    clearConsoleMessagesState,
    useAlternateBuffer,
    refreshStatic,
  });
  return { logger, handleClearScreen };
}

function useLayoutMeasure(p: AppLayoutParams) {
  const {
    consoleMessages,
    constrainHeight,
    setFooterHeight,
    footerHeight,
    confirmationRequest,
    terminalHeight,
  } = p;
  useSelectionDebugLogger({ confirmationRequest });
  const handleConfirmationSelect = useConfirmationSelection({
    confirmationRequest,
  });
  const { mainControlsRef, pendingHistoryItemRef, rootUiRef } =
    useLayoutMeasurement({
      enabled: true,
      copyShortcutEnabled: p.copyModeEnabled,
      setFooterHeight,
      terminalHeight,
      consoleMessages,
      showErrorDetails: p.showErrorDetails,
    });
  const staticExtraHeight = 3;
  const availableTerminalHeight = useMemo(
    () => terminalHeight - footerHeight - staticExtraHeight,
    [terminalHeight, footerHeight],
  );
  useFlickerDetector(rootUiRef, terminalHeight, constrainHeight);
  return {
    mainControlsRef,
    pendingHistoryItemRef,
    rootUiRef,
    availableTerminalHeight,
    handleConfirmationSelect,
  };
}

function useLayoutContext(p: AppLayoutParams) {
  const {
    uiRuntime,
    settings,
    runtimeMessageBus,
    consoleMessages,
    isAuthDialogOpen,
    isThemeDialogOpen,
    isEditorDialogOpen,
    isProviderDialogOpen,
    isToolsDialogOpen,
    isCreateProfileDialogOpen,
    showPrivacyNotice,
    isWelcomeDialogOpen,
    isFolderTrustDialogOpen,
    terminalHeight,
    terminalWidth,
    handleUserInputSubmit,
    handleSteer,
    interactiveRuntimeReady,
    vimModeEnabled,
    startupGuardsInitialized,
  } = p;
  const debugMode = uiRuntime.app.getDebugMode();
  const filteredConsoleMessages = useMemo(() => {
    if (debugMode) return consoleMessages;
    return consoleMessages.filter((msg) => msg.type !== 'debug');
  }, [consoleMessages, debugMode]);
  const branchName = useGitBranchName(uiRuntime.session.getTargetDir());
  const contextFileNames = useMemo(() => {
    const fromSettings = settings.merged.ui.contextFileName;
    if (fromSettings != null && fromSettings !== '')
      return Array.isArray(fromSettings) ? fromSettings : [fromSettings];
    return getAllLlxprtMdFilenames();
  }, [settings.merged.ui.contextFileName]);
  const initialPrompt = useMemo(() => uiRuntime.app.getQuestion(), [uiRuntime]);
  useInitialPromptSubmit({
    initialPrompt,
    submitPrompt: handleUserInputSubmit,
    agentClientPresent: Boolean(uiRuntime.agentClientSource.getAgentClient()),
    interactiveRuntimeReady,
    blockedByDialogs: {
      isAuthDialogOpen,
      isThemeDialogOpen,
      isEditorDialogOpen,
      isProviderDialogOpen,
      isToolsDialogOpen,
      isCreateProfileDialogOpen,
      showPrivacyNotice,
      isWelcomeDialogOpen,
      isFolderTrustDialogOpen,
    },
    startupGuardsInitialized,
  });
  const mainAreaWidth = calculateMainAreaWidth(terminalWidth, settings);
  const placeholder = usePowerShellPlaceholder({ vimModeEnabled });
  useEffect(() => {
    uiRuntime.shell.setPtyTerminalSize(mainAreaWidth, terminalHeight);
  }, [uiRuntime, mainAreaWidth, terminalHeight]);
  const activeHooks = useHookDisplayState(runtimeMessageBus);
  return {
    filteredConsoleMessages,
    branchName,
    contextFileNames,
    initialPrompt,
    mainAreaWidth,
    placeholder,
    activeHooks,
    handleSteer,
  };
}

function useLayoutMeasurementAndFlow(p: AppLayoutParams) {
  const measure = useLayoutMeasure(p);
  const context = useLayoutContext(p);
  return { ...measure, ...context };
}

export function useAppLayout(params: AppLayoutParams) {
  const kb = useLayoutKeybindingsAndHistory(params);
  const measure = useLayoutMeasurementAndFlow(params);
  return { ...kb, ...measure };
}

export type AppLayoutResult = ReturnType<typeof useAppLayout>;