full site update

node_modules/unstorage/drivers/azure-cosmos.mjs

@@ -0,0 +1,86 @@
+import { createRequiredError, defineDriver } from "./utils/index.mjs";
+import { CosmosClient } from "@azure/cosmos";
+import { DefaultAzureCredential } from "@azure/identity";
+const DRIVER_NAME = "azure-cosmos";
+export default defineDriver((opts) => {
+  let client;
+  const getCosmosClient = async () => {
+    if (client) {
+      return client;
+    }
+    if (!opts.endpoint) {
+      throw createRequiredError(DRIVER_NAME, "endpoint");
+    }
+    if (opts.accountKey) {
+      const cosmosClient = new CosmosClient({
+        endpoint: opts.endpoint,
+        key: opts.accountKey
+      });
+      const { database } = await cosmosClient.databases.createIfNotExists({
+        id: opts.databaseName || "unstorage"
+      });
+      const { container } = await database.containers.createIfNotExists({
+        id: opts.containerName || "unstorage"
+      });
+      client = container;
+    } else {
+      const credential = new DefaultAzureCredential();
+      const cosmosClient = new CosmosClient({
+        endpoint: opts.endpoint,
+        aadCredentials: credential
+      });
+      const { database } = await cosmosClient.databases.createIfNotExists({
+        id: opts.databaseName || "unstorage"
+      });
+      const { container } = await database.containers.createIfNotExists({
+        id: opts.containerName || "unstorage"
+      });
+      client = container;
+    }
+    return client;
+  };
+  return {
+    name: DRIVER_NAME,
+    options: opts,
+    getInstance: getCosmosClient,
+    async hasItem(key) {
+      const item = await (await getCosmosClient()).item(key).read();
+      return item.resource ? true : false;
+    },
+    async getItem(key) {
+      const item = await (await getCosmosClient()).item(key).read();
+      return item.resource ? item.resource.value : null;
+    },
+    async setItem(key, value) {
+      const modified = /* @__PURE__ */ new Date();
+      await (await getCosmosClient()).items.upsert(
+        { id: key, value, modified },
+        { consistencyLevel: "Session" }
+      );
+    },
+    async removeItem(key) {
+      await (await getCosmosClient()).item(key).delete({ consistencyLevel: "Session" });
+    },
+    async getKeys() {
+      const iterator = (await getCosmosClient()).items.query(
+        `SELECT { id } from c`
+      );
+      return (await iterator.fetchAll()).resources.map((item) => item.id);
+    },
+    async getMeta(key) {
+      const item = await (await getCosmosClient()).item(key).read();
+      return {
+        mtime: item.resource?.modified ? new Date(item.resource.modified) : void 0
+      };
+    },
+    async clear() {
+      const iterator = (await getCosmosClient()).items.query(
+        `SELECT { id } from c`
+      );
+      const items = (await iterator.fetchAll()).resources;
+      for (const item of items) {
+        await (await getCosmosClient()).item(item.id).delete({ consistencyLevel: "Session" });
+      }
+    }
+  };
+});