chronocat/packages/shell/scripts/codegen.cts

import { dump, load } from 'js-yaml'
import { mkdir, readFile, writeFile } from 'node:fs/promises'
import { resolve } from 'node:path'
import type { Definition, PartialArgs } from 'typescript-json-schema'
import { buildGenerator, programFromConfig } from 'typescript-json-schema'

const settings: PartialArgs = {
  aliasRef: true,
  titles: false,
  defaultProps: true,
  noExtraProps: true,
  propOrder: false,
  required: true,
  strictNullChecks: true,
  skipLibCheck: true,
}

void (async () => {
  console.log('Building generator...')

  const generator = buildGenerator(
    programFromConfig(resolve(__dirname, 'tsconfig.entity.json')),
    settings,
  )!

  console.log('Generating config schema...')

  const configSchema = generator.getSchemaForSymbol('ChronocatConfig')

  console.log('Modifying config schema...')

  configSchema.$id = 'https://chronocat.vercel.app/config-v0.schema.json'
  configSchema.title = 'Chronocat 配置'
  configSchema.description = 'Chronocat 配置（chronocat.yml）'

  const schemaString = JSON.stringify(configSchema).replaceAll(
    '"anyOf"',
    '"type":"object","discriminator":{"propertyName":"type"},"oneOf"',
  )

  console.log('Writing config schema...')

  void writeFile(
    resolve(__dirname, '../../docs/static/config-v0.schema.json'),
    schemaString,
  )

  console.log('Generating schemas for codegen...')

  const schemas = generator.getUserSymbols().map((x) => ({
    $id: x,
    ...generator.getSchemaForSymbol(x),
  }))

  console.log('Writing schemas for codegen...')

  await mkdir(resolve(__dirname, '../generated'), {
    recursive: true,
  })

  await writeFile(
    resolve(__dirname, '../generated/schemas.json'),
    JSON.stringify(schemas, null, 2).replaceAll(
      '"anyOf"',
      '"type":"object","discriminator":{"propertyName":"type"},"oneOf"',
    ),
  )

  console.log('Generating schemas for openapi...')

  let oapiDefinitions = generator.getSchemaForSymbols(
    generator.getUserSymbols(),
  )

  console.log('Modifying schemas for openapi...')

  oapiDefinitions = JSON.parse(
    JSON.stringify(oapiDefinitions)
      .replaceAll(
        '"anyOf"',
        '"type":"object","discriminator":{"propertyName":"type"},"oneOf"',
      )
      .replaceAll('#/definitions/', '#/components/schemas/'),
  ) as Definition

  console.log('Writing ChronocatConfig for openapi...')

  const openapiSource = load(
    await readFile(resolve(__dirname, '../static/openapi.yaml'), 'utf-8'),
  ) as {
    components: {
      schemas: unknown
    }
  }

  openapiSource.components.schemas = oapiDefinitions.definitions

  await writeFile(
    resolve(__dirname, '../../docs/static/openapi.yaml'),
    dump(openapiSource),
  )
})()