wiki-graph-sdk 0.8.1 → 0.8.2

package/dist/default-worker.cjs

@@ -750,15 +750,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) {
-  (0, import_platform.installWikiGraphPlatform)(nodeWikiGraphPlatform);
-  (0, import_platform.installWikiGraphStorage)(createNodeWikiGraphStorage(stateRoot));
+function ensureNodeWikiGraphPlatform() {
+  try {
+    (0, import_platform.getWikiGraphPlatform)();
+  } catch {
+    (0, import_platform.installWikiGraphPlatform)(nodeWikiGraphPlatform);
+  }
+  try {
+    (0, import_platform.getWikiGraphStorage)();
+  } catch {
+    (0, import_platform.installWikiGraphStorage)(createNodeWikiGraphStorage());
+  }
 }
 async function withNodeWikiGraphStorage(stateRoot, operation) {
   return await (0, import_platform.withWikiGraphStorage)(
@@ -1468,6 +1481,31 @@ function abortError(signal) {
 var import_worker = require("wiki-graph-core/worker");
 var import_wiki_graph_core7 = require("wiki-graph-core");
 
+// src/host.ts
+var import_platform2 = require("wiki-graph-core/platform");
+function prepareWikiGraphSDKHost(host) {
+  if (host?.platform === void 0) {
+    ensureNodeWikiGraphPlatform();
+    return;
+  }
+  (0, import_platform2.installWikiGraphPlatform)(host.platform);
+}
+function assertWikiGraphSDKHostOptions(options) {
+  if (options.host?.storage !== void 0 && options.stateDir !== void 0) {
+    throw new TypeError(
+      "Wiki Graph SDK host.storage and stateDir cannot be used together."
+    );
+  }
+}
+async function withWikiGraphSDKHost(options, operation) {
+  assertWikiGraphSDKHostOptions(options);
+  prepareWikiGraphSDKHost(options.host);
+  if (options.host?.storage !== void 0) {
+    return await (0, import_platform2.withWikiGraphStorage)(options.host.storage, operation);
+  }
+  return await withNodeWikiGraphStorage(options.stateDir, operation);
+}
+
 // src/archive/target.ts
 var import_promises2 = require("fs/promises");
 var import_path3 = require("path");
@@ -1683,10 +1721,12 @@ async function writeWikiGraphArchiveLocation(location, operation, options = {})
 // src/worker.ts
 var import_worker2 = require("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 (0, import_worker.runBuildJobWorker)(coreOptions)
   );
 }