import { parse } from 'yaml'
import { pascalCase } from "change-case";
import fs from "node:fs"

const modulesDir = "./src-tsp/modules"

const res = await fetch("https://blue-build.org/modules.json")
const modules = await res.json()

let moduleImports = []
let moduleModels = []

fs.existsSync(modulesDir) && fs.rmSync(modulesDir, { recursive: true });
fs.mkdirSync(modulesDir)

for (const module of modules) {
    if (module.tsp === "") continue
    const res = await fetch(module.tsp)
    let text = await res.text()

    text = text // add `...ModuleDefaults;` after the model type
        .split("\n")
        .flatMap(line => {
            if (line.trimStart().startsWith("model")) {
                return [
                    `@extension("additionalProperties", false)`,
                    line
                ];
            }
            if (line.trimStart().startsWith(`type: "${module.name}"`)) {
                return [
                    line,
                    '',
                    '    ...ModuleDefaults; // added by fetchModuleSchemas.js',
                ];
            } else {
                return line;
            }
        })
        .join('\n')

    fs.writeFileSync(`${modulesDir}/${module.name}.tsp`, text)

    moduleImports.push(`${module.name}.tsp`)
    moduleModels.push(`${pascalCase(module.name)}`)
}

fs.writeFileSync(`${modulesDir}/index.tsp`, 
moduleImports.map(m => `import "./${m}";`).join("\n") + `
import "@typespec/json-schema";

using TypeSpec.JsonSchema;

@oneOf
union RepoModule {
${moduleModels.map(m => `  ${m}Module`).join(",\n")}
}`
)