@itwin/core-backend 3.4.0-dev.48 → 3.4.0-dev.50

package/lib/cjs/CloudStorageBackend.js

@@ -1,165 +1,165 @@
-"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 */
-class CloudStorageService {
-    initialize() { }
-    terminate() { }
-    async download(_name) { return undefined; }
-    makeDescriptor(id) {
-        return { name: id.name, provider: this.id };
-    }
-}
-exports.CloudStorageService = CloudStorageService;
-/** @beta */
-class AzureBlobStorage extends CloudStorageService {
-    constructor(credentials) {
-        var _a;
-        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 = (_a = credentials.baseUrl) !== null && _a !== void 0 ? _a : `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) {
-        var _a, _b, _c;
-        try {
-            await this.ensureContainer(container);
-            const containerClient = this._service.getContainerClient(container);
-            const blocks = containerClient.getBlockBlobClient(name);
-            const blobHTTPHeaders = {
-                blobContentType: (_a = options === null || options === void 0 ? void 0 : options.type) !== null && _a !== void 0 ? _a : "application/octet-stream",
-                blobCacheControl: (_b = options === null || options === void 0 ? void 0 : options.cacheControl) !== null && _b !== void 0 ? _b : "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 (_c = result.etag) !== null && _c !== void 0 ? _c : "";
-        }
-        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;
-/** @internal */
-class CloudStorageTileUploader {
-    constructor() {
-        this._activeUploads = new Map();
-    }
-    get activeUploads() {
-        return this._activeUploads.values();
-    }
-    async uploadToCache(id, content, containerKey, resourceKey, metadata) {
-        var _a;
-        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 ((_a = IModelHost_1.IModelHost.tileCacheService) === null || _a === void 0 ? void 0 : _a.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 */
+class CloudStorageService {
+    initialize() { }
+    terminate() { }
+    async download(_name) { return undefined; }
+    makeDescriptor(id) {
+        return { name: id.name, provider: this.id };
+    }
+}
+exports.CloudStorageService = CloudStorageService;
+/** @beta */
+class AzureBlobStorage extends CloudStorageService {
+    constructor(credentials) {
+        var _a;
+        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 = (_a = credentials.baseUrl) !== null && _a !== void 0 ? _a : `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) {
+        var _a, _b, _c;
+        try {
+            await this.ensureContainer(container);
+            const containerClient = this._service.getContainerClient(container);
+            const blocks = containerClient.getBlockBlobClient(name);
+            const blobHTTPHeaders = {
+                blobContentType: (_a = options === null || options === void 0 ? void 0 : options.type) !== null && _a !== void 0 ? _a : "application/octet-stream",
+                blobCacheControl: (_b = options === null || options === void 0 ? void 0 : options.cacheControl) !== null && _b !== void 0 ? _b : "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 (_c = result.etag) !== null && _c !== void 0 ? _c : "";
+        }
+        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;
+/** @internal */
+class CloudStorageTileUploader {
+    constructor() {
+        this._activeUploads = new Map();
+    }
+    get activeUploads() {
+        return this._activeUploads.values();
+    }
+    async uploadToCache(id, content, containerKey, resourceKey, metadata) {
+        var _a;
+        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 ((_a = IModelHost_1.IModelHost.tileCacheService) === null || _a === void 0 ? void 0 : _a.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

package/lib/cjs/CloudStorageBackend.js.map

@@ -1 +1 @@
-{"version":3,"file":"CloudStorageBackend.js","sourceRoot":"","sources":["../../src/CloudStorageBackend.ts"],"names":[],"mappings":";;;AAAA;;;+FAG+F;AAC/F,mCAA+C;AAC/C,6BAA6B;AAC7B,6CAA6C;AAC7C,sDAAyD;AACzD,oDAG4B;AAC5B,mEAAgE;AAChE,6CAA0C;AAgB1C,YAAY;AACZ,MAAsB,mBAAmB;IAChC,UAAU,KAAW,CAAC;IACtB,SAAS,KAAW,CAAC;IAIrB,KAAK,CAAC,QAAQ,CAAC,KAAa,IAAmC,OAAO,SAAS,CAAC,CAAC,CAAC;IAE/E,cAAc,CAAC,EAAmC;QAC1D,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;IAC9C,CAAC;CACF;AAXD,kDAWC;AAED,YAAY;AACZ,MAAa,gBAAiB,SAAQ,mBAAmB;IAKvD,YAAmB,WAAwC;;QACzD,KAAK,EAAE,CAAC;QAcM,OAAE,GAAG,kCAAoB,CAAC,KAAK,CAAC;QAZ9C,IAAI,CAAC,WAAW,CAAC,OAAO,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE;YAClD,MAAM,IAAI,yBAAW,CAAC,2BAAa,CAAC,KAAK,EAAE,6CAA6C,CAAC,CAAC;SAC3F;QAED,IAAI,CAAC,QAAQ,GAAG,MAAA,WAAW,CAAC,OAAO,mCAAI,WAAW,WAAW,CAAC,OAAO,wBAAwB,CAAC;QAE9F,IAAI,CAAC,WAAW,GAAG,IAAI,KAAK,CAAC,0BAA0B,CAAC,WAAW,CAAC,OAAO,EAAE,WAAW,CAAC,SAAS,CAAC,CAAC;QACpG,MAAM,OAAO,GAAiC,EAAE,CAAC;QACjD,MAAM,QAAQ,GAAG,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QAC9D,IAAI,CAAC,QAAQ,GAAG,IAAI,KAAK,CAAC,iBAAiB,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACvE,CAAC;IAIM,kBAAkB,CAAC,EAAmC,EAAE,MAAY,EAAE,QAAiB;QAC5F,MAAM,MAAM,GAAiC;YAC3C,aAAa,EAAE,EAAE,CAAC,IAAI;YACtB,WAAW,EAAE,KAAK,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC;YACtD,SAAS,EAAE,MAAM;SAClB,CAAC;QAEF,IAAI,QAAQ,IAAI,QAAQ,KAAK,WAAW,IAAI,QAAQ,KAAK,WAAW,IAAI,QAAQ,KAAK,KAAK,EAAE;YAC1F,MAAM,CAAC,OAAO,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;SACtC;QAED,MAAM,KAAK,GAAG,KAAK,CAAC,8BAA8B,CAAC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QAC7E,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACnC,GAAG,CAAC,QAAQ,GAAG,GAAG,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;QAChE,GAAG,CAAC,MAAM,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;QAE9B,MAAM,SAAS,GAA6B;YAC1C,UAAU,EAAE,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC;YACnC,KAAK,EAAE,CAAC;YACR,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE;YACzB,GAAG,EAAE,GAAG,CAAC,QAAQ,EAAE;SACpB,CAAC;QAEF,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,KAAK,CAAC,eAAe,CAAC,IAAY;QACvC,IAAI;YACF,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;YACzD,MAAM,SAAS,CAAC,MAAM,EAAE,CAAC;YACzB,OAAO;SACR;QAAC,OAAO,QAAQ,EAAE;YACjB,IAAI;gBACF,MAAM,GAAG,GAAG,QAA2B,CAAC;gBACxC,IAAI,GAAG,CAAC,UAAU,KAAK,GAAG,IAAI,GAAG,CAAC,IAAI,KAAK,wBAAwB,EAAE;oBACnE,OAAO;iBACR;aACF;YAAC,MAAM,GAAG;YAEX,IAAI,OAAO,CAAC,QAAQ,CAAC,KAAK,WAAW,IAAI,QAAQ,EAAE;gBACjD,MAAM,QAAQ,CAAC;aAChB;YAED,MAAM,IAAI,yBAAW,CAAC,2BAAa,CAAC,KAAK,EAAE,+BAA+B,IAAI,IAAI,CAAC,CAAC;SACrF;IACH,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,SAAiB,EAAE,IAAY,EAAE,IAAgB,EAAE,OAAmC,EAAE,QAAiB;;QAC3H,IAAI;YACF,MAAM,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YAEtC,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;YACpE,MAAM,MAAM,GAAG,eAAe,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;YACxD,MAAM,eAAe,GAAG;gBACtB,eAAe,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,mCAAI,0BAA0B;gBAC5D,gBAAgB,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,YAAY,mCAAI,sCAAsC;aAClF,CAAC;YAEF,MAAM,WAAW,GAAuC,QAAQ,CAAC,CAAC;gBAChE;oBACE,eAAe;oBACf,QAAQ,EAAE,EAAE,GAAG,QAAQ,EAAE;iBAC1B,CAAC,CAAC,CAAC;gBACF,eAAe;aAChB,CAAC;YAEJ,MAAM,UAAU,GAAG,IAAI,oBAAW,EAAE,CAAC;YACrC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAErB,IAAI,MAAgB,CAAC;YAErB,IAAI,OAAO,IAAI,OAAO,CAAC,eAAe,KAAK,MAAM,EAAE;gBACjD,IAAI,SAAS,KAAK,WAAW,CAAC,eAAe;oBAC3C,MAAM,IAAI,yBAAW,CAAC,2BAAa,CAAC,KAAK,EAAE,wCAAwC,CAAC,CAAC;gBAEvF,WAAW,CAAC,eAAe,CAAC,mBAAmB,GAAG,OAAO,CAAC,eAAe,CAAC;gBAC1E,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;gBACrC,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;aACtC;iBAAM;gBACL,MAAM,GAAG,UAAU,CAAC;aACrB;YAED,MAAM,SAAS,GAAG,GAAG,GAAG,IAAI,GAAG,IAAI,CAAC;YACpC,MAAM,WAAW,GAAG,CAAC,CAAC;YACtB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;YACtF,OAAO,MAAA,MAAM,CAAC,IAAI,mCAAI,EAAE,CAAC;SAC1B;QAAC,OAAO,QAAQ,EAAE;YACjB,IAAI,OAAO,CAAC,QAAQ,CAAC,KAAK,WAAW,IAAI,QAAQ,EAAE;gBACjD,MAAM,QAAQ,CAAC;aAChB;YAED,MAAM,IAAI,yBAAW,CAAC,2BAAa,CAAC,KAAK,EAAE,qBAAqB,IAAI,IAAI,CAAC,CAAC;SAC3E;IACH,CAAC;CACF;AApHD,4CAoHC;AAED,gBAAgB;AAChB,MAAa,wBAAwB;IAArC;QACU,mBAAc,GAA+B,IAAI,GAAG,EAAE,CAAC;IAyCjE,CAAC;IAvCC,IAAW,aAAa;QACtB,OAAO,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC;IACtC,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,EAAyB,EAAE,OAAmB,EAAE,YAAoB,EAAE,WAAmB,EAAE,QAAiB;;QACtI,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC;QAEpD,IAAI;YACF,MAAM,OAAO,GAA8B,EAAE,CAAC;YAC9C,IAAI,uBAAU,CAAC,mBAAmB,EAAE;gBAClC,OAAO,CAAC,eAAe,GAAG,MAAM,CAAC;aAClC;YAED,MAAM,QAAQ,GAAG,EAAE,GAAG,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,OAAO,CAAC,UAAU,EAAE,QAAQ,EAAE,uBAAU,CAAC,mBAAmB,EAAE,CAAC;YACtJ,MAAM,UAAU,GAAG,IAAI,yBAAU,CAAC,uCAAuC,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,CAAC;YAC3F,MAAM,CAAA,MAAA,uBAAU,CAAC,gBAAgB,0CAAE,MAAM,CAAC,YAAY,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAA,CAAC;YACjG,UAAU,CAAC,OAAO,EAAE,CAAC;SACtB;QAAC,OAAO,GAAG,EAAE;YACZ,qBAAM,CAAC,QAAQ,CAAC,6CAAqB,CAAC,gBAAgB,EAAE,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;SACxH;QAED,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,YAAY,GAAG,WAAW,CAAC,CAAC;IACzD,CAAC;IAEM,KAAK,CAAC,SAAS,CAAC,UAA0B,EAAE,MAAc,EAAE,SAAiB,EAAE,OAAmB,EAAE,IAAwB,EAAE,QAAiB;QACpJ,MAAM,EAAE,GAA0B,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;QAE1E,MAAM,KAAK,GAAG,mCAAqB,CAAC,QAAQ,EAAE,CAAC;QAC/C,MAAM,YAAY,GAAG,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;QACjD,MAAM,WAAW,GAAG,KAAK,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;QAC/C,MAAM,GAAG,GAAG,YAAY,GAAG,WAAW,CAAC;QAEvC,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC;YAC9B,OAAO;QAET,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC;QACrF,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QACtC,MAAM,OAAO,CAAC;IAChB,CAAC;CACF;AA1CD,4DA0CC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n* See LICENSE.md in the project root for license terms and full copyright notice.\n*--------------------------------------------------------------------------------------------*/\nimport { PassThrough, Readable } from \"stream\";\nimport * as zlib from \"zlib\";\nimport * as Azure from \"@azure/storage-blob\";\nimport { Logger, PerfLogger } from \"@itwin/core-bentley\";\nimport {\n  BentleyStatus, CloudStorageContainerDescriptor, CloudStorageContainerUrl, CloudStorageProvider, CloudStorageTileCache, IModelError, IModelRpcProps,\n  TileContentIdentifier,\n} from \"@itwin/core-common\";\nimport { BackendLoggerCategory } from \"./BackendLoggerCategory\";\nimport { IModelHost } from \"./IModelHost\";\n\n/** @beta */\nexport interface AzureBlobStorageCredentials {\n  account: string;\n  accessKey: string;\n  baseUrl?: string;\n}\n\n/** @beta */\nexport interface CloudStorageUploadOptions {\n  type?: string;\n  cacheControl?: string;\n  contentEncoding?: \"gzip\";\n}\n\n/** @beta */\nexport abstract class CloudStorageService {\n  public initialize(): void { }\n  public terminate(): void { }\n  public abstract id: CloudStorageProvider;\n  public abstract obtainContainerUrl(id: CloudStorageContainerDescriptor, expiry: Date, clientIp?: string): CloudStorageContainerUrl;\n  public abstract upload(container: string, name: string, data: Uint8Array, options?: CloudStorageUploadOptions, metadata?: object): Promise<string>;\n  public async download(_name: string): Promise<Readable | undefined> { return undefined; }\n\n  protected makeDescriptor(id: CloudStorageContainerDescriptor) {\n    return { name: id.name, provider: this.id };\n  }\n}\n\n/** @beta */\nexport class AzureBlobStorage extends CloudStorageService {\n  private _service: Azure.BlobServiceClient;\n  private _credential: Azure.StorageSharedKeyCredential;\n  private _baseUrl: string;\n\n  public constructor(credentials: AzureBlobStorageCredentials) {\n    super();\n\n    if (!credentials.account || !credentials.accessKey) {\n      throw new IModelError(BentleyStatus.ERROR, \"Invalid credentials for Azure blob storage.\");\n    }\n\n    this._baseUrl = credentials.baseUrl ?? `https://${credentials.account}.blob.core.windows.net`;\n\n    this._credential = new Azure.StorageSharedKeyCredential(credentials.account, credentials.accessKey);\n    const options: Azure.StoragePipelineOptions = {};\n    const pipeline = Azure.newPipeline(this._credential, options);\n    this._service = new Azure.BlobServiceClient(this._baseUrl, pipeline);\n  }\n\n  public readonly id = CloudStorageProvider.Azure;\n\n  public obtainContainerUrl(id: CloudStorageContainerDescriptor, expiry: Date, clientIp?: string): CloudStorageContainerUrl {\n    const policy: Azure.BlobSASSignatureValues = {\n      containerName: id.name,\n      permissions: Azure.ContainerSASPermissions.parse(\"rl\"),\n      expiresOn: expiry,\n    };\n\n    if (clientIp && clientIp !== \"localhost\" && clientIp !== \"127.0.0.1\" && clientIp !== \"::1\") {\n      policy.ipRange = { start: clientIp };\n    }\n\n    const token = Azure.generateBlobSASQueryParameters(policy, this._credential);\n    const url = new URL(this._baseUrl);\n    url.pathname = `${url.pathname.replace(/\\/*$/, \"\")}/${id.name}`;\n    url.search = token.toString();\n\n    const urlObject: CloudStorageContainerUrl = {\n      descriptor: this.makeDescriptor(id),\n      valid: 0,\n      expires: expiry.getTime(),\n      url: url.toString(),\n    };\n\n    return urlObject;\n  }\n\n  public async ensureContainer(name: string): Promise<void> {\n    try {\n      const container = this._service.getContainerClient(name);\n      await container.create();\n      return;\n    } catch (maybeErr) {\n      try {\n        const err = maybeErr as Azure.RestError;\n        if (err.statusCode === 409 && err.code === \"ContainerAlreadyExists\") {\n          return;\n        }\n      } catch { }\n\n      if (typeof (maybeErr) !== \"undefined\" && maybeErr) {\n        throw maybeErr;\n      }\n\n      throw new IModelError(BentleyStatus.ERROR, `Unable to create container \"${name}\".`);\n    }\n  }\n\n  public async upload(container: string, name: string, data: Uint8Array, options?: CloudStorageUploadOptions, metadata?: object): Promise<string> {\n    try {\n      await this.ensureContainer(container);\n\n      const containerClient = this._service.getContainerClient(container);\n      const blocks = containerClient.getBlockBlobClient(name);\n      const blobHTTPHeaders = {\n        blobContentType: options?.type ?? \"application/octet-stream\",\n        blobCacheControl: options?.cacheControl ?? \"private, max-age=31536000, immutable\",\n      };\n\n      const blobOptions: Azure.BlockBlobUploadStreamOptions = metadata ?\n        {\n          blobHTTPHeaders,\n          metadata: { ...metadata },\n        } : {\n          blobHTTPHeaders,\n        };\n\n      const dataStream = new PassThrough();\n      dataStream.end(data);\n\n      let source: Readable;\n\n      if (options && options.contentEncoding === \"gzip\") {\n        if (undefined === blobOptions.blobHTTPHeaders)\n          throw new IModelError(BentleyStatus.ERROR, \"Blob HTTP headers object is undefined.\");\n\n        blobOptions.blobHTTPHeaders.blobContentEncoding = options.contentEncoding;\n        const compressor = zlib.createGzip();\n        source = dataStream.pipe(compressor);\n      } else {\n        source = dataStream;\n      }\n\n      const blockSize = 100 * 1024 * 1024;\n      const concurrency = 1;\n      const result = await blocks.uploadStream(source, blockSize, concurrency, blobOptions);\n      return result.etag ?? \"\";\n    } catch (maybeErr) {\n      if (typeof (maybeErr) !== \"undefined\" && maybeErr) {\n        throw maybeErr;\n      }\n\n      throw new IModelError(BentleyStatus.ERROR, `Unable to upload \"${name}\".`);\n    }\n  }\n}\n\n/** @internal */\nexport class CloudStorageTileUploader {\n  private _activeUploads: Map<string, Promise<void>> = new Map();\n\n  public get activeUploads(): Iterable<Promise<void>> {\n    return this._activeUploads.values();\n  }\n\n  private async uploadToCache(id: TileContentIdentifier, content: Uint8Array, containerKey: string, resourceKey: string, metadata?: object,) {\n    await new Promise((resolve) => setTimeout(resolve));\n\n    try {\n      const options: CloudStorageUploadOptions = {};\n      if (IModelHost.compressCachedTiles) {\n        options.contentEncoding = \"gzip\";\n      }\n\n      const perfInfo = { ...id.tokenProps, treeId: id.treeId, contentId: id.contentId, size: content.byteLength, compress: IModelHost.compressCachedTiles };\n      const perfLogger = new PerfLogger(\"Uploading tile to external tile cache\", () => perfInfo);\n      await IModelHost.tileCacheService?.upload(containerKey, resourceKey, content, options, metadata);\n      perfLogger.dispose();\n    } catch (err) {\n      Logger.logError(BackendLoggerCategory.IModelTileUpload, (err instanceof Error) ? err.toString() : JSON.stringify(err));\n    }\n\n    this._activeUploads.delete(containerKey + resourceKey);\n  }\n\n  public async cacheTile(tokenProps: IModelRpcProps, treeId: string, contentId: string, content: Uint8Array, guid: string | undefined, metadata?: object): Promise<void> {\n    const id: TileContentIdentifier = { tokenProps, treeId, contentId, guid };\n\n    const cache = CloudStorageTileCache.getCache();\n    const containerKey = cache.formContainerName(id);\n    const resourceKey = cache.formResourceName(id);\n    const key = containerKey + resourceKey;\n\n    if (this._activeUploads.has(key))\n      return;\n\n    const promise = this.uploadToCache(id, content, containerKey, resourceKey, metadata);\n    this._activeUploads.set(key, promise);\n    await promise;\n  }\n}\n"]}
+{"version":3,"file":"CloudStorageBackend.js","sourceRoot":"","sources":["../../src/CloudStorageBackend.ts"],"names":[],"mappings":";;;AAAA;;;+FAG+F;AAC/F,mCAA+C;AAC/C,6BAA6B;AAC7B,6CAA6C;AAC7C,sDAAyD;AACzD,oDAG4B;AAC5B,mEAAgE;AAChE,6CAA0C;AAgB1C,YAAY;AACZ,MAAsB,mBAAmB;IAChC,UAAU,KAAW,CAAC;IACtB,SAAS,KAAW,CAAC;IAIrB,KAAK,CAAC,QAAQ,CAAC,KAAa,IAAmC,OAAO,SAAS,CAAC,CAAC,CAAC;IAE/E,cAAc,CAAC,EAAmC;QAC1D,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;IAC9C,CAAC;CACF;AAXD,kDAWC;AAED,YAAY;AACZ,MAAa,gBAAiB,SAAQ,mBAAmB;IAKvD,YAAmB,WAAwC;;QACzD,KAAK,EAAE,CAAC;QAcM,OAAE,GAAG,kCAAoB,CAAC,KAAK,CAAC;QAZ9C,IAAI,CAAC,WAAW,CAAC,OAAO,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE;YAClD,MAAM,IAAI,yBAAW,CAAC,2BAAa,CAAC,KAAK,EAAE,6CAA6C,CAAC,CAAC;SAC3F;QAED,IAAI,CAAC,QAAQ,GAAG,MAAA,WAAW,CAAC,OAAO,mCAAI,WAAW,WAAW,CAAC,OAAO,wBAAwB,CAAC;QAE9F,IAAI,CAAC,WAAW,GAAG,IAAI,KAAK,CAAC,0BAA0B,CAAC,WAAW,CAAC,OAAO,EAAE,WAAW,CAAC,SAAS,CAAC,CAAC;QACpG,MAAM,OAAO,GAAiC,EAAE,CAAC;QACjD,MAAM,QAAQ,GAAG,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QAC9D,IAAI,CAAC,QAAQ,GAAG,IAAI,KAAK,CAAC,iBAAiB,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACvE,CAAC;IAIM,kBAAkB,CAAC,EAAmC,EAAE,MAAY,EAAE,QAAiB;QAC5F,MAAM,MAAM,GAAiC;YAC3C,aAAa,EAAE,EAAE,CAAC,IAAI;YACtB,WAAW,EAAE,KAAK,CAAC,uBAAuB,CAAC,KAAK,CAAC,IAAI,CAAC;YACtD,SAAS,EAAE,MAAM;SAClB,CAAC;QAEF,IAAI,QAAQ,IAAI,QAAQ,KAAK,WAAW,IAAI,QAAQ,KAAK,WAAW,IAAI,QAAQ,KAAK,KAAK,EAAE;YAC1F,MAAM,CAAC,OAAO,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;SACtC;QAED,MAAM,KAAK,GAAG,KAAK,CAAC,8BAA8B,CAAC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QAC7E,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACnC,GAAG,CAAC,QAAQ,GAAG,GAAG,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;QAChE,GAAG,CAAC,MAAM,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;QAE9B,MAAM,SAAS,GAA6B;YAC1C,UAAU,EAAE,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC;YACnC,KAAK,EAAE,CAAC;YACR,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE;YACzB,GAAG,EAAE,GAAG,CAAC,QAAQ,EAAE;SACpB,CAAC;QAEF,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,KAAK,CAAC,eAAe,CAAC,IAAY;QACvC,IAAI;YACF,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;YACzD,MAAM,SAAS,CAAC,MAAM,EAAE,CAAC;YACzB,OAAO;SACR;QAAC,OAAO,QAAQ,EAAE;YACjB,IAAI;gBACF,MAAM,GAAG,GAAG,QAA2B,CAAC;gBACxC,IAAI,GAAG,CAAC,UAAU,KAAK,GAAG,IAAI,GAAG,CAAC,IAAI,KAAK,wBAAwB,EAAE;oBACnE,OAAO;iBACR;aACF;YAAC,MAAM,GAAG;YAEX,IAAI,OAAO,CAAC,QAAQ,CAAC,KAAK,WAAW,IAAI,QAAQ,EAAE;gBACjD,MAAM,QAAQ,CAAC;aAChB;YAED,MAAM,IAAI,yBAAW,CAAC,2BAAa,CAAC,KAAK,EAAE,+BAA+B,IAAI,IAAI,CAAC,CAAC;SACrF;IACH,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,SAAiB,EAAE,IAAY,EAAE,IAAgB,EAAE,OAAmC,EAAE,QAAiB;;QAC3H,IAAI;YACF,MAAM,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YAEtC,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;YACpE,MAAM,MAAM,GAAG,eAAe,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;YACxD,MAAM,eAAe,GAAG;gBACtB,eAAe,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI,mCAAI,0BAA0B;gBAC5D,gBAAgB,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,YAAY,mCAAI,sCAAsC;aAClF,CAAC;YAEF,MAAM,WAAW,GAAuC,QAAQ,CAAC,CAAC;gBAChE;oBACE,eAAe;oBACf,QAAQ,EAAE,EAAE,GAAG,QAAQ,EAAE;iBAC1B,CAAC,CAAC,CAAC;gBACF,eAAe;aAChB,CAAC;YAEJ,MAAM,UAAU,GAAG,IAAI,oBAAW,EAAE,CAAC;YACrC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAErB,IAAI,MAAgB,CAAC;YAErB,IAAI,OAAO,IAAI,OAAO,CAAC,eAAe,KAAK,MAAM,EAAE;gBACjD,IAAI,SAAS,KAAK,WAAW,CAAC,eAAe;oBAC3C,MAAM,IAAI,yBAAW,CAAC,2BAAa,CAAC,KAAK,EAAE,wCAAwC,CAAC,CAAC;gBAEvF,WAAW,CAAC,eAAe,CAAC,mBAAmB,GAAG,OAAO,CAAC,eAAe,CAAC;gBAC1E,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;gBACrC,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;aACtC;iBAAM;gBACL,MAAM,GAAG,UAAU,CAAC;aACrB;YAED,MAAM,SAAS,GAAG,GAAG,GAAG,IAAI,GAAG,IAAI,CAAC;YACpC,MAAM,WAAW,GAAG,CAAC,CAAC;YACtB,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;YACtF,OAAO,MAAA,MAAM,CAAC,IAAI,mCAAI,EAAE,CAAC;SAC1B;QAAC,OAAO,QAAQ,EAAE;YACjB,IAAI,OAAO,CAAC,QAAQ,CAAC,KAAK,WAAW,IAAI,QAAQ,EAAE;gBACjD,MAAM,QAAQ,CAAC;aAChB;YAED,MAAM,IAAI,yBAAW,CAAC,2BAAa,CAAC,KAAK,EAAE,qBAAqB,IAAI,IAAI,CAAC,CAAC;SAC3E;IACH,CAAC;CACF;AApHD,4CAoHC;AAED,gBAAgB;AAChB,MAAa,wBAAwB;IAArC;QACU,mBAAc,GAA+B,IAAI,GAAG,EAAE,CAAC;IAyCjE,CAAC;IAvCC,IAAW,aAAa;QACtB,OAAO,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC;IACtC,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,EAAyB,EAAE,OAAmB,EAAE,YAAoB,EAAE,WAAmB,EAAE,QAAiB;;QACtI,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC;QAEpD,IAAI;YACF,MAAM,OAAO,GAA8B,EAAE,CAAC;YAC9C,IAAI,uBAAU,CAAC,mBAAmB,EAAE;gBAClC,OAAO,CAAC,eAAe,GAAG,MAAM,CAAC;aAClC;YAED,MAAM,QAAQ,GAAG,EAAE,GAAG,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,OAAO,CAAC,UAAU,EAAE,QAAQ,EAAE,uBAAU,CAAC,mBAAmB,EAAE,CAAC;YACtJ,MAAM,UAAU,GAAG,IAAI,yBAAU,CAAC,uCAAuC,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,CAAC;YAC3F,MAAM,CAAA,MAAA,uBAAU,CAAC,gBAAgB,0CAAE,MAAM,CAAC,YAAY,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAA,CAAC;YACjG,UAAU,CAAC,OAAO,EAAE,CAAC;SACtB;QAAC,OAAO,GAAG,EAAE;YACZ,qBAAM,CAAC,QAAQ,CAAC,6CAAqB,CAAC,gBAAgB,EAAE,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;SACxH;QAED,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,YAAY,GAAG,WAAW,CAAC,CAAC;IACzD,CAAC;IAEM,KAAK,CAAC,SAAS,CAAC,UAA0B,EAAE,MAAc,EAAE,SAAiB,EAAE,OAAmB,EAAE,IAAwB,EAAE,QAAiB;QACpJ,MAAM,EAAE,GAA0B,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;QAE1E,MAAM,KAAK,GAAG,mCAAqB,CAAC,QAAQ,EAAE,CAAC;QAC/C,MAAM,YAAY,GAAG,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;QACjD,MAAM,WAAW,GAAG,KAAK,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;QAC/C,MAAM,GAAG,GAAG,YAAY,GAAG,WAAW,CAAC;QAEvC,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC;YAC9B,OAAO;QAET,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC;QACrF,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QACtC,MAAM,OAAO,CAAC;IAChB,CAAC;CACF;AA1CD,4DA0CC","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\nimport { PassThrough, Readable } from \"stream\";\r\nimport * as zlib from \"zlib\";\r\nimport * as Azure from \"@azure/storage-blob\";\r\nimport { Logger, PerfLogger } from \"@itwin/core-bentley\";\r\nimport {\r\n  BentleyStatus, CloudStorageContainerDescriptor, CloudStorageContainerUrl, CloudStorageProvider, CloudStorageTileCache, IModelError, IModelRpcProps,\r\n  TileContentIdentifier,\r\n} from \"@itwin/core-common\";\r\nimport { BackendLoggerCategory } from \"./BackendLoggerCategory\";\r\nimport { IModelHost } from \"./IModelHost\";\r\n\r\n/** @beta */\r\nexport interface AzureBlobStorageCredentials {\r\n  account: string;\r\n  accessKey: string;\r\n  baseUrl?: string;\r\n}\r\n\r\n/** @beta */\r\nexport interface CloudStorageUploadOptions {\r\n  type?: string;\r\n  cacheControl?: string;\r\n  contentEncoding?: \"gzip\";\r\n}\r\n\r\n/** @beta */\r\nexport abstract class CloudStorageService {\r\n  public initialize(): void { }\r\n  public terminate(): void { }\r\n  public abstract id: CloudStorageProvider;\r\n  public abstract obtainContainerUrl(id: CloudStorageContainerDescriptor, expiry: Date, clientIp?: string): CloudStorageContainerUrl;\r\n  public abstract upload(container: string, name: string, data: Uint8Array, options?: CloudStorageUploadOptions, metadata?: object): Promise<string>;\r\n  public async download(_name: string): Promise<Readable | undefined> { return undefined; }\r\n\r\n  protected makeDescriptor(id: CloudStorageContainerDescriptor) {\r\n    return { name: id.name, provider: this.id };\r\n  }\r\n}\r\n\r\n/** @beta */\r\nexport class AzureBlobStorage extends CloudStorageService {\r\n  private _service: Azure.BlobServiceClient;\r\n  private _credential: Azure.StorageSharedKeyCredential;\r\n  private _baseUrl: string;\r\n\r\n  public constructor(credentials: AzureBlobStorageCredentials) {\r\n    super();\r\n\r\n    if (!credentials.account || !credentials.accessKey) {\r\n      throw new IModelError(BentleyStatus.ERROR, \"Invalid credentials for Azure blob storage.\");\r\n    }\r\n\r\n    this._baseUrl = credentials.baseUrl ?? `https://${credentials.account}.blob.core.windows.net`;\r\n\r\n    this._credential = new Azure.StorageSharedKeyCredential(credentials.account, credentials.accessKey);\r\n    const options: Azure.StoragePipelineOptions = {};\r\n    const pipeline = Azure.newPipeline(this._credential, options);\r\n    this._service = new Azure.BlobServiceClient(this._baseUrl, pipeline);\r\n  }\r\n\r\n  public readonly id = CloudStorageProvider.Azure;\r\n\r\n  public obtainContainerUrl(id: CloudStorageContainerDescriptor, expiry: Date, clientIp?: string): CloudStorageContainerUrl {\r\n    const policy: Azure.BlobSASSignatureValues = {\r\n      containerName: id.name,\r\n      permissions: Azure.ContainerSASPermissions.parse(\"rl\"),\r\n      expiresOn: expiry,\r\n    };\r\n\r\n    if (clientIp && clientIp !== \"localhost\" && clientIp !== \"127.0.0.1\" && clientIp !== \"::1\") {\r\n      policy.ipRange = { start: clientIp };\r\n    }\r\n\r\n    const token = Azure.generateBlobSASQueryParameters(policy, this._credential);\r\n    const url = new URL(this._baseUrl);\r\n    url.pathname = `${url.pathname.replace(/\\/*$/, \"\")}/${id.name}`;\r\n    url.search = token.toString();\r\n\r\n    const urlObject: CloudStorageContainerUrl = {\r\n      descriptor: this.makeDescriptor(id),\r\n      valid: 0,\r\n      expires: expiry.getTime(),\r\n      url: url.toString(),\r\n    };\r\n\r\n    return urlObject;\r\n  }\r\n\r\n  public async ensureContainer(name: string): Promise<void> {\r\n    try {\r\n      const container = this._service.getContainerClient(name);\r\n      await container.create();\r\n      return;\r\n    } catch (maybeErr) {\r\n      try {\r\n        const err = maybeErr as Azure.RestError;\r\n        if (err.statusCode === 409 && err.code === \"ContainerAlreadyExists\") {\r\n          return;\r\n        }\r\n      } catch { }\r\n\r\n      if (typeof (maybeErr) !== \"undefined\" && maybeErr) {\r\n        throw maybeErr;\r\n      }\r\n\r\n      throw new IModelError(BentleyStatus.ERROR, `Unable to create container \"${name}\".`);\r\n    }\r\n  }\r\n\r\n  public async upload(container: string, name: string, data: Uint8Array, options?: CloudStorageUploadOptions, metadata?: object): Promise<string> {\r\n    try {\r\n      await this.ensureContainer(container);\r\n\r\n      const containerClient = this._service.getContainerClient(container);\r\n      const blocks = containerClient.getBlockBlobClient(name);\r\n      const blobHTTPHeaders = {\r\n        blobContentType: options?.type ?? \"application/octet-stream\",\r\n        blobCacheControl: options?.cacheControl ?? \"private, max-age=31536000, immutable\",\r\n      };\r\n\r\n      const blobOptions: Azure.BlockBlobUploadStreamOptions = metadata ?\r\n        {\r\n          blobHTTPHeaders,\r\n          metadata: { ...metadata },\r\n        } : {\r\n          blobHTTPHeaders,\r\n        };\r\n\r\n      const dataStream = new PassThrough();\r\n      dataStream.end(data);\r\n\r\n      let source: Readable;\r\n\r\n      if (options && options.contentEncoding === \"gzip\") {\r\n        if (undefined === blobOptions.blobHTTPHeaders)\r\n          throw new IModelError(BentleyStatus.ERROR, \"Blob HTTP headers object is undefined.\");\r\n\r\n        blobOptions.blobHTTPHeaders.blobContentEncoding = options.contentEncoding;\r\n        const compressor = zlib.createGzip();\r\n        source = dataStream.pipe(compressor);\r\n      } else {\r\n        source = dataStream;\r\n      }\r\n\r\n      const blockSize = 100 * 1024 * 1024;\r\n      const concurrency = 1;\r\n      const result = await blocks.uploadStream(source, blockSize, concurrency, blobOptions);\r\n      return result.etag ?? \"\";\r\n    } catch (maybeErr) {\r\n      if (typeof (maybeErr) !== \"undefined\" && maybeErr) {\r\n        throw maybeErr;\r\n      }\r\n\r\n      throw new IModelError(BentleyStatus.ERROR, `Unable to upload \"${name}\".`);\r\n    }\r\n  }\r\n}\r\n\r\n/** @internal */\r\nexport class CloudStorageTileUploader {\r\n  private _activeUploads: Map<string, Promise<void>> = new Map();\r\n\r\n  public get activeUploads(): Iterable<Promise<void>> {\r\n    return this._activeUploads.values();\r\n  }\r\n\r\n  private async uploadToCache(id: TileContentIdentifier, content: Uint8Array, containerKey: string, resourceKey: string, metadata?: object,) {\r\n    await new Promise((resolve) => setTimeout(resolve));\r\n\r\n    try {\r\n      const options: CloudStorageUploadOptions = {};\r\n      if (IModelHost.compressCachedTiles) {\r\n        options.contentEncoding = \"gzip\";\r\n      }\r\n\r\n      const perfInfo = { ...id.tokenProps, treeId: id.treeId, contentId: id.contentId, size: content.byteLength, compress: IModelHost.compressCachedTiles };\r\n      const perfLogger = new PerfLogger(\"Uploading tile to external tile cache\", () => perfInfo);\r\n      await IModelHost.tileCacheService?.upload(containerKey, resourceKey, content, options, metadata);\r\n      perfLogger.dispose();\r\n    } catch (err) {\r\n      Logger.logError(BackendLoggerCategory.IModelTileUpload, (err instanceof Error) ? err.toString() : JSON.stringify(err));\r\n    }\r\n\r\n    this._activeUploads.delete(containerKey + resourceKey);\r\n  }\r\n\r\n  public async cacheTile(tokenProps: IModelRpcProps, treeId: string, contentId: string, content: Uint8Array, guid: string | undefined, metadata?: object): Promise<void> {\r\n    const id: TileContentIdentifier = { tokenProps, treeId, contentId, guid };\r\n\r\n    const cache = CloudStorageTileCache.getCache();\r\n    const containerKey = cache.formContainerName(id);\r\n    const resourceKey = cache.formResourceName(id);\r\n    const key = containerKey + resourceKey;\r\n\r\n    if (this._activeUploads.has(key))\r\n      return;\r\n\r\n    const promise = this.uploadToCache(id, content, containerKey, resourceKey, metadata);\r\n    this._activeUploads.set(key, promise);\r\n    await promise;\r\n  }\r\n}\r\n"]}