full site update

node_modules/unstorage/drivers/azure-storage-blob.mjs

@@ -0,0 +1,154 @@
+import { createError, defineDriver } from "./utils/index.mjs";
+import {
+  BlobServiceClient,
+  ContainerClient,
+  StorageSharedKeyCredential
+} from "@azure/storage-blob";
+import { DefaultAzureCredential } from "@azure/identity";
+const DRIVER_NAME = "azure-storage-blob";
+export default defineDriver((opts) => {
+  let containerClient;
+  const endpointSuffix = opts.endpointSuffix || ".blob.core.windows.net";
+  const getContainerClient = () => {
+    if (containerClient) {
+      return containerClient;
+    }
+    if (!opts.connectionString && !opts.sasUrl && !opts.accountName) {
+      throw createError(DRIVER_NAME, "missing accountName");
+    }
+    let serviceClient;
+    if (opts.accountKey) {
+      const credential = new StorageSharedKeyCredential(
+        opts.accountName,
+        opts.accountKey
+      );
+      serviceClient = new BlobServiceClient(
+        `https://${opts.accountName}${endpointSuffix}`,
+        credential
+      );
+    } else if (opts.sasUrl) {
+      if (opts.containerName && opts.sasUrl.includes(`${opts.containerName}?`)) {
+        containerClient = new ContainerClient(`${opts.sasUrl}`);
+        return containerClient;
+      }
+      serviceClient = new BlobServiceClient(opts.sasUrl);
+    } else if (opts.sasKey) {
+      if (opts.containerName) {
+        containerClient = new ContainerClient(
+          `https://${opts.accountName}${endpointSuffix}/${opts.containerName}?${opts.sasKey}`
+        );
+        return containerClient;
+      }
+      serviceClient = new BlobServiceClient(
+        `https://${opts.accountName}${endpointSuffix}?${opts.sasKey}`
+      );
+    } else if (opts.connectionString) {
+      serviceClient = BlobServiceClient.fromConnectionString(
+        opts.connectionString
+      );
+    } else {
+      const credential = new DefaultAzureCredential();
+      serviceClient = new BlobServiceClient(
+        `https://${opts.accountName}${endpointSuffix}`,
+        credential
+      );
+    }
+    containerClient = serviceClient.getContainerClient(
+      opts.containerName || "unstorage"
+    );
+    containerClient.createIfNotExists();
+    return containerClient;
+  };
+  return {
+    name: DRIVER_NAME,
+    options: opts,
+    getInstance: getContainerClient,
+    async hasItem(key) {
+      return await getContainerClient().getBlockBlobClient(key).exists();
+    },
+    async getItem(key) {
+      try {
+        const blob = await getContainerClient().getBlockBlobClient(key).download();
+        if (isBrowser) {
+          return blob.blobBody ? await blobToString(await blob.blobBody) : null;
+        }
+        return blob.readableStreamBody ? (await streamToBuffer(blob.readableStreamBody)).toString() : null;
+      } catch {
+        return null;
+      }
+    },
+    async getItemRaw(key) {
+      try {
+        const blob = await getContainerClient().getBlockBlobClient(key).download();
+        if (isBrowser) {
+          return blob.blobBody ? await blobToString(await blob.blobBody) : null;
+        }
+        return blob.readableStreamBody ? await streamToBuffer(blob.readableStreamBody) : null;
+      } catch {
+        return null;
+      }
+    },
+    async setItem(key, value) {
+      await getContainerClient().getBlockBlobClient(key).upload(value, Buffer.byteLength(value));
+    },
+    async setItemRaw(key, value) {
+      await getContainerClient().getBlockBlobClient(key).upload(value, Buffer.byteLength(value));
+    },
+    async removeItem(key) {
+      await getContainerClient().getBlockBlobClient(key).deleteIfExists({ deleteSnapshots: "include" });
+    },
+    async getKeys() {
+      const iterator = getContainerClient().listBlobsFlat().byPage({ maxPageSize: 1e3 });
+      const keys = [];
+      for await (const page of iterator) {
+        const pageKeys = page.segment.blobItems.map((blob) => blob.name);
+        keys.push(...pageKeys);
+      }
+      return keys;
+    },
+    async getMeta(key) {
+      const blobProperties = await getContainerClient().getBlockBlobClient(key).getProperties();
+      return {
+        mtime: blobProperties.lastModified,
+        atime: blobProperties.lastAccessed,
+        cr: blobProperties.createdOn,
+        ...blobProperties.metadata
+      };
+    },
+    async clear() {
+      const iterator = getContainerClient().listBlobsFlat().byPage({ maxPageSize: 1e3 });
+      for await (const page of iterator) {
+        await Promise.all(
+          page.segment.blobItems.map(
+            async (blob) => await getContainerClient().deleteBlob(blob.name, {
+              deleteSnapshots: "include"
+            })
+          )
+        );
+      }
+    }
+  };
+});
+const isBrowser = typeof window !== "undefined";
+async function streamToBuffer(readableStream) {
+  return new Promise((resolve, reject) => {
+    const chunks = [];
+    readableStream.on("data", (data) => {
+      chunks.push(data instanceof Buffer ? data : Buffer.from(data));
+    });
+    readableStream.on("end", () => {
+      resolve(Buffer.concat(chunks));
+    });
+    readableStream.on("error", reject);
+  });
+}
+async function blobToString(blob) {
+  const fileReader = new FileReader();
+  return new Promise((resolve, reject) => {
+    fileReader.onloadend = (ev) => {
+      resolve(ev.target?.result);
+    };
+    fileReader.onerror = reject;
+    fileReader.readAsText(blob);
+  });
+}