'use client'

import { useCallback, useEffect, useMemo, useRef, useState } from 'react'
import type { RunStreamEvent } from '@/lib/novel-promotion/run-stream/types'
import { applyRunStreamEvent } from './state-machine'
import { subscribeRecoveredRun } from './recovered-run-subscription'
import { executeRunRequest } from './run-request-executor'
import { deriveRunStreamView } from './run-stream-view'
import type { RunResult, RunState, RunStreamView, UseRunStreamStateOptions } from './types'
import { apiFetch } from '@/lib/api-fetch'
import { startRecoveryProbe } from './recovery-probe'

export type {
  RunResult,
  RunState,
  RunStepState,
  UseRunStreamStateOptions,
} from './types'

const TASK_STREAM_TIMEOUT_MS = 1000 * 60 * 30

export function useRunStreamState<TParams extends Record<string, unknown>>(
  options: UseRunStreamStateOptions<TParams>,
): RunStreamView {
  const {
    projectId,
    endpoint,
    storageKeyPrefix,
    storageScopeKey,
    buildRequestBody,
    validateParams,
    resolveActiveRunId,
  } = options
  const [runState, setRunState] = useState<RunState | null>(null)
  const runStateRef = useRef<RunState | null>(null)
  const [clock, setClock] = useState(() => Date.now())
  const [isLiveRunning, setIsLiveRunning] = useState(false)
  const [isRecoveredRunning, setIsRecoveredRunning] = useState(false)
  const abortRef = useRef<AbortController | null>(null)
  const finalResultRef = useRef<RunResult | null>(null)
  const resolveActiveRunIdRef = useRef(resolveActiveRunId)
  const storageKey = useMemo(() => {
    if (storageScopeKey) {
      return `${storageKeyPrefix}:${projectId}:${storageScopeKey}`
    }
    return `${storageKeyPrefix}:${projectId}`
  }, [projectId, storageKeyPrefix, storageScopeKey])

  const applyEvent = useCallback((event: RunStreamEvent) => {
    setRunState((prev) => applyRunStreamEvent(prev, event))
  }, [])

  useEffect(() => {
    runStateRef.current = runState
  }, [runState])

  useEffect(() => {
    resolveActiveRunIdRef.current = resolveActiveRunId
  }, [resolveActiveRunId])

  useEffect(() => {
    if (!projectId || !resolveActiveRunIdRef.current) return

    if (runStateRef.current) return

    return startRecoveryProbe({
      projectId,
      storageKey,
      storageScopeKey,
      hasRunState: () => runStateRef.current !== null,
      resolveActiveRunId: (context) =>
        resolveActiveRunIdRef.current?.(context) ?? Promise.resolve(null),
      onRecovered: (activeRunId) => {
        const now = Date.now()
        setRunState((prev) => {
          if (prev) return prev
          return {
            runId: activeRunId,
            status: 'running',
            startedAt: now,
            updatedAt: now,
            terminalAt: null,
            errorMessage: '',
            summary: null,
            payload: null,
            stepsById: {},
            stepOrder: [],
            activeStepId: null,
            selectedStepId: null,
          }
        })
        setIsRecoveredRunning(true)
      },
    })
  }, [projectId, storageKey, storageScopeKey])

  useEffect(() => {
    if (!projectId || !isRecoveredRunning || isLiveRunning) return
    const runId = runState?.runId || ''
    if (!runId || runState?.status !== 'running') return

    return subscribeRecoveredRun({
      runId,
      taskStreamTimeoutMs: TASK_STREAM_TIMEOUT_MS,
      applyAndCapture: applyEvent,
      onSettled: () => {
        setIsRecoveredRunning(false)
      },
    })
  }, [
    applyEvent,
    isLiveRunning,
    isRecoveredRunning,
    projectId,
    runState?.runId,
    runState?.status,
  ])

  useEffect(() => {
    if (!isRecoveredRunning) return
    if (!runState) {
      setIsRecoveredRunning(false)
      return
    }
    if (runState.status === 'completed' || runState.status === 'failed') {
      setIsRecoveredRunning(false)
    }
  }, [isRecoveredRunning, runState, runState?.status])

  const run = useCallback(
    async (params: TParams): Promise<RunResult> => {
      if (!projectId) {
        throw new Error('projectId is required')
      }
      validateParams?.(params)

      abortRef.current?.abort()
      setIsRecoveredRunning(false)
      setIsLiveRunning(true)
      const controller = new AbortController()
      abortRef.current = controller
      finalResultRef.current = null

      try {
        const requestBody = buildRequestBody(params)
        return await executeRunRequest({
          endpointUrl: endpoint(projectId),
          requestBody,
          controller,
          taskStreamTimeoutMs: TASK_STREAM_TIMEOUT_MS,
          applyAndCapture: applyEvent,
          finalResultRef,
        })
      } finally {
        if (abortRef.current === controller) {
          abortRef.current = null
        }
        setIsLiveRunning(false)
      }
    },
    [
      applyEvent,
      buildRequestBody,
      endpoint,
      projectId,
      validateParams,
    ],
  )

  const retryStep = useCallback(async (params: {
    stepId: string
    modelOverride?: string
    reason?: string
  }): Promise<RunResult> => {
    const runId = runStateRef.current?.runId || ''
    if (!runId) {
      throw new Error('runId is required')
    }
    const stepId = params.stepId.trim()
    if (!stepId) {
      throw new Error('stepId is required')
    }

    const response = await apiFetch(
      `/api/runs/${runId}/steps/${encodeURIComponent(stepId)}/retry`,
      {
        method: 'POST',
        headers: { 'content-type': 'application/json' },
        body: JSON.stringify({
          modelOverride: params.modelOverride || undefined,
          reason: params.reason || undefined,
        }),
      },
    )
    const payload = await response.json().catch(() => null)
    if (!response.ok) {
      const errorMessage =
        payload && typeof payload === 'object' && typeof (payload as { error?: { message?: unknown } }).error?.message === 'string'
          ? (payload as { error: { message: string } }).error.message
          : 'retry step failed'
      throw new Error(errorMessage)
    }

    applyEvent({
      runId,
      event: 'run.start',
      ts: new Date().toISOString(),
      status: 'running',
      message: 'retrying failed step',
    })
    setIsRecoveredRunning(true)
    return {
      runId,
      status: 'running',
      summary: null,
      payload: payload && typeof payload === 'object' ? (payload as Record<string, unknown>) : null,
      errorMessage: '',
    }
  }, [applyEvent])

  const stop = useCallback(() => {
    const runningRunId = runState?.status === 'running' ? runState.runId : ''
    if (runningRunId) {
      void apiFetch(`/api/runs/${runningRunId}/cancel`, {
        method: 'POST',
      }).catch(() => null)
      applyEvent({
        runId: runningRunId,
        event: 'run.error',
        ts: new Date().toISOString(),
        status: 'failed',
        message: 'aborted',
      })
    }
    abortRef.current?.abort()
    abortRef.current = null
    setIsLiveRunning(false)
  }, [applyEvent, runState?.runId, runState?.status])

  const reset = useCallback(() => {
    stop()
    setRunState(null)
    finalResultRef.current = null
    setIsRecoveredRunning(false)
  }, [stop])

  useEffect(() => {
    const timer = window.setInterval(() => setClock(Date.now()), 500)
    return () => window.clearInterval(timer)
  }, [])

  const view = useMemo(() => {
    return deriveRunStreamView({
      runState,
      isLiveRunning,
      clock,
    })
  }, [clock, isLiveRunning, runState])

  const selectStep = useCallback((stepId: string) => {
    setRunState((prev) => {
      if (!prev || !prev.stepsById[stepId]) return prev
      return {
        ...prev,
        selectedStepId: stepId,
      }
    })
  }, [])

  return {
    runState,
    runId: runState?.runId || '',
    status: runState?.status || 'idle',
    isRunning: isLiveRunning,
    isRecoveredRunning,
    isVisible: view.isVisible,
    errorMessage: runState?.errorMessage || '',
    summary: runState?.summary || null,
    payload: runState?.payload || null,
    stages: view.stages,
    orderedSteps: view.orderedSteps,
    activeStepId: view.activeStepId,
    selectedStep: view.selectedStep,
    outputText: view.outputText,
    overallProgress: view.overallProgress,
    activeMessage: view.activeMessage,
    run: run as (params: Record<string, unknown>) => Promise<RunResult>,
    retryStep,
    stop,
    reset,
    selectStep,
  }
}
