feat: expose underlying client instance (#446)

build.config.ts

@@ -21,4 +21,5 @@ export default defineBuildConfig({
       declaration: false,
     },
   ],
+  externals: ["mongodb"],
 });

src/drivers/azure-app-configuration.ts

@@ -68,6 +68,7 @@ export default defineDriver((opts: AzureAppConfigurationOptions = {}) => {
   return {
     name: DRIVER_NAME,
     options: opts,
+    getInstance: getClient,
     async hasItem(key) {
       try {
         await getClient().getConfigurationSetting({

src/drivers/azure-cosmos.ts

@@ -86,6 +86,7 @@ export default defineDriver((opts: AzureCosmosOptions) => {
   return {
     name: DRIVER_NAME,
     options: opts,
+    getInstance: getCosmosClient,
     async hasItem(key) {
       const item = await (await getCosmosClient())
         .item(key)

src/drivers/azure-key-vault.ts

@@ -45,6 +45,7 @@ export default defineDriver((opts: AzureKeyVaultOptions) => {
   return {
     name: DRIVER_NAME,
     options: opts,
+    getInstance: getKeyVaultClient,
     async hasItem(key) {
       try {
         await getKeyVaultClient().getSecret(encode(key));

src/drivers/azure-storage-blob.ts

@@ -81,6 +81,7 @@ export default defineDriver((opts: AzureStorageBlobOptions) => {
   return {
     name: DRIVER_NAME,
     options: opts,
+    getInstance: getContainerClient,
     async hasItem(key) {
       return await getContainerClient().getBlockBlobClient(key).exists();
     },

src/drivers/azure-storage-table.ts

@@ -105,6 +105,7 @@ export default defineDriver((opts: AzureStorageTableOptions) => {
   return {
     name: DRIVER_NAME,
     options: opts,
+    getInstance: getClient,
     async hasItem(key) {
       try {
         await getClient().getEntity(partitionKey, key);

src/drivers/capacitor-preferences.ts

@@ -8,43 +8,46 @@ export interface CapacitorPreferencesOptions {
   base?: string;
 }
 
-export default defineDriver<CapacitorPreferencesOptions>((opts) => {
-  const base = normalizeKey(opts?.base || "");
-  const resolveKey = (key: string) => joinKeys(base, key);
+export default defineDriver<CapacitorPreferencesOptions, typeof Preferences>(
+  (opts) => {
+    const base = normalizeKey(opts?.base || "");
+    const resolveKey = (key: string) => joinKeys(base, key);
 
-  return {
-    name: DRIVER_NAME,
-    options: opts,
-    hasItem(key) {
-      return Preferences.keys().then((r) => r.keys.includes(resolveKey(key)));
-    },
-    getItem(key) {
-      return Preferences.get({ key: resolveKey(key) }).then((r) => r.value);
-    },
-    getItemRaw(key) {
-      return Preferences.get({ key: resolveKey(key) }).then((r) => r.value);
-    },
-    setItem(key, value) {
-      return Preferences.set({ key: resolveKey(key), value });
-    },
-    setItemRaw(key, value) {
-      return Preferences.set({ key: resolveKey(key), value });
-    },
-    removeItem(key) {
-      return Preferences.remove({ key: resolveKey(key) });
-    },
-    async getKeys() {
-      const { keys } = await Preferences.keys();
-      return keys.map((key) => key.slice(base.length));
-    },
-    async clear(prefix) {
-      const { keys } = await Preferences.keys();
-      const _prefix = resolveKey(prefix || "");
-      await Promise.all(
-        keys
-          .filter((key) => key.startsWith(_prefix))
-          .map((key) => Preferences.remove({ key }))
-      );
-    },
-  };
-});
+    return {
+      name: DRIVER_NAME,
+      options: opts,
+      getInstance: () => Preferences,
+      hasItem(key) {
+        return Preferences.keys().then((r) => r.keys.includes(resolveKey(key)));
+      },
+      getItem(key) {
+        return Preferences.get({ key: resolveKey(key) }).then((r) => r.value);
+      },
+      getItemRaw(key) {
+        return Preferences.get({ key: resolveKey(key) }).then((r) => r.value);
+      },
+      setItem(key, value) {
+        return Preferences.set({ key: resolveKey(key), value });
+      },
+      setItemRaw(key, value) {
+        return Preferences.set({ key: resolveKey(key), value });
+      },
+      removeItem(key) {
+        return Preferences.remove({ key: resolveKey(key) });
+      },
+      async getKeys() {
+        const { keys } = await Preferences.keys();
+        return keys.map((key) => key.slice(base.length));
+      },
+      async clear(prefix) {
+        const { keys } = await Preferences.keys();
+        const _prefix = resolveKey(prefix || "");
+        await Promise.all(
+          keys
+            .filter((key) => key.startsWith(_prefix))
+            .map((key) => Preferences.remove({ key }))
+        );
+      },
+    };
+  }
+);

src/drivers/cloudflare-kv-binding.ts

@@ -25,6 +25,7 @@ export default defineDriver((opts: KVOptions) => {
   return {
     name: DRIVER_NAME,
     options: opts,
+    getInstance: () => getKVBinding(opts.binding),
     async hasItem(key) {
       key = r(key);
       const binding = getKVBinding(opts.binding);

src/drivers/cloudflare-r2-binding.ts

@@ -25,6 +25,7 @@ export default defineDriver((opts: CloudflareR2Options = {}) => {
   return {
     name: DRIVER_NAME,
     options: opts,
+    getInstance: () => getR2Binding(opts.binding),
     async hasItem(key) {
       key = r(key);
       const binding = getR2Binding(opts.binding);

src/drivers/localstorage.ts

@@ -32,6 +32,7 @@ export default defineDriver((opts: LocalStorageOptions = {}) => {
   return {
     name: DRIVER_NAME,
     options: opts,
+    getInstance: () => opts.localStorage!,
     hasItem(key) {
       return Object.prototype.hasOwnProperty.call(opts.localStorage!, r(key));
     },

src/drivers/lru-cache.ts

@@ -25,6 +25,7 @@ export default defineDriver((opts: LRUDriverOptions = {}) => {
   return {
     name: DRIVER_NAME,
     options: opts,
+    getInstance: () => cache,
     hasItem(key) {
       return cache.has(key);
     },

src/drivers/memory.ts

@@ -2,12 +2,12 @@ import { defineDriver } from "./utils";
 
 const DRIVER_NAME = "memory";
 
-export default defineDriver<void>(() => {
+export default defineDriver<void, Map<string, any>>(() => {
   const data = new Map<string, any>();
 
   return {
     name: DRIVER_NAME,
-    options: {},
+    getInstance: () => data,
     hasItem(key) {
       return data.has(key);
     },

src/drivers/mongodb.ts

@@ -39,6 +39,7 @@ export default defineDriver((opts: MongoDbOptions) => {
   return {
     name: DRIVER_NAME,
     options: opts,
+    getInstance: getMongoCollection,
     async hasItem(key) {
       const result = await getMongoCollection().findOne({ key });
       return !!result;

src/drivers/netlify-blobs.ts

@@ -42,73 +42,73 @@ export type NetlifyStoreOptions =
   | NetlifyDeployStoreOptions
   | NetlifyNamedStoreOptions;
 
-export default defineDriver(
-  ({ deployScoped, name, ...opts }: NetlifyStoreOptions) => {
-    let store: Store;
+export default defineDriver((options: NetlifyStoreOptions) => {
+  const { deployScoped, name, ...opts } = options;
+  let store: Store;
 
-    const getClient = () => {
-      if (!store) {
-        if (deployScoped) {
-          if (name) {
-            throw createError(
-              DRIVER_NAME,
-              "deploy-scoped stores cannot have a name"
-            );
-          }
-          store = getDeployStore({ fetch, ...opts });
-        } else {
-          if (!name) {
-            throw createRequiredError(DRIVER_NAME, "name");
-          }
-          // Ensures that reserved characters are encoded
-          store = getStore({ name: encodeURIComponent(name), fetch, ...opts });
+  const getClient = () => {
+    if (!store) {
+      if (deployScoped) {
+        if (name) {
+          throw createError(
+            DRIVER_NAME,
+            "deploy-scoped stores cannot have a name"
+          );
         }
+        store = getDeployStore({ fetch, ...options });
+      } else {
+        if (!name) {
+          throw createRequiredError(DRIVER_NAME, "name");
+        }
+        // Ensures that reserved characters are encoded
+        store = getStore({ name: encodeURIComponent(name), fetch, ...opts });
       }
-      return store;
-    };
+    }
+    return store;
+  };
 
-    return {
-      name: DRIVER_NAME,
-      options: {},
-      async hasItem(key) {
-        return getClient().getMetadata(key).then(Boolean);
-      },
-      getItem: (key, tops?: GetOptions) => {
-        // @ts-expect-error has trouble with the overloaded types
-        return getClient().get(key, tops);
-      },
-      getMeta(key) {
-        return getClient().getMetadata(key);
-      },
-      getItemRaw(key, topts?: GetOptions) {
-        // @ts-expect-error has trouble with the overloaded types
-        return getClient().get(key, { type: topts?.type ?? "arrayBuffer" });
-      },
-      setItem(key, value, topts?: SetOptions) {
-        return getClient().set(key, value, topts);
-      },
-      setItemRaw(key, value: string | ArrayBuffer | Blob, topts?: SetOptions) {
-        return getClient().set(key, value, topts);
-      },
-      removeItem(key) {
-        return getClient().delete(key);
-      },
-      async getKeys(
-        base?: string,
-        tops?: Omit<ListOptions, "prefix" | "paginate">
-      ) {
-        return (await getClient().list({ ...tops, prefix: base })).blobs.map(
-          (item) => item.key
-        );
-      },
-      async clear(base?: string) {
-        const client = getClient();
-        return Promise.allSettled(
-          (await client.list({ prefix: base })).blobs.map((item) =>
-            client.delete(item.key)
-          )
-        ).then(() => {});
-      },
-    };
-  }
-);
+  return {
+    name: DRIVER_NAME,
+    options,
+    getInstance: getClient,
+    async hasItem(key) {
+      return getClient().getMetadata(key).then(Boolean);
+    },
+    getItem: (key, tops?: GetOptions) => {
+      // @ts-expect-error has trouble with the overloaded types
+      return getClient().get(key, tops);
+    },
+    getMeta(key) {
+      return getClient().getMetadata(key);
+    },
+    getItemRaw(key, topts?: GetOptions) {
+      // @ts-expect-error has trouble with the overloaded types
+      return getClient().get(key, { type: topts?.type ?? "arrayBuffer" });
+    },
+    setItem(key, value, topts?: SetOptions) {
+      return getClient().set(key, value, topts);
+    },
+    setItemRaw(key, value: string | ArrayBuffer | Blob, topts?: SetOptions) {
+      return getClient().set(key, value, topts);
+    },
+    removeItem(key) {
+      return getClient().delete(key);
+    },
+    async getKeys(
+      base?: string,
+      tops?: Omit<ListOptions, "prefix" | "paginate">
+    ) {
+      return (await getClient().list({ ...tops, prefix: base })).blobs.map(
+        (item) => item.key
+      );
+    },
+    async clear(base?: string) {
+      const client = getClient();
+      return Promise.allSettled(
+        (await client.list({ prefix: base })).blobs.map((item) =>
+          client.delete(item.key)
+        )
+      ).then(() => {});
+    },
+  };
+});

src/drivers/planetscale.ts

@@ -50,6 +50,7 @@ export default defineDriver((opts: PlanetscaleDriverOptions = {}) => {
   return {
     name: DRIVER_NAME,
     options: opts,
+    getInstance: getConnection,
     hasItem: async (key) => {
       const res = await getConnection().execute(
         `SELECT EXISTS (SELECT 1 FROM ${opts.table} WHERE id = :key) as value;`,

src/drivers/redis.ts

@@ -35,7 +35,7 @@ export interface RedisOptions extends _RedisOptions {
 
 const DRIVER_NAME = "redis";
 
-export default defineDriver((opts: RedisOptions = {}) => {
+export default defineDriver((opts: RedisOptions) => {
   let redisClient: Redis | Cluster;
   const getRedisClient = () => {
     if (redisClient) {
@@ -58,6 +58,7 @@ export default defineDriver((opts: RedisOptions = {}) => {
   return {
     name: DRIVER_NAME,
     options: opts,
+    getInstance: getRedisClient,
     async hasItem(key) {
       return Boolean(await getRedisClient().exists(p(key)));
     },

src/drivers/session-storage.ts

@@ -32,6 +32,7 @@ export default defineDriver((opts: SessionStorageOptions = {}) => {
   return {
     name: DRIVER_NAME,
     options: opts,
+    getInstance: () => opts.sessionStorage!,
     hasItem(key) {
       return Object.prototype.hasOwnProperty.call(opts.sessionStorage, r(key));
     },

src/drivers/utils/index.ts

@@ -1,11 +1,13 @@
 import type { Driver } from "../..";
 
-type DriverFactory<T> = (opts: T) => Driver;
+type DriverFactory<OptionsT, InstanceT> = (
+  opts: OptionsT
+) => Driver<OptionsT, InstanceT>;
 interface ErrorOptions {}
 
-export function defineDriver<T = any>(
-  factory: DriverFactory<T>
-): DriverFactory<T> {
+export function defineDriver<OptionsT = any, InstanceT = never>(
+  factory: DriverFactory<OptionsT, InstanceT>
+): DriverFactory<OptionsT, InstanceT> {
   return factory;
 }