wiki-graph-sdk 0.8.1 → 0.8.3

package/dist/default-worker.js

@@ -45,6 +45,9 @@ function getRuntimeContextStorage() {
   });
   return storage;
 }
+async function withWikiGraphSDKRuntimeContext(context, operation) {
+  return await runtimeContext.run(context, operation);
+}
 function getWikiGraphSDKRuntimeContext() {
   return runtimeContext.getStore() ?? {
     cwd: process.cwd(),
@@ -54,7 +57,8 @@ function getWikiGraphSDKRuntimeContext() {
   };
 }
 function getWikiGraphSDKStateDir() {
-  return getWikiGraphSDKRuntimeContext().stateDir;
+  const context = getWikiGraphSDKRuntimeContext();
+  return context.host?.stateDir ?? context.stateDir;
 }
 function getWikiGraphSDKProviders() {
   return getWikiGraphSDKRuntimeContext().providers ?? {};
@@ -75,6 +79,7 @@ import * as yauzl from "yauzl";
 import * as yazl from "yazl";
 import { Environment, Loader } from "nunjucks";
 import {
+  getWikiGraphPlatform,
   getWikiGraphStorage,
   installWikiGraphPlatform,
   installWikiGraphStorage,
@@ -736,15 +741,28 @@ function resolveNodeDataDirectory() {
     directory = parent;
   }
 }
-function createNodeWikiGraphStorage(stateRoot = pathModule.join(os.homedir(), ".wikigraph")) {
+function createNodeWikiGraphStorage(stateRootOrOptions = pathModule.join(
+  os.homedir(),
+  ".wikigraph"
+)) {
+  const libraryRoot = typeof stateRootOrOptions === "string" ? stateRootOrOptions : stateRootOrOptions.libraryRoot;
+  const documentStoreRoot = typeof stateRootOrOptions === "string" ? pathModule.join(stateRootOrOptions, "documents") : stateRootOrOptions.documentStoreRoot;
   return {
-    library: new NodeDirectory(stateRoot),
-    documentStore: new NodeDirectory(pathModule.join(stateRoot, "documents"))
+    library: new NodeDirectory(libraryRoot),
+    documentStore: new NodeDirectory(documentStoreRoot)
   };
 }
-function installNodeWikiGraphPlatform(stateRoot) {
-  installWikiGraphPlatform(nodeWikiGraphPlatform);
-  installWikiGraphStorage(createNodeWikiGraphStorage(stateRoot));
+function ensureNodeWikiGraphPlatform() {
+  try {
+    getWikiGraphPlatform();
+  } catch {
+    installWikiGraphPlatform(nodeWikiGraphPlatform);
+  }
+  try {
+    getWikiGraphStorage();
+  } catch {
+    installWikiGraphStorage(createNodeWikiGraphStorage());
+  }
 }
 async function withNodeWikiGraphStorage(stateRoot, operation) {
   return await withWikiGraphStorage(
@@ -754,6 +772,7 @@ async function withNodeWikiGraphStorage(stateRoot, operation) {
 }
 
 // src/local-config.ts
+import { mkdir as mkdir2 } from "fs/promises";
 import { join as join2 } from "path";
 import { homedir as homedir2 } from "os";
 var LOCAL_CONFIG_SCHEMA_SQL = `
@@ -783,13 +802,10 @@ function validateLocalConfigSection(section, value) {
   }
 }
 async function withLocalConfigDatabase(operation) {
+  const stateDir = getWikiGraphSDKStateDir() ?? join2(homedir2(), ".wikigraph");
+  await mkdir2(stateDir, { recursive: true });
   const database = await openSharedStateDatabase(
-    new NodeFile(
-      join2(
-        getWikiGraphSDKStateDir() ?? join2(homedir2(), ".wikigraph"),
-        "core.sqlite"
-      )
-    ),
+    new NodeFile(join2(stateDir, "core.sqlite")),
     LOCAL_CONFIG_SCHEMA_SQL
   );
   try {
@@ -1466,6 +1482,55 @@ import {
   WikiGraphArchiveFile as WikiGraphArchiveFile2
 } from "wiki-graph-core";
 
+// src/host.ts
+import {
+  installWikiGraphPlatform as installWikiGraphPlatform2,
+  withWikiGraphStorage as withWikiGraphStorage2
+} from "wiki-graph-core/platform";
+function prepareWikiGraphSDKHost(host) {
+  if (host?.platform === void 0) {
+    ensureNodeWikiGraphPlatform();
+    return;
+  }
+  installWikiGraphPlatform2(host.platform);
+}
+function assertWikiGraphSDKHostOptions(options) {
+  if (options.host?.storage !== void 0 && options.host.stateDir === void 0) {
+    throw new TypeError(
+      "Wiki Graph SDK host.storage requires host.stateDir so SDK configuration cannot fall back to the default user state directory."
+    );
+  }
+  if (options.stateDir !== void 0 && (options.host?.stateDir !== void 0 || options.host?.storage !== void 0)) {
+    throw new TypeError(
+      "Wiki Graph SDK host state/storage and top-level stateDir cannot be used together. Put the state directory in host.stateDir when providing host.storage."
+    );
+  }
+}
+async function withWikiGraphSDKHost(options, operation) {
+  const current = getWikiGraphSDKRuntimeContext();
+  const effective = {
+    ...options.host === void 0 ? current.host === void 0 ? {} : { host: current.host } : { host: options.host },
+    ...options.stateDir === void 0 ? current.stateDir === void 0 ? {} : { stateDir: current.stateDir } : { stateDir: options.stateDir }
+  };
+  assertWikiGraphSDKHostOptions(effective);
+  prepareWikiGraphSDKHost(effective.host);
+  const runWithContext = async () => await withWikiGraphSDKRuntimeContext(
+    {
+      ...current,
+      ...effective.host === void 0 ? {} : { host: effective.host },
+      ...effective.stateDir === void 0 ? {} : { stateDir: effective.stateDir }
+    },
+    operation
+  );
+  if (effective.host?.storage !== void 0) {
+    return await withWikiGraphStorage2(effective.host.storage, runWithContext);
+  }
+  return await withNodeWikiGraphStorage(
+    effective.host?.stateDir ?? effective.stateDir,
+    runWithContext
+  );
+}
+
 // src/archive/target.ts
 import { realpath, stat as stat2 } from "fs/promises";
 import { basename as basename2, dirname as dirname2, isAbsolute as isAbsolute3, join as join3, relative, resolve as resolve3 } from "path";
@@ -1697,10 +1762,12 @@ import {
   writeChapterJobInputFile as writeChapterJobInputFile2
 } from "wiki-graph-core/worker";
 async function runBuildJobWorker(options) {
-  installNodeWikiGraphPlatform();
-  const { stateDir, ...coreOptions } = options;
-  await withNodeWikiGraphStorage(
-    stateDir,
+  const { host, stateDir, ...coreOptions } = options;
+  await withWikiGraphSDKHost(
+    {
+      ...host === void 0 ? {} : { host },
+      ...stateDir === void 0 ? {} : { stateDir }
+    },
     async () => await runCoreBuildJobWorker(coreOptions)
   );
 }