Mark IJbema
74 lines
2.5 KiB
TypeScript
74 lines
2.5 KiB
TypeScript
import { Server } from "../../server/server"
|
|
import { PublicApi } from "../../server/routes/instance/httpapi/public"
|
|
import type { CommandModule } from "yargs"
|
|
import { OpenApi } from "effect/unstable/httpapi"
|
|
|
|
type Args = {
|
|
httpapi: boolean
|
|
}
|
|
|
|
export const GenerateCommand = {
|
|
command: "generate",
|
|
builder: (yargs) =>
|
|
yargs.option("httpapi", {
|
|
type: "boolean",
|
|
default: false,
|
|
description: "Generate OpenAPI from the experimental Effect HttpApi contract",
|
|
}),
|
|
handler: async (args) => {
|
|
const specs = args.httpapi ? OpenApi.fromApi(PublicApi) : await Server.openapi()
|
|
// kilocode_change start
|
|
specs.info.title = "kilo"
|
|
specs.info.description = "kilo api"
|
|
// kilocode_change end
|
|
for (const item of Object.values(specs.paths)) {
|
|
for (const method of ["get", "post", "put", "delete", "patch"] as const) {
|
|
const operation = item[method]
|
|
if (!operation?.operationId) continue
|
|
operation["x-codeSamples"] = [
|
|
// kilocode_change start
|
|
{
|
|
lang: "js",
|
|
source: [
|
|
`import { createKiloClient } from "@kilocode/sdk`,
|
|
``,
|
|
`const client = createKiloClient()`,
|
|
`await client.${operation.operationId}({`,
|
|
` ...`,
|
|
`})`,
|
|
].join("\n"),
|
|
},
|
|
// kilocode_change end,
|
|
]
|
|
}
|
|
}
|
|
const raw = JSON.stringify(specs, null, 2)
|
|
// kilocode_change start - replace upstream product name in all descriptions
|
|
.replaceAll("OpenCode", "Kilo")
|
|
.replaceAll("opencode.local", "kilo.local")
|
|
.replaceAll("opencode serve", "kilo serve")
|
|
.replaceAll("https://opencode.ai/", "https://kilo.ai/")
|
|
// kilocode_change end
|
|
|
|
// Format through prettier so output is byte-identical to committed file
|
|
// regardless of whether ./script/format.ts runs afterward.
|
|
const prettier = await import("prettier")
|
|
const babel = await import("prettier/plugins/babel")
|
|
const estree = await import("prettier/plugins/estree")
|
|
const format = prettier.format ?? prettier.default?.format
|
|
const json = await format(raw, {
|
|
parser: "json",
|
|
plugins: [babel.default ?? babel, estree.default ?? estree],
|
|
printWidth: 120,
|
|
})
|
|
|
|
// Wait for stdout to finish writing before process.exit() is called
|
|
await new Promise<void>((resolve, reject) => {
|
|
process.stdout.write(json, (err) => {
|
|
if (err) reject(err)
|
|
else resolve()
|
|
})
|
|
})
|
|
},
|
|
} satisfies CommandModule<object, Args>
|