storageBuckets.keys()

virtualstate · Dec 1, 2023 · 0779f4f · 0779f4f
1 parent d3eafaf
commit 0779f4f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/src/storage-buckets/internal.ts b/src/storage-buckets/internal.ts
@@ -6,6 +6,14 @@ export interface InternalBucket {
     delete(): Promise<void>
 }
 
+export async function listInternalStorageBucketNames() {
+    const keys = await listKeyValueStoreIndex();
+    const buckets = keys
+        .filter(key => key.startsWith("bucket:"))
+        .map(key => key.split(":")[1]);
+    return ["default", ...new Set(buckets)];
+}
+
 export function getInternalStorageBucket(bucket: string = "default"): InternalBucket {
     const isDefault = bucket === "default";
     const prefix = isDefault ? "" : `bucket:${bucket}:`

diff --git a/src/storage-buckets/manager.ts b/src/storage-buckets/manager.ts
@@ -1,6 +1,6 @@
 import {StorageBucket, StorageBucketManager, StorageBucketOptions} from "./types";
 import {Promise} from "@virtualstate/promise";
-import {getInternalStorageBucket} from "./internal";
+import {getInternalStorageBucket, listInternalStorageBucketNames} from "./internal";
 import {DurableCacheStorage} from "../fetch";
 import {getOrigin} from "../listen";
 import {getConfig} from "../config";
@@ -83,7 +83,10 @@ export class DurableStorageBucketManager implements StorageBucketManager {
     }
 
     async keys(): Promise<string[]> {
-        return [];
+        return [...new Set([
+            ...this.buckets.keys(),
+            ...await listInternalStorageBucketNames()
+        ])];
     }
 
     async open(name: string, options?: StorageBucketOptions): Promise<StorageBucket> {