openAgent

1#!/usr/bin/env bun
2 
3import { Config } from "@/config/config"
4import { ConfigV1 } from "@opencode-ai/core/v1/config/config"
5import { TuiConfig } from "@opencode-ai/tui/config"
6import { Schema } from "effect"
7 
8type JsonSchema = Record<string, unknown>
9const MODEL_REF = "https://models.dev/model-schema.json#/$defs/Model"
10 
11function generateEffect(schema: Schema.Top) {
12  const document = Schema.toJsonSchemaDocument(schema)
13  const normalized = normalize({
14    $schema: "https://json-schema.org/draft/2020-12/schema",
15    ...document.schema,
16    $defs: document.definitions,
17  })
18  if (!isRecord(normalized)) throw new Error("schema generator produced a non-object schema")
19  const restored = restoreModelRefs(normalized)
20  if (!isRecord(restored)) throw new Error("schema generator produced a non-object schema")
21  restored.allowComments = true
22  restored.allowTrailingCommas = true
23  return restored
24}
25 
26function normalize(value: unknown): unknown {
27  if (Array.isArray(value)) return value.map(normalize)
28  if (!isRecord(value)) return value
29 
30  const schema = Object.fromEntries(Object.entries(value).map(([key, item]) => [key, normalize(item)]))
31 
32  if (Array.isArray(schema.anyOf)) {
33    const anyOf = schema.anyOf.filter((item) => !isRecord(item) || item.type !== "null")
34    if (anyOf.length !== schema.anyOf.length) {
35      const { anyOf: _, ...rest } = schema
36      if (anyOf.length === 1 && isRecord(anyOf[0])) return normalize({ ...anyOf[0], ...rest })
37      return { ...rest, anyOf }
38    }
39  }
40 
41  if (Array.isArray(schema.allOf) && schema.allOf.length === 1 && isRecord(schema.allOf[0])) {
42    const { allOf: _, ...rest } = schema
43    return normalize({ ...schema.allOf[0], ...rest })
44  }
45 
46  if (schema.type === "integer" && schema.maximum === undefined) {
47    return { ...schema, maximum: Number.MAX_SAFE_INTEGER }
48  }
49 
50  return schema
51}
52 
53function restoreModelRefs(value: unknown, key?: string): unknown {
54  if (Array.isArray(value)) return value.map((item) => restoreModelRefs(item))
55  if (!isRecord(value)) return value
56 
57  const schema = Object.fromEntries(Object.entries(value).map(([name, item]) => [name, restoreModelRefs(item, name)]))
58  if ((key === "model" || key === "small_model") && schema.type === "string") {
59    return { ...schema, $ref: MODEL_REF }
60  }
61  return schema
62}
63 
64function isRecord(value: unknown): value is JsonSchema {
65  return typeof value === "object" && value !== null && !Array.isArray(value)
66}
67 
68const configFile = process.argv[2]
69const tuiFile = process.argv[3]
70 
71console.log(configFile)
72await Bun.write(configFile, JSON.stringify(generateEffect(ConfigV1.Info), null, 2))
73 
74if (tuiFile) {
75  console.log(tuiFile)
76  await Bun.write(tuiFile, JSON.stringify(generateEffect(TuiConfig.Info), null, 2))
77}
78