wehub-resource-sync
66 lines
2.2 KiB
TypeScript
66 lines
2.2 KiB
TypeScript
#!/usr/bin/env bun
|
|
// Generates src/generated.ts (model family union) and
|
|
// src/snapshot.js (the bundled data snapshot) from this repository's TOMLs.
|
|
|
|
import path from "node:path"
|
|
import { generateCatalog, ModelFamilyValues } from "@models.dev/core"
|
|
|
|
const root = path.join(import.meta.dirname, "..", "..", "..")
|
|
const src = path.join(import.meta.dirname, "..", "src")
|
|
|
|
function sortRecord<T>(record: Record<string, T>): Record<string, T> {
|
|
return Object.fromEntries(Object.entries(record).sort(([a], [b]) => (a < b ? -1 : a > b ? 1 : 0)))
|
|
}
|
|
|
|
/** Deterministic catalog: provider, per-provider model, and metadata keys sorted. */
|
|
export async function loadCatalog() {
|
|
const catalog = await generateCatalog(root)
|
|
const providers = sortRecord(
|
|
Object.fromEntries(
|
|
Object.entries(catalog.providers).map(([id, provider]) => [id, { ...provider, models: sortRecord(provider.models) }]),
|
|
),
|
|
)
|
|
return { providers, models: sortRecord(catalog.models) }
|
|
}
|
|
|
|
/** The exact JSON payload embedded in src/snapshot.js. Used by publish to diff against npm. */
|
|
export function snapshotPayload(catalog: Awaited<ReturnType<typeof loadCatalog>>) {
|
|
return JSON.stringify(catalog)
|
|
}
|
|
|
|
function union(values: string[]) {
|
|
return values.map((value) => ` | ${JSON.stringify(value)}`).join("\n")
|
|
}
|
|
|
|
export async function generate() {
|
|
const catalog = await loadCatalog()
|
|
|
|
const families = [...new Set<string>(ModelFamilyValues)].sort()
|
|
await Bun.write(
|
|
path.join(src, "generated.ts"),
|
|
`// Generated by script/generate.ts. Do not edit; run \`bun run generate\` in packages/sdk.
|
|
|
|
/** Model family identifiers used to group related models. */
|
|
export type ModelFamily =
|
|
${union(families)}
|
|
`,
|
|
)
|
|
|
|
await Bun.write(
|
|
path.join(src, "snapshot.js"),
|
|
`// Generated by script/generate.ts. Do not edit; run \`bun run generate\` in packages/sdk.
|
|
const data = /* @__PURE__ */ JSON.parse(${JSON.stringify(snapshotPayload(catalog))})
|
|
export const providers = data.providers
|
|
export const models = data.models
|
|
export const generatedAt = ${JSON.stringify(new Date().toISOString())}
|
|
export default data
|
|
`,
|
|
)
|
|
|
|
}
|
|
|
|
if (import.meta.main) {
|
|
await generate()
|
|
console.log("generated src/generated.ts and src/snapshot.js")
|
|
}
|