import type { RobotMetaInput } from './_instructions-primitives.ts'

import { z } from 'zod'

import { stackVersions } from '../stackVersions.ts'
import {
  bitrateSchema,
  interpolateRobot,
  robotBase,
  robotFFmpegAudio,
  robotUse,
  robotUseWithHiddenFields,
  sampleRateSchema,
} from './_instructions-primitives.ts'

export const meta: RobotMetaInput = {
  bytescount: 4,
  discount_factor: 0.25,
  discount_pct: 75,
  example_code: {
    steps: {
      merged: {
        robot: '/audio/merge',
        preset: 'mp3',
        use: {
          steps: [
            {
              name: ':original',
              fields: 'first_audio_file',
              as: 'audio',
            },
            {
              name: ':original',
              fields: 'second_audio_file',
              as: 'audio',
            },
            {
              name: ':original',
              fields: 'third_audio_file',
              as: 'audio',
            },
          ],
        },
        ffmpeg_stack: stackVersions.ffmpeg.recommendedVersion,
      },
    },
  },
  example_code_description:
    'If you have a form with 3 file input fields and wish to overlay the uploaded audios, instruct Transloadit using the `name` attribute of each input field. Use this attribute as the value for the `fields` key in the JSON, and set `as` to `audio`:',
  minimum_charge: 0,
  output_factor: 0.8,
  override_lvl1: 'Audio Encoding',
  purpose_sentence: 'overlays several audio files on top of each other',
  purpose_verb: 'merge',
  purpose_word: 'merge',
  purpose_words: 'Merge audio files into one',
  service_slug: 'audio-encoding',
  slot_count: 20,
  title: 'Merge audio files into one',
  typical_file_size_mb: 3.8,
  typical_file_type: 'audio file',
  uses_tools: ['ffmpeg'],
  name: 'AudioMergeRobot',
  priceFactor: 4,
  queueSlotCount: 20,
  isAllowedForUrlTransform: true,
  trackOutputFileSize: true,
  isInternal: false,
  removeJobResultFilesFromDiskRightAfterStoringOnS3: false,
  stage: 'ga',
}

export const robotAudioMergeInstructionsSchema = robotBase
  .merge(robotUse)
  .merge(robotFFmpegAudio)
  .extend({
    robot: z.literal('/audio/merge'),
    bitrate: bitrateSchema.optional().describe(`
Bit rate of the resulting audio file, in bits per second. If not specified will default to the bit rate of the input audio file.
`),
    sample_rate: sampleRateSchema.optional().describe(`
Sample rate of the resulting audio file, in Hertz. If not specified will default to the sample rate of the input audio file.
`),
    duration: z
      .enum(['first', 'longest', 'shortest'])
      .default('longest')
      .describe(`
Duration of the output file compared to the duration of all merged audio files. Can be \`"first"\` (duration of the first input file), \`"shortest"\` (duration of the shortest audio file) or \`"longest"\` for the duration of the longest input file.
`),
    loop: z
      .boolean()
      .default(false)
      .describe(`
Specifies if any input files that do not match the target duration should be looped to match it. Useful for audio merging where your overlay file is typically much shorter than the main audio file.
`),
    volume: z
      .enum(['average', 'sum'])
      .default('average')
      .describe(`
Valid values are \`"average"\` and \`"sum"\` here. \`"average"\` means each input is scaled 1/n (n is the number of inputs) or \`"sum"\` which means each individual audio stays on the same volume, but since we merge tracks 'on top' of each other, this could result in very loud output.
`),
  })
  .strict()

export const robotAudioMergeInstructionsWithHiddenFieldsSchema = robotAudioMergeInstructionsSchema
  .omit({ use: true })
  .merge(robotUseWithHiddenFields)
  .extend({
    result: z
      .union([z.literal('debug'), robotAudioMergeInstructionsSchema.shape.result])
      .optional(),
  })

export type RobotAudioMergeInstructions = z.infer<typeof robotAudioMergeInstructionsSchema>
export type RobotAudioMergeInstructionsWithHiddenFields = z.infer<
  typeof robotAudioMergeInstructionsWithHiddenFieldsSchema
>

export const interpolatableRobotAudioMergeInstructionsSchema = interpolateRobot(
  robotAudioMergeInstructionsSchema,
)
export type InterpolatableRobotAudioMergeInstructions =
  InterpolatableRobotAudioMergeInstructionsInput

export type InterpolatableRobotAudioMergeInstructionsInput = z.input<
  typeof interpolatableRobotAudioMergeInstructionsSchema
>

export const interpolatableRobotAudioMergeInstructionsWithHiddenFieldsSchema = interpolateRobot(
  robotAudioMergeInstructionsWithHiddenFieldsSchema,
)
export type InterpolatableRobotAudioMergeInstructionsWithHiddenFields = z.infer<
  typeof interpolatableRobotAudioMergeInstructionsWithHiddenFieldsSchema
>
export type InterpolatableRobotAudioMergeInstructionsWithHiddenFieldsInput = z.input<
  typeof interpolatableRobotAudioMergeInstructionsWithHiddenFieldsSchema
>