@supatype/cli 0.1.0-alpha.10

package/src/app-config.ts

@@ -0,0 +1,128 @@
+import { readFileSync, writeFileSync, existsSync } from "node:fs"
+import { resolve } from "node:path"
+import { Node, Project, QuoteKind, SyntaxKind, type ObjectLiteralExpression, type SourceFile } from "ts-morph"
+
+export interface UpdateAppConfigInput {
+  mode: "none" | "static" | "proxy"
+  staticDir?: string
+  upstream?: string
+}
+
+export function updateAppConfigInProject(cwd: string, input: UpdateAppConfigInput): string {
+  const configPath = resolve(cwd, "supatype.config.ts")
+  if (!existsSync(configPath)) {
+    throw new Error("supatype.config.ts not found. Run: supatype init")
+  }
+  const next = updateAppConfigAst(configPath, input)
+  writeFileSync(configPath, next, "utf8")
+  return configPath
+}
+
+function updateAppConfigAst(configPath: string, input: UpdateAppConfigInput): string {
+  const project = new Project({
+    useInMemoryFileSystem: false,
+    skipAddingFilesFromTsConfig: true,
+    manipulationSettings: {
+      quoteKind: QuoteKind.Double,
+    },
+  })
+  const srcText = readFileSync(configPath, "utf8")
+  const sourceFile = project.createSourceFile(configPath, srcText, { overwrite: true })
+  const rootObject = getRootConfigObject(sourceFile)
+
+  const appProperty = rootObject.getProperty("app")
+
+  if (appProperty === undefined) {
+    rootObject.addPropertyAssignment({
+      name: "app",
+      initializer: renderAppInitializer(input),
+    })
+  } else if (Node.isPropertyAssignment(appProperty)) {
+    const init = appProperty.getInitializer()
+    if (init && Node.isObjectLiteralExpression(init)) {
+      patchAppObject(init, input)
+    } else {
+      appProperty.setInitializer(renderAppInitializer(input))
+    }
+  } else {
+    appProperty.remove()
+    rootObject.addPropertyAssignment({
+      name: "app",
+      initializer: renderAppInitializer(input),
+    })
+  }
+
+  return sourceFile.getFullText()
+}
+
+function getRootConfigObject(sourceFile: SourceFile): ObjectLiteralExpression {
+  const exportAssignment = sourceFile.getFirstDescendantByKind(SyntaxKind.ExportAssignment)
+  if (!exportAssignment) {
+    throw new Error("Could not find default export in supatype.config.ts.")
+  }
+  const expr = exportAssignment.getExpression()
+  if (Node.isObjectLiteralExpression(expr)) {
+    return expr
+  }
+  if (Node.isCallExpression(expr)) {
+    const [firstArg] = expr.getArguments()
+    if (firstArg && Node.isObjectLiteralExpression(firstArg)) {
+      return firstArg
+    }
+  }
+  throw new Error(
+    "supatype.config.ts must export an object literal or defineConfig({...}).",
+  )
+}
+
+function renderAppInitializer(input: UpdateAppConfigInput): string {
+  if (input.mode === "proxy") {
+    return `{
+      mode: "proxy",
+      upstream: "${input.upstream ?? "http://localhost:3000"}",
+    }`
+  }
+  if (input.mode === "static") {
+    return `{
+      mode: "static",
+      static_dir: "${input.staticDir ?? "./dist"}",
+    }`
+  }
+  return `{
+    mode: "none",
+  }`
+}
+
+function patchAppObject(appObj: ObjectLiteralExpression, input: UpdateAppConfigInput): void {
+  upsertStringProperty(appObj, "mode", input.mode)
+
+  if (input.mode === "proxy") {
+    upsertStringProperty(appObj, "upstream", input.upstream ?? "http://localhost:3000")
+    removePropertyIfPresent(appObj, "static_dir")
+    return
+  }
+
+  if (input.mode === "static") {
+    upsertStringProperty(appObj, "static_dir", input.staticDir ?? "./dist")
+    removePropertyIfPresent(appObj, "upstream")
+    return
+  }
+
+  removePropertyIfPresent(appObj, "upstream")
+  removePropertyIfPresent(appObj, "static_dir")
+}
+
+function upsertStringProperty(obj: ObjectLiteralExpression, key: string, value: string): void {
+  const existing = obj.getProperty(key)
+  if (existing && Node.isPropertyAssignment(existing)) {
+    existing.setInitializer(`"${value}"`)
+    return
+  }
+  if (existing) existing.remove()
+  obj.addPropertyAssignment({ name: key, initializer: `"${value}"` })
+}
+
+function removePropertyIfPresent(obj: ObjectLiteralExpression, key: string): void {
+  const prop = obj.getProperty(key)
+  if (prop) prop.remove()
+}

package/src/augmentation-generator.ts

@@ -0,0 +1,126 @@
+interface AstModel {
+  tableName: string
+  fields: Record<string, Record<string, unknown>>
+}
+
+interface AstShape {
+  models?: AstModel[]
+}
+
+export function generateClientAugmentation(ast: unknown): string {
+  const parsed = ast as AstShape
+  const models = [...(parsed.models ?? [])].sort((a, b) => a.tableName.localeCompare(b.tableName))
+
+  const lines: string[] = []
+  lines.push("// Generated by supatype CLI — do not edit manually.")
+  lines.push("")
+  lines.push("declare module \"@supatype/client\" {")
+  lines.push("  interface SupatypeModels {")
+
+  for (const model of models) {
+    lines.push(`    ${quoteKey(model.tableName)}: {`)
+    lines.push(`      Row: ${generateRowType(model.fields)}`)
+    lines.push(`      Insert: ${generateInsertType(model.fields)}`)
+    lines.push(`      Update: ${generateUpdateType(model.fields)}`)
+    lines.push("    }")
+  }
+
+  lines.push("  }")
+  lines.push("}")
+  lines.push("")
+  lines.push("export {}")
+  lines.push("")
+
+  return lines.join("\n")
+}
+
+function generateRowType(fields: Record<string, Record<string, unknown>>): string {
+  const entries = Object.entries(fields).sort(([a], [b]) => a.localeCompare(b))
+  if (entries.length === 0) return "Record<string, unknown>"
+  const body = entries
+    .map(([name, meta]) => `  ${quoteKey(name)}: ${toTsType(meta)}`)
+    .join("\n")
+  return `{\n${body}\n}`
+}
+
+function insertColumnOptionalOnInsert(meta: Record<string, unknown>): boolean {
+  return meta["default"] !== undefined || meta["serverGenerated"] === true
+}
+
+function generateInsertType(fields: Record<string, Record<string, unknown>>): string {
+  const entries = Object.entries(fields).sort(([a], [b]) => a.localeCompare(b))
+  if (entries.length === 0) return "Record<string, unknown>"
+  const body = entries
+    .map(([name, meta]) => {
+      const required = meta["required"] === true && !insertColumnOptionalOnInsert(meta)
+      return `  ${quoteKey(name)}${required ? "" : "?"}: ${toTsType(meta)}`
+    })
+    .join("\n")
+  return `{\n${body}\n}`
+}
+
+function generateUpdateType(fields: Record<string, Record<string, unknown>>): string {
+  const entries = Object.entries(fields).sort(([a], [b]) => a.localeCompare(b))
+  if (entries.length === 0) return "Record<string, unknown>"
+  const body = entries
+    .map(([name, meta]) => `  ${quoteKey(name)}?: ${toTsType(meta)}`)
+    .join("\n")
+  return `{\n${body}\n}`
+}
+
+function toTsType(meta: Record<string, unknown>): string {
+  const kind = typeof meta["kind"] === "string" ? meta["kind"] : "json"
+  const required = meta["required"] === true
+  const base = (() => {
+    switch (kind) {
+      case "uuid":
+      case "slug":
+      case "text":
+      case "email":
+      case "url":
+      case "date":
+      case "timestamp":
+      case "datetime":
+      case "money":
+      case "xml":
+      case "interval":
+        return "string"
+      case "integer":
+      case "smallInt":
+      case "serial":
+      case "float":
+      case "decimal":
+        return "number"
+      case "bigSerial":
+      case "bigInt":
+        return "bigint"
+      case "boolean":
+        return "boolean"
+      case "json":
+      case "blocks":
+      case "geo":
+      case "vector":
+      case "relation":
+      case "array":
+      case "image":
+      case "file":
+        return "Record<string, unknown>"
+      case "richText":
+        return "(import(\"@supatype/types/lexical\").SerializedEditorState | string)"
+      case "enum": {
+        const values = Array.isArray(meta["values"]) ? meta["values"] as unknown[] : []
+        const quoted = values
+          .filter((v): v is string => typeof v === "string")
+          .map((v) => JSON.stringify(v))
+        return quoted.length > 0 ? quoted.join(" | ") : "string"
+      }
+      default:
+        return "unknown"
+    }
+  })()
+  return required ? base : `${base} | null`
+}
+
+function quoteKey(key: string): string {
+  return /^[A-Za-z_$][A-Za-z0-9_$]*$/.test(key) ? key : JSON.stringify(key)
+}