import { mkdir, readFile, writeFile } from 'node:fs/promises'

import { dirname, resolve } from 'node:path'

import { fileURLToPath } from 'node:url'

import { compile } from 'json-schema-to-typescript'

import { formatGeneratedSource } from './format-generated-source'

const SCRIPT_DIR = dirname(fileURLToPath(import.meta.url))

const ROOT = resolve(SCRIPT_DIR, '..')

const DEFAULT_CONTRACT_PATH = resolve(

ROOT,

'../copilot/copilot/contracts/request-trace-v1.schema.json'

)

const OUTPUT_PATH = resolve(ROOT, 'apps/sim/lib/copilot/generated/request-trace-v1.ts')

function generateRuntimeConstants(schema: Record<string, unknown>): string {

const defs = (schema.$defs ?? schema.definitions ?? {}) as Record<string, unknown>

const lines: string[] = []

for (const [name, def] of Object.entries(defs)) {

if (!def || typeof def !== 'object') continue

const defObj = def as Record<string, unknown>

const enumValues = defObj.enum

if (!Array.isArray(enumValues) || enumValues.length === 0) continue

if (!enumValues.every((v) => typeof v === 'string')) continue

const entries = (enumValues as string[])

.map((v) => ` ${JSON.stringify(v)}: ${JSON.stringify(v)}`)

.join(',\n')

lines.push(`export const ${name} = {\n${entries},\n} as const;\n`)

}

return lines.join('\n')

}

async function main() {

const checkOnly = process.argv.includes('--check')

const inputPathArg = process.argv.find((arg) => arg.startsWith('--input='))

const inputPath = inputPathArg

? resolve(ROOT, inputPathArg.slice('--input='.length))

: DEFAULT_CONTRACT_PATH

const raw = await readFile(inputPath, 'utf8')

const schema = JSON.parse(raw)

const types = await compile(schema, 'RequestTraceV1SimReport', {

bannerComment: '// AUTO-GENERATED FILE. DO NOT EDIT.\n//',

unreachableDefinitions: true,

additionalProperties: false,

})

const constants = generateRuntimeConstants(schema)

const rendered = formatGeneratedSource(

constants ? `${types}\n${constants}\n` : types,

OUTPUT_PATH,

ROOT

)

if (checkOnly) {

const existing = await readFile(OUTPUT_PATH, 'utf8').catch(() => null)

if (existing !== rendered) {

throw new Error(

`Generated request trace contract is stale. Run: bun run trace-contracts:generate`

)

}

console.log('Request trace contract is up to date.')

return

}

await mkdir(dirname(OUTPUT_PATH), { recursive: true })

await writeFile(OUTPUT_PATH, rendered, 'utf8')

console.log(`Generated request trace types -> ${OUTPUT_PATH}`)

}

main().catch((err) => {

console.error(err)

process.exit(1)

})