import { Database, FileText, Link2, Play, Save, Search, ShieldCheck, SlidersHorizontal } from 'lucide-react'
import { AppProvider } from './context/AppContext'
import { useApp } from './context/useApp'
import { AppShell } from './components/layout/AppShell'
import { screenDescription, screenTitle } from './components/layout/workspaceMeta'
import { Dashboard } from './pages/dashboard/dashboard'
import { ModelLibrary } from './pages/model-library/ModelLibrary'
import { InputLibrary } from './pages/input-library/InputLibrary'
import { ResultsExplorer } from './pages/results-explorer/ResultsExplorer'
import { ScenarioAnalysis } from './pages/scenario-analysis/ScenarioAnalysis'
import { FormulaLibrary } from './pages/formula-library/FormulaLibrary'
import { ProjectionSetsLibrary } from './pages/projection-sets-library/ProjectionSetsLibrary'
import { ReportsCompleted } from './pages/reports-completed/ReportsCompleted'
import { ReserveInvestigation } from './pages/reserve-investigation/ReserveInvestigation'
import { ModelWorkspace } from './pages/model-workspace/ModelWorkspace'
import { InputWorkflowScreen } from './pages/input-workflow-screens/InputWorkflowScreens'
import { ProjectionSetupProgress } from './pages/projection-setup-progress/ProjectionSetupProgress'
import { RunSetWorkflow } from './pages/run-set-workflow/RunSetWorkflow'
import { MentorWorkspace } from './pages/mentor-workspace/MentorWorkspace'
import { AdminSettings } from './pages/admin-settings/AdminSettings'
import { DataManager } from './pages/data-manager/DataManager'
import { ScreenErrorBoundary } from './components/ui/ScreenErrorBoundary'
import type { Model } from './types'
import { useState } from 'react'
import { ApiProvider } from './api/ApiProvider'
import { AuthGate } from './api/AuthGate'
import { WorkspaceGate, WorkspaceProvider } from './api/WorkspaceContext'

function RoutedApp() {
  const { activeScreen, notify, setActiveScreen, updateProjectionInputs, projectionSetupActive, setProjectionSetupActive, activeExecutionRunSetId } = useApp()
  const [modelQuery, setModelQuery] = useState('')
  const [formulaQuery, setFormulaQuery] = useState('')
  const [formulaLineageActive, setFormulaLineageActive] = useState(false)
  const [formulaDetailActive, setFormulaDetailActive] = useState(false)
  const [resultsCompareActive, setResultsCompareActive] = useState(false)
  const [resultsExplorerKey, setResultsExplorerKey] = useState(0)
  const [investigationSelectorActive, setInvestigationSelectorActive] = useState(true)
  const [selectedModel, setSelectedModel] = useState<Model | null>(null)
  const [startProjectionCreate, setStartProjectionCreate] = useState(false)
  const [dataManagerCategory, setDataManagerCategory] = useState<'liability_inforce' | 'assumption_table' | 'factor_table' | 'scenario'>('liability_inforce')
  const [inputWorkflowPage, setInputWorkflowPage] = useState('Input Library')
  const isModelWorkspace = activeScreen === 'Models' && selectedModel !== null
  const startProjectionSetup = () => {
    setProjectionSetupActive(true)
    setStartProjectionCreate(true)
    setActiveScreen('Projections')
  }
  const endProjectionSetup = () => {
    setProjectionSetupActive(false)
    setStartProjectionCreate(false)
    setActiveScreen('Projections')
  }
  const openModel = (model: Model) => {
    setSelectedModel(model)
    if (projectionSetupActive) updateProjectionInputs(current => ({
      ...current,
      model,
      draft: current.model?.name === model.name
        ? current.draft
        : { ...current.draft, formulaConfirmed: false, scenarioCompatible: false },
    }))
  }
  const openDataManager = (category: 'liability_inforce' | 'assumption_table' | 'factor_table' | 'scenario') => {
    setDataManagerCategory(category)
    setActiveScreen('Data Manager')
  }
  const openInputWorkflow = (page: string) => {
    if (page === 'Import Inforce') {
      openDataManager('liability_inforce')
      return
    }
    setProjectionSetupActive(true)
    setStartProjectionCreate(true)
    setInputWorkflowPage(page)
    setActiveScreen('Inputs')
  }
  const openScenarioLibrary = () => {
    setProjectionSetupActive(true)
    setStartProjectionCreate(true)
    setActiveScreen('Scenarios')
  }
  const actions = (
    <button
      className="button button-secondary dashboard-search-button max-[760px]:min-w-0 max-[760px]:px-2.5 max-[760px]:py-2"
      onClick={() => notify('Search opened')}
    >
      Search
      <Search size={18} />
    </button>
  )
  const investigationActions = (
    <div className="investigation-header-actions flex flex-wrap items-center gap-2">
      <button className="investigation-header-button" onClick={() => notify('Investigation search opened')}><Search size={16} />Search</button>
      <button className="investigation-header-button" onClick={() => notify('Investigation link copied')}><Link2 size={16} />Share</button>
      <button className="investigation-header-button" onClick={() => notify('Trace view options opened')}><SlidersHorizontal size={16} />View</button>
      <button className="investigation-save-button" onClick={() => notify('Investigation saved')}><Save size={16} />Save investigation</button>
    </div>
  )

  return (
    <AppShell
      title={isModelWorkspace ? selectedModel.name : activeScreen === 'Runs' ? activeExecutionRunSetId ? 'Execution Center' : 'Run Set Library' : screenTitle(activeScreen)}
      description={isModelWorkspace ? 'Understand the model structure, dependencies, interfaces, and reusable logic.' : activeScreen === 'Runs' ? activeExecutionRunSetId ? 'Monitor submitted runs, progress, events, and execution provenance.' : 'Review Run Set submissions and monitor every individual run.' : screenDescription(activeScreen)}
      eyebrow={isModelWorkspace ? <button className="mw-breadcrumb" onClick={() => setSelectedModel(null)}>Models / Model workspace</button> : activeScreen === 'Dashboard' ? 'Home' : activeScreen}
      topbarVariant={isModelWorkspace ? 'model-workspace' : activeScreen === 'Mentor' ? 'mentor' : activeScreen === 'Admin / Settings' ? 'admin' : activeScreen === 'Models' || activeScreen === 'Formulas' ? 'library' : 'default'}
      searchPlaceholder={activeScreen === 'Formulas' ? 'Search formula groups, versions, owners, dependencies...' : undefined}
      className={`${isModelWorkspace ? 'model-workspace-page-shell' : activeScreen === 'Mentor' ? 'mentor-shell' : activeScreen === 'Admin / Settings' ? 'admin-shell' : activeScreen === 'Inputs' ? 'input-shell' : activeScreen === 'Results' ? `results-shell ${resultsCompareActive ? 'results-compare-shell' : ''}` : activeScreen === 'Scenarios' ? 'scenario-shell' : activeScreen === 'Formulas' ? `formula-shell ${formulaLineageActive ? 'formula-lineage-shell-active' : ''} ${formulaDetailActive ? 'formula-detail-shell-active' : ''}` : activeScreen === 'Runs' ? 'execution-shell' : activeScreen === 'Projections' ? 'projection-shell' : activeScreen === 'Reports' ? 'report-shell' : activeScreen === 'Investigate' ? `investigation-shell ${investigationSelectorActive ? 'investigation-selector-shell' : ''}` : ''} ${projectionSetupActive ? 'projection-flow-active' : ''}`}
      modelName={isModelWorkspace ? selectedModel.name : undefined}
      modelVersion={isModelWorkspace ? selectedModel.version : undefined}
      onNavigate={(screen) => {
        if (screen !== activeScreen) {
          setFormulaLineageActive(false)
          setFormulaDetailActive(false)
          setResultsCompareActive(false)
        }
        if (screen === 'Results') {
          setResultsCompareActive(false)
          setResultsExplorerKey((key) => key + 1)
        }
        if (screen === 'Investigate') setInvestigationSelectorActive(true)
      }}
      searchValue={activeScreen === 'Formulas' ? formulaQuery : modelQuery}
      onSearchChange={activeScreen === 'Formulas' ? setFormulaQuery : setModelQuery}
      actions={isModelWorkspace ? <ModelWorkspaceActions onValidate={() => notify('Model validation started')} onCreate={startProjectionSetup} /> : activeScreen === 'Inputs' ? <InputTrustSignal /> : activeScreen === 'Investigate' && !investigationSelectorActive ? investigationActions : activeScreen === 'Dashboard' || activeScreen === 'Results' || activeScreen === 'Projections' || activeScreen === 'Reports' ? actions : undefined}
      workflowPanel={projectionSetupActive ? <ProjectionSetupProgress onNavigateToInput={openInputWorkflow} onNavigateToScenario={openScenarioLibrary} onReturnToSetup={startProjectionSetup} onClose={endProjectionSetup} /> : undefined}
    >
      <ScreenErrorBoundary key={activeScreen}>
      {activeScreen === 'Dashboard' ? (
        <Dashboard />
      ) : activeScreen === 'Models' ? (
        isModelWorkspace ? (
          <ModelWorkspace key={selectedModel?.id} model={selectedModel} onNotify={notify} onCreateProjection={startProjectionSetup} onBack={() => setActiveScreen('Models')} />
        ) : (
          <ModelLibrary query={modelQuery} onNotify={notify} onOpenModel={openModel} />
        )
      ) : activeScreen === 'Inputs' ? (
        inputWorkflowPage === 'Input Library' ? (
          <InputLibrary
            onNotify={notify}
            onNavigate={(page: string) => setInputWorkflowPage(page)}
            onOpenInforceImport={() => openDataManager('liability_inforce')}
          />
        ) : (
          <InputWorkflowScreen
            page={inputWorkflowPage}
            onNavigate={(page: string) => setInputWorkflowPage(page)}
            onReturnToProjection={startProjectionSetup}
            onNotify={notify}
            onOpenDataManager={openDataManager}
          />
        )
      ) : activeScreen === 'Data Manager' ? (
        <DataManager initialCategory={dataManagerCategory} />
      ) : activeScreen === 'Scenarios' ? (
        <ScenarioAnalysis onNotify={notify} onOpenProjectionSetup={startProjectionSetup} onOpenDataManager={openDataManager} />
      ) : activeScreen === 'Formulas' ? (
        <FormulaLibrary query={formulaQuery} onNotify={notify} onLineageChange={setFormulaLineageActive} onDetailChange={setFormulaDetailActive} />
      ) : activeScreen === 'Runs' ? (
        <RunSetWorkflow onNotify={notify} />
      ) : activeScreen === 'Projections' ? (
        <ProjectionSetsLibrary onNotify={notify} startCreate={startProjectionCreate} onCreateOpened={() => setStartProjectionCreate(false)} onSetupStarted={() => setProjectionSetupActive(true)} onSetupFinished={() => setProjectionSetupActive(false)} onNavigateToInput={openInputWorkflow} onNavigateToScenario={openScenarioLibrary} />
      ) : activeScreen === 'Results' ? (
        <ResultsExplorer key={resultsExplorerKey} onNotify={notify} onCompareChange={setResultsCompareActive} />
      ) : activeScreen === 'Reports' ? (
        <ReportsCompleted onNotify={notify} />
      ) : activeScreen === 'Investigate' ? (
        <ReserveInvestigation onNotify={notify} onViewChange={(view: string) => setInvestigationSelectorActive(view === 'selector')} />
      ) : activeScreen === 'Mentor' ? (
        <MentorWorkspace onOpenInvestigate={() => setActiveScreen('Investigate')} />
      ) : activeScreen === 'Admin / Settings' ? (
        <AdminSettings onNotify={notify} />
      ) : (
        <section className="panel">
          <h2 className="section-title">Workspace cleared</h2>
          <p className="muted">Page content has been removed.</p>
        </section>
      )}
      </ScreenErrorBoundary>
    </AppShell>
  )
}

function ModelWorkspaceActions({ onValidate, onCreate }: { onValidate: () => void; onCreate: () => void }) {
  return (
    <div className="model-workspace-header-actions">
      <button className="model-workspace-secondary-action" onClick={onValidate}><ShieldCheck size={19} />Validate model</button>
      <button className="model-workspace-primary-action" onClick={onCreate}><Play size={19} />Create Projection Set</button>
    </div>
  )
}

function InputTrustSignal() {
  return (
    <div className="input-trust-signal max-[760px]:hidden">
      <div className="input-trust-art" aria-hidden="true">
        <span><Database size={28} /></span>
        <i><FileText size={20} /></i>
      </div>
      <div>
        <b>Trusted inputs<br />for reliable projections</b>
        <small>Validated. Mapped. Model-ready.</small>
      </div>
    </div>
  )
}

export default function App() {
  return (
    <ApiProvider>
      <AuthGate>
        <WorkspaceProvider>
          <WorkspaceGate>
            <AppProvider>
              <RoutedApp />
            </AppProvider>
          </WorkspaceGate>
        </WorkspaceProvider>
      </AuthGate>
    </ApiProvider>
  )
}