mirror of
https://github.com/anomalyco/opencode.git
synced 2026-07-23 02:36:57 -04:00
Kit Langton
419 lines
17 KiB
TypeScript
Executable File
419 lines
17 KiB
TypeScript
Executable File
#!/usr/bin/env bun
|
|
import { fileURLToPath } from "url"
|
|
|
|
const dir = fileURLToPath(new URL("..", import.meta.url))
|
|
process.chdir(dir)
|
|
|
|
import { $ } from "bun"
|
|
import path from "path"
|
|
|
|
import { createClient } from "@hey-api/openapi-ts"
|
|
|
|
const opencode = path.resolve(dir, "../../opencode")
|
|
const client = path.resolve(dir, "../../client")
|
|
|
|
await $`bun dev generate > ${dir}/openapi.json`.cwd(opencode)
|
|
await $`bun -e ${`
|
|
import { OpenApi } from "effect/unstable/httpapi"
|
|
import { ClientApi } from "@opencode-ai/protocol/client"
|
|
|
|
const output = process.argv.at(-1)
|
|
if (!output) throw new Error("Missing OpenAPI output path")
|
|
await Bun.write(output, JSON.stringify(OpenApi.fromApi(ClientApi)))
|
|
`} ${path.join(dir, "openapi-v2.json")}`.cwd(client)
|
|
|
|
type OpenApiDocument = {
|
|
components?: { schemas?: Record<string, unknown> }
|
|
paths?: Record<string, unknown>
|
|
[key: string]: unknown
|
|
}
|
|
|
|
const document = (await Bun.file("./openapi.json").json()) as OpenApiDocument
|
|
const v2Document = (await Bun.file("./openapi-v2.json").json()) as OpenApiDocument
|
|
normalizeComponentNames(v2Document)
|
|
deduplicateEquivalentComponent(v2Document, "Shell", "Shell1")
|
|
renameCollidingComponents(document, v2Document)
|
|
document.paths = { ...document.paths, ...v2Document.paths }
|
|
document.components = {
|
|
...document.components,
|
|
schemas: { ...document.components?.schemas, ...v2Document.components?.schemas },
|
|
}
|
|
inlineTypedAllOfConstraints(document)
|
|
const schemas = document.components?.schemas
|
|
if (schemas) {
|
|
const reachable = new Set<string>()
|
|
const visit = (value: unknown) => {
|
|
if (Array.isArray(value)) {
|
|
value.forEach(visit)
|
|
return
|
|
}
|
|
if (typeof value !== "object" || value === null) return
|
|
for (const [key, child] of Object.entries(value)) {
|
|
if (key === "$ref" && typeof child === "string" && child.startsWith("#/components/schemas/")) {
|
|
const name = child.slice("#/components/schemas/".length)
|
|
if (reachable.has(name)) continue
|
|
reachable.add(name)
|
|
visit(schemas[name])
|
|
} else {
|
|
visit(child)
|
|
}
|
|
}
|
|
}
|
|
visit({ ...document, components: { ...document.components, schemas: undefined } })
|
|
for (const name of Object.keys(schemas)) {
|
|
if (
|
|
/^(SessionAgentSelected|SessionModelSelected|SessionMoved|SessionRenamed|SessionForked|SessionInputPromoted|SessionInputAdmitted|SessionExecutionStarted|SessionExecutionSucceeded|SessionExecutionFailed|SessionExecutionInterrupted|SessionInstructionsUpdated|SessionSynthetic|SessionSkillActivated|SessionShellStarted|SessionShellEnded|SessionStepStarted|SessionStepEnded|SessionStepFailed|SessionTextStarted|SessionTextDelta|SessionTextEnded|SessionToolInputStarted|SessionToolInputDelta|SessionToolInputEnded|SessionToolCalled|SessionToolProgress|SessionToolSuccess|SessionToolFailed|SessionRetryScheduled|SessionCompactionStarted|SessionCompactionDelta|SessionCompactionEnded|SessionRevertStaged|SessionRevertCleared|SessionRevertCommitted)\d+$/.test(
|
|
name,
|
|
) &&
|
|
!reachable.has(name)
|
|
)
|
|
delete schemas[name]
|
|
}
|
|
await Bun.write("./openapi.json", JSON.stringify(document))
|
|
}
|
|
|
|
await createClient({
|
|
input: "./openapi.json",
|
|
output: {
|
|
path: "./src/v2/gen",
|
|
tsConfigPath: path.join(dir, "tsconfig.json"),
|
|
clean: true,
|
|
},
|
|
plugins: [
|
|
{
|
|
name: "@hey-api/typescript",
|
|
exportFromIndex: false,
|
|
},
|
|
{
|
|
name: "@hey-api/sdk",
|
|
instance: "OpencodeClient",
|
|
exportFromIndex: false,
|
|
auth: false,
|
|
paramsStructure: "flat",
|
|
},
|
|
{
|
|
name: "@hey-api/client-fetch",
|
|
exportFromIndex: false,
|
|
baseUrl: "http://localhost:4096",
|
|
},
|
|
],
|
|
})
|
|
|
|
const generatedTypesPath = "./src/v2/gen/types.gen.ts"
|
|
const generatedTypes = await Bun.file(generatedTypesPath).text()
|
|
if (
|
|
/export type (SessionAgentSelected|SessionModelSelected|SessionMoved|SessionRenamed|SessionForked|SessionInputPromoted|SessionInputAdmitted|SessionExecutionStarted|SessionExecutionSucceeded|SessionExecutionFailed|SessionExecutionInterrupted|SessionInstructionsUpdated|SessionSynthetic|SessionSkillActivated|SessionShellStarted|SessionShellEnded|SessionStepStarted|SessionStepEnded|SessionStepFailed|SessionTextStarted|SessionTextDelta|SessionTextEnded|SessionToolInputStarted|SessionToolInputDelta|SessionToolInputEnded|SessionToolCalled|SessionToolProgress|SessionToolSuccess|SessionToolFailed|SessionRetryScheduled|SessionCompactionStarted|SessionCompactionDelta|SessionCompactionEnded|SessionRevertStaged|SessionRevertCleared|SessionRevertCommitted)\d+ =/.test(
|
|
generatedTypes,
|
|
)
|
|
) {
|
|
throw new Error("Session history generated duplicate Session event variants")
|
|
}
|
|
const sessionErrorTypesPatched = deduplicateEquivalentGeneratedTypes(
|
|
generatedTypes,
|
|
"SessionStructuredError",
|
|
/^SessionStructuredError\d+$/,
|
|
)
|
|
const obsoleteSessionNext = [...sessionErrorTypesPatched.matchAll(/export type (SessionNext\w*) =/g)].map(
|
|
(match) => match[1],
|
|
)
|
|
if (obsoleteSessionNext.length > 0) {
|
|
throw new Error(`Obsolete SessionNext generated type noise reintroduced: ${obsoleteSessionNext.join(", ")}`)
|
|
}
|
|
const logTypesPatched = sessionErrorTypesPatched.replace(
|
|
/(export type V2SessionLogData = \{[\s\S]*?query\?: \{\s*after\?: )string/,
|
|
"$1number",
|
|
)
|
|
if (logTypesPatched === sessionErrorTypesPatched) {
|
|
throw new Error("Session log numeric query patch did not apply")
|
|
}
|
|
const sessionListTypesPatched = logTypesPatched.replace(
|
|
/(export type V2SessionListData = \{[\s\S]*?query\?: \{[\s\S]*?limit\?: )string( \| null)/,
|
|
"$1number$2",
|
|
)
|
|
if (sessionListTypesPatched === logTypesPatched) {
|
|
throw new Error("Session list numeric query patch did not apply")
|
|
}
|
|
const sessionMessagesTypesPatched = sessionListTypesPatched.replace(
|
|
/(export type V2MessageListData = \{[\s\S]*?query\?: \{[\s\S]*?limit\?: )string( \| null)/,
|
|
"$1number$2",
|
|
)
|
|
if (sessionMessagesTypesPatched === sessionListTypesPatched) {
|
|
throw new Error("Session messages numeric query patch did not apply")
|
|
}
|
|
const eventSubscribeTypesPatched = sessionMessagesTypesPatched.replace(
|
|
/(export type V2EventSubscribeResponses = \{\s*\/\*\*[\s\S]*?\*\/\s*200: )\{\s*id: string \| null;?\s*event: string;?\s*data: (?:V2EventStream(?:V2)?|V2EventJsonString);?\s*\};?/,
|
|
"$1V2Event",
|
|
)
|
|
if (eventSubscribeTypesPatched === sessionMessagesTypesPatched) {
|
|
throw new Error("Event subscribe response patch did not apply")
|
|
}
|
|
if (/SessionStructuredError\d/.test(eventSubscribeTypesPatched)) {
|
|
throw new Error("Session structured error generated a name-mangled duplicate")
|
|
}
|
|
if (/\bSessionNext\w*\b/.test(eventSubscribeTypesPatched)) {
|
|
throw new Error("Obsolete SessionNext generated type noise reintroduced")
|
|
}
|
|
if (/export type Shell\d+V2 =/.test(eventSubscribeTypesPatched)) {
|
|
throw new Error("Shell generated a name-mangled duplicate")
|
|
}
|
|
await Bun.write(generatedTypesPath, eventSubscribeTypesPatched)
|
|
|
|
const querySerializerPath = "./src/v2/gen/client/utils.gen.ts"
|
|
const querySerializerSource = await Bun.file(querySerializerPath).text()
|
|
const querySerializerPatched = querySerializerSource.replace(
|
|
/if \(value === undefined \|\| value === null\) \{\s*continue;?\s*\}/,
|
|
"if (value === undefined) {\n continue;\n }\n\n if (value === null) {\n search.push(`${name}=null`);\n continue;\n }",
|
|
)
|
|
if (querySerializerPatched === querySerializerSource) {
|
|
throw new Error(
|
|
`Query serializer null patch did not apply; @hey-api/openapi-ts output may have changed (${querySerializerPath})`,
|
|
)
|
|
}
|
|
await Bun.write(querySerializerPath, querySerializerPatched)
|
|
|
|
const generatedSdkPath = "./src/v2/gen/sdk.gen.ts"
|
|
const generatedSdk = await Bun.file(generatedSdkPath).text()
|
|
const logSdkPatched = generatedSdk.replace(
|
|
/(Read the session log[\s\S]*?parameters: \{[\s\S]*?after\?: )string(\s*\|\s*null)?/,
|
|
"$1number$2",
|
|
)
|
|
if (logSdkPatched === generatedSdk) {
|
|
throw new Error("Session log numeric SDK patch did not apply")
|
|
}
|
|
const sessionListSdkPatched = logSdkPatched.replace(
|
|
/(List sessions[\s\S]*?parameters\?: \{[\s\S]*?limit\?: )string( \| null)/,
|
|
"$1number$2",
|
|
)
|
|
if (sessionListSdkPatched === logSdkPatched) {
|
|
throw new Error("Session list numeric SDK patch did not apply")
|
|
}
|
|
const sessionMessagesSdkPatched = sessionListSdkPatched.replace(
|
|
/(Get session messages[\s\S]*?parameters: \{[\s\S]*?limit\?: )string( \| null)/,
|
|
"$1number$2",
|
|
)
|
|
if (sessionMessagesSdkPatched === sessionListSdkPatched) {
|
|
throw new Error("Session messages numeric SDK patch did not apply")
|
|
}
|
|
await Bun.write(generatedSdkPath, sessionMessagesSdkPatched)
|
|
|
|
// Patch a @hey-api/openapi-ts codegen bug: SseFn incorrectly passes the
|
|
// endpoint's TError into the second generic of ServerSentEventsResult, which
|
|
// is the AsyncGenerator's TReturn slot. Iterator return values have nothing
|
|
// to do with HTTP errors, and any consumer that calls `.return()` or returns
|
|
// from a mock generator gets type-checked against the wrong shape. Drop the
|
|
// arg so TReturn defaults to void.
|
|
const sseTypesPath = "./src/v2/gen/client/types.gen.ts"
|
|
const sseTypesFile = Bun.file(sseTypesPath)
|
|
const sseTypesSource = await sseTypesFile.text()
|
|
const sseTypesPatched = sseTypesSource.replace(
|
|
"=> Promise<ServerSentEventsResult<TData, TError>>",
|
|
"=> Promise<ServerSentEventsResult<TData>>",
|
|
)
|
|
if (sseTypesPatched === sseTypesSource) {
|
|
throw new Error(`SseFn patch did not apply; @hey-api/openapi-ts output may have changed (${sseTypesPath})`)
|
|
}
|
|
await Bun.write(sseTypesPath, sseTypesPatched)
|
|
|
|
await $`bun prettier --write src/gen`
|
|
await $`bun prettier --write src/v2`
|
|
await $`rm -rf dist`
|
|
await $`bun tsc`
|
|
await $`rm openapi.json openapi-v2.json`
|
|
|
|
function renameCollidingComponents(target: OpenApiDocument, source: OpenApiDocument) {
|
|
const targetSchemas = target.components?.schemas
|
|
const sourceSchemas = source.components?.schemas
|
|
if (!targetSchemas || !sourceSchemas) return
|
|
|
|
const renames = new Map<string, string>()
|
|
for (const name of Object.keys(sourceSchemas)) {
|
|
if (!Object.hasOwn(targetSchemas, name)) continue
|
|
if (JSON.stringify(normalizeSchema(sourceSchemas[name])) === JSON.stringify(normalizeSchema(targetSchemas[name]))) {
|
|
delete sourceSchemas[name]
|
|
continue
|
|
}
|
|
let renamed = `${name}V2`
|
|
let index = 2
|
|
while (Object.hasOwn(targetSchemas, renamed) || Object.hasOwn(sourceSchemas, renamed)) {
|
|
renamed = `${name}V2${index}`
|
|
index++
|
|
}
|
|
renames.set(name, renamed)
|
|
}
|
|
if (renames.size === 0) return
|
|
|
|
source.components = {
|
|
...source.components,
|
|
schemas: Object.fromEntries(
|
|
Object.entries(sourceSchemas).map(([name, schema]) => [renames.get(name) ?? name, rewriteRefs(schema, renames)]),
|
|
),
|
|
}
|
|
source.paths = rewriteRefs(source.paths, renames) as Record<string, unknown> | undefined
|
|
}
|
|
|
|
function normalizeComponentNames(document: OpenApiDocument) {
|
|
const schemas = document.components?.schemas
|
|
if (!schemas) return
|
|
|
|
const canonical = new Map(Object.entries(schemas))
|
|
const renames = new Map<string, string>()
|
|
for (const name of Object.keys(schemas)) {
|
|
const next = componentTypeName(name)
|
|
if (next === name) continue
|
|
const existing = canonical.get(next)
|
|
if (existing !== undefined) {
|
|
if (JSON.stringify(normalizeSchema(schemas[name])) !== JSON.stringify(normalizeSchema(existing))) continue
|
|
renames.set(name, next)
|
|
continue
|
|
}
|
|
renames.set(name, next)
|
|
canonical.set(next, schemas[name])
|
|
}
|
|
if (renames.size === 0) return
|
|
|
|
const renamed = new Set<string>()
|
|
document.components = {
|
|
...document.components,
|
|
schemas: Object.fromEntries(
|
|
[
|
|
...Object.entries(schemas).filter(([name]) => !renames.has(name)),
|
|
...Object.entries(schemas).flatMap(([name, schema]) => {
|
|
const next = renames.get(name)
|
|
if (!next || Object.hasOwn(schemas, next) || renamed.has(next)) return []
|
|
renamed.add(next)
|
|
return [[next, schema] as const]
|
|
}),
|
|
].map(([name, schema]) => [name, rewriteRefs(schema, renames)]),
|
|
),
|
|
}
|
|
document.paths = rewriteRefs(document.paths, renames) as Record<string, unknown> | undefined
|
|
}
|
|
|
|
function componentTypeName(name: string) {
|
|
if (!name.includes(".")) return name
|
|
return name
|
|
.split(".")
|
|
.filter((part) => !/^\d+$/.test(part))
|
|
.map((part) => part.slice(0, 1).toUpperCase() + part.slice(1))
|
|
.join("")
|
|
}
|
|
|
|
function deduplicateEquivalentComponent(document: OpenApiDocument, canonical: string, duplicate: string) {
|
|
const schemas = document.components?.schemas
|
|
if (!schemas?.[canonical] || !schemas[duplicate]) return
|
|
if (JSON.stringify(normalizeSchema(schemas[canonical])) !== JSON.stringify(normalizeSchema(schemas[duplicate]))) {
|
|
throw new Error(`${duplicate} no longer has the same wire shape as ${canonical}`)
|
|
}
|
|
|
|
const renames = new Map([[duplicate, canonical]])
|
|
const rewritten = rewriteRefs(schemas, renames) as Record<string, unknown>
|
|
delete rewritten[duplicate]
|
|
document.components = { ...document.components, schemas: rewritten }
|
|
document.paths = rewriteRefs(document.paths, renames) as Record<string, unknown> | undefined
|
|
}
|
|
|
|
function deduplicateEquivalentGeneratedTypes(source: string, canonical: string, duplicates: RegExp) {
|
|
const canonicalType = generatedType(source, canonical)
|
|
if (!canonicalType) throw new Error(`Generated canonical type missing: ${canonical}`)
|
|
const names = [...source.matchAll(/export type (\w+) =/g)]
|
|
.map((match) => match[1])
|
|
.filter((name): name is string => name !== undefined && duplicates.test(name))
|
|
|
|
return names.reduce((patched, name) => {
|
|
const duplicate = generatedType(patched, name)
|
|
const currentCanonical = generatedType(patched, canonical)
|
|
if (!duplicate || !currentCanonical) throw new Error(`Generated type declaration missing while comparing ${name}`)
|
|
if (normalizeGeneratedType(currentCanonical.shape) !== normalizeGeneratedType(duplicate.shape)) {
|
|
throw new Error(`${name} no longer has the same generated type shape as ${canonical}`)
|
|
}
|
|
return (patched.slice(0, duplicate.start) + patched.slice(duplicate.end)).replaceAll(name, canonical)
|
|
}, source)
|
|
}
|
|
|
|
function generatedType(source: string, name: string) {
|
|
const start = source.indexOf(`export type ${name} =`)
|
|
if (start === -1) return undefined
|
|
const next = source.indexOf("\n\nexport type ", start + 1)
|
|
const shapeEnd = next === -1 ? source.length : next
|
|
return {
|
|
start,
|
|
end: next === -1 ? source.length : next + 2,
|
|
shape: source.slice(source.indexOf("=", start) + 1, shapeEnd),
|
|
}
|
|
}
|
|
|
|
function normalizeGeneratedType(shape: string) {
|
|
return shape.replaceAll(/\s/g, "")
|
|
}
|
|
|
|
function normalizeSchema(value: unknown, key?: string): unknown {
|
|
if (Array.isArray(value)) {
|
|
const flattened =
|
|
key === "anyOf"
|
|
? value.flatMap((item) =>
|
|
typeof item === "object" && item !== null && Object.keys(item).length === 1 && "anyOf" in item
|
|
? Array.isArray(item.anyOf)
|
|
? item.anyOf
|
|
: [item]
|
|
: [item],
|
|
)
|
|
: value
|
|
const expanded =
|
|
key === "anyOf"
|
|
? flattened.flatMap((item) => {
|
|
if (typeof item !== "object" || item === null || !("type" in item) || !("enum" in item)) return [item]
|
|
if (Object.keys(item).some((property) => property !== "type" && property !== "enum")) return [item]
|
|
if (!Array.isArray(item.enum)) return [item]
|
|
return item.enum.map((member) => ({ type: item.type, enum: [member] }))
|
|
})
|
|
: flattened
|
|
const normalized = expanded.map((item) => normalizeSchema(item))
|
|
if (key !== "anyOf" && key !== "required" && key !== "enum") return normalized
|
|
return [...new Map(normalized.map((item) => [JSON.stringify(item), item])).values()].sort((a, b) =>
|
|
JSON.stringify(a).localeCompare(JSON.stringify(b)),
|
|
)
|
|
}
|
|
if (typeof value !== "object" || value === null) return value
|
|
return Object.fromEntries(
|
|
Object.entries(value)
|
|
.sort(([left], [right]) => left.localeCompare(right))
|
|
.map(([property, child]) => [property, normalizeSchema(child, property)]),
|
|
)
|
|
}
|
|
|
|
function rewriteRefs(value: unknown, renames: Map<string, string>): unknown {
|
|
if (Array.isArray(value)) return value.map((item) => rewriteRefs(item, renames))
|
|
if (typeof value !== "object" || value === null) return value
|
|
|
|
return Object.fromEntries(
|
|
Object.entries(value).map(([key, child]) => {
|
|
if (key !== "$ref" || typeof child !== "string") return [key, rewriteRefs(child, renames)]
|
|
const prefix = "#/components/schemas/"
|
|
if (!child.startsWith(prefix)) return [key, child]
|
|
return [key, `${prefix}${renames.get(child.slice(prefix.length)) ?? child.slice(prefix.length)}`]
|
|
}),
|
|
)
|
|
}
|
|
|
|
function inlineTypedAllOfConstraints(value: unknown): void {
|
|
if (Array.isArray(value)) {
|
|
value.forEach(inlineTypedAllOfConstraints)
|
|
return
|
|
}
|
|
if (typeof value !== "object" || value === null) return
|
|
|
|
const schema = value as { allOf?: unknown; type?: unknown; [key: string]: unknown }
|
|
if (typeof schema.type === "string" && Array.isArray(schema.allOf) && schema.allOf.every(isConstraintSchema)) {
|
|
for (const item of schema.allOf) Object.assign(schema, item)
|
|
delete schema.allOf
|
|
}
|
|
Object.values(schema).forEach(inlineTypedAllOfConstraints)
|
|
}
|
|
|
|
function isConstraintSchema(value: unknown): value is Record<string, unknown> {
|
|
if (typeof value !== "object" || value === null || Array.isArray(value)) return false
|
|
return !Object.keys(value).some(
|
|
(key) => key === "$ref" || key === "type" || key === "allOf" || key === "anyOf" || key === "oneOf",
|
|
)
|
|
}
|