wiki-graph-sdk 0.8.1 → 0.8.2

package/dist/default-worker.js

@@ -75,6 +75,7 @@ import * as yauzl from "yauzl";
 import * as yazl from "yazl";
 import { Environment, Loader } from "nunjucks";
 import {
+  getWikiGraphPlatform,
   getWikiGraphStorage,
   installWikiGraphPlatform,
   installWikiGraphStorage,
@@ -736,15 +737,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(
@@ -1466,6 +1480,34 @@ 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.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 withWikiGraphStorage2(options.host.storage, operation);
+  }
+  return await withNodeWikiGraphStorage(options.stateDir, operation);
+}
+
 // 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 +1739,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)
   );
 }