@itwin/core-backend 4.0.0-dev.52 → 4.0.0-dev.55

package/lib/cjs/CloudStorageBackend.js

@@ -1,171 +1,171 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.CloudStorageTileUploader = exports.AzureBlobStorage = exports.CloudStorageService = void 0;
-/*---------------------------------------------------------------------------------------------
-* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
-* See LICENSE.md in the project root for license terms and full copyright notice.
-*--------------------------------------------------------------------------------------------*/
-const stream_1 = require("stream");
-const zlib = require("zlib");
-const Azure = require("@azure/storage-blob");
-const core_bentley_1 = require("@itwin/core-bentley");
-const core_common_1 = require("@itwin/core-common");
-const BackendLoggerCategory_1 = require("./BackendLoggerCategory");
-const IModelHost_1 = require("./IModelHost");
-/**
- * @beta
- * @deprecated in 3.x. Use [[TileStorage]] instead.
- */
-class CloudStorageService {
-    initialize() { }
-    terminate() { }
-    async download(_name) { return undefined; }
-    makeDescriptor(id) {
-        return { name: id.name, provider: this.id };
-    }
-}
-exports.CloudStorageService = CloudStorageService;
-/**
- * @beta
- * @deprecated in 3.x. Use `@itwin/object-storage-oss` instead, see https://github.com/iTwin/object-storage for more details.
- */
-class AzureBlobStorage extends CloudStorageService {
-    constructor(credentials) {
-        super();
-        this.id = core_common_1.CloudStorageProvider.Azure;
-        if (!credentials.account || !credentials.accessKey) {
-            throw new core_common_1.IModelError(core_common_1.BentleyStatus.ERROR, "Invalid credentials for Azure blob storage.");
-        }
-        this._baseUrl = credentials.baseUrl ?? `https://${credentials.account}.blob.core.windows.net`;
-        this._credential = new Azure.StorageSharedKeyCredential(credentials.account, credentials.accessKey);
-        const options = {};
-        const pipeline = Azure.newPipeline(this._credential, options);
-        this._service = new Azure.BlobServiceClient(this._baseUrl, pipeline);
-    }
-    obtainContainerUrl(id, expiry, clientIp) {
-        const policy = {
-            containerName: id.name,
-            permissions: Azure.ContainerSASPermissions.parse("rl"),
-            expiresOn: expiry,
-        };
-        if (clientIp && clientIp !== "localhost" && clientIp !== "127.0.0.1" && clientIp !== "::1") {
-            policy.ipRange = { start: clientIp };
-        }
-        const token = Azure.generateBlobSASQueryParameters(policy, this._credential);
-        const url = new URL(this._baseUrl);
-        url.pathname = `${url.pathname.replace(/\/*$/, "")}/${id.name}`;
-        url.search = token.toString();
-        const urlObject = {
-            descriptor: this.makeDescriptor(id),
-            valid: 0,
-            expires: expiry.getTime(),
-            url: url.toString(),
-        };
-        return urlObject;
-    }
-    async ensureContainer(name) {
-        try {
-            const container = this._service.getContainerClient(name);
-            await container.create();
-            return;
-        }
-        catch (maybeErr) {
-            try {
-                const err = maybeErr;
-                if (err.statusCode === 409 && err.code === "ContainerAlreadyExists") {
-                    return;
-                }
-            }
-            catch { }
-            if (typeof (maybeErr) !== "undefined" && maybeErr) {
-                throw maybeErr;
-            }
-            throw new core_common_1.IModelError(core_common_1.BentleyStatus.ERROR, `Unable to create container "${name}".`);
-        }
-    }
-    async upload(container, name, data, options, metadata) {
-        try {
-            await this.ensureContainer(container);
-            const containerClient = this._service.getContainerClient(container);
-            const blocks = containerClient.getBlockBlobClient(name);
-            const blobHTTPHeaders = {
-                blobContentType: options?.type ?? "application/octet-stream",
-                blobCacheControl: options?.cacheControl ?? "private, max-age=31536000, immutable",
-            };
-            const blobOptions = metadata ?
-                {
-                    blobHTTPHeaders,
-                    metadata: { ...metadata },
-                } : {
-                blobHTTPHeaders,
-            };
-            const dataStream = new stream_1.PassThrough();
-            dataStream.end(data);
-            let source;
-            if (options && options.contentEncoding === "gzip") {
-                if (undefined === blobOptions.blobHTTPHeaders)
-                    throw new core_common_1.IModelError(core_common_1.BentleyStatus.ERROR, "Blob HTTP headers object is undefined.");
-                blobOptions.blobHTTPHeaders.blobContentEncoding = options.contentEncoding;
-                const compressor = zlib.createGzip();
-                source = dataStream.pipe(compressor);
-            }
-            else {
-                source = dataStream;
-            }
-            const blockSize = 100 * 1024 * 1024;
-            const concurrency = 1;
-            const result = await blocks.uploadStream(source, blockSize, concurrency, blobOptions);
-            return result.etag ?? "";
-        }
-        catch (maybeErr) {
-            if (typeof (maybeErr) !== "undefined" && maybeErr) {
-                throw maybeErr;
-            }
-            throw new core_common_1.IModelError(core_common_1.BentleyStatus.ERROR, `Unable to upload "${name}".`);
-        }
-    }
-}
-exports.AzureBlobStorage = AzureBlobStorage;
-/**
- * @beta
- * @deprecated in 3.x. Use [[TileStorage]] instead.
- */
-class CloudStorageTileUploader {
-    constructor() {
-        this._activeUploads = new Map();
-    }
-    get activeUploads() {
-        return this._activeUploads.values();
-    }
-    async uploadToCache(id, content, containerKey, resourceKey, metadata) {
-        await new Promise((resolve) => setTimeout(resolve));
-        try {
-            const options = {};
-            if (IModelHost_1.IModelHost.compressCachedTiles) {
-                options.contentEncoding = "gzip";
-            }
-            const perfInfo = { ...id.tokenProps, treeId: id.treeId, contentId: id.contentId, size: content.byteLength, compress: IModelHost_1.IModelHost.compressCachedTiles };
-            const perfLogger = new core_bentley_1.PerfLogger("Uploading tile to external tile cache", () => perfInfo);
-            await IModelHost_1.IModelHost.tileCacheService?.upload(containerKey, resourceKey, content, options, metadata);
-            perfLogger.dispose();
-        }
-        catch (err) {
-            core_bentley_1.Logger.logError(BackendLoggerCategory_1.BackendLoggerCategory.IModelTileUpload, (err instanceof Error) ? err.toString() : JSON.stringify(err));
-        }
-        this._activeUploads.delete(containerKey + resourceKey);
-    }
-    async cacheTile(tokenProps, treeId, contentId, content, guid, metadata) {
-        const id = { tokenProps, treeId, contentId, guid };
-        const cache = core_common_1.CloudStorageTileCache.getCache();
-        const containerKey = cache.formContainerName(id);
-        const resourceKey = cache.formResourceName(id);
-        const key = containerKey + resourceKey;
-        if (this._activeUploads.has(key))
-            return;
-        const promise = this.uploadToCache(id, content, containerKey, resourceKey, metadata);
-        this._activeUploads.set(key, promise);
-        await promise;
-    }
-}
-exports.CloudStorageTileUploader = CloudStorageTileUploader;
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CloudStorageTileUploader = exports.AzureBlobStorage = exports.CloudStorageService = void 0;
+/*---------------------------------------------------------------------------------------------
+* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+* See LICENSE.md in the project root for license terms and full copyright notice.
+*--------------------------------------------------------------------------------------------*/
+const stream_1 = require("stream");
+const zlib = require("zlib");
+const Azure = require("@azure/storage-blob");
+const core_bentley_1 = require("@itwin/core-bentley");
+const core_common_1 = require("@itwin/core-common");
+const BackendLoggerCategory_1 = require("./BackendLoggerCategory");
+const IModelHost_1 = require("./IModelHost");
+/**
+ * @beta
+ * @deprecated in 3.x. Use [[TileStorage]] instead.
+ */
+class CloudStorageService {
+    initialize() { }
+    terminate() { }
+    async download(_name) { return undefined; }
+    makeDescriptor(id) {
+        return { name: id.name, provider: this.id };
+    }
+}
+exports.CloudStorageService = CloudStorageService;
+/**
+ * @beta
+ * @deprecated in 3.x. Use `@itwin/object-storage-oss` instead, see https://github.com/iTwin/object-storage for more details.
+ */
+class AzureBlobStorage extends CloudStorageService {
+    constructor(credentials) {
+        super();
+        this.id = core_common_1.CloudStorageProvider.Azure;
+        if (!credentials.account || !credentials.accessKey) {
+            throw new core_common_1.IModelError(core_common_1.BentleyStatus.ERROR, "Invalid credentials for Azure blob storage.");
+        }
+        this._baseUrl = credentials.baseUrl ?? `https://${credentials.account}.blob.core.windows.net`;
+        this._credential = new Azure.StorageSharedKeyCredential(credentials.account, credentials.accessKey);
+        const options = {};
+        const pipeline = Azure.newPipeline(this._credential, options);
+        this._service = new Azure.BlobServiceClient(this._baseUrl, pipeline);
+    }
+    obtainContainerUrl(id, expiry, clientIp) {
+        const policy = {
+            containerName: id.name,
+            permissions: Azure.ContainerSASPermissions.parse("rl"),
+            expiresOn: expiry,
+        };
+        if (clientIp && clientIp !== "localhost" && clientIp !== "127.0.0.1" && clientIp !== "::1") {
+            policy.ipRange = { start: clientIp };
+        }
+        const token = Azure.generateBlobSASQueryParameters(policy, this._credential);
+        const url = new URL(this._baseUrl);
+        url.pathname = `${url.pathname.replace(/\/*$/, "")}/${id.name}`;
+        url.search = token.toString();
+        const urlObject = {
+            descriptor: this.makeDescriptor(id),
+            valid: 0,
+            expires: expiry.getTime(),
+            url: url.toString(),
+        };
+        return urlObject;
+    }
+    async ensureContainer(name) {
+        try {
+            const container = this._service.getContainerClient(name);
+            await container.create();
+            return;
+        }
+        catch (maybeErr) {
+            try {
+                const err = maybeErr;
+                if (err.statusCode === 409 && err.code === "ContainerAlreadyExists") {
+                    return;
+                }
+            }
+            catch { }
+            if (typeof (maybeErr) !== "undefined" && maybeErr) {
+                throw maybeErr;
+            }
+            throw new core_common_1.IModelError(core_common_1.BentleyStatus.ERROR, `Unable to create container "${name}".`);
+        }
+    }
+    async upload(container, name, data, options, metadata) {
+        try {
+            await this.ensureContainer(container);
+            const containerClient = this._service.getContainerClient(container);
+            const blocks = containerClient.getBlockBlobClient(name);
+            const blobHTTPHeaders = {
+                blobContentType: options?.type ?? "application/octet-stream",
+                blobCacheControl: options?.cacheControl ?? "private, max-age=31536000, immutable",
+            };
+            const blobOptions = metadata ?
+                {
+                    blobHTTPHeaders,
+                    metadata: { ...metadata },
+                } : {
+                blobHTTPHeaders,
+            };
+            const dataStream = new stream_1.PassThrough();
+            dataStream.end(data);
+            let source;
+            if (options && options.contentEncoding === "gzip") {
+                if (undefined === blobOptions.blobHTTPHeaders)
+                    throw new core_common_1.IModelError(core_common_1.BentleyStatus.ERROR, "Blob HTTP headers object is undefined.");
+                blobOptions.blobHTTPHeaders.blobContentEncoding = options.contentEncoding;
+                const compressor = zlib.createGzip();
+                source = dataStream.pipe(compressor);
+            }
+            else {
+                source = dataStream;
+            }
+            const blockSize = 100 * 1024 * 1024;
+            const concurrency = 1;
+            const result = await blocks.uploadStream(source, blockSize, concurrency, blobOptions);
+            return result.etag ?? "";
+        }
+        catch (maybeErr) {
+            if (typeof (maybeErr) !== "undefined" && maybeErr) {
+                throw maybeErr;
+            }
+            throw new core_common_1.IModelError(core_common_1.BentleyStatus.ERROR, `Unable to upload "${name}".`);
+        }
+    }
+}
+exports.AzureBlobStorage = AzureBlobStorage;
+/**
+ * @beta
+ * @deprecated in 3.x. Use [[TileStorage]] instead.
+ */
+class CloudStorageTileUploader {
+    constructor() {
+        this._activeUploads = new Map();
+    }
+    get activeUploads() {
+        return this._activeUploads.values();
+    }
+    async uploadToCache(id, content, containerKey, resourceKey, metadata) {
+        await new Promise((resolve) => setTimeout(resolve));
+        try {
+            const options = {};
+            if (IModelHost_1.IModelHost.compressCachedTiles) {
+                options.contentEncoding = "gzip";
+            }
+            const perfInfo = { ...id.tokenProps, treeId: id.treeId, contentId: id.contentId, size: content.byteLength, compress: IModelHost_1.IModelHost.compressCachedTiles };
+            const perfLogger = new core_bentley_1.PerfLogger("Uploading tile to external tile cache", () => perfInfo);
+            await IModelHost_1.IModelHost.tileCacheService?.upload(containerKey, resourceKey, content, options, metadata);
+            perfLogger.dispose();
+        }
+        catch (err) {
+            core_bentley_1.Logger.logError(BackendLoggerCategory_1.BackendLoggerCategory.IModelTileUpload, (err instanceof Error) ? err.toString() : JSON.stringify(err));
+        }
+        this._activeUploads.delete(containerKey + resourceKey);
+    }
+    async cacheTile(tokenProps, treeId, contentId, content, guid, metadata) {
+        const id = { tokenProps, treeId, contentId, guid };
+        const cache = core_common_1.CloudStorageTileCache.getCache();
+        const containerKey = cache.formContainerName(id);
+        const resourceKey = cache.formResourceName(id);
+        const key = containerKey + resourceKey;
+        if (this._activeUploads.has(key))
+            return;
+        const promise = this.uploadToCache(id, content, containerKey, resourceKey, metadata);
+        this._activeUploads.set(key, promise);
+        await promise;
+    }
+}
+exports.CloudStorageTileUploader = CloudStorageTileUploader;
 //# sourceMappingURL=CloudStorageBackend.js.map