import { injectGeneration } from './inject-generation'
import { reconstructSpeechResult } from '@tanstack/ai-client'
import type { Signal } from '@angular/core'
import type { TTSResult } from '@tanstack/ai'
import type {
  GenerationClientState,
  GenerationPersistenceOptions,
  InferGenerationOutputFromReturn,
  SpeechGenerateInput,
} from '@tanstack/ai-client'
import type {
  InjectGenerationOptions,
  InjectGenerationResult,
} from './inject-generation'

export type InjectGenerateSpeechOptions<TOutput = TTSResult> = Omit<
  InjectGenerationOptions<SpeechGenerateInput, TTSResult, TOutput>,
  'onResult' | 'reconstructResult'
> & {
  onResult?: (result: TTSResult) => TOutput | null | void
}

export interface InjectGenerateSpeechResult<TOutput = TTSResult> extends Omit<
  InjectGenerationResult<TOutput>,
  'generate'
> {
  generate: (input: SpeechGenerateInput) => Promise<void>
  result: Signal<TOutput | null>
  isLoading: Signal<boolean>
  error: Signal<Error | undefined>
  status: Signal<GenerationClientState>
}

export function injectGenerateSpeech<TTransformed = void>(
  options: Omit<
    InjectGenerateSpeechOptions,
    'onResult' | 'persistence' | 'threadId'
  > & {
    onResult?: (result: TTSResult) => TTransformed
  } & GenerationPersistenceOptions,
): InjectGenerateSpeechResult<
  InferGenerationOutputFromReturn<TTSResult, TTransformed>
> {
  const devtools = {
    ...options.devtools,
    framework: 'angular' as const,
    hookName: 'injectGenerateSpeech',
    outputKind: 'audio' as const,
  }
  const generation = injectGeneration<
    SpeechGenerateInput,
    TTSResult,
    TTransformed
  >({
    ...options,
    devtools,
    reconstructResult: reconstructSpeechResult,
  })
  return generation
}