okengine 0.11.0 → 0.11.2

package/src/shared/lazy-src.ts

@@ -0,0 +1,79 @@
+/**
+ * Sync-load package-local modules that Bun.build leaves as runtime requires
+ * (not inlined). Bun `src/` first; published `dist/` chunks for `"import"`.
+ *
+ * Uses Bun APIs (`import.meta.dir`, `import.meta.require`) and try/catch load
+ * instead of exists()+require — Bun’s preferred pattern (the exists syscall
+ * is an extra round trip).
+ */
+
+import { dirname, join } from "node:path";
+
+let cachedRoot: string | undefined;
+
+/**
+ * Try a sync `import.meta.require`; return `undefined` only when the file is missing.
+ *
+ * @typeParam T - Module namespace shape
+ * @param path - Absolute module path
+ */
+function tryRequire<T>(path: string): T | undefined {
+  try {
+    return import.meta.require(path) as T;
+  } catch (err) {
+    const code =
+      err !== null && typeof err === "object" && "code" in err
+        ? String((err as { code?: unknown }).code)
+        : undefined;
+    if (
+      code === "MODULE_NOT_FOUND" ||
+      code === "ENOENT" ||
+      (err instanceof Error && /Cannot find module|ENOENT/i.test(err.message))
+    ) {
+      return undefined;
+    }
+    throw err;
+  }
+}
+
+/**
+ * Walk up from this module to the okengine package root (`package.json` name).
+ */
+export function okengineRoot(): string {
+  if (cachedRoot) return cachedRoot;
+  let dir = import.meta.dir;
+  for (let i = 0; i < 12; i++) {
+    const pkg = tryRequire<{ name?: string }>(join(dir, "package.json"));
+    if (pkg?.name === "okengine") {
+      cachedRoot = dir;
+      return dir;
+    }
+    const parent = dirname(dir);
+    if (parent === dir) break;
+    dir = parent;
+  }
+  throw new Error("okengine: package root not found from " + import.meta.dir);
+}
+
+/**
+ * Synchronously load `src/<rel>.ts` (Bun) or published `dist/<distName>.js`.
+ *
+ * @typeParam T - Module namespace shape
+ * @param relSrc - Path under `src/` without extension (e.g. `auth/config`)
+ * @param distName - Filename under `dist/` without extension (e.g. `auth-config`)
+ */
+export function requirePackageModule<T>(relSrc: string, distName: string): T {
+  const root = okengineRoot();
+  const srcPath = join(root, "src", `${relSrc}.ts`);
+  const distPath = join(root, "dist", `${distName}.js`);
+  // Bun native TS — prefer source so local `oke` / tests stay on one graph.
+  if (typeof Bun !== "undefined") {
+    const fromSrc = tryRequire<T>(srcPath);
+    if (fromSrc !== undefined) return fromSrc;
+  }
+  const fromDist = tryRequire<T>(distPath);
+  if (fromDist !== undefined) return fromDist;
+  const fallback = tryRequire<T>(srcPath);
+  if (fallback !== undefined) return fallback;
+  throw new Error(`okengine: missing lazy module src/${relSrc}.ts or dist/${distName}.js`);
+}

package/src/test/create-test-app.ts

@@ -14,9 +14,6 @@
  * ```
  */
 
-import { mkdtemp, rm } from "node:fs/promises";
-import { tmpdir } from "node:os";
-import { join } from "node:path";
 import {
   createMockAiDriver,
   createChannelInbox,
@@ -215,15 +212,26 @@ export async function createTestApp<App extends OkeApp>(
     return out;
   };
 
-  // Unique on-disk PGLite datadir per harness — `memory://` is not reliably
-  // isolated across boots in one worker (IF NOT EXISTS / row leaks).
-  const prevPgliteUrl = process.env.OKE_PGLITE_URL;
-  const pgliteDir = await mkdtemp(join(tmpdir(), "oke-pglite-"));
-  process.env.OKE_PGLITE_URL = pgliteDir;
+  // Harness contract is in-process memory SQL (same as vault/channel/ai/runs).
+  // Global `STORE_SQL_DEFAULTS.test` is PGLite for real app boots — do not inherit
+  // that here, or every insert pays cold WASM+pgvector init (~seconds). Dialects
+  // that need PGLite pass `boot.config.drivers.store.sql` explicitly.
+  const bootConfig = options.boot?.config;
+  const config = {
+    ...(bootConfig ?? {}),
+    drivers: {
+      ...(bootConfig?.drivers ?? {}),
+      store: {
+        ...(bootConfig?.drivers?.store ?? {}),
+        sql: bootConfig?.drivers?.store?.sql ?? { test: "memory" },
+      },
+    },
+  };
 
   await app.boot({
     ...(options.boot ?? {}),
     env: "test",
+    config,
     // Test-only opt-out (honoured because env is "test" below). Production
     // boots never skip posture via this flag.
     unguardedHttp: options.boot?.unguardedHttp ?? appOpts.gate?.unguardedHttp ?? "allow",
@@ -331,9 +339,6 @@ export async function createTestApp<App extends OkeApp>(
     },
     async close() {
       await app.bootResult?.close();
-      if (prevPgliteUrl === undefined) delete process.env.OKE_PGLITE_URL;
-      else process.env.OKE_PGLITE_URL = prevPgliteUrl;
-      await rm(pgliteDir, { recursive: true, force: true }).catch(() => undefined);
     },
   };
 

package/src/test/reset-element-registries.ts

@@ -1,22 +1,26 @@
 /**
  * Global safety net for the `store.sql` / `store.files` / `vault.secret` /
- * `signal()` / `channel.<medium>().template()` auto-registries
- * (`src/kernel/element-registries.ts`).
+ * `signal()` / `channel.<medium>().template()` / `ai.model`·prompt·embed·agent
+ * auto-registries (`src/kernel/element-registries.ts`).
  *
  * Unlike `on()` bindings — created almost exclusively to wire a real app —
- * these four factories are called throughout the suite as bare value
- * constructors, completely unrelated to booting an app (hundreds of call
- * sites across `src/elements/*.test.ts`). Since the registries are plain
- * module-level arrays shared by every test file in one `bun test` process,
- * leaving cleanup to per-file discipline (the convention `on.ts` relies on —
- * see `resetBindings()` / `registry: "ignore"`) would let stray decls from
- * one file silently reach a default-registry `oke()` call in a completely
+ * these factories are called throughout the suite as bare value constructors,
+ * completely unrelated to booting an app (hundreds of call sites across
+ * `src/elements/*.test.ts`). Since the registries are plain module-level
+ * arrays shared by every test file in one `bun test` process, leaving cleanup
+ * to per-file discipline (the convention `on.ts` relies on — see
+ * `resetBindings()` / `registry: "ignore"`) would let stray decls from one
+ * file silently reach a default-registry `oke()` call in a completely
  * unrelated file. Reset after every test, globally, via `bunfig.toml`
  * `[test].preload`.
  */
 
 import { afterEach } from "bun:test";
 import {
+  aiAgentRegistry,
+  aiEmbedRegistry,
+  aiModelRegistry,
+  aiPromptRegistry,
   channelTemplateRegistry,
   requiredEnvRegistry,
   secretRegistry,
@@ -30,4 +34,8 @@ afterEach(() => {
   requiredEnvRegistry.length = 0;
   signalRegistry.length = 0;
   channelTemplateRegistry.length = 0;
+  aiModelRegistry.length = 0;
+  aiPromptRegistry.length = 0;
+  aiEmbedRegistry.length = 0;
+  aiAgentRegistry.length = 0;
 });