import { FileHelper } from '@start9labs/start-sdk'
import { z } from 'zod'
import { sdk } from '../sdk'

export const sparkConfigSchema = z.object({
  spark1_host: z.string().catch(''),
  spark1_user: z.string().catch(''),
  spark2_host: z.string().catch(''),
  spark2_user: z.string().catch(''),
  // Optional per-service overrides. Blank => use spark2_host / spark2_user.
  parakeet_host: z.string().catch(''),
  parakeet_user: z.string().catch(''),
  parakeet_container: z.string().catch(''),
  magpie_host: z.string().catch(''),
  magpie_user: z.string().catch(''),
  magpie_container: z.string().catch(''),
})

export type SparkConfig = z.infer<typeof sparkConfigSchema>

export const sparkConfigYaml = FileHelper.yaml(
  { base: sdk.volumes.main, subpath: 'config.yaml' },
  sparkConfigSchema,
)