import {restoreLoras,loraUse} from './lora-memory.mjs'
import { restoreInput, sectionsFrom } from './contracts.mjs'
import { normalizeStyles } from './styles.mjs'
import { QWEN21_ASPECT_IDS as ASPECT_IDS, resolveQwen21Size } from './qwen21-size.mjs'
export { resolveQwen21Size, QWEN21_SIZE_TABLE } from './qwen21-size.mjs'

export const aspectOptions = [
  {id:'auto',label:'Automatic',hint:'Match start or hero still'},
  {id:'16:9',label:'16:9',hint:'Widescreen'},
  {id:'9:16',label:'9:16',hint:'Tall'},
  {id:'1:1',label:'1:1',hint:'Square'},
  {id:'4:3',label:'4:3',hint:'Standard'},
  {id:'3:4',label:'3:4',hint:'Portrait standard'},
  {id:'3:2',label:'3:2',hint:'Photo'},
  {id:'2:3',label:'2:3',hint:'Portrait photo'},
  {id:'21:9',label:'21:9',hint:'Ultrawide'},
  {id:'9:21',label:'9:21',hint:'Ultra-tall'}
]

export function resolveSize(aspect, source, fallback = {width:1344,height:768}, engine = '') {
  if (engine === 'qwen21') return resolveQwen21Size(aspect, source)
  const presets = {'16:9':[1344,768],'9:16':[768,1344],'1:1':[960,960]}
  if (presets[aspect]) return {width:presets[aspect][0],height:presets[aspect][1]}
  if (!source?.width || !source?.height) return {width:fallback.width || 1344,height:fallback.height || 768}
  const factor = Math.min(1344/Math.max(source.width,source.height),768/Math.min(source.width,source.height))
  return {width:Math.max(256,Math.round(source.width*factor/32)*32),height:Math.max(256,Math.round(source.height*factor/32)*32)}
}

export { ASPECT_IDS }

export function bindClip(form, clip, xaigen = false, memory) {
  if (!clip?.id) return form
  const saved = clip.studio2 || clip.job || {}
  const settings = saved.settings || clip.settings || {}
  const next = {...form, startClipId:clip.id, mode:'extend', settings:{...form.settings}}
  next.settings.aspect=ASPECT_IDS.has(settings.aspect)?settings.aspect:(form.settings?.aspect || 'auto')
  for (const key of ['duration','width','height','fps','loraStack','turbo','steps','cfg']) {
    if (settings[key] == null) continue
    try { next.settings[key] = typeof structuredClone === 'function' ? structuredClone(settings[key]) : JSON.parse(JSON.stringify(settings[key])) }
    catch { next.settings[key] = settings[key] }
  }
  next.engine = xaigen && (saved.engine || clip.engine)==='ltx' ? 'ltx' : 'minimax'
  if (saved.identityStillId) next.identityStillId = saved.identityStillId
  for (const key of ['lockFace','lockOutfit']) if (typeof saved[key] === 'boolean') next[key] = saved[key]
  return restoreLoras(next,settings.loraStack,xaigen,memory)
}
export function restoreAsset(form, item, xaigen = false, memory) {
  if (item.kind==='clip' || item.studio2?.kind==='video') {
    const next=bindClip(form,item,xaigen,memory)
    const seed=item.studio2?.settings?.seed ?? item.settings?.seed ?? item.seed
    if (seed != null) next.settings.seed=seed
    next.settings.seedMode='random'
    return {...next,mode:'extend'}
  }
  const restored = restoreInput(item)
  const next = {...form,...restored,settings:{...form.settings,...restored.settings}}
  next.settings.seedMode='random'
  next.engine=restored.engine==='qwen21'?'qwen21':restored.engine==='krea'?'krea':'flux'
  next.mode=['generate','edit','compose','refine','iterate'].includes(restored.mode)?restored.mode:'edit'
  next.imageAId=item.id
  next.startClipId=''
  next.settings.aspect=item.studio2?.settings?.aspect || item.settings?.aspect || 'auto'
  return restoreLoras(next,item.studio2?.settings?.loraStack ?? item.settings?.loraStack,xaigen,memory)
}
export function extendSelected(form, clip, xaigen = false, memory) {
  const next = { ...form, settings:{...form.settings}, mode: 'extend', engine: xaigen && form.engine === 'ltx' ? 'ltx' : 'minimax' }
  const target = clip?.id ? clip : null
  if (!target?.id) return restoreLoras(next,loraUse(form)===loraUse(next)?form.settings?.loraStack:[],xaigen,memory)
  // Already bound to this clip: switching task must not stomp the user's overrides
  // (locks, CFG, etc.) by re-copying the clip's saved settings over them.
  if (form.startClipId === target.id) return restoreLoras(next,loraUse(form)===loraUse(next)?form.settings?.loraStack:[],xaigen,memory)
  try { return bindClip(next, target, xaigen, memory) }
  catch { return { ...next, startClipId: target.id } }
}
export function changeSource(form) {
  return form.mode==='extend' ? form : {...form,startClipId:'',settings:{...form.settings,aspect:'auto'}}
}

export function useAsHero(form, item) { return {...form,identityStillId:item.id} }
export function useAsSource(form, item) { return {...form,imageAId:item.id} }

/** Hydrate the live bench from a saved Studio 2 request (Load). Does not enqueue. */
export function hydrateStudio2Job(form, saved, { stillIds = new Set(), clipIds = new Set(), xaigen = false, memory } = {}) {
  const src = structuredClone(saved?.studio2 || saved || {})
  const missing = []
  const keepStill = (id, label) => {
    const value = id || ''
    if (!value) return ''
    if (stillIds.has(value)) return value
    missing.push(`${label} missing from library`)
    return ''
  }
  const keepClip = (id, label) => {
    const value = id || ''
    if (!value) return ''
    if (clipIds.has(value)) return value
    missing.push(`${label} missing from library`)
    return ''
  }
  const modeRaw = src.mode === 'iterate' ? 'edit' : src.mode
  const mode = ['generate','edit','compose','refine','video','extend','music'].includes(modeRaw) ? modeRaw : 'edit'
  let engine = src.engine === 'qwen21' ? 'qwen21' : src.engine === 'krea' ? 'krea' : src.engine === 'ltx' ? 'ltx' : src.engine === 'minimax' ? 'minimax' : 'flux'
  if (['video','extend'].includes(mode)) engine = xaigen && engine === 'ltx' ? 'ltx' : 'minimax'
  else if (engine === 'minimax' || engine === 'ltx') engine = 'flux'
  const settings = structuredClone(src.settings || {})
  const aspect = ASPECT_IDS.has(settings.aspect) ? settings.aspect : 'auto'
  const loraStack = Array.isArray(settings.loraStack) ? settings.loraStack.map(item => ({
    name: item.name,
    strengthModel: Number(item.strengthModel ?? item.strength ?? 1),
    strengthClip: Number(item.strengthClip ?? item.strengthModel ?? item.strength ?? 1)
  })) : []
  const guides = (Array.isArray(src.guides) ? src.guides : []).map(g => {
    const stillId = keepStill(g.stillId, 'guide still')
    return stillId ? { stillId, frame: g.frame } : null
  }).filter(Boolean)
  const next = {
    ...form,
    mode,
    engine,
    enhancePrompt: engine === 'qwen21' && !!src.enhancePrompt,
    turbo: engine === 'qwen21' && !!src.turbo,
    lockFace: src.lockFace !== false,
    lockOutfit: src.lockOutfit !== false,
    hideInput: !!src.hideInput,
    folderId: src.folderId || form.folderId,
    identityStillId: keepStill(src.identityStillId, 'hero'),
    imageAId: keepStill(src.imageAId, 'start'),
    imageBId: keepStill(src.imageBId, 'second still'),
    maskId: keepStill(src.maskId, 'mask'),
    startClipId: keepClip(src.startClipId, 'start clip'),
    endStillId: keepStill(src.endStillId, 'end still'),
    guides,
    imageStyles: normalizeStyles(src.imageStyles || { positive: [], negative: [] }),
    promptSections: sectionsFrom(src.promptSections || { action: src.compiledPrompt || src.prompt || '' }),
    settings: {
      ...form.settings,
      ...settings,
      aspect,
      seed: settings.seed ?? form.settings?.seed ?? null,
      seedMode: 'locked',
      loraStack
    }
  }
  return { form: restoreLoras(next, loraStack, xaigen, memory), missing: missing[0] || '', warnings: missing }
}

export function restoreAll(form,item,xaigen=false,memory) {
 const restored=restoreInput(item),next={...form,...restored,settings:{...form.settings,...restored.settings,seedMode:'random'}}
 if(item.kind==='clip') return {...bindClip(next,item,xaigen,memory),mode:'extend'}
 next.mode=restored.mode==='iterate'?'edit':restored.mode || 'edit'
 next.engine=restored.engine==='qwen21'?'qwen21':restored.engine==='krea'?'krea':'flux'
 next.imageAId=item.id
 next.startClipId=''
 return restoreLoras(next,item.studio2?.settings?.loraStack ?? item.settings?.loraStack,xaigen,memory)
}