@metamask/shield-controller 1.0.0 → 1.1.0

package/CHANGELOG.md

@@ -7,6 +7,12 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [1.1.0]
+
+### Fixed
+
+- Fixed and optimized shield-coverage-result polling with Cockatiel Policy from Controller-utils. ([#6847](https://github.com/MetaMask/core/pull/6847))
+
 ## [1.0.0]
 
 ### Added
@@ -99,7 +105,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 - Initial release of the shield-controller package ([#6137](https://github.com/MetaMask/core/pull/6137)
 
-[Unreleased]: https://github.com/MetaMask/core/compare/@metamask/shield-controller@1.0.0...HEAD
+[Unreleased]: https://github.com/MetaMask/core/compare/@metamask/shield-controller@1.1.0...HEAD
+[1.1.0]: https://github.com/MetaMask/core/compare/@metamask/shield-controller@1.0.0...@metamask/shield-controller@1.1.0
 [1.0.0]: https://github.com/MetaMask/core/compare/@metamask/shield-controller@0.4.0...@metamask/shield-controller@1.0.0
 [0.4.0]: https://github.com/MetaMask/core/compare/@metamask/shield-controller@0.3.2...@metamask/shield-controller@0.4.0
 [0.3.2]: https://github.com/MetaMask/core/compare/@metamask/shield-controller@0.3.1...@metamask/shield-controller@0.3.2

package/dist/backend.cjs

@@ -10,26 +10,30 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
     if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
     return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
 };
-var _ShieldRemoteBackend_instances, _ShieldRemoteBackend_getAccessToken, _ShieldRemoteBackend_getCoverageResultTimeout, _ShieldRemoteBackend_getCoverageResultPollInterval, _ShieldRemoteBackend_baseUrl, _ShieldRemoteBackend_fetch, _ShieldRemoteBackend_initCoverageCheck, _ShieldRemoteBackend_getCoverageResult, _ShieldRemoteBackend_createHeaders;
+var _ShieldRemoteBackend_instances, _ShieldRemoteBackend_getAccessToken, _ShieldRemoteBackend_baseUrl, _ShieldRemoteBackend_fetch, _ShieldRemoteBackend_pollingPolicy, _ShieldRemoteBackend_initCoverageCheck, _ShieldRemoteBackend_getCoverageResult, _ShieldRemoteBackend_createHeaders;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.parseSignatureRequestMethod = exports.ShieldRemoteBackend = void 0;
+const controller_utils_1 = require("@metamask/controller-utils");
 const signature_controller_1 = require("@metamask/signature-controller");
 const constants_1 = require("./constants.cjs");
+const polling_with_policy_1 = require("./polling-with-policy.cjs");
 class ShieldRemoteBackend {
     constructor({ getAccessToken, getCoverageResultTimeout = 5000, // milliseconds
     getCoverageResultPollInterval = 1000, // milliseconds
     baseUrl, fetch: fetchFn, }) {
         _ShieldRemoteBackend_instances.add(this);
         _ShieldRemoteBackend_getAccessToken.set(this, void 0);
-        _ShieldRemoteBackend_getCoverageResultTimeout.set(this, void 0);
-        _ShieldRemoteBackend_getCoverageResultPollInterval.set(this, void 0);
         _ShieldRemoteBackend_baseUrl.set(this, void 0);
         _ShieldRemoteBackend_fetch.set(this, void 0);
+        _ShieldRemoteBackend_pollingPolicy.set(this, void 0);
         __classPrivateFieldSet(this, _ShieldRemoteBackend_getAccessToken, getAccessToken, "f");
-        __classPrivateFieldSet(this, _ShieldRemoteBackend_getCoverageResultTimeout, getCoverageResultTimeout, "f");
-        __classPrivateFieldSet(this, _ShieldRemoteBackend_getCoverageResultPollInterval, getCoverageResultPollInterval, "f");
         __classPrivateFieldSet(this, _ShieldRemoteBackend_baseUrl, baseUrl, "f");
         __classPrivateFieldSet(this, _ShieldRemoteBackend_fetch, fetchFn, "f");
+        const { backoff, maxRetries } = computePollingIntervalAndRetryCount(getCoverageResultTimeout, getCoverageResultPollInterval);
+        __classPrivateFieldSet(this, _ShieldRemoteBackend_pollingPolicy, new polling_with_policy_1.PollingWithCockatielPolicy({
+            backoff,
+            maxRetries,
+        }), "f");
     }
     async checkCoverage(req) {
         let { coverageId } = req;
@@ -38,9 +42,7 @@ class ShieldRemoteBackend {
             ({ coverageId } = await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_initCoverageCheck).call(this, 'v1/transaction/coverage/init', reqBody));
         }
         const txCoverageResultUrl = `${__classPrivateFieldGet(this, _ShieldRemoteBackend_baseUrl, "f")}/v1/transaction/coverage/result`;
-        const coverageResult = await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_getCoverageResult).call(this, coverageId, {
-            coverageResultUrl: txCoverageResultUrl,
-        });
+        const coverageResult = await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_getCoverageResult).call(this, req.txMeta.id, coverageId, txCoverageResultUrl);
         return {
             coverageId,
             message: coverageResult.message,
@@ -55,9 +57,7 @@ class ShieldRemoteBackend {
             ({ coverageId } = await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_initCoverageCheck).call(this, 'v1/signature/coverage/init', reqBody));
         }
         const signatureCoverageResultUrl = `${__classPrivateFieldGet(this, _ShieldRemoteBackend_baseUrl, "f")}/v1/signature/coverage/result`;
-        const coverageResult = await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_getCoverageResult).call(this, coverageId, {
-            coverageResultUrl: signatureCoverageResultUrl,
-        });
+        const coverageResult = await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_getCoverageResult).call(this, req.signatureRequest.id, coverageId, signatureCoverageResultUrl);
         return {
             coverageId,
             message: coverageResult.message,
@@ -72,6 +72,8 @@ class ShieldRemoteBackend {
             status: req.status,
             ...initBody,
         };
+        // cancel the pending get coverage result request
+        __classPrivateFieldGet(this, _ShieldRemoteBackend_pollingPolicy, "f").abortPendingRequest(req.signatureRequest.id);
         const res = await __classPrivateFieldGet(this, _ShieldRemoteBackend_fetch, "f").call(this, `${__classPrivateFieldGet(this, _ShieldRemoteBackend_baseUrl, "f")}/v1/signature/coverage/log`, {
             method: 'POST',
             headers: await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_createHeaders).call(this),
@@ -88,6 +90,8 @@ class ShieldRemoteBackend {
             status: req.status,
             ...initBody,
         };
+        // cancel the pending get coverage result request
+        __classPrivateFieldGet(this, _ShieldRemoteBackend_pollingPolicy, "f").abortPendingRequest(req.txMeta.id);
         const res = await __classPrivateFieldGet(this, _ShieldRemoteBackend_fetch, "f").call(this, `${__classPrivateFieldGet(this, _ShieldRemoteBackend_baseUrl, "f")}/v1/transaction/coverage/log`, {
             method: 'POST',
             headers: await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_createHeaders).call(this),
@@ -99,7 +103,7 @@ class ShieldRemoteBackend {
     }
 }
 exports.ShieldRemoteBackend = ShieldRemoteBackend;
-_ShieldRemoteBackend_getAccessToken = new WeakMap(), _ShieldRemoteBackend_getCoverageResultTimeout = new WeakMap(), _ShieldRemoteBackend_getCoverageResultPollInterval = new WeakMap(), _ShieldRemoteBackend_baseUrl = new WeakMap(), _ShieldRemoteBackend_fetch = new WeakMap(), _ShieldRemoteBackend_instances = new WeakSet(), _ShieldRemoteBackend_initCoverageCheck = async function _ShieldRemoteBackend_initCoverageCheck(path, reqBody) {
+_ShieldRemoteBackend_getAccessToken = new WeakMap(), _ShieldRemoteBackend_baseUrl = new WeakMap(), _ShieldRemoteBackend_fetch = new WeakMap(), _ShieldRemoteBackend_pollingPolicy = new WeakMap(), _ShieldRemoteBackend_instances = new WeakSet(), _ShieldRemoteBackend_initCoverageCheck = async function _ShieldRemoteBackend_initCoverageCheck(path, reqBody) {
     const res = await __classPrivateFieldGet(this, _ShieldRemoteBackend_fetch, "f").call(this, `${__classPrivateFieldGet(this, _ShieldRemoteBackend_baseUrl, "f")}/${path}`, {
         method: 'POST',
         headers: await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_createHeaders).call(this),
@@ -109,40 +113,33 @@ _ShieldRemoteBackend_getAccessToken = new WeakMap(), _ShieldRemoteBackend_getCov
         throw new Error(`Failed to init coverage check: ${res.status}`);
     }
     return (await res.json());
-}, _ShieldRemoteBackend_getCoverageResult = async function _ShieldRemoteBackend_getCoverageResult(coverageId, configs) {
+}, _ShieldRemoteBackend_getCoverageResult = async function _ShieldRemoteBackend_getCoverageResult(requestId, coverageId, coverageResultUrl) {
     const reqBody = {
         coverageId,
     };
-    const timeout = configs?.timeout ?? __classPrivateFieldGet(this, _ShieldRemoteBackend_getCoverageResultTimeout, "f");
-    const pollInterval = configs?.pollInterval ?? __classPrivateFieldGet(this, _ShieldRemoteBackend_getCoverageResultPollInterval, "f");
     const headers = await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_createHeaders).call(this);
-    return await new Promise((resolve, reject) => {
-        let timeoutReached = false;
-        setTimeout(() => {
-            timeoutReached = true;
-            reject(new Error('Timeout waiting for coverage result'));
-        }, timeout);
-        const poll = async () => {
-            // The timeoutReached variable is modified in the timeout callback.
-            // eslint-disable-next-line no-unmodified-loop-condition
-            while (!timeoutReached) {
-                const startTime = Date.now();
-                const res = await __classPrivateFieldGet(this, _ShieldRemoteBackend_fetch, "f").call(this, configs.coverageResultUrl, {
-                    method: 'POST',
-                    headers,
-                    body: JSON.stringify(reqBody),
-                });
-                if (res.status === 200) {
-                    return (await res.json());
-                }
-                await sleep(pollInterval - (Date.now() - startTime));
-            }
-            // The following line will not have an effect as the upper level promise
-            // will already be rejected by now.
-            throw new Error('unexpected error');
-        };
-        poll().then(resolve).catch(reject);
-    });
+    const getCoverageResultFn = async (signal) => {
+        const res = await __classPrivateFieldGet(this, _ShieldRemoteBackend_fetch, "f").call(this, coverageResultUrl, {
+            method: 'POST',
+            headers,
+            body: JSON.stringify(reqBody),
+            signal,
+        });
+        if (res.status === 200) {
+            return (await res.json());
+        }
+        // parse the error message from the response body
+        let errorMessage = 'Timeout waiting for coverage result';
+        try {
+            const errorJson = await res.json();
+            errorMessage = `Failed to get coverage result: ${errorJson.message || errorJson.status}`;
+        }
+        catch {
+            errorMessage = `Failed to get coverage result: ${res.status}`;
+        }
+        throw new controller_utils_1.HttpError(res.status, errorMessage);
+    };
+    return __classPrivateFieldGet(this, _ShieldRemoteBackend_pollingPolicy, "f").start(requestId, getCoverageResultFn);
 }, _ShieldRemoteBackend_createHeaders = async function _ShieldRemoteBackend_createHeaders() {
     const accessToken = await __classPrivateFieldGet(this, _ShieldRemoteBackend_getAccessToken, "f").call(this);
     return {
@@ -150,15 +147,6 @@ _ShieldRemoteBackend_getAccessToken = new WeakMap(), _ShieldRemoteBackend_getCov
         Authorization: `Bearer ${accessToken}`,
     };
 };
-/**
- * Sleep for a specified amount of time.
- *
- * @param ms - The number of milliseconds to sleep.
- * @returns A promise that resolves after the specified amount of time.
- */
-async function sleep(ms) {
-    return new Promise((resolve) => setTimeout(resolve, ms));
-}
 /**
  * Make the body for the init coverage check request.
  *
@@ -220,4 +208,22 @@ function parseSignatureRequestMethod(signatureRequest) {
     return signatureRequest.type;
 }
 exports.parseSignatureRequestMethod = parseSignatureRequestMethod;
+/**
+ * Compute the polling interval and retry count for the Cockatiel policy based on the timeout and poll interval given.
+ *
+ * @param timeout - The timeout in milliseconds.
+ * @param pollInterval - The poll interval in milliseconds.
+ * @returns The polling interval and retry count.
+ */
+function computePollingIntervalAndRetryCount(timeout, pollInterval) {
+    const backoff = new controller_utils_1.ConstantBackoff(pollInterval);
+    const computedMaxRetries = Math.floor(timeout / pollInterval) + 1;
+    const maxRetries = isNaN(computedMaxRetries) || !isFinite(computedMaxRetries)
+        ? controller_utils_1.DEFAULT_MAX_RETRIES
+        : computedMaxRetries;
+    return {
+        backoff,
+        maxRetries,
+    };
+}
 //# sourceMappingURL=backend.cjs.map

package/dist/backend.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"backend.cjs","sourceRoot":"","sources":["../src/backend.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,yEAIwC;AAIxC,+CAAmD;AA+CnD,MAAa,mBAAmB;IAW9B,YAAY,EACV,cAAc,EACd,wBAAwB,GAAG,IAAI,EAAE,eAAe;IAChD,6BAA6B,GAAG,IAAI,EAAE,eAAe;IACrD,OAAO,EACP,KAAK,EAAE,OAAO,GAOf;;QAtBQ,sDAAuC;QAEvC,gEAAkC;QAElC,qEAAuC;QAEvC,+CAAiB;QAEjB,6CAAgC;QAevC,uBAAA,IAAI,uCAAmB,cAAc,MAAA,CAAC;QACtC,uBAAA,IAAI,iDAA6B,wBAAwB,MAAA,CAAC;QAC1D,uBAAA,IAAI,sDAAkC,6BAA6B,MAAA,CAAC;QACpE,uBAAA,IAAI,gCAAY,OAAO,MAAA,CAAC;QACxB,uBAAA,IAAI,8BAAU,OAAO,MAAA,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,GAAyB;QAC3C,IAAI,EAAE,UAAU,EAAE,GAAG,GAAG,CAAC;QACzB,IAAI,CAAC,UAAU,EAAE;YACf,MAAM,OAAO,GAAG,yBAAyB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACtD,CAAC,EAAE,UAAU,EAAE,GAAG,MAAM,uBAAA,IAAI,8EAAmB,MAAvB,IAAI,EAC1B,8BAA8B,EAC9B,OAAO,CACR,CAAC,CAAC;SACJ;QAED,MAAM,mBAAmB,GAAG,GAAG,uBAAA,IAAI,oCAAS,iCAAiC,CAAC;QAC9E,MAAM,cAAc,GAAG,MAAM,uBAAA,IAAI,8EAAmB,MAAvB,IAAI,EAAoB,UAAU,EAAE;YAC/D,iBAAiB,EAAE,mBAAmB;SACvC,CAAC,CAAC;QACH,OAAO;YACL,UAAU;YACV,OAAO,EAAE,cAAc,CAAC,OAAO;YAC/B,UAAU,EAAE,cAAc,CAAC,UAAU;YACrC,MAAM,EAAE,cAAc,CAAC,MAAM;SAC9B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,sBAAsB,CAC1B,GAAkC;QAElC,IAAI,EAAE,UAAU,EAAE,GAAG,GAAG,CAAC;QACzB,IAAI,CAAC,UAAU,EAAE;YACf,MAAM,OAAO,GAAG,kCAAkC,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;YACzE,CAAC,EAAE,UAAU,EAAE,GAAG,MAAM,uBAAA,IAAI,8EAAmB,MAAvB,IAAI,EAC1B,4BAA4B,EAC5B,OAAO,CACR,CAAC,CAAC;SACJ;QAED,MAAM,0BAA0B,GAAG,GAAG,uBAAA,IAAI,oCAAS,+BAA+B,CAAC;QACnF,MAAM,cAAc,GAAG,MAAM,uBAAA,IAAI,8EAAmB,MAAvB,IAAI,EAAoB,UAAU,EAAE;YAC/D,iBAAiB,EAAE,0BAA0B;SAC9C,CAAC,CAAC;QACH,OAAO;YACL,UAAU;YACV,OAAO,EAAE,cAAc,CAAC,OAAO;YAC/B,UAAU,EAAE,cAAc,CAAC,UAAU;YACrC,MAAM,EAAE,cAAc,CAAC,MAAM;SAC9B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,GAAwB;QACzC,MAAM,QAAQ,GAAG,kCAAkC,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QAC1E,MAAM,IAAI,GAAG;YACX,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,GAAG,QAAQ;SACZ,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,uBAAA,IAAI,kCAAO,MAAX,IAAI,EACpB,GAAG,uBAAA,IAAI,oCAAS,4BAA4B,EAC5C;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,MAAM,uBAAA,IAAI,0EAAe,MAAnB,IAAI,CAAiB;YACpC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CACF,CAAC;QACF,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;YACtB,MAAM,IAAI,KAAK,CAAC,4BAA4B,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;SAC3D;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,GAA0B;QAC7C,MAAM,QAAQ,GAAG,yBAAyB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvD,MAAM,IAAI,GAAG;YACX,eAAe,EAAE,GAAG,CAAC,eAAe;YACpC,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,GAAG,QAAQ;SACZ,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,uBAAA,IAAI,kCAAO,MAAX,IAAI,EACpB,GAAG,uBAAA,IAAI,oCAAS,8BAA8B,EAC9C;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,MAAM,uBAAA,IAAI,0EAAe,MAAnB,IAAI,CAAiB;YACpC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CACF,CAAC;QACF,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;YACtB,MAAM,IAAI,KAAK,CAAC,8BAA8B,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;SAC7D;IACH,CAAC;CAwEF;AA7LD,kDA6LC;2WAtEC,KAAK,iDACH,IAAY,EACZ,OAAgB;IAEhB,MAAM,GAAG,GAAG,MAAM,uBAAA,IAAI,kCAAO,MAAX,IAAI,EAAQ,GAAG,uBAAA,IAAI,oCAAS,IAAI,IAAI,EAAE,EAAE;QACxD,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,MAAM,uBAAA,IAAI,0EAAe,MAAnB,IAAI,CAAiB;QACpC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;KAC9B,CAAC,CAAC;IACH,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;QACtB,MAAM,IAAI,KAAK,CAAC,kCAAkC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;KACjE;IACD,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA8B,CAAC;AACzD,CAAC,2CAED,KAAK,iDACH,UAAkB,EAClB,OAIC;IAED,MAAM,OAAO,GAA6B;QACxC,UAAU;KACX,CAAC;IAEF,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,uBAAA,IAAI,qDAA0B,CAAC;IACnE,MAAM,YAAY,GAChB,OAAO,EAAE,YAAY,IAAI,uBAAA,IAAI,0DAA+B,CAAC;IAE/D,MAAM,OAAO,GAAG,MAAM,uBAAA,IAAI,0EAAe,MAAnB,IAAI,CAAiB,CAAC;IAC5C,OAAO,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC3C,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,UAAU,CAAC,GAAG,EAAE;YACd,cAAc,GAAG,IAAI,CAAC;YACtB,MAAM,CAAC,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC,CAAC;QAC3D,CAAC,EAAE,OAAO,CAAC,CAAC;QAEZ,MAAM,IAAI,GAAG,KAAK,IAAwC,EAAE;YAC1D,mEAAmE;YACnE,wDAAwD;YACxD,OAAO,CAAC,cAAc,EAAE;gBACtB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAC7B,MAAM,GAAG,GAAG,MAAM,uBAAA,IAAI,kCAAO,MAAX,IAAI,EAAQ,OAAO,CAAC,iBAAiB,EAAE;oBACvD,MAAM,EAAE,MAAM;oBACd,OAAO;oBACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;iBAC9B,CAAC,CAAC;gBACH,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;oBACtB,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA8B,CAAC;iBACxD;gBACD,MAAM,KAAK,CAAC,YAAY,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC;aACtD;YACD,wEAAwE;YACxE,mCAAmC;YACnC,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACtC,CAAC,CAAC;QAEF,IAAI,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;AACL,CAAC,uCAED,KAAK;IACH,MAAM,WAAW,GAAG,MAAM,uBAAA,IAAI,2CAAgB,MAApB,IAAI,CAAkB,CAAC;IACjD,OAAO;QACL,cAAc,EAAE,kBAAkB;QAClC,aAAa,EAAE,UAAU,WAAW,EAAE;KACvC,CAAC;AACJ,CAAC;AAGH;;;;;GAKG;AACH,KAAK,UAAU,KAAK,CAAC,EAAU;IAC7B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC;AAED;;;;;GAKG;AACH,SAAS,yBAAyB,CAChC,MAAuB;IAEvB,OAAO;QACL,QAAQ,EAAE;YACR;gBACE,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;gBAC1B,EAAE,EAAE,MAAM,CAAC,QAAQ,CAAC,EAAE;gBACtB,KAAK,EAAE,MAAM,CAAC,QAAQ,CAAC,KAAK;gBAC5B,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;gBAC1B,KAAK,EAAE,MAAM,CAAC,QAAQ,CAAC,KAAK;aAC7B;SACF;QACD,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM;KACtB,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,SAAS,kCAAkC,CACzC,gBAAkC;IAElC,sEAAsE;IACtE,2FAA2F;IAC3F,iHAAiH;IACjH,MAAM,MAAM,GAAG,2BAA2B,CAAC,gBAAgB,CAAC,CAAC;IAE7D,OAAO;QACL,OAAO,EAAE,gBAAgB,CAAC,OAAO;QACjC,IAAI,EAAE,gBAAgB,CAAC,aAAa,CAAC,IAAI;QACzC,IAAI,EAAE,gBAAgB,CAAC,aAAa,CAAC,IAAI;QACzC,MAAM;QACN,MAAM,EAAE,gBAAgB,CAAC,aAAa,CAAC,MAAM;KAC9C,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,SAAgB,2BAA2B,CACzC,gBAAkC;IAElC,IAAI,gBAAgB,CAAC,IAAI,KAAK,2CAAoB,CAAC,SAAS,EAAE;QAC5D,QAAQ,gBAAgB,CAAC,OAAO,EAAE;YAChC,KAAK,gCAAoB,CAAC,EAAE;gBAC1B,OAAO,gCAAS,CAAC,eAAe,CAAC;YACnC,KAAK,gCAAoB,CAAC,EAAE;gBAC1B,OAAO,gCAAS,CAAC,eAAe,CAAC;YACnC,KAAK,gCAAoB,CAAC,EAAE,CAAC;YAC7B;gBACE,OAAO,2CAAoB,CAAC,SAAS,CAAC;SACzC;KACF;IAED,OAAO,gBAAgB,CAAC,IAAI,CAAC;AAC/B,CAAC;AAhBD,kEAgBC","sourcesContent":["import {\n  EthMethod,\n  SignatureRequestType,\n  type SignatureRequest,\n} from '@metamask/signature-controller';\nimport type { TransactionMeta } from '@metamask/transaction-controller';\nimport type { Json } from '@metamask/utils';\n\nimport { SignTypedDataVersion } from './constants';\nimport type {\n  CheckCoverageRequest,\n  CheckSignatureCoverageRequest,\n  CoverageResult,\n  CoverageStatus,\n  LogSignatureRequest,\n  LogTransactionRequest,\n  ShieldBackend,\n} from './types';\n\nexport type InitCoverageCheckRequest = {\n  txParams: [\n    {\n      from: string;\n      to?: string;\n      value?: string;\n      data?: string;\n      nonce?: string;\n    },\n  ];\n  chainId: string;\n  origin?: string;\n};\n\nexport type InitSignatureCoverageCheckRequest = {\n  chainId: string;\n  data: Json;\n  from: string;\n  method: string;\n  origin?: string;\n};\n\nexport type InitCoverageCheckResponse = {\n  coverageId: string;\n};\n\nexport type GetCoverageResultRequest = {\n  coverageId: string;\n};\n\nexport type GetCoverageResultResponse = {\n  message?: string;\n  reasonCode?: string;\n  status: CoverageStatus;\n};\n\nexport class ShieldRemoteBackend implements ShieldBackend {\n  readonly #getAccessToken: () => Promise<string>;\n\n  readonly #getCoverageResultTimeout: number;\n\n  readonly #getCoverageResultPollInterval: number;\n\n  readonly #baseUrl: string;\n\n  readonly #fetch: typeof globalThis.fetch;\n\n  constructor({\n    getAccessToken,\n    getCoverageResultTimeout = 5000, // milliseconds\n    getCoverageResultPollInterval = 1000, // milliseconds\n    baseUrl,\n    fetch: fetchFn,\n  }: {\n    getAccessToken: () => Promise<string>;\n    getCoverageResultTimeout?: number;\n    getCoverageResultPollInterval?: number;\n    baseUrl: string;\n    fetch: typeof globalThis.fetch;\n  }) {\n    this.#getAccessToken = getAccessToken;\n    this.#getCoverageResultTimeout = getCoverageResultTimeout;\n    this.#getCoverageResultPollInterval = getCoverageResultPollInterval;\n    this.#baseUrl = baseUrl;\n    this.#fetch = fetchFn;\n  }\n\n  async checkCoverage(req: CheckCoverageRequest): Promise<CoverageResult> {\n    let { coverageId } = req;\n    if (!coverageId) {\n      const reqBody = makeInitCoverageCheckBody(req.txMeta);\n      ({ coverageId } = await this.#initCoverageCheck(\n        'v1/transaction/coverage/init',\n        reqBody,\n      ));\n    }\n\n    const txCoverageResultUrl = `${this.#baseUrl}/v1/transaction/coverage/result`;\n    const coverageResult = await this.#getCoverageResult(coverageId, {\n      coverageResultUrl: txCoverageResultUrl,\n    });\n    return {\n      coverageId,\n      message: coverageResult.message,\n      reasonCode: coverageResult.reasonCode,\n      status: coverageResult.status,\n    };\n  }\n\n  async checkSignatureCoverage(\n    req: CheckSignatureCoverageRequest,\n  ): Promise<CoverageResult> {\n    let { coverageId } = req;\n    if (!coverageId) {\n      const reqBody = makeInitSignatureCoverageCheckBody(req.signatureRequest);\n      ({ coverageId } = await this.#initCoverageCheck(\n        'v1/signature/coverage/init',\n        reqBody,\n      ));\n    }\n\n    const signatureCoverageResultUrl = `${this.#baseUrl}/v1/signature/coverage/result`;\n    const coverageResult = await this.#getCoverageResult(coverageId, {\n      coverageResultUrl: signatureCoverageResultUrl,\n    });\n    return {\n      coverageId,\n      message: coverageResult.message,\n      reasonCode: coverageResult.reasonCode,\n      status: coverageResult.status,\n    };\n  }\n\n  async logSignature(req: LogSignatureRequest): Promise<void> {\n    const initBody = makeInitSignatureCoverageCheckBody(req.signatureRequest);\n    const body = {\n      signature: req.signature,\n      status: req.status,\n      ...initBody,\n    };\n\n    const res = await this.#fetch(\n      `${this.#baseUrl}/v1/signature/coverage/log`,\n      {\n        method: 'POST',\n        headers: await this.#createHeaders(),\n        body: JSON.stringify(body),\n      },\n    );\n    if (res.status !== 200) {\n      throw new Error(`Failed to log signature: ${res.status}`);\n    }\n  }\n\n  async logTransaction(req: LogTransactionRequest): Promise<void> {\n    const initBody = makeInitCoverageCheckBody(req.txMeta);\n    const body = {\n      transactionHash: req.transactionHash,\n      status: req.status,\n      ...initBody,\n    };\n\n    const res = await this.#fetch(\n      `${this.#baseUrl}/v1/transaction/coverage/log`,\n      {\n        method: 'POST',\n        headers: await this.#createHeaders(),\n        body: JSON.stringify(body),\n      },\n    );\n    if (res.status !== 200) {\n      throw new Error(`Failed to log transaction: ${res.status}`);\n    }\n  }\n\n  async #initCoverageCheck(\n    path: string,\n    reqBody: unknown,\n  ): Promise<InitCoverageCheckResponse> {\n    const res = await this.#fetch(`${this.#baseUrl}/${path}`, {\n      method: 'POST',\n      headers: await this.#createHeaders(),\n      body: JSON.stringify(reqBody),\n    });\n    if (res.status !== 200) {\n      throw new Error(`Failed to init coverage check: ${res.status}`);\n    }\n    return (await res.json()) as InitCoverageCheckResponse;\n  }\n\n  async #getCoverageResult(\n    coverageId: string,\n    configs: {\n      coverageResultUrl: string;\n      timeout?: number;\n      pollInterval?: number;\n    },\n  ): Promise<GetCoverageResultResponse> {\n    const reqBody: GetCoverageResultRequest = {\n      coverageId,\n    };\n\n    const timeout = configs?.timeout ?? this.#getCoverageResultTimeout;\n    const pollInterval =\n      configs?.pollInterval ?? this.#getCoverageResultPollInterval;\n\n    const headers = await this.#createHeaders();\n    return await new Promise((resolve, reject) => {\n      let timeoutReached = false;\n      setTimeout(() => {\n        timeoutReached = true;\n        reject(new Error('Timeout waiting for coverage result'));\n      }, timeout);\n\n      const poll = async (): Promise<GetCoverageResultResponse> => {\n        // The timeoutReached variable is modified in the timeout callback.\n        // eslint-disable-next-line no-unmodified-loop-condition\n        while (!timeoutReached) {\n          const startTime = Date.now();\n          const res = await this.#fetch(configs.coverageResultUrl, {\n            method: 'POST',\n            headers,\n            body: JSON.stringify(reqBody),\n          });\n          if (res.status === 200) {\n            return (await res.json()) as GetCoverageResultResponse;\n          }\n          await sleep(pollInterval - (Date.now() - startTime));\n        }\n        // The following line will not have an effect as the upper level promise\n        // will already be rejected by now.\n        throw new Error('unexpected error');\n      };\n\n      poll().then(resolve).catch(reject);\n    });\n  }\n\n  async #createHeaders() {\n    const accessToken = await this.#getAccessToken();\n    return {\n      'Content-Type': 'application/json',\n      Authorization: `Bearer ${accessToken}`,\n    };\n  }\n}\n\n/**\n * Sleep for a specified amount of time.\n *\n * @param ms - The number of milliseconds to sleep.\n * @returns A promise that resolves after the specified amount of time.\n */\nasync function sleep(ms: number) {\n  return new Promise((resolve) => setTimeout(resolve, ms));\n}\n\n/**\n * Make the body for the init coverage check request.\n *\n * @param txMeta - The transaction metadata.\n * @returns The body for the init coverage check request.\n */\nfunction makeInitCoverageCheckBody(\n  txMeta: TransactionMeta,\n): InitCoverageCheckRequest {\n  return {\n    txParams: [\n      {\n        from: txMeta.txParams.from,\n        to: txMeta.txParams.to,\n        value: txMeta.txParams.value,\n        data: txMeta.txParams.data,\n        nonce: txMeta.txParams.nonce,\n      },\n    ],\n    chainId: txMeta.chainId,\n    origin: txMeta.origin,\n  };\n}\n\n/**\n * Make the body for the init signature coverage check request.\n *\n * @param signatureRequest - The signature request.\n * @returns The body for the init signature coverage check request.\n */\nfunction makeInitSignatureCoverageCheckBody(\n  signatureRequest: SignatureRequest,\n): InitSignatureCoverageCheckRequest {\n  // TODO: confirm that do we still need to validate the signature data?\n  // signature controller already validates the signature data before adding it to the state.\n  // @link https://github.com/MetaMask/core/blob/main/packages/signature-controller/src/SignatureController.ts#L408\n  const method = parseSignatureRequestMethod(signatureRequest);\n\n  return {\n    chainId: signatureRequest.chainId,\n    data: signatureRequest.messageParams.data,\n    from: signatureRequest.messageParams.from,\n    method,\n    origin: signatureRequest.messageParams.origin,\n  };\n}\n\n/**\n * Parse the JSON-RPC method from the signature request.\n *\n * @param signatureRequest - The signature request.\n * @returns The JSON-RPC method.\n */\nexport function parseSignatureRequestMethod(\n  signatureRequest: SignatureRequest,\n): string {\n  if (signatureRequest.type === SignatureRequestType.TypedSign) {\n    switch (signatureRequest.version) {\n      case SignTypedDataVersion.V3:\n        return EthMethod.SignTypedDataV3;\n      case SignTypedDataVersion.V4:\n        return EthMethod.SignTypedDataV4;\n      case SignTypedDataVersion.V1:\n      default:\n        return SignatureRequestType.TypedSign;\n    }\n  }\n\n  return signatureRequest.type;\n}\n"]}
+{"version":3,"file":"backend.cjs","sourceRoot":"","sources":["../src/backend.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,iEAIoC;AACpC,yEAIwC;AAIxC,+CAAmD;AACnD,mEAAmE;AA+CnE,MAAa,mBAAmB;IAS9B,YAAY,EACV,cAAc,EACd,wBAAwB,GAAG,IAAI,EAAE,eAAe;IAChD,6BAA6B,GAAG,IAAI,EAAE,eAAe;IACrD,OAAO,EACP,KAAK,EAAE,OAAO,GAOf;;QApBQ,sDAAuC;QAEvC,+CAAiB;QAEjB,6CAAgC;QAEhC,qDAA2C;QAelD,uBAAA,IAAI,uCAAmB,cAAc,MAAA,CAAC;QACtC,uBAAA,IAAI,gCAAY,OAAO,MAAA,CAAC;QACxB,uBAAA,IAAI,8BAAU,OAAO,MAAA,CAAC;QAEtB,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,mCAAmC,CACjE,wBAAwB,EACxB,6BAA6B,CAC9B,CAAC;QAEF,uBAAA,IAAI,sCAAkB,IAAI,gDAA0B,CAAC;YACnD,OAAO;YACP,UAAU;SACX,CAAC,MAAA,CAAC;IACL,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,GAAyB;QAC3C,IAAI,EAAE,UAAU,EAAE,GAAG,GAAG,CAAC;QACzB,IAAI,CAAC,UAAU,EAAE;YACf,MAAM,OAAO,GAAG,yBAAyB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACtD,CAAC,EAAE,UAAU,EAAE,GAAG,MAAM,uBAAA,IAAI,8EAAmB,MAAvB,IAAI,EAC1B,8BAA8B,EAC9B,OAAO,CACR,CAAC,CAAC;SACJ;QAED,MAAM,mBAAmB,GAAG,GAAG,uBAAA,IAAI,oCAAS,iCAAiC,CAAC;QAC9E,MAAM,cAAc,GAAG,MAAM,uBAAA,IAAI,8EAAmB,MAAvB,IAAI,EAC/B,GAAG,CAAC,MAAM,CAAC,EAAE,EACb,UAAU,EACV,mBAAmB,CACpB,CAAC;QACF,OAAO;YACL,UAAU;YACV,OAAO,EAAE,cAAc,CAAC,OAAO;YAC/B,UAAU,EAAE,cAAc,CAAC,UAAU;YACrC,MAAM,EAAE,cAAc,CAAC,MAAM;SAC9B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,sBAAsB,CAC1B,GAAkC;QAElC,IAAI,EAAE,UAAU,EAAE,GAAG,GAAG,CAAC;QACzB,IAAI,CAAC,UAAU,EAAE;YACf,MAAM,OAAO,GAAG,kCAAkC,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;YACzE,CAAC,EAAE,UAAU,EAAE,GAAG,MAAM,uBAAA,IAAI,8EAAmB,MAAvB,IAAI,EAC1B,4BAA4B,EAC5B,OAAO,CACR,CAAC,CAAC;SACJ;QAED,MAAM,0BAA0B,GAAG,GAAG,uBAAA,IAAI,oCAAS,+BAA+B,CAAC;QACnF,MAAM,cAAc,GAAG,MAAM,uBAAA,IAAI,8EAAmB,MAAvB,IAAI,EAC/B,GAAG,CAAC,gBAAgB,CAAC,EAAE,EACvB,UAAU,EACV,0BAA0B,CAC3B,CAAC;QACF,OAAO;YACL,UAAU;YACV,OAAO,EAAE,cAAc,CAAC,OAAO;YAC/B,UAAU,EAAE,cAAc,CAAC,UAAU;YACrC,MAAM,EAAE,cAAc,CAAC,MAAM;SAC9B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,GAAwB;QACzC,MAAM,QAAQ,GAAG,kCAAkC,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QAC1E,MAAM,IAAI,GAAG;YACX,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,GAAG,QAAQ;SACZ,CAAC;QAEF,iDAAiD;QACjD,uBAAA,IAAI,0CAAe,CAAC,mBAAmB,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;QAEjE,MAAM,GAAG,GAAG,MAAM,uBAAA,IAAI,kCAAO,MAAX,IAAI,EACpB,GAAG,uBAAA,IAAI,oCAAS,4BAA4B,EAC5C;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,MAAM,uBAAA,IAAI,0EAAe,MAAnB,IAAI,CAAiB;YACpC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CACF,CAAC;QACF,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;YACtB,MAAM,IAAI,KAAK,CAAC,4BAA4B,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;SAC3D;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,GAA0B;QAC7C,MAAM,QAAQ,GAAG,yBAAyB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvD,MAAM,IAAI,GAAG;YACX,eAAe,EAAE,GAAG,CAAC,eAAe;YACpC,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,GAAG,QAAQ;SACZ,CAAC;QAEF,iDAAiD;QACjD,uBAAA,IAAI,0CAAe,CAAC,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAEvD,MAAM,GAAG,GAAG,MAAM,uBAAA,IAAI,kCAAO,MAAX,IAAI,EACpB,GAAG,uBAAA,IAAI,oCAAS,8BAA8B,EAC9C;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,MAAM,uBAAA,IAAI,0EAAe,MAAnB,IAAI,CAAiB;YACpC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CACF,CAAC;QACF,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;YACtB,MAAM,IAAI,KAAK,CAAC,8BAA8B,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;SAC7D;IACH,CAAC;CA4DF;AAjMD,kDAiMC;4RA1DC,KAAK,iDACH,IAAY,EACZ,OAAgB;IAEhB,MAAM,GAAG,GAAG,MAAM,uBAAA,IAAI,kCAAO,MAAX,IAAI,EAAQ,GAAG,uBAAA,IAAI,oCAAS,IAAI,IAAI,EAAE,EAAE;QACxD,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,MAAM,uBAAA,IAAI,0EAAe,MAAnB,IAAI,CAAiB;QACpC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;KAC9B,CAAC,CAAC;IACH,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;QACtB,MAAM,IAAI,KAAK,CAAC,kCAAkC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;KACjE;IACD,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA8B,CAAC;AACzD,CAAC,2CAED,KAAK,iDACH,SAAiB,EACjB,UAAkB,EAClB,iBAAyB;IAEzB,MAAM,OAAO,GAA6B;QACxC,UAAU;KACX,CAAC;IAEF,MAAM,OAAO,GAAG,MAAM,uBAAA,IAAI,0EAAe,MAAnB,IAAI,CAAiB,CAAC;IAE5C,MAAM,mBAAmB,GAAG,KAAK,EAAE,MAAmB,EAAE,EAAE;QACxD,MAAM,GAAG,GAAG,MAAM,uBAAA,IAAI,kCAAO,MAAX,IAAI,EAAQ,iBAAiB,EAAE;YAC/C,MAAM,EAAE,MAAM;YACd,OAAO;YACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;YAC7B,MAAM;SACP,CAAC,CAAC;QACH,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;YACtB,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA8B,CAAC;SACxD;QAED,iDAAiD;QACjD,IAAI,YAAY,GAAG,qCAAqC,CAAC;QACzD,IAAI;YACF,MAAM,SAAS,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;YACnC,YAAY,GAAG,kCAAkC,SAAS,CAAC,OAAO,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;SAC1F;QAAC,MAAM;YACN,YAAY,GAAG,kCAAkC,GAAG,CAAC,MAAM,EAAE,CAAC;SAC/D;QACD,MAAM,IAAI,4BAAS,CAAC,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;IAChD,CAAC,CAAC;IAEF,OAAO,uBAAA,IAAI,0CAAe,CAAC,KAAK,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC;AACnE,CAAC,uCAED,KAAK;IACH,MAAM,WAAW,GAAG,MAAM,uBAAA,IAAI,2CAAgB,MAApB,IAAI,CAAkB,CAAC;IACjD,OAAO;QACL,cAAc,EAAE,kBAAkB;QAClC,aAAa,EAAE,UAAU,WAAW,EAAE;KACvC,CAAC;AACJ,CAAC;AAGH;;;;;GAKG;AACH,SAAS,yBAAyB,CAChC,MAAuB;IAEvB,OAAO;QACL,QAAQ,EAAE;YACR;gBACE,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;gBAC1B,EAAE,EAAE,MAAM,CAAC,QAAQ,CAAC,EAAE;gBACtB,KAAK,EAAE,MAAM,CAAC,QAAQ,CAAC,KAAK;gBAC5B,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;gBAC1B,KAAK,EAAE,MAAM,CAAC,QAAQ,CAAC,KAAK;aAC7B;SACF;QACD,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM;KACtB,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,SAAS,kCAAkC,CACzC,gBAAkC;IAElC,sEAAsE;IACtE,2FAA2F;IAC3F,iHAAiH;IACjH,MAAM,MAAM,GAAG,2BAA2B,CAAC,gBAAgB,CAAC,CAAC;IAE7D,OAAO;QACL,OAAO,EAAE,gBAAgB,CAAC,OAAO;QACjC,IAAI,EAAE,gBAAgB,CAAC,aAAa,CAAC,IAAI;QACzC,IAAI,EAAE,gBAAgB,CAAC,aAAa,CAAC,IAAI;QACzC,MAAM;QACN,MAAM,EAAE,gBAAgB,CAAC,aAAa,CAAC,MAAM;KAC9C,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,SAAgB,2BAA2B,CACzC,gBAAkC;IAElC,IAAI,gBAAgB,CAAC,IAAI,KAAK,2CAAoB,CAAC,SAAS,EAAE;QAC5D,QAAQ,gBAAgB,CAAC,OAAO,EAAE;YAChC,KAAK,gCAAoB,CAAC,EAAE;gBAC1B,OAAO,gCAAS,CAAC,eAAe,CAAC;YACnC,KAAK,gCAAoB,CAAC,EAAE;gBAC1B,OAAO,gCAAS,CAAC,eAAe,CAAC;YACnC,KAAK,gCAAoB,CAAC,EAAE,CAAC;YAC7B;gBACE,OAAO,2CAAoB,CAAC,SAAS,CAAC;SACzC;KACF;IAED,OAAO,gBAAgB,CAAC,IAAI,CAAC;AAC/B,CAAC;AAhBD,kEAgBC;AAED;;;;;;GAMG;AACH,SAAS,mCAAmC,CAC1C,OAAe,EACf,YAAoB;IAEpB,MAAM,OAAO,GAAG,IAAI,kCAAe,CAAC,YAAY,CAAC,CAAC;IAClD,MAAM,kBAAkB,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;IAElE,MAAM,UAAU,GACd,KAAK,CAAC,kBAAkB,CAAC,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QACxD,CAAC,CAAC,sCAAmB;QACrB,CAAC,CAAC,kBAAkB,CAAC;IAEzB,OAAO;QACL,OAAO;QACP,UAAU;KACX,CAAC;AACJ,CAAC","sourcesContent":["import {\n  ConstantBackoff,\n  DEFAULT_MAX_RETRIES,\n  HttpError,\n} from '@metamask/controller-utils';\nimport {\n  EthMethod,\n  SignatureRequestType,\n  type SignatureRequest,\n} from '@metamask/signature-controller';\nimport type { TransactionMeta } from '@metamask/transaction-controller';\nimport type { Json } from '@metamask/utils';\n\nimport { SignTypedDataVersion } from './constants';\nimport { PollingWithCockatielPolicy } from './polling-with-policy';\nimport type {\n  CheckCoverageRequest,\n  CheckSignatureCoverageRequest,\n  CoverageResult,\n  CoverageStatus,\n  LogSignatureRequest,\n  LogTransactionRequest,\n  ShieldBackend,\n} from './types';\n\nexport type InitCoverageCheckRequest = {\n  txParams: [\n    {\n      from: string;\n      to?: string;\n      value?: string;\n      data?: string;\n      nonce?: string;\n    },\n  ];\n  chainId: string;\n  origin?: string;\n};\n\nexport type InitSignatureCoverageCheckRequest = {\n  chainId: string;\n  data: Json;\n  from: string;\n  method: string;\n  origin?: string;\n};\n\nexport type InitCoverageCheckResponse = {\n  coverageId: string;\n};\n\nexport type GetCoverageResultRequest = {\n  coverageId: string;\n};\n\nexport type GetCoverageResultResponse = {\n  message?: string;\n  reasonCode?: string;\n  status: CoverageStatus;\n};\n\nexport class ShieldRemoteBackend implements ShieldBackend {\n  readonly #getAccessToken: () => Promise<string>;\n\n  readonly #baseUrl: string;\n\n  readonly #fetch: typeof globalThis.fetch;\n\n  readonly #pollingPolicy: PollingWithCockatielPolicy;\n\n  constructor({\n    getAccessToken,\n    getCoverageResultTimeout = 5000, // milliseconds\n    getCoverageResultPollInterval = 1000, // milliseconds\n    baseUrl,\n    fetch: fetchFn,\n  }: {\n    getAccessToken: () => Promise<string>;\n    getCoverageResultTimeout?: number;\n    getCoverageResultPollInterval?: number;\n    baseUrl: string;\n    fetch: typeof globalThis.fetch;\n  }) {\n    this.#getAccessToken = getAccessToken;\n    this.#baseUrl = baseUrl;\n    this.#fetch = fetchFn;\n\n    const { backoff, maxRetries } = computePollingIntervalAndRetryCount(\n      getCoverageResultTimeout,\n      getCoverageResultPollInterval,\n    );\n\n    this.#pollingPolicy = new PollingWithCockatielPolicy({\n      backoff,\n      maxRetries,\n    });\n  }\n\n  async checkCoverage(req: CheckCoverageRequest): Promise<CoverageResult> {\n    let { coverageId } = req;\n    if (!coverageId) {\n      const reqBody = makeInitCoverageCheckBody(req.txMeta);\n      ({ coverageId } = await this.#initCoverageCheck(\n        'v1/transaction/coverage/init',\n        reqBody,\n      ));\n    }\n\n    const txCoverageResultUrl = `${this.#baseUrl}/v1/transaction/coverage/result`;\n    const coverageResult = await this.#getCoverageResult(\n      req.txMeta.id,\n      coverageId,\n      txCoverageResultUrl,\n    );\n    return {\n      coverageId,\n      message: coverageResult.message,\n      reasonCode: coverageResult.reasonCode,\n      status: coverageResult.status,\n    };\n  }\n\n  async checkSignatureCoverage(\n    req: CheckSignatureCoverageRequest,\n  ): Promise<CoverageResult> {\n    let { coverageId } = req;\n    if (!coverageId) {\n      const reqBody = makeInitSignatureCoverageCheckBody(req.signatureRequest);\n      ({ coverageId } = await this.#initCoverageCheck(\n        'v1/signature/coverage/init',\n        reqBody,\n      ));\n    }\n\n    const signatureCoverageResultUrl = `${this.#baseUrl}/v1/signature/coverage/result`;\n    const coverageResult = await this.#getCoverageResult(\n      req.signatureRequest.id,\n      coverageId,\n      signatureCoverageResultUrl,\n    );\n    return {\n      coverageId,\n      message: coverageResult.message,\n      reasonCode: coverageResult.reasonCode,\n      status: coverageResult.status,\n    };\n  }\n\n  async logSignature(req: LogSignatureRequest): Promise<void> {\n    const initBody = makeInitSignatureCoverageCheckBody(req.signatureRequest);\n    const body = {\n      signature: req.signature,\n      status: req.status,\n      ...initBody,\n    };\n\n    // cancel the pending get coverage result request\n    this.#pollingPolicy.abortPendingRequest(req.signatureRequest.id);\n\n    const res = await this.#fetch(\n      `${this.#baseUrl}/v1/signature/coverage/log`,\n      {\n        method: 'POST',\n        headers: await this.#createHeaders(),\n        body: JSON.stringify(body),\n      },\n    );\n    if (res.status !== 200) {\n      throw new Error(`Failed to log signature: ${res.status}`);\n    }\n  }\n\n  async logTransaction(req: LogTransactionRequest): Promise<void> {\n    const initBody = makeInitCoverageCheckBody(req.txMeta);\n    const body = {\n      transactionHash: req.transactionHash,\n      status: req.status,\n      ...initBody,\n    };\n\n    // cancel the pending get coverage result request\n    this.#pollingPolicy.abortPendingRequest(req.txMeta.id);\n\n    const res = await this.#fetch(\n      `${this.#baseUrl}/v1/transaction/coverage/log`,\n      {\n        method: 'POST',\n        headers: await this.#createHeaders(),\n        body: JSON.stringify(body),\n      },\n    );\n    if (res.status !== 200) {\n      throw new Error(`Failed to log transaction: ${res.status}`);\n    }\n  }\n\n  async #initCoverageCheck(\n    path: string,\n    reqBody: unknown,\n  ): Promise<InitCoverageCheckResponse> {\n    const res = await this.#fetch(`${this.#baseUrl}/${path}`, {\n      method: 'POST',\n      headers: await this.#createHeaders(),\n      body: JSON.stringify(reqBody),\n    });\n    if (res.status !== 200) {\n      throw new Error(`Failed to init coverage check: ${res.status}`);\n    }\n    return (await res.json()) as InitCoverageCheckResponse;\n  }\n\n  async #getCoverageResult(\n    requestId: string,\n    coverageId: string,\n    coverageResultUrl: string,\n  ): Promise<GetCoverageResultResponse> {\n    const reqBody: GetCoverageResultRequest = {\n      coverageId,\n    };\n\n    const headers = await this.#createHeaders();\n\n    const getCoverageResultFn = async (signal: AbortSignal) => {\n      const res = await this.#fetch(coverageResultUrl, {\n        method: 'POST',\n        headers,\n        body: JSON.stringify(reqBody),\n        signal,\n      });\n      if (res.status === 200) {\n        return (await res.json()) as GetCoverageResultResponse;\n      }\n\n      // parse the error message from the response body\n      let errorMessage = 'Timeout waiting for coverage result';\n      try {\n        const errorJson = await res.json();\n        errorMessage = `Failed to get coverage result: ${errorJson.message || errorJson.status}`;\n      } catch {\n        errorMessage = `Failed to get coverage result: ${res.status}`;\n      }\n      throw new HttpError(res.status, errorMessage);\n    };\n\n    return this.#pollingPolicy.start(requestId, getCoverageResultFn);\n  }\n\n  async #createHeaders() {\n    const accessToken = await this.#getAccessToken();\n    return {\n      'Content-Type': 'application/json',\n      Authorization: `Bearer ${accessToken}`,\n    };\n  }\n}\n\n/**\n * Make the body for the init coverage check request.\n *\n * @param txMeta - The transaction metadata.\n * @returns The body for the init coverage check request.\n */\nfunction makeInitCoverageCheckBody(\n  txMeta: TransactionMeta,\n): InitCoverageCheckRequest {\n  return {\n    txParams: [\n      {\n        from: txMeta.txParams.from,\n        to: txMeta.txParams.to,\n        value: txMeta.txParams.value,\n        data: txMeta.txParams.data,\n        nonce: txMeta.txParams.nonce,\n      },\n    ],\n    chainId: txMeta.chainId,\n    origin: txMeta.origin,\n  };\n}\n\n/**\n * Make the body for the init signature coverage check request.\n *\n * @param signatureRequest - The signature request.\n * @returns The body for the init signature coverage check request.\n */\nfunction makeInitSignatureCoverageCheckBody(\n  signatureRequest: SignatureRequest,\n): InitSignatureCoverageCheckRequest {\n  // TODO: confirm that do we still need to validate the signature data?\n  // signature controller already validates the signature data before adding it to the state.\n  // @link https://github.com/MetaMask/core/blob/main/packages/signature-controller/src/SignatureController.ts#L408\n  const method = parseSignatureRequestMethod(signatureRequest);\n\n  return {\n    chainId: signatureRequest.chainId,\n    data: signatureRequest.messageParams.data,\n    from: signatureRequest.messageParams.from,\n    method,\n    origin: signatureRequest.messageParams.origin,\n  };\n}\n\n/**\n * Parse the JSON-RPC method from the signature request.\n *\n * @param signatureRequest - The signature request.\n * @returns The JSON-RPC method.\n */\nexport function parseSignatureRequestMethod(\n  signatureRequest: SignatureRequest,\n): string {\n  if (signatureRequest.type === SignatureRequestType.TypedSign) {\n    switch (signatureRequest.version) {\n      case SignTypedDataVersion.V3:\n        return EthMethod.SignTypedDataV3;\n      case SignTypedDataVersion.V4:\n        return EthMethod.SignTypedDataV4;\n      case SignTypedDataVersion.V1:\n      default:\n        return SignatureRequestType.TypedSign;\n    }\n  }\n\n  return signatureRequest.type;\n}\n\n/**\n * Compute the polling interval and retry count for the Cockatiel policy based on the timeout and poll interval given.\n *\n * @param timeout - The timeout in milliseconds.\n * @param pollInterval - The poll interval in milliseconds.\n * @returns The polling interval and retry count.\n */\nfunction computePollingIntervalAndRetryCount(\n  timeout: number,\n  pollInterval: number,\n) {\n  const backoff = new ConstantBackoff(pollInterval);\n  const computedMaxRetries = Math.floor(timeout / pollInterval) + 1;\n\n  const maxRetries =\n    isNaN(computedMaxRetries) || !isFinite(computedMaxRetries)\n      ? DEFAULT_MAX_RETRIES\n      : computedMaxRetries;\n\n  return {\n    backoff,\n    maxRetries,\n  };\n}\n"]}

package/dist/backend.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"backend.d.cts","sourceRoot":"","sources":["../src/backend.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,KAAK,gBAAgB,EACtB,uCAAuC;AAExC,OAAO,KAAK,EAAE,IAAI,EAAE,wBAAwB;AAG5C,OAAO,KAAK,EACV,oBAAoB,EACpB,6BAA6B,EAC7B,cAAc,EACd,cAAc,EACd,mBAAmB,EACnB,qBAAqB,EACrB,aAAa,EACd,oBAAgB;AAEjB,MAAM,MAAM,wBAAwB,GAAG;IACrC,QAAQ,EAAE;QACR;YACE,IAAI,EAAE,MAAM,CAAC;YACb,EAAE,CAAC,EAAE,MAAM,CAAC;YACZ,KAAK,CAAC,EAAE,MAAM,CAAC;YACf,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,KAAK,CAAC,EAAE,MAAM,CAAC;SAChB;KACF,CAAC;IACF,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,iCAAiC,GAAG;IAC9C,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,IAAI,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,UAAU,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IACrC,UAAU,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,cAAc,CAAC;CACxB,CAAC;AAEF,qBAAa,mBAAoB,YAAW,aAAa;;gBAW3C,EACV,cAAc,EACd,wBAA+B,EAAE,eAAe;IAChD,6BAAoC,EAAE,eAAe;IACrD,OAAO,EACP,KAAK,EAAE,OAAO,GACf,EAAE;QACD,cAAc,EAAE,MAAM,OAAO,CAAC,MAAM,CAAC,CAAC;QACtC,wBAAwB,CAAC,EAAE,MAAM,CAAC;QAClC,6BAA6B,CAAC,EAAE,MAAM,CAAC;QACvC,OAAO,EAAE,MAAM,CAAC;QAChB,KAAK,EAAE,OAAO,UAAU,CAAC,KAAK,CAAC;KAChC;IAQK,aAAa,CAAC,GAAG,EAAE,oBAAoB,GAAG,OAAO,CAAC,cAAc,CAAC;IAsBjE,sBAAsB,CAC1B,GAAG,EAAE,6BAA6B,GACjC,OAAO,CAAC,cAAc,CAAC;IAsBpB,YAAY,CAAC,GAAG,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBrD,cAAc,CAAC,GAAG,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;CA2FhE;AA2DD;;;;;GAKG;AACH,wBAAgB,2BAA2B,CACzC,gBAAgB,EAAE,gBAAgB,GACjC,MAAM,CAcR"}
+{"version":3,"file":"backend.d.cts","sourceRoot":"","sources":["../src/backend.ts"],"names":[],"mappings":"AAKA,OAAO,EAGL,KAAK,gBAAgB,EACtB,uCAAuC;AAExC,OAAO,KAAK,EAAE,IAAI,EAAE,wBAAwB;AAI5C,OAAO,KAAK,EACV,oBAAoB,EACpB,6BAA6B,EAC7B,cAAc,EACd,cAAc,EACd,mBAAmB,EACnB,qBAAqB,EACrB,aAAa,EACd,oBAAgB;AAEjB,MAAM,MAAM,wBAAwB,GAAG;IACrC,QAAQ,EAAE;QACR;YACE,IAAI,EAAE,MAAM,CAAC;YACb,EAAE,CAAC,EAAE,MAAM,CAAC;YACZ,KAAK,CAAC,EAAE,MAAM,CAAC;YACf,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,KAAK,CAAC,EAAE,MAAM,CAAC;SAChB;KACF,CAAC;IACF,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,iCAAiC,GAAG;IAC9C,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,IAAI,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,UAAU,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IACrC,UAAU,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,cAAc,CAAC;CACxB,CAAC;AAEF,qBAAa,mBAAoB,YAAW,aAAa;;gBAS3C,EACV,cAAc,EACd,wBAA+B,EAAE,eAAe;IAChD,6BAAoC,EAAE,eAAe;IACrD,OAAO,EACP,KAAK,EAAE,OAAO,GACf,EAAE;QACD,cAAc,EAAE,MAAM,OAAO,CAAC,MAAM,CAAC,CAAC;QACtC,wBAAwB,CAAC,EAAE,MAAM,CAAC;QAClC,6BAA6B,CAAC,EAAE,MAAM,CAAC;QACvC,OAAO,EAAE,MAAM,CAAC;QAChB,KAAK,EAAE,OAAO,UAAU,CAAC,KAAK,CAAC;KAChC;IAgBK,aAAa,CAAC,GAAG,EAAE,oBAAoB,GAAG,OAAO,CAAC,cAAc,CAAC;IAwBjE,sBAAsB,CAC1B,GAAG,EAAE,6BAA6B,GACjC,OAAO,CAAC,cAAc,CAAC;IAwBpB,YAAY,CAAC,GAAG,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAwBrD,cAAc,CAAC,GAAG,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;CAkFhE;AAiDD;;;;;GAKG;AACH,wBAAgB,2BAA2B,CACzC,gBAAgB,EAAE,gBAAgB,GACjC,MAAM,CAcR"}

package/dist/backend.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"backend.d.mts","sourceRoot":"","sources":["../src/backend.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,KAAK,gBAAgB,EACtB,uCAAuC;AAExC,OAAO,KAAK,EAAE,IAAI,EAAE,wBAAwB;AAG5C,OAAO,KAAK,EACV,oBAAoB,EACpB,6BAA6B,EAC7B,cAAc,EACd,cAAc,EACd,mBAAmB,EACnB,qBAAqB,EACrB,aAAa,EACd,oBAAgB;AAEjB,MAAM,MAAM,wBAAwB,GAAG;IACrC,QAAQ,EAAE;QACR;YACE,IAAI,EAAE,MAAM,CAAC;YACb,EAAE,CAAC,EAAE,MAAM,CAAC;YACZ,KAAK,CAAC,EAAE,MAAM,CAAC;YACf,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,KAAK,CAAC,EAAE,MAAM,CAAC;SAChB;KACF,CAAC;IACF,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,iCAAiC,GAAG;IAC9C,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,IAAI,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,UAAU,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IACrC,UAAU,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,cAAc,CAAC;CACxB,CAAC;AAEF,qBAAa,mBAAoB,YAAW,aAAa;;gBAW3C,EACV,cAAc,EACd,wBAA+B,EAAE,eAAe;IAChD,6BAAoC,EAAE,eAAe;IACrD,OAAO,EACP,KAAK,EAAE,OAAO,GACf,EAAE;QACD,cAAc,EAAE,MAAM,OAAO,CAAC,MAAM,CAAC,CAAC;QACtC,wBAAwB,CAAC,EAAE,MAAM,CAAC;QAClC,6BAA6B,CAAC,EAAE,MAAM,CAAC;QACvC,OAAO,EAAE,MAAM,CAAC;QAChB,KAAK,EAAE,OAAO,UAAU,CAAC,KAAK,CAAC;KAChC;IAQK,aAAa,CAAC,GAAG,EAAE,oBAAoB,GAAG,OAAO,CAAC,cAAc,CAAC;IAsBjE,sBAAsB,CAC1B,GAAG,EAAE,6BAA6B,GACjC,OAAO,CAAC,cAAc,CAAC;IAsBpB,YAAY,CAAC,GAAG,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBrD,cAAc,CAAC,GAAG,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;CA2FhE;AA2DD;;;;;GAKG;AACH,wBAAgB,2BAA2B,CACzC,gBAAgB,EAAE,gBAAgB,GACjC,MAAM,CAcR"}
+{"version":3,"file":"backend.d.mts","sourceRoot":"","sources":["../src/backend.ts"],"names":[],"mappings":"AAKA,OAAO,EAGL,KAAK,gBAAgB,EACtB,uCAAuC;AAExC,OAAO,KAAK,EAAE,IAAI,EAAE,wBAAwB;AAI5C,OAAO,KAAK,EACV,oBAAoB,EACpB,6BAA6B,EAC7B,cAAc,EACd,cAAc,EACd,mBAAmB,EACnB,qBAAqB,EACrB,aAAa,EACd,oBAAgB;AAEjB,MAAM,MAAM,wBAAwB,GAAG;IACrC,QAAQ,EAAE;QACR;YACE,IAAI,EAAE,MAAM,CAAC;YACb,EAAE,CAAC,EAAE,MAAM,CAAC;YACZ,KAAK,CAAC,EAAE,MAAM,CAAC;YACf,IAAI,CAAC,EAAE,MAAM,CAAC;YACd,KAAK,CAAC,EAAE,MAAM,CAAC;SAChB;KACF,CAAC;IACF,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,iCAAiC,GAAG;IAC9C,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,IAAI,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,UAAU,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,wBAAwB,GAAG;IACrC,UAAU,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG;IACtC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,cAAc,CAAC;CACxB,CAAC;AAEF,qBAAa,mBAAoB,YAAW,aAAa;;gBAS3C,EACV,cAAc,EACd,wBAA+B,EAAE,eAAe;IAChD,6BAAoC,EAAE,eAAe;IACrD,OAAO,EACP,KAAK,EAAE,OAAO,GACf,EAAE;QACD,cAAc,EAAE,MAAM,OAAO,CAAC,MAAM,CAAC,CAAC;QACtC,wBAAwB,CAAC,EAAE,MAAM,CAAC;QAClC,6BAA6B,CAAC,EAAE,MAAM,CAAC;QACvC,OAAO,EAAE,MAAM,CAAC;QAChB,KAAK,EAAE,OAAO,UAAU,CAAC,KAAK,CAAC;KAChC;IAgBK,aAAa,CAAC,GAAG,EAAE,oBAAoB,GAAG,OAAO,CAAC,cAAc,CAAC;IAwBjE,sBAAsB,CAC1B,GAAG,EAAE,6BAA6B,GACjC,OAAO,CAAC,cAAc,CAAC;IAwBpB,YAAY,CAAC,GAAG,EAAE,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAwBrD,cAAc,CAAC,GAAG,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;CAkFhE;AAiDD;;;;;GAKG;AACH,wBAAgB,2BAA2B,CACzC,gBAAgB,EAAE,gBAAgB,GACjC,MAAM,CAcR"}

package/dist/backend.mjs

@@ -9,24 +9,28 @@ var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (
     if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
     return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
 };
-var _ShieldRemoteBackend_instances, _ShieldRemoteBackend_getAccessToken, _ShieldRemoteBackend_getCoverageResultTimeout, _ShieldRemoteBackend_getCoverageResultPollInterval, _ShieldRemoteBackend_baseUrl, _ShieldRemoteBackend_fetch, _ShieldRemoteBackend_initCoverageCheck, _ShieldRemoteBackend_getCoverageResult, _ShieldRemoteBackend_createHeaders;
+var _ShieldRemoteBackend_instances, _ShieldRemoteBackend_getAccessToken, _ShieldRemoteBackend_baseUrl, _ShieldRemoteBackend_fetch, _ShieldRemoteBackend_pollingPolicy, _ShieldRemoteBackend_initCoverageCheck, _ShieldRemoteBackend_getCoverageResult, _ShieldRemoteBackend_createHeaders;
+import { ConstantBackoff, DEFAULT_MAX_RETRIES, HttpError } from "@metamask/controller-utils";
 import { EthMethod, SignatureRequestType } from "@metamask/signature-controller";
 import { SignTypedDataVersion } from "./constants.mjs";
+import { PollingWithCockatielPolicy } from "./polling-with-policy.mjs";
 export class ShieldRemoteBackend {
     constructor({ getAccessToken, getCoverageResultTimeout = 5000, // milliseconds
     getCoverageResultPollInterval = 1000, // milliseconds
     baseUrl, fetch: fetchFn, }) {
         _ShieldRemoteBackend_instances.add(this);
         _ShieldRemoteBackend_getAccessToken.set(this, void 0);
-        _ShieldRemoteBackend_getCoverageResultTimeout.set(this, void 0);
-        _ShieldRemoteBackend_getCoverageResultPollInterval.set(this, void 0);
         _ShieldRemoteBackend_baseUrl.set(this, void 0);
         _ShieldRemoteBackend_fetch.set(this, void 0);
+        _ShieldRemoteBackend_pollingPolicy.set(this, void 0);
         __classPrivateFieldSet(this, _ShieldRemoteBackend_getAccessToken, getAccessToken, "f");
-        __classPrivateFieldSet(this, _ShieldRemoteBackend_getCoverageResultTimeout, getCoverageResultTimeout, "f");
-        __classPrivateFieldSet(this, _ShieldRemoteBackend_getCoverageResultPollInterval, getCoverageResultPollInterval, "f");
         __classPrivateFieldSet(this, _ShieldRemoteBackend_baseUrl, baseUrl, "f");
         __classPrivateFieldSet(this, _ShieldRemoteBackend_fetch, fetchFn, "f");
+        const { backoff, maxRetries } = computePollingIntervalAndRetryCount(getCoverageResultTimeout, getCoverageResultPollInterval);
+        __classPrivateFieldSet(this, _ShieldRemoteBackend_pollingPolicy, new PollingWithCockatielPolicy({
+            backoff,
+            maxRetries,
+        }), "f");
     }
     async checkCoverage(req) {
         let { coverageId } = req;
@@ -35,9 +39,7 @@ export class ShieldRemoteBackend {
             ({ coverageId } = await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_initCoverageCheck).call(this, 'v1/transaction/coverage/init', reqBody));
         }
         const txCoverageResultUrl = `${__classPrivateFieldGet(this, _ShieldRemoteBackend_baseUrl, "f")}/v1/transaction/coverage/result`;
-        const coverageResult = await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_getCoverageResult).call(this, coverageId, {
-            coverageResultUrl: txCoverageResultUrl,
-        });
+        const coverageResult = await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_getCoverageResult).call(this, req.txMeta.id, coverageId, txCoverageResultUrl);
         return {
             coverageId,
             message: coverageResult.message,
@@ -52,9 +54,7 @@ export class ShieldRemoteBackend {
             ({ coverageId } = await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_initCoverageCheck).call(this, 'v1/signature/coverage/init', reqBody));
         }
         const signatureCoverageResultUrl = `${__classPrivateFieldGet(this, _ShieldRemoteBackend_baseUrl, "f")}/v1/signature/coverage/result`;
-        const coverageResult = await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_getCoverageResult).call(this, coverageId, {
-            coverageResultUrl: signatureCoverageResultUrl,
-        });
+        const coverageResult = await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_getCoverageResult).call(this, req.signatureRequest.id, coverageId, signatureCoverageResultUrl);
         return {
             coverageId,
             message: coverageResult.message,
@@ -69,6 +69,8 @@ export class ShieldRemoteBackend {
             status: req.status,
             ...initBody,
         };
+        // cancel the pending get coverage result request
+        __classPrivateFieldGet(this, _ShieldRemoteBackend_pollingPolicy, "f").abortPendingRequest(req.signatureRequest.id);
         const res = await __classPrivateFieldGet(this, _ShieldRemoteBackend_fetch, "f").call(this, `${__classPrivateFieldGet(this, _ShieldRemoteBackend_baseUrl, "f")}/v1/signature/coverage/log`, {
             method: 'POST',
             headers: await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_createHeaders).call(this),
@@ -85,6 +87,8 @@ export class ShieldRemoteBackend {
             status: req.status,
             ...initBody,
         };
+        // cancel the pending get coverage result request
+        __classPrivateFieldGet(this, _ShieldRemoteBackend_pollingPolicy, "f").abortPendingRequest(req.txMeta.id);
         const res = await __classPrivateFieldGet(this, _ShieldRemoteBackend_fetch, "f").call(this, `${__classPrivateFieldGet(this, _ShieldRemoteBackend_baseUrl, "f")}/v1/transaction/coverage/log`, {
             method: 'POST',
             headers: await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_createHeaders).call(this),
@@ -95,7 +99,7 @@ export class ShieldRemoteBackend {
         }
     }
 }
-_ShieldRemoteBackend_getAccessToken = new WeakMap(), _ShieldRemoteBackend_getCoverageResultTimeout = new WeakMap(), _ShieldRemoteBackend_getCoverageResultPollInterval = new WeakMap(), _ShieldRemoteBackend_baseUrl = new WeakMap(), _ShieldRemoteBackend_fetch = new WeakMap(), _ShieldRemoteBackend_instances = new WeakSet(), _ShieldRemoteBackend_initCoverageCheck = async function _ShieldRemoteBackend_initCoverageCheck(path, reqBody) {
+_ShieldRemoteBackend_getAccessToken = new WeakMap(), _ShieldRemoteBackend_baseUrl = new WeakMap(), _ShieldRemoteBackend_fetch = new WeakMap(), _ShieldRemoteBackend_pollingPolicy = new WeakMap(), _ShieldRemoteBackend_instances = new WeakSet(), _ShieldRemoteBackend_initCoverageCheck = async function _ShieldRemoteBackend_initCoverageCheck(path, reqBody) {
     const res = await __classPrivateFieldGet(this, _ShieldRemoteBackend_fetch, "f").call(this, `${__classPrivateFieldGet(this, _ShieldRemoteBackend_baseUrl, "f")}/${path}`, {
         method: 'POST',
         headers: await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_createHeaders).call(this),
@@ -105,40 +109,33 @@ _ShieldRemoteBackend_getAccessToken = new WeakMap(), _ShieldRemoteBackend_getCov
         throw new Error(`Failed to init coverage check: ${res.status}`);
     }
     return (await res.json());
-}, _ShieldRemoteBackend_getCoverageResult = async function _ShieldRemoteBackend_getCoverageResult(coverageId, configs) {
+}, _ShieldRemoteBackend_getCoverageResult = async function _ShieldRemoteBackend_getCoverageResult(requestId, coverageId, coverageResultUrl) {
     const reqBody = {
         coverageId,
     };
-    const timeout = configs?.timeout ?? __classPrivateFieldGet(this, _ShieldRemoteBackend_getCoverageResultTimeout, "f");
-    const pollInterval = configs?.pollInterval ?? __classPrivateFieldGet(this, _ShieldRemoteBackend_getCoverageResultPollInterval, "f");
     const headers = await __classPrivateFieldGet(this, _ShieldRemoteBackend_instances, "m", _ShieldRemoteBackend_createHeaders).call(this);
-    return await new Promise((resolve, reject) => {
-        let timeoutReached = false;
-        setTimeout(() => {
-            timeoutReached = true;
-            reject(new Error('Timeout waiting for coverage result'));
-        }, timeout);
-        const poll = async () => {
-            // The timeoutReached variable is modified in the timeout callback.
-            // eslint-disable-next-line no-unmodified-loop-condition
-            while (!timeoutReached) {
-                const startTime = Date.now();
-                const res = await __classPrivateFieldGet(this, _ShieldRemoteBackend_fetch, "f").call(this, configs.coverageResultUrl, {
-                    method: 'POST',
-                    headers,
-                    body: JSON.stringify(reqBody),
-                });
-                if (res.status === 200) {
-                    return (await res.json());
-                }
-                await sleep(pollInterval - (Date.now() - startTime));
-            }
-            // The following line will not have an effect as the upper level promise
-            // will already be rejected by now.
-            throw new Error('unexpected error');
-        };
-        poll().then(resolve).catch(reject);
-    });
+    const getCoverageResultFn = async (signal) => {
+        const res = await __classPrivateFieldGet(this, _ShieldRemoteBackend_fetch, "f").call(this, coverageResultUrl, {
+            method: 'POST',
+            headers,
+            body: JSON.stringify(reqBody),
+            signal,
+        });
+        if (res.status === 200) {
+            return (await res.json());
+        }
+        // parse the error message from the response body
+        let errorMessage = 'Timeout waiting for coverage result';
+        try {
+            const errorJson = await res.json();
+            errorMessage = `Failed to get coverage result: ${errorJson.message || errorJson.status}`;
+        }
+        catch {
+            errorMessage = `Failed to get coverage result: ${res.status}`;
+        }
+        throw new HttpError(res.status, errorMessage);
+    };
+    return __classPrivateFieldGet(this, _ShieldRemoteBackend_pollingPolicy, "f").start(requestId, getCoverageResultFn);
 }, _ShieldRemoteBackend_createHeaders = async function _ShieldRemoteBackend_createHeaders() {
     const accessToken = await __classPrivateFieldGet(this, _ShieldRemoteBackend_getAccessToken, "f").call(this);
     return {
@@ -146,15 +143,6 @@ _ShieldRemoteBackend_getAccessToken = new WeakMap(), _ShieldRemoteBackend_getCov
         Authorization: `Bearer ${accessToken}`,
     };
 };
-/**
- * Sleep for a specified amount of time.
- *
- * @param ms - The number of milliseconds to sleep.
- * @returns A promise that resolves after the specified amount of time.
- */
-async function sleep(ms) {
-    return new Promise((resolve) => setTimeout(resolve, ms));
-}
 /**
  * Make the body for the init coverage check request.
  *
@@ -215,4 +203,22 @@ export function parseSignatureRequestMethod(signatureRequest) {
     }
     return signatureRequest.type;
 }
+/**
+ * Compute the polling interval and retry count for the Cockatiel policy based on the timeout and poll interval given.
+ *
+ * @param timeout - The timeout in milliseconds.
+ * @param pollInterval - The poll interval in milliseconds.
+ * @returns The polling interval and retry count.
+ */
+function computePollingIntervalAndRetryCount(timeout, pollInterval) {
+    const backoff = new ConstantBackoff(pollInterval);
+    const computedMaxRetries = Math.floor(timeout / pollInterval) + 1;
+    const maxRetries = isNaN(computedMaxRetries) || !isFinite(computedMaxRetries)
+        ? DEFAULT_MAX_RETRIES
+        : computedMaxRetries;
+    return {
+        backoff,
+        maxRetries,
+    };
+}
 //# sourceMappingURL=backend.mjs.map

package/dist/backend.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"backend.mjs","sourceRoot":"","sources":["../src/backend.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EACL,SAAS,EACT,oBAAoB,EAErB,uCAAuC;AAIxC,OAAO,EAAE,oBAAoB,EAAE,wBAAoB;AA+CnD,MAAM,OAAO,mBAAmB;IAW9B,YAAY,EACV,cAAc,EACd,wBAAwB,GAAG,IAAI,EAAE,eAAe;IAChD,6BAA6B,GAAG,IAAI,EAAE,eAAe;IACrD,OAAO,EACP,KAAK,EAAE,OAAO,GAOf;;QAtBQ,sDAAuC;QAEvC,gEAAkC;QAElC,qEAAuC;QAEvC,+CAAiB;QAEjB,6CAAgC;QAevC,uBAAA,IAAI,uCAAmB,cAAc,MAAA,CAAC;QACtC,uBAAA,IAAI,iDAA6B,wBAAwB,MAAA,CAAC;QAC1D,uBAAA,IAAI,sDAAkC,6BAA6B,MAAA,CAAC;QACpE,uBAAA,IAAI,gCAAY,OAAO,MAAA,CAAC;QACxB,uBAAA,IAAI,8BAAU,OAAO,MAAA,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,GAAyB;QAC3C,IAAI,EAAE,UAAU,EAAE,GAAG,GAAG,CAAC;QACzB,IAAI,CAAC,UAAU,EAAE;YACf,MAAM,OAAO,GAAG,yBAAyB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACtD,CAAC,EAAE,UAAU,EAAE,GAAG,MAAM,uBAAA,IAAI,8EAAmB,MAAvB,IAAI,EAC1B,8BAA8B,EAC9B,OAAO,CACR,CAAC,CAAC;SACJ;QAED,MAAM,mBAAmB,GAAG,GAAG,uBAAA,IAAI,oCAAS,iCAAiC,CAAC;QAC9E,MAAM,cAAc,GAAG,MAAM,uBAAA,IAAI,8EAAmB,MAAvB,IAAI,EAAoB,UAAU,EAAE;YAC/D,iBAAiB,EAAE,mBAAmB;SACvC,CAAC,CAAC;QACH,OAAO;YACL,UAAU;YACV,OAAO,EAAE,cAAc,CAAC,OAAO;YAC/B,UAAU,EAAE,cAAc,CAAC,UAAU;YACrC,MAAM,EAAE,cAAc,CAAC,MAAM;SAC9B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,sBAAsB,CAC1B,GAAkC;QAElC,IAAI,EAAE,UAAU,EAAE,GAAG,GAAG,CAAC;QACzB,IAAI,CAAC,UAAU,EAAE;YACf,MAAM,OAAO,GAAG,kCAAkC,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;YACzE,CAAC,EAAE,UAAU,EAAE,GAAG,MAAM,uBAAA,IAAI,8EAAmB,MAAvB,IAAI,EAC1B,4BAA4B,EAC5B,OAAO,CACR,CAAC,CAAC;SACJ;QAED,MAAM,0BAA0B,GAAG,GAAG,uBAAA,IAAI,oCAAS,+BAA+B,CAAC;QACnF,MAAM,cAAc,GAAG,MAAM,uBAAA,IAAI,8EAAmB,MAAvB,IAAI,EAAoB,UAAU,EAAE;YAC/D,iBAAiB,EAAE,0BAA0B;SAC9C,CAAC,CAAC;QACH,OAAO;YACL,UAAU;YACV,OAAO,EAAE,cAAc,CAAC,OAAO;YAC/B,UAAU,EAAE,cAAc,CAAC,UAAU;YACrC,MAAM,EAAE,cAAc,CAAC,MAAM;SAC9B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,GAAwB;QACzC,MAAM,QAAQ,GAAG,kCAAkC,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QAC1E,MAAM,IAAI,GAAG;YACX,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,GAAG,QAAQ;SACZ,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,uBAAA,IAAI,kCAAO,MAAX,IAAI,EACpB,GAAG,uBAAA,IAAI,oCAAS,4BAA4B,EAC5C;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,MAAM,uBAAA,IAAI,0EAAe,MAAnB,IAAI,CAAiB;YACpC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CACF,CAAC;QACF,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;YACtB,MAAM,IAAI,KAAK,CAAC,4BAA4B,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;SAC3D;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,GAA0B;QAC7C,MAAM,QAAQ,GAAG,yBAAyB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvD,MAAM,IAAI,GAAG;YACX,eAAe,EAAE,GAAG,CAAC,eAAe;YACpC,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,GAAG,QAAQ;SACZ,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,uBAAA,IAAI,kCAAO,MAAX,IAAI,EACpB,GAAG,uBAAA,IAAI,oCAAS,8BAA8B,EAC9C;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,MAAM,uBAAA,IAAI,0EAAe,MAAnB,IAAI,CAAiB;YACpC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CACF,CAAC;QACF,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;YACtB,MAAM,IAAI,KAAK,CAAC,8BAA8B,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;SAC7D;IACH,CAAC;CAwEF;2WAtEC,KAAK,iDACH,IAAY,EACZ,OAAgB;IAEhB,MAAM,GAAG,GAAG,MAAM,uBAAA,IAAI,kCAAO,MAAX,IAAI,EAAQ,GAAG,uBAAA,IAAI,oCAAS,IAAI,IAAI,EAAE,EAAE;QACxD,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,MAAM,uBAAA,IAAI,0EAAe,MAAnB,IAAI,CAAiB;QACpC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;KAC9B,CAAC,CAAC;IACH,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;QACtB,MAAM,IAAI,KAAK,CAAC,kCAAkC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;KACjE;IACD,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA8B,CAAC;AACzD,CAAC,2CAED,KAAK,iDACH,UAAkB,EAClB,OAIC;IAED,MAAM,OAAO,GAA6B;QACxC,UAAU;KACX,CAAC;IAEF,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,uBAAA,IAAI,qDAA0B,CAAC;IACnE,MAAM,YAAY,GAChB,OAAO,EAAE,YAAY,IAAI,uBAAA,IAAI,0DAA+B,CAAC;IAE/D,MAAM,OAAO,GAAG,MAAM,uBAAA,IAAI,0EAAe,MAAnB,IAAI,CAAiB,CAAC;IAC5C,OAAO,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC3C,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,UAAU,CAAC,GAAG,EAAE;YACd,cAAc,GAAG,IAAI,CAAC;YACtB,MAAM,CAAC,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC,CAAC;QAC3D,CAAC,EAAE,OAAO,CAAC,CAAC;QAEZ,MAAM,IAAI,GAAG,KAAK,IAAwC,EAAE;YAC1D,mEAAmE;YACnE,wDAAwD;YACxD,OAAO,CAAC,cAAc,EAAE;gBACtB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAC7B,MAAM,GAAG,GAAG,MAAM,uBAAA,IAAI,kCAAO,MAAX,IAAI,EAAQ,OAAO,CAAC,iBAAiB,EAAE;oBACvD,MAAM,EAAE,MAAM;oBACd,OAAO;oBACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;iBAC9B,CAAC,CAAC;gBACH,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;oBACtB,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA8B,CAAC;iBACxD;gBACD,MAAM,KAAK,CAAC,YAAY,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC;aACtD;YACD,wEAAwE;YACxE,mCAAmC;YACnC,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACtC,CAAC,CAAC;QAEF,IAAI,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;AACL,CAAC,uCAED,KAAK;IACH,MAAM,WAAW,GAAG,MAAM,uBAAA,IAAI,2CAAgB,MAApB,IAAI,CAAkB,CAAC;IACjD,OAAO;QACL,cAAc,EAAE,kBAAkB;QAClC,aAAa,EAAE,UAAU,WAAW,EAAE;KACvC,CAAC;AACJ,CAAC;AAGH;;;;;GAKG;AACH,KAAK,UAAU,KAAK,CAAC,EAAU;IAC7B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC;AAED;;;;;GAKG;AACH,SAAS,yBAAyB,CAChC,MAAuB;IAEvB,OAAO;QACL,QAAQ,EAAE;YACR;gBACE,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;gBAC1B,EAAE,EAAE,MAAM,CAAC,QAAQ,CAAC,EAAE;gBACtB,KAAK,EAAE,MAAM,CAAC,QAAQ,CAAC,KAAK;gBAC5B,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;gBAC1B,KAAK,EAAE,MAAM,CAAC,QAAQ,CAAC,KAAK;aAC7B;SACF;QACD,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM;KACtB,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,SAAS,kCAAkC,CACzC,gBAAkC;IAElC,sEAAsE;IACtE,2FAA2F;IAC3F,iHAAiH;IACjH,MAAM,MAAM,GAAG,2BAA2B,CAAC,gBAAgB,CAAC,CAAC;IAE7D,OAAO;QACL,OAAO,EAAE,gBAAgB,CAAC,OAAO;QACjC,IAAI,EAAE,gBAAgB,CAAC,aAAa,CAAC,IAAI;QACzC,IAAI,EAAE,gBAAgB,CAAC,aAAa,CAAC,IAAI;QACzC,MAAM;QACN,MAAM,EAAE,gBAAgB,CAAC,aAAa,CAAC,MAAM;KAC9C,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,2BAA2B,CACzC,gBAAkC;IAElC,IAAI,gBAAgB,CAAC,IAAI,KAAK,oBAAoB,CAAC,SAAS,EAAE;QAC5D,QAAQ,gBAAgB,CAAC,OAAO,EAAE;YAChC,KAAK,oBAAoB,CAAC,EAAE;gBAC1B,OAAO,SAAS,CAAC,eAAe,CAAC;YACnC,KAAK,oBAAoB,CAAC,EAAE;gBAC1B,OAAO,SAAS,CAAC,eAAe,CAAC;YACnC,KAAK,oBAAoB,CAAC,EAAE,CAAC;YAC7B;gBACE,OAAO,oBAAoB,CAAC,SAAS,CAAC;SACzC;KACF;IAED,OAAO,gBAAgB,CAAC,IAAI,CAAC;AAC/B,CAAC","sourcesContent":["import {\n  EthMethod,\n  SignatureRequestType,\n  type SignatureRequest,\n} from '@metamask/signature-controller';\nimport type { TransactionMeta } from '@metamask/transaction-controller';\nimport type { Json } from '@metamask/utils';\n\nimport { SignTypedDataVersion } from './constants';\nimport type {\n  CheckCoverageRequest,\n  CheckSignatureCoverageRequest,\n  CoverageResult,\n  CoverageStatus,\n  LogSignatureRequest,\n  LogTransactionRequest,\n  ShieldBackend,\n} from './types';\n\nexport type InitCoverageCheckRequest = {\n  txParams: [\n    {\n      from: string;\n      to?: string;\n      value?: string;\n      data?: string;\n      nonce?: string;\n    },\n  ];\n  chainId: string;\n  origin?: string;\n};\n\nexport type InitSignatureCoverageCheckRequest = {\n  chainId: string;\n  data: Json;\n  from: string;\n  method: string;\n  origin?: string;\n};\n\nexport type InitCoverageCheckResponse = {\n  coverageId: string;\n};\n\nexport type GetCoverageResultRequest = {\n  coverageId: string;\n};\n\nexport type GetCoverageResultResponse = {\n  message?: string;\n  reasonCode?: string;\n  status: CoverageStatus;\n};\n\nexport class ShieldRemoteBackend implements ShieldBackend {\n  readonly #getAccessToken: () => Promise<string>;\n\n  readonly #getCoverageResultTimeout: number;\n\n  readonly #getCoverageResultPollInterval: number;\n\n  readonly #baseUrl: string;\n\n  readonly #fetch: typeof globalThis.fetch;\n\n  constructor({\n    getAccessToken,\n    getCoverageResultTimeout = 5000, // milliseconds\n    getCoverageResultPollInterval = 1000, // milliseconds\n    baseUrl,\n    fetch: fetchFn,\n  }: {\n    getAccessToken: () => Promise<string>;\n    getCoverageResultTimeout?: number;\n    getCoverageResultPollInterval?: number;\n    baseUrl: string;\n    fetch: typeof globalThis.fetch;\n  }) {\n    this.#getAccessToken = getAccessToken;\n    this.#getCoverageResultTimeout = getCoverageResultTimeout;\n    this.#getCoverageResultPollInterval = getCoverageResultPollInterval;\n    this.#baseUrl = baseUrl;\n    this.#fetch = fetchFn;\n  }\n\n  async checkCoverage(req: CheckCoverageRequest): Promise<CoverageResult> {\n    let { coverageId } = req;\n    if (!coverageId) {\n      const reqBody = makeInitCoverageCheckBody(req.txMeta);\n      ({ coverageId } = await this.#initCoverageCheck(\n        'v1/transaction/coverage/init',\n        reqBody,\n      ));\n    }\n\n    const txCoverageResultUrl = `${this.#baseUrl}/v1/transaction/coverage/result`;\n    const coverageResult = await this.#getCoverageResult(coverageId, {\n      coverageResultUrl: txCoverageResultUrl,\n    });\n    return {\n      coverageId,\n      message: coverageResult.message,\n      reasonCode: coverageResult.reasonCode,\n      status: coverageResult.status,\n    };\n  }\n\n  async checkSignatureCoverage(\n    req: CheckSignatureCoverageRequest,\n  ): Promise<CoverageResult> {\n    let { coverageId } = req;\n    if (!coverageId) {\n      const reqBody = makeInitSignatureCoverageCheckBody(req.signatureRequest);\n      ({ coverageId } = await this.#initCoverageCheck(\n        'v1/signature/coverage/init',\n        reqBody,\n      ));\n    }\n\n    const signatureCoverageResultUrl = `${this.#baseUrl}/v1/signature/coverage/result`;\n    const coverageResult = await this.#getCoverageResult(coverageId, {\n      coverageResultUrl: signatureCoverageResultUrl,\n    });\n    return {\n      coverageId,\n      message: coverageResult.message,\n      reasonCode: coverageResult.reasonCode,\n      status: coverageResult.status,\n    };\n  }\n\n  async logSignature(req: LogSignatureRequest): Promise<void> {\n    const initBody = makeInitSignatureCoverageCheckBody(req.signatureRequest);\n    const body = {\n      signature: req.signature,\n      status: req.status,\n      ...initBody,\n    };\n\n    const res = await this.#fetch(\n      `${this.#baseUrl}/v1/signature/coverage/log`,\n      {\n        method: 'POST',\n        headers: await this.#createHeaders(),\n        body: JSON.stringify(body),\n      },\n    );\n    if (res.status !== 200) {\n      throw new Error(`Failed to log signature: ${res.status}`);\n    }\n  }\n\n  async logTransaction(req: LogTransactionRequest): Promise<void> {\n    const initBody = makeInitCoverageCheckBody(req.txMeta);\n    const body = {\n      transactionHash: req.transactionHash,\n      status: req.status,\n      ...initBody,\n    };\n\n    const res = await this.#fetch(\n      `${this.#baseUrl}/v1/transaction/coverage/log`,\n      {\n        method: 'POST',\n        headers: await this.#createHeaders(),\n        body: JSON.stringify(body),\n      },\n    );\n    if (res.status !== 200) {\n      throw new Error(`Failed to log transaction: ${res.status}`);\n    }\n  }\n\n  async #initCoverageCheck(\n    path: string,\n    reqBody: unknown,\n  ): Promise<InitCoverageCheckResponse> {\n    const res = await this.#fetch(`${this.#baseUrl}/${path}`, {\n      method: 'POST',\n      headers: await this.#createHeaders(),\n      body: JSON.stringify(reqBody),\n    });\n    if (res.status !== 200) {\n      throw new Error(`Failed to init coverage check: ${res.status}`);\n    }\n    return (await res.json()) as InitCoverageCheckResponse;\n  }\n\n  async #getCoverageResult(\n    coverageId: string,\n    configs: {\n      coverageResultUrl: string;\n      timeout?: number;\n      pollInterval?: number;\n    },\n  ): Promise<GetCoverageResultResponse> {\n    const reqBody: GetCoverageResultRequest = {\n      coverageId,\n    };\n\n    const timeout = configs?.timeout ?? this.#getCoverageResultTimeout;\n    const pollInterval =\n      configs?.pollInterval ?? this.#getCoverageResultPollInterval;\n\n    const headers = await this.#createHeaders();\n    return await new Promise((resolve, reject) => {\n      let timeoutReached = false;\n      setTimeout(() => {\n        timeoutReached = true;\n        reject(new Error('Timeout waiting for coverage result'));\n      }, timeout);\n\n      const poll = async (): Promise<GetCoverageResultResponse> => {\n        // The timeoutReached variable is modified in the timeout callback.\n        // eslint-disable-next-line no-unmodified-loop-condition\n        while (!timeoutReached) {\n          const startTime = Date.now();\n          const res = await this.#fetch(configs.coverageResultUrl, {\n            method: 'POST',\n            headers,\n            body: JSON.stringify(reqBody),\n          });\n          if (res.status === 200) {\n            return (await res.json()) as GetCoverageResultResponse;\n          }\n          await sleep(pollInterval - (Date.now() - startTime));\n        }\n        // The following line will not have an effect as the upper level promise\n        // will already be rejected by now.\n        throw new Error('unexpected error');\n      };\n\n      poll().then(resolve).catch(reject);\n    });\n  }\n\n  async #createHeaders() {\n    const accessToken = await this.#getAccessToken();\n    return {\n      'Content-Type': 'application/json',\n      Authorization: `Bearer ${accessToken}`,\n    };\n  }\n}\n\n/**\n * Sleep for a specified amount of time.\n *\n * @param ms - The number of milliseconds to sleep.\n * @returns A promise that resolves after the specified amount of time.\n */\nasync function sleep(ms: number) {\n  return new Promise((resolve) => setTimeout(resolve, ms));\n}\n\n/**\n * Make the body for the init coverage check request.\n *\n * @param txMeta - The transaction metadata.\n * @returns The body for the init coverage check request.\n */\nfunction makeInitCoverageCheckBody(\n  txMeta: TransactionMeta,\n): InitCoverageCheckRequest {\n  return {\n    txParams: [\n      {\n        from: txMeta.txParams.from,\n        to: txMeta.txParams.to,\n        value: txMeta.txParams.value,\n        data: txMeta.txParams.data,\n        nonce: txMeta.txParams.nonce,\n      },\n    ],\n    chainId: txMeta.chainId,\n    origin: txMeta.origin,\n  };\n}\n\n/**\n * Make the body for the init signature coverage check request.\n *\n * @param signatureRequest - The signature request.\n * @returns The body for the init signature coverage check request.\n */\nfunction makeInitSignatureCoverageCheckBody(\n  signatureRequest: SignatureRequest,\n): InitSignatureCoverageCheckRequest {\n  // TODO: confirm that do we still need to validate the signature data?\n  // signature controller already validates the signature data before adding it to the state.\n  // @link https://github.com/MetaMask/core/blob/main/packages/signature-controller/src/SignatureController.ts#L408\n  const method = parseSignatureRequestMethod(signatureRequest);\n\n  return {\n    chainId: signatureRequest.chainId,\n    data: signatureRequest.messageParams.data,\n    from: signatureRequest.messageParams.from,\n    method,\n    origin: signatureRequest.messageParams.origin,\n  };\n}\n\n/**\n * Parse the JSON-RPC method from the signature request.\n *\n * @param signatureRequest - The signature request.\n * @returns The JSON-RPC method.\n */\nexport function parseSignatureRequestMethod(\n  signatureRequest: SignatureRequest,\n): string {\n  if (signatureRequest.type === SignatureRequestType.TypedSign) {\n    switch (signatureRequest.version) {\n      case SignTypedDataVersion.V3:\n        return EthMethod.SignTypedDataV3;\n      case SignTypedDataVersion.V4:\n        return EthMethod.SignTypedDataV4;\n      case SignTypedDataVersion.V1:\n      default:\n        return SignatureRequestType.TypedSign;\n    }\n  }\n\n  return signatureRequest.type;\n}\n"]}
+{"version":3,"file":"backend.mjs","sourceRoot":"","sources":["../src/backend.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EACL,eAAe,EACf,mBAAmB,EACnB,SAAS,EACV,mCAAmC;AACpC,OAAO,EACL,SAAS,EACT,oBAAoB,EAErB,uCAAuC;AAIxC,OAAO,EAAE,oBAAoB,EAAE,wBAAoB;AACnD,OAAO,EAAE,0BAA0B,EAAE,kCAA8B;AA+CnE,MAAM,OAAO,mBAAmB;IAS9B,YAAY,EACV,cAAc,EACd,wBAAwB,GAAG,IAAI,EAAE,eAAe;IAChD,6BAA6B,GAAG,IAAI,EAAE,eAAe;IACrD,OAAO,EACP,KAAK,EAAE,OAAO,GAOf;;QApBQ,sDAAuC;QAEvC,+CAAiB;QAEjB,6CAAgC;QAEhC,qDAA2C;QAelD,uBAAA,IAAI,uCAAmB,cAAc,MAAA,CAAC;QACtC,uBAAA,IAAI,gCAAY,OAAO,MAAA,CAAC;QACxB,uBAAA,IAAI,8BAAU,OAAO,MAAA,CAAC;QAEtB,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,mCAAmC,CACjE,wBAAwB,EACxB,6BAA6B,CAC9B,CAAC;QAEF,uBAAA,IAAI,sCAAkB,IAAI,0BAA0B,CAAC;YACnD,OAAO;YACP,UAAU;SACX,CAAC,MAAA,CAAC;IACL,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,GAAyB;QAC3C,IAAI,EAAE,UAAU,EAAE,GAAG,GAAG,CAAC;QACzB,IAAI,CAAC,UAAU,EAAE;YACf,MAAM,OAAO,GAAG,yBAAyB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACtD,CAAC,EAAE,UAAU,EAAE,GAAG,MAAM,uBAAA,IAAI,8EAAmB,MAAvB,IAAI,EAC1B,8BAA8B,EAC9B,OAAO,CACR,CAAC,CAAC;SACJ;QAED,MAAM,mBAAmB,GAAG,GAAG,uBAAA,IAAI,oCAAS,iCAAiC,CAAC;QAC9E,MAAM,cAAc,GAAG,MAAM,uBAAA,IAAI,8EAAmB,MAAvB,IAAI,EAC/B,GAAG,CAAC,MAAM,CAAC,EAAE,EACb,UAAU,EACV,mBAAmB,CACpB,CAAC;QACF,OAAO;YACL,UAAU;YACV,OAAO,EAAE,cAAc,CAAC,OAAO;YAC/B,UAAU,EAAE,cAAc,CAAC,UAAU;YACrC,MAAM,EAAE,cAAc,CAAC,MAAM;SAC9B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,sBAAsB,CAC1B,GAAkC;QAElC,IAAI,EAAE,UAAU,EAAE,GAAG,GAAG,CAAC;QACzB,IAAI,CAAC,UAAU,EAAE;YACf,MAAM,OAAO,GAAG,kCAAkC,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;YACzE,CAAC,EAAE,UAAU,EAAE,GAAG,MAAM,uBAAA,IAAI,8EAAmB,MAAvB,IAAI,EAC1B,4BAA4B,EAC5B,OAAO,CACR,CAAC,CAAC;SACJ;QAED,MAAM,0BAA0B,GAAG,GAAG,uBAAA,IAAI,oCAAS,+BAA+B,CAAC;QACnF,MAAM,cAAc,GAAG,MAAM,uBAAA,IAAI,8EAAmB,MAAvB,IAAI,EAC/B,GAAG,CAAC,gBAAgB,CAAC,EAAE,EACvB,UAAU,EACV,0BAA0B,CAC3B,CAAC;QACF,OAAO;YACL,UAAU;YACV,OAAO,EAAE,cAAc,CAAC,OAAO;YAC/B,UAAU,EAAE,cAAc,CAAC,UAAU;YACrC,MAAM,EAAE,cAAc,CAAC,MAAM;SAC9B,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,GAAwB;QACzC,MAAM,QAAQ,GAAG,kCAAkC,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QAC1E,MAAM,IAAI,GAAG;YACX,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,GAAG,QAAQ;SACZ,CAAC;QAEF,iDAAiD;QACjD,uBAAA,IAAI,0CAAe,CAAC,mBAAmB,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;QAEjE,MAAM,GAAG,GAAG,MAAM,uBAAA,IAAI,kCAAO,MAAX,IAAI,EACpB,GAAG,uBAAA,IAAI,oCAAS,4BAA4B,EAC5C;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,MAAM,uBAAA,IAAI,0EAAe,MAAnB,IAAI,CAAiB;YACpC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CACF,CAAC;QACF,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;YACtB,MAAM,IAAI,KAAK,CAAC,4BAA4B,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;SAC3D;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,GAA0B;QAC7C,MAAM,QAAQ,GAAG,yBAAyB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACvD,MAAM,IAAI,GAAG;YACX,eAAe,EAAE,GAAG,CAAC,eAAe;YACpC,MAAM,EAAE,GAAG,CAAC,MAAM;YAClB,GAAG,QAAQ;SACZ,CAAC;QAEF,iDAAiD;QACjD,uBAAA,IAAI,0CAAe,CAAC,mBAAmB,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAEvD,MAAM,GAAG,GAAG,MAAM,uBAAA,IAAI,kCAAO,MAAX,IAAI,EACpB,GAAG,uBAAA,IAAI,oCAAS,8BAA8B,EAC9C;YACE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,MAAM,uBAAA,IAAI,0EAAe,MAAnB,IAAI,CAAiB;YACpC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CACF,CAAC;QACF,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;YACtB,MAAM,IAAI,KAAK,CAAC,8BAA8B,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;SAC7D;IACH,CAAC;CA4DF;4RA1DC,KAAK,iDACH,IAAY,EACZ,OAAgB;IAEhB,MAAM,GAAG,GAAG,MAAM,uBAAA,IAAI,kCAAO,MAAX,IAAI,EAAQ,GAAG,uBAAA,IAAI,oCAAS,IAAI,IAAI,EAAE,EAAE;QACxD,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,MAAM,uBAAA,IAAI,0EAAe,MAAnB,IAAI,CAAiB;QACpC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;KAC9B,CAAC,CAAC;IACH,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;QACtB,MAAM,IAAI,KAAK,CAAC,kCAAkC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;KACjE;IACD,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA8B,CAAC;AACzD,CAAC,2CAED,KAAK,iDACH,SAAiB,EACjB,UAAkB,EAClB,iBAAyB;IAEzB,MAAM,OAAO,GAA6B;QACxC,UAAU;KACX,CAAC;IAEF,MAAM,OAAO,GAAG,MAAM,uBAAA,IAAI,0EAAe,MAAnB,IAAI,CAAiB,CAAC;IAE5C,MAAM,mBAAmB,GAAG,KAAK,EAAE,MAAmB,EAAE,EAAE;QACxD,MAAM,GAAG,GAAG,MAAM,uBAAA,IAAI,kCAAO,MAAX,IAAI,EAAQ,iBAAiB,EAAE;YAC/C,MAAM,EAAE,MAAM;YACd,OAAO;YACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;YAC7B,MAAM;SACP,CAAC,CAAC;QACH,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE;YACtB,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA8B,CAAC;SACxD;QAED,iDAAiD;QACjD,IAAI,YAAY,GAAG,qCAAqC,CAAC;QACzD,IAAI;YACF,MAAM,SAAS,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;YACnC,YAAY,GAAG,kCAAkC,SAAS,CAAC,OAAO,IAAI,SAAS,CAAC,MAAM,EAAE,CAAC;SAC1F;QAAC,MAAM;YACN,YAAY,GAAG,kCAAkC,GAAG,CAAC,MAAM,EAAE,CAAC;SAC/D;QACD,MAAM,IAAI,SAAS,CAAC,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;IAChD,CAAC,CAAC;IAEF,OAAO,uBAAA,IAAI,0CAAe,CAAC,KAAK,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC;AACnE,CAAC,uCAED,KAAK;IACH,MAAM,WAAW,GAAG,MAAM,uBAAA,IAAI,2CAAgB,MAApB,IAAI,CAAkB,CAAC;IACjD,OAAO;QACL,cAAc,EAAE,kBAAkB;QAClC,aAAa,EAAE,UAAU,WAAW,EAAE;KACvC,CAAC;AACJ,CAAC;AAGH;;;;;GAKG;AACH,SAAS,yBAAyB,CAChC,MAAuB;IAEvB,OAAO;QACL,QAAQ,EAAE;YACR;gBACE,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;gBAC1B,EAAE,EAAE,MAAM,CAAC,QAAQ,CAAC,EAAE;gBACtB,KAAK,EAAE,MAAM,CAAC,QAAQ,CAAC,KAAK;gBAC5B,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;gBAC1B,KAAK,EAAE,MAAM,CAAC,QAAQ,CAAC,KAAK;aAC7B;SACF;QACD,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM;KACtB,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,SAAS,kCAAkC,CACzC,gBAAkC;IAElC,sEAAsE;IACtE,2FAA2F;IAC3F,iHAAiH;IACjH,MAAM,MAAM,GAAG,2BAA2B,CAAC,gBAAgB,CAAC,CAAC;IAE7D,OAAO;QACL,OAAO,EAAE,gBAAgB,CAAC,OAAO;QACjC,IAAI,EAAE,gBAAgB,CAAC,aAAa,CAAC,IAAI;QACzC,IAAI,EAAE,gBAAgB,CAAC,aAAa,CAAC,IAAI;QACzC,MAAM;QACN,MAAM,EAAE,gBAAgB,CAAC,aAAa,CAAC,MAAM;KAC9C,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,2BAA2B,CACzC,gBAAkC;IAElC,IAAI,gBAAgB,CAAC,IAAI,KAAK,oBAAoB,CAAC,SAAS,EAAE;QAC5D,QAAQ,gBAAgB,CAAC,OAAO,EAAE;YAChC,KAAK,oBAAoB,CAAC,EAAE;gBAC1B,OAAO,SAAS,CAAC,eAAe,CAAC;YACnC,KAAK,oBAAoB,CAAC,EAAE;gBAC1B,OAAO,SAAS,CAAC,eAAe,CAAC;YACnC,KAAK,oBAAoB,CAAC,EAAE,CAAC;YAC7B;gBACE,OAAO,oBAAoB,CAAC,SAAS,CAAC;SACzC;KACF;IAED,OAAO,gBAAgB,CAAC,IAAI,CAAC;AAC/B,CAAC;AAED;;;;;;GAMG;AACH,SAAS,mCAAmC,CAC1C,OAAe,EACf,YAAoB;IAEpB,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,YAAY,CAAC,CAAC;IAClD,MAAM,kBAAkB,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;IAElE,MAAM,UAAU,GACd,KAAK,CAAC,kBAAkB,CAAC,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QACxD,CAAC,CAAC,mBAAmB;QACrB,CAAC,CAAC,kBAAkB,CAAC;IAEzB,OAAO;QACL,OAAO;QACP,UAAU;KACX,CAAC;AACJ,CAAC","sourcesContent":["import {\n  ConstantBackoff,\n  DEFAULT_MAX_RETRIES,\n  HttpError,\n} from '@metamask/controller-utils';\nimport {\n  EthMethod,\n  SignatureRequestType,\n  type SignatureRequest,\n} from '@metamask/signature-controller';\nimport type { TransactionMeta } from '@metamask/transaction-controller';\nimport type { Json } from '@metamask/utils';\n\nimport { SignTypedDataVersion } from './constants';\nimport { PollingWithCockatielPolicy } from './polling-with-policy';\nimport type {\n  CheckCoverageRequest,\n  CheckSignatureCoverageRequest,\n  CoverageResult,\n  CoverageStatus,\n  LogSignatureRequest,\n  LogTransactionRequest,\n  ShieldBackend,\n} from './types';\n\nexport type InitCoverageCheckRequest = {\n  txParams: [\n    {\n      from: string;\n      to?: string;\n      value?: string;\n      data?: string;\n      nonce?: string;\n    },\n  ];\n  chainId: string;\n  origin?: string;\n};\n\nexport type InitSignatureCoverageCheckRequest = {\n  chainId: string;\n  data: Json;\n  from: string;\n  method: string;\n  origin?: string;\n};\n\nexport type InitCoverageCheckResponse = {\n  coverageId: string;\n};\n\nexport type GetCoverageResultRequest = {\n  coverageId: string;\n};\n\nexport type GetCoverageResultResponse = {\n  message?: string;\n  reasonCode?: string;\n  status: CoverageStatus;\n};\n\nexport class ShieldRemoteBackend implements ShieldBackend {\n  readonly #getAccessToken: () => Promise<string>;\n\n  readonly #baseUrl: string;\n\n  readonly #fetch: typeof globalThis.fetch;\n\n  readonly #pollingPolicy: PollingWithCockatielPolicy;\n\n  constructor({\n    getAccessToken,\n    getCoverageResultTimeout = 5000, // milliseconds\n    getCoverageResultPollInterval = 1000, // milliseconds\n    baseUrl,\n    fetch: fetchFn,\n  }: {\n    getAccessToken: () => Promise<string>;\n    getCoverageResultTimeout?: number;\n    getCoverageResultPollInterval?: number;\n    baseUrl: string;\n    fetch: typeof globalThis.fetch;\n  }) {\n    this.#getAccessToken = getAccessToken;\n    this.#baseUrl = baseUrl;\n    this.#fetch = fetchFn;\n\n    const { backoff, maxRetries } = computePollingIntervalAndRetryCount(\n      getCoverageResultTimeout,\n      getCoverageResultPollInterval,\n    );\n\n    this.#pollingPolicy = new PollingWithCockatielPolicy({\n      backoff,\n      maxRetries,\n    });\n  }\n\n  async checkCoverage(req: CheckCoverageRequest): Promise<CoverageResult> {\n    let { coverageId } = req;\n    if (!coverageId) {\n      const reqBody = makeInitCoverageCheckBody(req.txMeta);\n      ({ coverageId } = await this.#initCoverageCheck(\n        'v1/transaction/coverage/init',\n        reqBody,\n      ));\n    }\n\n    const txCoverageResultUrl = `${this.#baseUrl}/v1/transaction/coverage/result`;\n    const coverageResult = await this.#getCoverageResult(\n      req.txMeta.id,\n      coverageId,\n      txCoverageResultUrl,\n    );\n    return {\n      coverageId,\n      message: coverageResult.message,\n      reasonCode: coverageResult.reasonCode,\n      status: coverageResult.status,\n    };\n  }\n\n  async checkSignatureCoverage(\n    req: CheckSignatureCoverageRequest,\n  ): Promise<CoverageResult> {\n    let { coverageId } = req;\n    if (!coverageId) {\n      const reqBody = makeInitSignatureCoverageCheckBody(req.signatureRequest);\n      ({ coverageId } = await this.#initCoverageCheck(\n        'v1/signature/coverage/init',\n        reqBody,\n      ));\n    }\n\n    const signatureCoverageResultUrl = `${this.#baseUrl}/v1/signature/coverage/result`;\n    const coverageResult = await this.#getCoverageResult(\n      req.signatureRequest.id,\n      coverageId,\n      signatureCoverageResultUrl,\n    );\n    return {\n      coverageId,\n      message: coverageResult.message,\n      reasonCode: coverageResult.reasonCode,\n      status: coverageResult.status,\n    };\n  }\n\n  async logSignature(req: LogSignatureRequest): Promise<void> {\n    const initBody = makeInitSignatureCoverageCheckBody(req.signatureRequest);\n    const body = {\n      signature: req.signature,\n      status: req.status,\n      ...initBody,\n    };\n\n    // cancel the pending get coverage result request\n    this.#pollingPolicy.abortPendingRequest(req.signatureRequest.id);\n\n    const res = await this.#fetch(\n      `${this.#baseUrl}/v1/signature/coverage/log`,\n      {\n        method: 'POST',\n        headers: await this.#createHeaders(),\n        body: JSON.stringify(body),\n      },\n    );\n    if (res.status !== 200) {\n      throw new Error(`Failed to log signature: ${res.status}`);\n    }\n  }\n\n  async logTransaction(req: LogTransactionRequest): Promise<void> {\n    const initBody = makeInitCoverageCheckBody(req.txMeta);\n    const body = {\n      transactionHash: req.transactionHash,\n      status: req.status,\n      ...initBody,\n    };\n\n    // cancel the pending get coverage result request\n    this.#pollingPolicy.abortPendingRequest(req.txMeta.id);\n\n    const res = await this.#fetch(\n      `${this.#baseUrl}/v1/transaction/coverage/log`,\n      {\n        method: 'POST',\n        headers: await this.#createHeaders(),\n        body: JSON.stringify(body),\n      },\n    );\n    if (res.status !== 200) {\n      throw new Error(`Failed to log transaction: ${res.status}`);\n    }\n  }\n\n  async #initCoverageCheck(\n    path: string,\n    reqBody: unknown,\n  ): Promise<InitCoverageCheckResponse> {\n    const res = await this.#fetch(`${this.#baseUrl}/${path}`, {\n      method: 'POST',\n      headers: await this.#createHeaders(),\n      body: JSON.stringify(reqBody),\n    });\n    if (res.status !== 200) {\n      throw new Error(`Failed to init coverage check: ${res.status}`);\n    }\n    return (await res.json()) as InitCoverageCheckResponse;\n  }\n\n  async #getCoverageResult(\n    requestId: string,\n    coverageId: string,\n    coverageResultUrl: string,\n  ): Promise<GetCoverageResultResponse> {\n    const reqBody: GetCoverageResultRequest = {\n      coverageId,\n    };\n\n    const headers = await this.#createHeaders();\n\n    const getCoverageResultFn = async (signal: AbortSignal) => {\n      const res = await this.#fetch(coverageResultUrl, {\n        method: 'POST',\n        headers,\n        body: JSON.stringify(reqBody),\n        signal,\n      });\n      if (res.status === 200) {\n        return (await res.json()) as GetCoverageResultResponse;\n      }\n\n      // parse the error message from the response body\n      let errorMessage = 'Timeout waiting for coverage result';\n      try {\n        const errorJson = await res.json();\n        errorMessage = `Failed to get coverage result: ${errorJson.message || errorJson.status}`;\n      } catch {\n        errorMessage = `Failed to get coverage result: ${res.status}`;\n      }\n      throw new HttpError(res.status, errorMessage);\n    };\n\n    return this.#pollingPolicy.start(requestId, getCoverageResultFn);\n  }\n\n  async #createHeaders() {\n    const accessToken = await this.#getAccessToken();\n    return {\n      'Content-Type': 'application/json',\n      Authorization: `Bearer ${accessToken}`,\n    };\n  }\n}\n\n/**\n * Make the body for the init coverage check request.\n *\n * @param txMeta - The transaction metadata.\n * @returns The body for the init coverage check request.\n */\nfunction makeInitCoverageCheckBody(\n  txMeta: TransactionMeta,\n): InitCoverageCheckRequest {\n  return {\n    txParams: [\n      {\n        from: txMeta.txParams.from,\n        to: txMeta.txParams.to,\n        value: txMeta.txParams.value,\n        data: txMeta.txParams.data,\n        nonce: txMeta.txParams.nonce,\n      },\n    ],\n    chainId: txMeta.chainId,\n    origin: txMeta.origin,\n  };\n}\n\n/**\n * Make the body for the init signature coverage check request.\n *\n * @param signatureRequest - The signature request.\n * @returns The body for the init signature coverage check request.\n */\nfunction makeInitSignatureCoverageCheckBody(\n  signatureRequest: SignatureRequest,\n): InitSignatureCoverageCheckRequest {\n  // TODO: confirm that do we still need to validate the signature data?\n  // signature controller already validates the signature data before adding it to the state.\n  // @link https://github.com/MetaMask/core/blob/main/packages/signature-controller/src/SignatureController.ts#L408\n  const method = parseSignatureRequestMethod(signatureRequest);\n\n  return {\n    chainId: signatureRequest.chainId,\n    data: signatureRequest.messageParams.data,\n    from: signatureRequest.messageParams.from,\n    method,\n    origin: signatureRequest.messageParams.origin,\n  };\n}\n\n/**\n * Parse the JSON-RPC method from the signature request.\n *\n * @param signatureRequest - The signature request.\n * @returns The JSON-RPC method.\n */\nexport function parseSignatureRequestMethod(\n  signatureRequest: SignatureRequest,\n): string {\n  if (signatureRequest.type === SignatureRequestType.TypedSign) {\n    switch (signatureRequest.version) {\n      case SignTypedDataVersion.V3:\n        return EthMethod.SignTypedDataV3;\n      case SignTypedDataVersion.V4:\n        return EthMethod.SignTypedDataV4;\n      case SignTypedDataVersion.V1:\n      default:\n        return SignatureRequestType.TypedSign;\n    }\n  }\n\n  return signatureRequest.type;\n}\n\n/**\n * Compute the polling interval and retry count for the Cockatiel policy based on the timeout and poll interval given.\n *\n * @param timeout - The timeout in milliseconds.\n * @param pollInterval - The poll interval in milliseconds.\n * @returns The polling interval and retry count.\n */\nfunction computePollingIntervalAndRetryCount(\n  timeout: number,\n  pollInterval: number,\n) {\n  const backoff = new ConstantBackoff(pollInterval);\n  const computedMaxRetries = Math.floor(timeout / pollInterval) + 1;\n\n  const maxRetries =\n    isNaN(computedMaxRetries) || !isFinite(computedMaxRetries)\n      ? DEFAULT_MAX_RETRIES\n      : computedMaxRetries;\n\n  return {\n    backoff,\n    maxRetries,\n  };\n}\n"]}

package/dist/polling-with-policy.cjs

@@ -0,0 +1,76 @@
+"use strict";
+var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
+    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
+};
+var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
+    if (kind === "m") throw new TypeError("Private method is not writable");
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
+    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
+};
+var _PollingWithCockatielPolicy_instances, _PollingWithCockatielPolicy_policy, _PollingWithCockatielPolicy_requestEntry, _PollingWithCockatielPolicy_addNewRequestEntry, _PollingWithCockatielPolicy_cleanup, _PollingWithCockatielPolicy_shouldRetry;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PollingWithCockatielPolicy = void 0;
+const controller_utils_1 = require("@metamask/controller-utils");
+const cockatiel_1 = require("cockatiel");
+class PollingWithCockatielPolicy {
+    constructor(policyOptions = {}) {
+        _PollingWithCockatielPolicy_instances.add(this);
+        _PollingWithCockatielPolicy_policy.set(this, void 0);
+        _PollingWithCockatielPolicy_requestEntry.set(this, new Map());
+        const retryFilterPolicy = (0, cockatiel_1.handleWhen)(__classPrivateFieldGet(this, _PollingWithCockatielPolicy_instances, "m", _PollingWithCockatielPolicy_shouldRetry));
+        __classPrivateFieldSet(this, _PollingWithCockatielPolicy_policy, (0, controller_utils_1.createServicePolicy)({
+            ...policyOptions,
+            retryFilterPolicy,
+        }), "f");
+    }
+    async start(requestId, requestFn) {
+        const abortController = __classPrivateFieldGet(this, _PollingWithCockatielPolicy_instances, "m", _PollingWithCockatielPolicy_addNewRequestEntry).call(this, requestId);
+        try {
+            const result = await __classPrivateFieldGet(this, _PollingWithCockatielPolicy_policy, "f").execute(async ({ signal }) => {
+                return requestFn(signal);
+            }, abortController.signal);
+            return result;
+        }
+        catch (error) {
+            if (abortController.signal.aborted) {
+                throw new Error('Request cancelled');
+            }
+            throw error;
+        }
+        finally {
+            // Only cleanup if this abort controller is still active. If a new request with the same
+            // requestId started while this one was running, it would have replaced with a new abort controller.
+            // We must not delete the new request's controller when this older request finishes.
+            if (abortController === __classPrivateFieldGet(this, _PollingWithCockatielPolicy_requestEntry, "f").get(requestId)) {
+                __classPrivateFieldGet(this, _PollingWithCockatielPolicy_instances, "m", _PollingWithCockatielPolicy_cleanup).call(this, requestId);
+            }
+        }
+    }
+    abortPendingRequest(requestId) {
+        const abortController = __classPrivateFieldGet(this, _PollingWithCockatielPolicy_requestEntry, "f").get(requestId);
+        abortController?.abort();
+        __classPrivateFieldGet(this, _PollingWithCockatielPolicy_instances, "m", _PollingWithCockatielPolicy_cleanup).call(this, requestId);
+    }
+}
+exports.PollingWithCockatielPolicy = PollingWithCockatielPolicy;
+_PollingWithCockatielPolicy_policy = new WeakMap(), _PollingWithCockatielPolicy_requestEntry = new WeakMap(), _PollingWithCockatielPolicy_instances = new WeakSet(), _PollingWithCockatielPolicy_addNewRequestEntry = function _PollingWithCockatielPolicy_addNewRequestEntry(requestId) {
+    // abort the previous request if it exists
+    this.abortPendingRequest(requestId);
+    // create a new abort controller for the new request
+    const abortController = new AbortController();
+    __classPrivateFieldGet(this, _PollingWithCockatielPolicy_requestEntry, "f").set(requestId, abortController);
+    return abortController;
+}, _PollingWithCockatielPolicy_cleanup = function _PollingWithCockatielPolicy_cleanup(requestId) {
+    __classPrivateFieldGet(this, _PollingWithCockatielPolicy_requestEntry, "f").delete(requestId);
+}, _PollingWithCockatielPolicy_shouldRetry = function _PollingWithCockatielPolicy_shouldRetry(error) {
+    if (error instanceof controller_utils_1.HttpError) {
+        // Note: we don't retry on 5xx errors, only on 4xx errors.
+        // but we won't retry on 400 coz it means that the request body is invalid.
+        return error.httpStatus > 400 && error.httpStatus < 500;
+    }
+    return false;
+};
+//# sourceMappingURL=polling-with-policy.cjs.map

package/dist/polling-with-policy.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"polling-with-policy.cjs","sourceRoot":"","sources":["../src/polling-with-policy.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,iEAKoC;AACpC,yCAAuC;AAMvC,MAAa,0BAA0B;IAKrC,YAAY,gBAA4C,EAAE;;QAJjD,qDAAuB;QAEvB,mDAAgB,IAAI,GAAG,EAA2B,EAAC;QAG1D,MAAM,iBAAiB,GAAG,IAAA,sBAAU,EAAC,uBAAA,IAAI,sFAAa,CAAC,CAAC;QACxD,uBAAA,IAAI,sCAAW,IAAA,sCAAmB,EAAC;YACjC,GAAG,aAAa;YAChB,iBAAiB;SAClB,CAAC,MAAA,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK,CAAa,SAAiB,EAAE,SAAgC;QACzE,MAAM,eAAe,GAAG,uBAAA,IAAI,6FAAoB,MAAxB,IAAI,EAAqB,SAAS,CAAC,CAAC;QAE5D,IAAI;YACF,MAAM,MAAM,GAAG,MAAM,uBAAA,IAAI,0CAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;gBAC7D,OAAO,SAAS,CAAC,MAAM,CAAC,CAAC;YAC3B,CAAC,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC;YAC3B,OAAO,MAAM,CAAC;SACf;QAAC,OAAO,KAAK,EAAE;YACd,IAAI,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE;gBAClC,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;aACtC;YACD,MAAM,KAAK,CAAC;SACb;gBAAS;YACR,wFAAwF;YACxF,oGAAoG;YACpG,oFAAoF;YACpF,IAAI,eAAe,KAAK,uBAAA,IAAI,gDAAc,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE;gBACzD,uBAAA,IAAI,kFAAS,MAAb,IAAI,EAAU,SAAS,CAAC,CAAC;aAC1B;SACF;IACH,CAAC;IAED,mBAAmB,CAAC,SAAiB;QACnC,MAAM,eAAe,GAAG,uBAAA,IAAI,gDAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC1D,eAAe,EAAE,KAAK,EAAE,CAAC;QACzB,uBAAA,IAAI,kFAAS,MAAb,IAAI,EAAU,SAAS,CAAC,CAAC;IAC3B,CAAC;CAwBF;AAhED,gEAgEC;8QAtBqB,SAAiB;IACnC,0CAA0C;IAC1C,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;IAEpC,oDAAoD;IACpD,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;IAC9C,uBAAA,IAAI,gDAAc,CAAC,GAAG,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;IACnD,OAAO,eAAe,CAAC;AACzB,CAAC,qFAEQ,SAAiB;IACxB,uBAAA,IAAI,gDAAc,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;AACvC,CAAC,6FAEY,KAAY;IACvB,IAAI,KAAK,YAAY,4BAAS,EAAE;QAC9B,0DAA0D;QAC1D,2EAA2E;QAC3E,OAAO,KAAK,CAAC,UAAU,GAAG,GAAG,IAAI,KAAK,CAAC,UAAU,GAAG,GAAG,CAAC;KACzD;IACD,OAAO,KAAK,CAAC;AACf,CAAC","sourcesContent":["import {\n  createServicePolicy,\n  HttpError,\n  type CreateServicePolicyOptions,\n  type ServicePolicy,\n} from '@metamask/controller-utils';\nimport { handleWhen } from 'cockatiel';\n\nexport type RequestFn<ReturnType> = (\n  signal: AbortSignal,\n) => Promise<ReturnType>;\n\nexport class PollingWithCockatielPolicy {\n  readonly #policy: ServicePolicy;\n\n  readonly #requestEntry = new Map<string, AbortController>();\n\n  constructor(policyOptions: CreateServicePolicyOptions = {}) {\n    const retryFilterPolicy = handleWhen(this.#shouldRetry);\n    this.#policy = createServicePolicy({\n      ...policyOptions,\n      retryFilterPolicy,\n    });\n  }\n\n  async start<ReturnType>(requestId: string, requestFn: RequestFn<ReturnType>) {\n    const abortController = this.#addNewRequestEntry(requestId);\n\n    try {\n      const result = await this.#policy.execute(async ({ signal }) => {\n        return requestFn(signal);\n      }, abortController.signal);\n      return result;\n    } catch (error) {\n      if (abortController.signal.aborted) {\n        throw new Error('Request cancelled');\n      }\n      throw error;\n    } finally {\n      // Only cleanup if this abort controller is still active. If a new request with the same\n      // requestId started while this one was running, it would have replaced with a new abort controller.\n      // We must not delete the new request's controller when this older request finishes.\n      if (abortController === this.#requestEntry.get(requestId)) {\n        this.#cleanup(requestId);\n      }\n    }\n  }\n\n  abortPendingRequest(requestId: string) {\n    const abortController = this.#requestEntry.get(requestId);\n    abortController?.abort();\n    this.#cleanup(requestId);\n  }\n\n  #addNewRequestEntry(requestId: string) {\n    // abort the previous request if it exists\n    this.abortPendingRequest(requestId);\n\n    // create a new abort controller for the new request\n    const abortController = new AbortController();\n    this.#requestEntry.set(requestId, abortController);\n    return abortController;\n  }\n\n  #cleanup(requestId: string) {\n    this.#requestEntry.delete(requestId);\n  }\n\n  #shouldRetry(error: Error): boolean {\n    if (error instanceof HttpError) {\n      // Note: we don't retry on 5xx errors, only on 4xx errors.\n      // but we won't retry on 400 coz it means that the request body is invalid.\n      return error.httpStatus > 400 && error.httpStatus < 500;\n    }\n    return false;\n  }\n}\n"]}

package/dist/polling-with-policy.d.cts

@@ -0,0 +1,9 @@
+import { type CreateServicePolicyOptions } from "@metamask/controller-utils";
+export type RequestFn<ReturnType> = (signal: AbortSignal) => Promise<ReturnType>;
+export declare class PollingWithCockatielPolicy {
+    #private;
+    constructor(policyOptions?: CreateServicePolicyOptions);
+    start<ReturnType>(requestId: string, requestFn: RequestFn<ReturnType>): Promise<ReturnType>;
+    abortPendingRequest(requestId: string): void;
+}
+//# sourceMappingURL=polling-with-policy.d.cts.map

package/dist/polling-with-policy.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"polling-with-policy.d.cts","sourceRoot":"","sources":["../src/polling-with-policy.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,KAAK,0BAA0B,EAEhC,mCAAmC;AAGpC,MAAM,MAAM,SAAS,CAAC,UAAU,IAAI,CAClC,MAAM,EAAE,WAAW,KAChB,OAAO,CAAC,UAAU,CAAC,CAAC;AAEzB,qBAAa,0BAA0B;;gBAKzB,aAAa,GAAE,0BAA+B;IAQpD,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,UAAU,CAAC;IAuB3E,mBAAmB,CAAC,SAAS,EAAE,MAAM;CA4BtC"}

package/dist/polling-with-policy.d.mts

@@ -0,0 +1,9 @@
+import { type CreateServicePolicyOptions } from "@metamask/controller-utils";
+export type RequestFn<ReturnType> = (signal: AbortSignal) => Promise<ReturnType>;
+export declare class PollingWithCockatielPolicy {
+    #private;
+    constructor(policyOptions?: CreateServicePolicyOptions);
+    start<ReturnType>(requestId: string, requestFn: RequestFn<ReturnType>): Promise<ReturnType>;
+    abortPendingRequest(requestId: string): void;
+}
+//# sourceMappingURL=polling-with-policy.d.mts.map

package/dist/polling-with-policy.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"polling-with-policy.d.mts","sourceRoot":"","sources":["../src/polling-with-policy.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,KAAK,0BAA0B,EAEhC,mCAAmC;AAGpC,MAAM,MAAM,SAAS,CAAC,UAAU,IAAI,CAClC,MAAM,EAAE,WAAW,KAChB,OAAO,CAAC,UAAU,CAAC,CAAC;AAEzB,qBAAa,0BAA0B;;gBAKzB,aAAa,GAAE,0BAA+B;IAQpD,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC,UAAU,CAAC;IAuB3E,mBAAmB,CAAC,SAAS,EAAE,MAAM;CA4BtC"}

package/dist/polling-with-policy.mjs

@@ -0,0 +1,72 @@
+var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
+    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
+};
+var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
+    if (kind === "m") throw new TypeError("Private method is not writable");
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
+    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
+};
+var _PollingWithCockatielPolicy_instances, _PollingWithCockatielPolicy_policy, _PollingWithCockatielPolicy_requestEntry, _PollingWithCockatielPolicy_addNewRequestEntry, _PollingWithCockatielPolicy_cleanup, _PollingWithCockatielPolicy_shouldRetry;
+import { createServicePolicy, HttpError } from "@metamask/controller-utils";
+import { handleWhen } from "cockatiel";
+export class PollingWithCockatielPolicy {
+    constructor(policyOptions = {}) {
+        _PollingWithCockatielPolicy_instances.add(this);
+        _PollingWithCockatielPolicy_policy.set(this, void 0);
+        _PollingWithCockatielPolicy_requestEntry.set(this, new Map());
+        const retryFilterPolicy = handleWhen(__classPrivateFieldGet(this, _PollingWithCockatielPolicy_instances, "m", _PollingWithCockatielPolicy_shouldRetry));
+        __classPrivateFieldSet(this, _PollingWithCockatielPolicy_policy, createServicePolicy({
+            ...policyOptions,
+            retryFilterPolicy,
+        }), "f");
+    }
+    async start(requestId, requestFn) {
+        const abortController = __classPrivateFieldGet(this, _PollingWithCockatielPolicy_instances, "m", _PollingWithCockatielPolicy_addNewRequestEntry).call(this, requestId);
+        try {
+            const result = await __classPrivateFieldGet(this, _PollingWithCockatielPolicy_policy, "f").execute(async ({ signal }) => {
+                return requestFn(signal);
+            }, abortController.signal);
+            return result;
+        }
+        catch (error) {
+            if (abortController.signal.aborted) {
+                throw new Error('Request cancelled');
+            }
+            throw error;
+        }
+        finally {
+            // Only cleanup if this abort controller is still active. If a new request with the same
+            // requestId started while this one was running, it would have replaced with a new abort controller.
+            // We must not delete the new request's controller when this older request finishes.
+            if (abortController === __classPrivateFieldGet(this, _PollingWithCockatielPolicy_requestEntry, "f").get(requestId)) {
+                __classPrivateFieldGet(this, _PollingWithCockatielPolicy_instances, "m", _PollingWithCockatielPolicy_cleanup).call(this, requestId);
+            }
+        }
+    }
+    abortPendingRequest(requestId) {
+        const abortController = __classPrivateFieldGet(this, _PollingWithCockatielPolicy_requestEntry, "f").get(requestId);
+        abortController?.abort();
+        __classPrivateFieldGet(this, _PollingWithCockatielPolicy_instances, "m", _PollingWithCockatielPolicy_cleanup).call(this, requestId);
+    }
+}
+_PollingWithCockatielPolicy_policy = new WeakMap(), _PollingWithCockatielPolicy_requestEntry = new WeakMap(), _PollingWithCockatielPolicy_instances = new WeakSet(), _PollingWithCockatielPolicy_addNewRequestEntry = function _PollingWithCockatielPolicy_addNewRequestEntry(requestId) {
+    // abort the previous request if it exists
+    this.abortPendingRequest(requestId);
+    // create a new abort controller for the new request
+    const abortController = new AbortController();
+    __classPrivateFieldGet(this, _PollingWithCockatielPolicy_requestEntry, "f").set(requestId, abortController);
+    return abortController;
+}, _PollingWithCockatielPolicy_cleanup = function _PollingWithCockatielPolicy_cleanup(requestId) {
+    __classPrivateFieldGet(this, _PollingWithCockatielPolicy_requestEntry, "f").delete(requestId);
+}, _PollingWithCockatielPolicy_shouldRetry = function _PollingWithCockatielPolicy_shouldRetry(error) {
+    if (error instanceof HttpError) {
+        // Note: we don't retry on 5xx errors, only on 4xx errors.
+        // but we won't retry on 400 coz it means that the request body is invalid.
+        return error.httpStatus > 400 && error.httpStatus < 500;
+    }
+    return false;
+};
+//# sourceMappingURL=polling-with-policy.mjs.map

package/dist/polling-with-policy.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"polling-with-policy.mjs","sourceRoot":"","sources":["../src/polling-with-policy.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,OAAO,EACL,mBAAmB,EACnB,SAAS,EAGV,mCAAmC;AACpC,OAAO,EAAE,UAAU,EAAE,kBAAkB;AAMvC,MAAM,OAAO,0BAA0B;IAKrC,YAAY,gBAA4C,EAAE;;QAJjD,qDAAuB;QAEvB,mDAAgB,IAAI,GAAG,EAA2B,EAAC;QAG1D,MAAM,iBAAiB,GAAG,UAAU,CAAC,uBAAA,IAAI,sFAAa,CAAC,CAAC;QACxD,uBAAA,IAAI,sCAAW,mBAAmB,CAAC;YACjC,GAAG,aAAa;YAChB,iBAAiB;SAClB,CAAC,MAAA,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK,CAAa,SAAiB,EAAE,SAAgC;QACzE,MAAM,eAAe,GAAG,uBAAA,IAAI,6FAAoB,MAAxB,IAAI,EAAqB,SAAS,CAAC,CAAC;QAE5D,IAAI;YACF,MAAM,MAAM,GAAG,MAAM,uBAAA,IAAI,0CAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;gBAC7D,OAAO,SAAS,CAAC,MAAM,CAAC,CAAC;YAC3B,CAAC,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC;YAC3B,OAAO,MAAM,CAAC;SACf;QAAC,OAAO,KAAK,EAAE;YACd,IAAI,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE;gBAClC,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;aACtC;YACD,MAAM,KAAK,CAAC;SACb;gBAAS;YACR,wFAAwF;YACxF,oGAAoG;YACpG,oFAAoF;YACpF,IAAI,eAAe,KAAK,uBAAA,IAAI,gDAAc,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE;gBACzD,uBAAA,IAAI,kFAAS,MAAb,IAAI,EAAU,SAAS,CAAC,CAAC;aAC1B;SACF;IACH,CAAC;IAED,mBAAmB,CAAC,SAAiB;QACnC,MAAM,eAAe,GAAG,uBAAA,IAAI,gDAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC1D,eAAe,EAAE,KAAK,EAAE,CAAC;QACzB,uBAAA,IAAI,kFAAS,MAAb,IAAI,EAAU,SAAS,CAAC,CAAC;IAC3B,CAAC;CAwBF;8QAtBqB,SAAiB;IACnC,0CAA0C;IAC1C,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;IAEpC,oDAAoD;IACpD,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;IAC9C,uBAAA,IAAI,gDAAc,CAAC,GAAG,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;IACnD,OAAO,eAAe,CAAC;AACzB,CAAC,qFAEQ,SAAiB;IACxB,uBAAA,IAAI,gDAAc,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;AACvC,CAAC,6FAEY,KAAY;IACvB,IAAI,KAAK,YAAY,SAAS,EAAE;QAC9B,0DAA0D;QAC1D,2EAA2E;QAC3E,OAAO,KAAK,CAAC,UAAU,GAAG,GAAG,IAAI,KAAK,CAAC,UAAU,GAAG,GAAG,CAAC;KACzD;IACD,OAAO,KAAK,CAAC;AACf,CAAC","sourcesContent":["import {\n  createServicePolicy,\n  HttpError,\n  type CreateServicePolicyOptions,\n  type ServicePolicy,\n} from '@metamask/controller-utils';\nimport { handleWhen } from 'cockatiel';\n\nexport type RequestFn<ReturnType> = (\n  signal: AbortSignal,\n) => Promise<ReturnType>;\n\nexport class PollingWithCockatielPolicy {\n  readonly #policy: ServicePolicy;\n\n  readonly #requestEntry = new Map<string, AbortController>();\n\n  constructor(policyOptions: CreateServicePolicyOptions = {}) {\n    const retryFilterPolicy = handleWhen(this.#shouldRetry);\n    this.#policy = createServicePolicy({\n      ...policyOptions,\n      retryFilterPolicy,\n    });\n  }\n\n  async start<ReturnType>(requestId: string, requestFn: RequestFn<ReturnType>) {\n    const abortController = this.#addNewRequestEntry(requestId);\n\n    try {\n      const result = await this.#policy.execute(async ({ signal }) => {\n        return requestFn(signal);\n      }, abortController.signal);\n      return result;\n    } catch (error) {\n      if (abortController.signal.aborted) {\n        throw new Error('Request cancelled');\n      }\n      throw error;\n    } finally {\n      // Only cleanup if this abort controller is still active. If a new request with the same\n      // requestId started while this one was running, it would have replaced with a new abort controller.\n      // We must not delete the new request's controller when this older request finishes.\n      if (abortController === this.#requestEntry.get(requestId)) {\n        this.#cleanup(requestId);\n      }\n    }\n  }\n\n  abortPendingRequest(requestId: string) {\n    const abortController = this.#requestEntry.get(requestId);\n    abortController?.abort();\n    this.#cleanup(requestId);\n  }\n\n  #addNewRequestEntry(requestId: string) {\n    // abort the previous request if it exists\n    this.abortPendingRequest(requestId);\n\n    // create a new abort controller for the new request\n    const abortController = new AbortController();\n    this.#requestEntry.set(requestId, abortController);\n    return abortController;\n  }\n\n  #cleanup(requestId: string) {\n    this.#requestEntry.delete(requestId);\n  }\n\n  #shouldRetry(error: Error): boolean {\n    if (error instanceof HttpError) {\n      // Note: we don't retry on 5xx errors, only on 4xx errors.\n      // but we won't retry on 400 coz it means that the request body is invalid.\n      return error.httpStatus > 400 && error.httpStatus < 500;\n    }\n    return false;\n  }\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@metamask/shield-controller",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "description": "Controller handling shield transaction coverage logic",
   "keywords": [
     "MetaMask",
@@ -48,8 +48,10 @@
   },
   "dependencies": {
     "@metamask/base-controller": "^9.0.0",
+    "@metamask/controller-utils": "^11.14.1",
     "@metamask/messenger": "^0.3.0",
-    "@metamask/utils": "^11.8.1"
+    "@metamask/utils": "^11.8.1",
+    "cockatiel": "^3.1.2"
   },
   "devDependencies": {
     "@babel/runtime": "^7.23.9",