@zhivex-ai/harness 1.2.0 → 1.3.0-rc.7

package/dist/index-wg1sd3at.js

@@ -0,0 +1,80 @@
+import {
+  bundledModelCatalog,
+  modelCatalogSchema
+} from "./index-xqxeypt5.js";
+
+// src/models/catalog-store.ts
+import { mkdir, readFile, rename, writeFile, unlink } from "node:fs/promises";
+import { homedir } from "node:os";
+import path from "node:path";
+import { createHash, randomUUID } from "node:crypto";
+var MAX_BYTES = 1024 * 1024;
+var TTL = 60 * 60 * 1000;
+async function loadModelCatalog(options = {}) {
+  const fallback = { catalog: bundledModelCatalog, source: "bundled", stale: false };
+  const source = options.url ?? process.env.ZHIVEX_MODEL_CATALOG_URL;
+  if (!source)
+    return fallback;
+  let url;
+  try {
+    url = new URL(source);
+    if (url.protocol !== "https:" || url.username || url.password || url.hash)
+      return { ...fallback, stale: true };
+  } catch {
+    return { ...fallback, stale: true };
+  }
+  const now = options.now ?? Date.now();
+  const directory = options.cacheDirectory ?? path.join(homedir(), ".zhivex", "model-catalog");
+  const file = path.join(directory, `${createHash("sha256").update(url.href).digest("hex")}.json`);
+  let cached;
+  try {
+    const raw = await readFile(file, "utf8");
+    if (Buffer.byteLength(raw) > MAX_BYTES + 1024)
+      throw new Error("CATALOG_TOO_LARGE");
+    const envelope = JSON.parse(raw);
+    if (envelope.url !== url.href || !Number.isFinite(envelope.fetchedAt))
+      throw new Error("INVALID_CACHE");
+    cached = { catalog: modelCatalogSchema.parse(envelope.catalog), source: "cache", stale: true };
+    if (!options.refresh && now >= envelope.fetchedAt && now - envelope.fetchedAt < TTL)
+      return { ...cached, stale: false };
+  } catch {}
+  try {
+    const response = await (options.fetcher ?? fetch)(url, {
+      signal: AbortSignal.timeout(3000),
+      redirect: "error",
+      headers: { accept: "application/json" }
+    });
+    if (!response.ok || !response.body)
+      throw new Error("CATALOG_FETCH_FAILED");
+    const reader = response.body.getReader();
+    const chunks = [];
+    let size = 0;
+    try {
+      for (;; ) {
+        const { value, done } = await reader.read();
+        if (done)
+          break;
+        size += value.byteLength;
+        if (size > MAX_BYTES)
+          throw new Error("CATALOG_TOO_LARGE");
+        chunks.push(value);
+      }
+    } finally {
+      await reader.cancel().catch(() => {});
+    }
+    const catalog = modelCatalogSchema.parse(JSON.parse(Buffer.concat(chunks).toString("utf8")));
+    const temporary = `${file}.${randomUUID()}.tmp`;
+    try {
+      await mkdir(directory, { recursive: true, mode: 448 });
+      await writeFile(temporary, JSON.stringify({ url: url.href, fetchedAt: now, catalog }), { mode: 384, flag: "wx" });
+      await rename(temporary, file);
+    } catch {} finally {
+      await unlink(temporary).catch(() => {});
+    }
+    return { catalog, source: "remote", stale: false };
+  } catch {
+    return cached ?? { ...fallback, stale: true };
+  }
+}
+
+export { loadModelCatalog };