import { createCustomOpenAIClient } from '../../infrastructure/ai-provider/provider-client-factory'
import { createChatCompletionText } from '../openai-stream'
import { buildCompletionParams, readMaxCompletionTokens } from '../../infrastructure/ai-provider/completion-policy'
import { createLogger } from '../../shared/logging/logger'
import { getRoleSystemPrompt, getRoleUserPrompt } from '../../prompts'
import { buildVisionUserMessage } from '../generation/multimodal-input'
import type { ConceptInput, CustomApiConfig, PromptLocale, PromptOverrides } from '../../types'
import { ServiceUnavailableError } from '../../shared/errors/errors'
import { ProblemFramingFormatError, type ProblemFramingPlan } from './parser'
import { parseProblemFramingWithRepair } from './format-repair'

const logger = createLogger('ProblemFraming')

const PLANNER_TEMPERATURE = parseFloat(process.env.PROBLEM_FRAMING_TEMPERATURE || '0.7')
const PLANNER_MAX_TOKENS = readMaxCompletionTokens('PROBLEM_FRAMING')
const PLANNER_FORMAT_RETRIES = parseInt(process.env.PROBLEM_FRAMING_FORMAT_RETRIES || '2', 10)
const FORMAT_REPAIR_INPUT_LIMIT = 16_000

export type { ProblemFramingPlan } from './parser'

interface ProblemFramingParams extends ConceptInput {
  feedback?: string
  feedbackHistory?: string[]
  currentPlan?: ProblemFramingPlan
  customApiConfig: CustomApiConfig
  locale?: PromptLocale
  promptOverrides?: PromptOverrides
}

export async function generateProblemFramingPlan(params: ProblemFramingParams): Promise<ProblemFramingPlan> {
  const locale = params.locale === 'en-US' ? 'en-US' : 'zh-CN'
  const client = createCustomOpenAIClient(params.customApiConfig)
  const model = (params.customApiConfig.model || '').trim()

  if (!model) {
    throw new Error('No model available')
  }

  logger.info('Problem framing started', {
    locale,
    conceptLength: params.concept.length,
    hasFeedback: !!params.feedback,
    hasCurrentPlan: !!params.currentPlan,
    hasImages: !!params.referenceImages?.length
  })

  const promptOverrides: PromptOverrides = { ...params.promptOverrides, locale }
  const systemPrompt = getRoleSystemPrompt('problemFraming', promptOverrides)
  const userPrompt = getRoleUserPrompt(
    'problemFraming',
    {
      concept: params.concept,
      instructions: params.feedback,
      feedbackHistory: params.feedbackHistory?.length ? params.feedbackHistory.map((item, index) => `${index + 1}. ${item}`).join('\n') : undefined,
      sceneDesign: params.currentPlan ? JSON.stringify(params.currentPlan, null, 2) : undefined
    },
    promptOverrides
  )

  const response = await createChatCompletionText(
    client,
    {
      model,
      messages: [
        { role: 'system', content: systemPrompt },
        { role: 'user', content: buildVisionUserMessage(userPrompt, params.referenceImages) }
      ],
      temperature: PLANNER_TEMPERATURE,
      ...buildCompletionParams(PLANNER_MAX_TOKENS, 'PROBLEM_FRAMING', client)
    },
    { fallbackToNonStream: true, usageLabel: 'problem-framing' }
  )

  let plan: ProblemFramingPlan
  try {
    plan = await parseProblemFramingWithRepair({
      content: response.content,
      locale,
      maxRepairAttempts: Math.max(0, PLANNER_FORMAT_RETRIES),
      requireDesignFields: !promptOverrides.roles?.problemFraming?.system,
      repair: async ({ content, error, attempt }) => {
        logger.warn('Problem framing format invalid, requesting format repair', {
          attempt,
          error,
          contentLength: content.length,
        })

        const repairResponse = await createChatCompletionText(
          client,
          {
            model,
            messages: [
              {
                role: 'system',
                content: locale === 'en-US'
                  ? 'Repair formatting only; preserve every decision. Output one complete plan with mode/headline/summary/question/answer/audience/background_rules/selected_route/alternative_routes/out_of_scope/special_cases/visual_motif/designer_hint, and 3–12 repeated step blocks with phase/title/content. phase is background, phenomenon, derivation, or closing. Never truncate later steps or erase fields. No other text.'
                  : '只修复格式，保留全部决定。输出完整plan，包含mode/headline/summary/question/answer/audience/background_rules/selected_route/alternative_routes/out_of_scope/special_cases/visual_motif/designer_hint，以及3–12个含phase/title/content的step；phase为background、phenomenon、derivation或closing。不得截掉后续步骤或删字段，标签外零文字。',
              },
              { role: 'user', content: buildVisionUserMessage(userPrompt, params.referenceImages) },
              {
                role: 'user',
                content: `Parser error: ${error}\n\nModel output:\n${content.slice(0, FORMAT_REPAIR_INPUT_LIMIT)}`,
              },
            ],
            temperature: 0,
            ...buildCompletionParams(PLANNER_MAX_TOKENS, 'PROBLEM_FRAMING', client),
          },
          {
            fallbackToNonStream: true,
            usageLabel: `problem-framing-format-repair-${attempt}`,
          },
        )

        return repairResponse.content
      },
    })
  } catch (error) {
    if (!(error instanceof ProblemFramingFormatError)) {
      throw error
    }
    logger.error('Problem framing format repair exhausted', {
      error: error instanceof Error ? error.message : String(error),
    })
    throw new ServiceUnavailableError(
      locale === 'en-US'
        ? 'The planning card format could not be recovered. Please retry.'
        : '规划卡格式修复失败，请重试。',
      { code: 'PLAN_FORMAT_INVALID', retryable: true },
    )
  }

  logger.info('Problem framing completed', {
    mode: plan.mode,
    headline: plan.headline,
    stepCount: plan.steps.length
  })

  return plan
}