@workglow/storage 0.1.1 → 0.2.0

package/dist/bun.js

@@ -14,16 +14,10 @@ import { BaseError } from "@workglow/util";
 
 class StorageError extends BaseError {
   static type = "StorageError";
-  constructor(message) {
-    super(message);
-  }
 }
 
 class StorageValidationError extends StorageError {
   static type = "StorageValidationError";
-  constructor(message) {
-    super(message);
-  }
 }
 
 class StorageEmptyCriteriaError extends StorageValidationError {
@@ -124,14 +118,10 @@ class BaseTabularStorage {
       }
     }
     if (autoGeneratedKeys.length > 1) {
-      throw new Error(`Multiple auto-generated keys detected: ${autoGeneratedKeys.join(", ")}. ` + `Only the first primary key column can be auto-generated.`);
+      throw new Error(`Multiple auto-generated keys detected: ${autoGeneratedKeys.join(", ")}. ` + `At most one primary key column can be auto-generated.`);
     }
     if (autoGeneratedKeys.length > 0) {
       const autoGenKeyName = autoGeneratedKeys[0];
-      const firstPrimaryKey = String(primaryKeyNames[0]);
-      if (autoGenKeyName !== firstPrimaryKey) {
-        throw new Error(`Auto-generated key "${autoGenKeyName}" must be the first primary key column. ` + `Current first primary key is "${firstPrimaryKey}".`);
-      }
       this.autoGeneratedKeyName = autoGenKeyName;
       this.autoGeneratedKeyStrategy = this.determineGenerationStrategy(autoGenKeyName, schema.properties[autoGenKeyName]);
     }
@@ -1113,12 +1103,11 @@ function hfFeatureToJsonSchema(feature) {
 import {
   createServiceToken as createServiceToken5,
   globalServiceRegistry,
+  registerInputCompactor,
   registerInputResolver
 } from "@workglow/util";
 var TABULAR_REPOSITORIES = createServiceToken5("storage.tabular.repositories");
-if (!globalServiceRegistry.has(TABULAR_REPOSITORIES)) {
-  globalServiceRegistry.register(TABULAR_REPOSITORIES, () => new Map, true);
-}
+globalServiceRegistry.registerIfAbsent(TABULAR_REPOSITORIES, () => new Map, true);
 function getGlobalTabularRepositories() {
   return globalServiceRegistry.get(TABULAR_REPOSITORIES);
 }
@@ -1138,6 +1127,14 @@ function resolveRepositoryFromRegistry(id, format, registry) {
   return repo;
 }
 registerInputResolver("storage:tabular", resolveRepositoryFromRegistry);
+registerInputCompactor("storage:tabular", (value, _format, registry) => {
+  const repos = registry.has(TABULAR_REPOSITORIES) ? registry.get(TABULAR_REPOSITORIES) : getGlobalTabularRepositories();
+  for (const [id, repo] of repos) {
+    if (repo === value)
+      return id;
+  }
+  return;
+});
 // src/util/traced.ts
 import { getTelemetryProvider, SpanStatusCode } from "@workglow/util";
 async function traced(spanName, storageName, fn) {
@@ -2237,6 +2234,54 @@ class EncryptedKvCredentialStore {
     await this.kv.deleteAll();
   }
 }
+// src/credentials/LazyEncryptedCredentialStore.ts
+class LazyEncryptedCredentialStore {
+  kv;
+  inner;
+  constructor(kv) {
+    this.kv = kv;
+  }
+  get isUnlocked() {
+    return this.inner !== undefined;
+  }
+  unlock(passphrase) {
+    this.inner = new EncryptedKvCredentialStore(this.kv, passphrase);
+  }
+  lock() {
+    this.inner = undefined;
+  }
+  async get(key) {
+    if (!this.inner)
+      return;
+    return this.inner.get(key);
+  }
+  async put(key, value, options) {
+    if (!this.inner) {
+      throw new Error("Credential store is locked. Call unlock() before storing credentials.");
+    }
+    return this.inner.put(key, value, options);
+  }
+  async delete(key) {
+    if (!this.inner)
+      return false;
+    return this.inner.delete(key);
+  }
+  async has(key) {
+    if (!this.inner)
+      return false;
+    return this.inner.has(key);
+  }
+  async keys() {
+    if (!this.inner)
+      return [];
+    return this.inner.keys();
+  }
+  async deleteAll() {
+    if (!this.inner)
+      return;
+    return this.inner.deleteAll();
+  }
+}
 // src/tabular/FsFolderTabularStorage.ts
 import { createServiceToken as createServiceToken12, getLogger as getLogger3, makeFingerprint as makeFingerprint5, sleep as sleep3, uuid4 as uuid43 } from "@workglow/util";
 import { mkdir, readdir, readFile, rm, writeFile } from "fs/promises";
@@ -8282,6 +8327,7 @@ export {
   POSTGRES_KV_REPOSITORY,
   MEMORY_TABULAR_REPOSITORY,
   MEMORY_KV_REPOSITORY,
+  LazyEncryptedCredentialStore,
   KvViaTabularStorage,
   KvStorage,
   KV_REPOSITORY,
@@ -8320,4 +8366,4 @@ export {
   BaseTabularStorage
 };
 
-//# debugId=962404C01663B8E764756E2164756E21
+//# debugId=58A5C10FFC88774F64756E2164756E21