@metamask/shield-controller 0.4.0 → 1.1.0

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/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAeA,2DAG4B;AAF1B,oHAAA,gBAAgB,OAAA;AAChB,mIAAA,+BAA+B,OAAA;AAEjC,yCAA6E;AAApE,8GAAA,mBAAmB,OAAA;AAAE,sHAAA,2BAA2B,OAAA","sourcesContent":["export type {\n  CoverageStatus,\n  LogSignatureRequest,\n  LogTransactionRequest,\n  NormalizeSignatureRequestFn,\n} from './types';\nexport type {\n  ShieldControllerActions,\n  ShieldControllerEvents,\n  ShieldControllerMessenger,\n  ShieldControllerState,\n  ShieldControllerCheckCoverageAction,\n  ShieldControllerCoverageResultReceivedEvent,\n  ShieldControllerStateChangeEvent,\n} from './ShieldController';\nexport {\n  ShieldController,\n  getDefaultShieldControllerState,\n} from './ShieldController';\nexport { ShieldRemoteBackend, parseSignatureRequestMethod } from './backend';\n"]}
+{"version":3,"file":"index.cjs","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAgBA,2DAG4B;AAF1B,oHAAA,gBAAgB,OAAA;AAChB,mIAAA,+BAA+B,OAAA;AAEjC,yCAA6E;AAApE,8GAAA,mBAAmB,OAAA;AAAE,sHAAA,2BAA2B,OAAA","sourcesContent":["export type {\n  CoverageStatus,\n  LogSignatureRequest,\n  LogTransactionRequest,\n  NormalizeSignatureRequestFn,\n} from './types';\nexport type {\n  ShieldControllerActions,\n  ShieldControllerEvents,\n  ShieldControllerMessenger,\n  ShieldControllerState,\n  ShieldControllerGetStateAction,\n  ShieldControllerCheckCoverageAction,\n  ShieldControllerCoverageResultReceivedEvent,\n  ShieldControllerStateChangeEvent,\n} from './ShieldController';\nexport {\n  ShieldController,\n  getDefaultShieldControllerState,\n} from './ShieldController';\nexport { ShieldRemoteBackend, parseSignatureRequestMethod } from './backend';\n"]}

package/dist/index.d.cts

@@ -1,5 +1,5 @@
 export type { CoverageStatus, LogSignatureRequest, LogTransactionRequest, NormalizeSignatureRequestFn, } from "./types.cjs";
-export type { ShieldControllerActions, ShieldControllerEvents, ShieldControllerMessenger, ShieldControllerState, ShieldControllerCheckCoverageAction, ShieldControllerCoverageResultReceivedEvent, ShieldControllerStateChangeEvent, } from "./ShieldController.cjs";
+export type { ShieldControllerActions, ShieldControllerEvents, ShieldControllerMessenger, ShieldControllerState, ShieldControllerGetStateAction, ShieldControllerCheckCoverageAction, ShieldControllerCoverageResultReceivedEvent, ShieldControllerStateChangeEvent, } from "./ShieldController.cjs";
 export { ShieldController, getDefaultShieldControllerState, } from "./ShieldController.cjs";
 export { ShieldRemoteBackend, parseSignatureRequestMethod } from "./backend.cjs";
 //# sourceMappingURL=index.d.cts.map

package/dist/index.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.cts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,YAAY,EACV,cAAc,EACd,mBAAmB,EACnB,qBAAqB,EACrB,2BAA2B,GAC5B,oBAAgB;AACjB,YAAY,EACV,uBAAuB,EACvB,sBAAsB,EACtB,yBAAyB,EACzB,qBAAqB,EACrB,mCAAmC,EACnC,2CAA2C,EAC3C,gCAAgC,GACjC,+BAA2B;AAC5B,OAAO,EACL,gBAAgB,EAChB,+BAA+B,GAChC,+BAA2B;AAC5B,OAAO,EAAE,mBAAmB,EAAE,2BAA2B,EAAE,sBAAkB"}
+{"version":3,"file":"index.d.cts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,YAAY,EACV,cAAc,EACd,mBAAmB,EACnB,qBAAqB,EACrB,2BAA2B,GAC5B,oBAAgB;AACjB,YAAY,EACV,uBAAuB,EACvB,sBAAsB,EACtB,yBAAyB,EACzB,qBAAqB,EACrB,8BAA8B,EAC9B,mCAAmC,EACnC,2CAA2C,EAC3C,gCAAgC,GACjC,+BAA2B;AAC5B,OAAO,EACL,gBAAgB,EAChB,+BAA+B,GAChC,+BAA2B;AAC5B,OAAO,EAAE,mBAAmB,EAAE,2BAA2B,EAAE,sBAAkB"}

package/dist/index.d.mts

@@ -1,5 +1,5 @@
 export type { CoverageStatus, LogSignatureRequest, LogTransactionRequest, NormalizeSignatureRequestFn, } from "./types.mjs";
-export type { ShieldControllerActions, ShieldControllerEvents, ShieldControllerMessenger, ShieldControllerState, ShieldControllerCheckCoverageAction, ShieldControllerCoverageResultReceivedEvent, ShieldControllerStateChangeEvent, } from "./ShieldController.mjs";
+export type { ShieldControllerActions, ShieldControllerEvents, ShieldControllerMessenger, ShieldControllerState, ShieldControllerGetStateAction, ShieldControllerCheckCoverageAction, ShieldControllerCoverageResultReceivedEvent, ShieldControllerStateChangeEvent, } from "./ShieldController.mjs";
 export { ShieldController, getDefaultShieldControllerState, } from "./ShieldController.mjs";
 export { ShieldRemoteBackend, parseSignatureRequestMethod } from "./backend.mjs";
 //# sourceMappingURL=index.d.mts.map

package/dist/index.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.mts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,YAAY,EACV,cAAc,EACd,mBAAmB,EACnB,qBAAqB,EACrB,2BAA2B,GAC5B,oBAAgB;AACjB,YAAY,EACV,uBAAuB,EACvB,sBAAsB,EACtB,yBAAyB,EACzB,qBAAqB,EACrB,mCAAmC,EACnC,2CAA2C,EAC3C,gCAAgC,GACjC,+BAA2B;AAC5B,OAAO,EACL,gBAAgB,EAChB,+BAA+B,GAChC,+BAA2B;AAC5B,OAAO,EAAE,mBAAmB,EAAE,2BAA2B,EAAE,sBAAkB"}
+{"version":3,"file":"index.d.mts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,YAAY,EACV,cAAc,EACd,mBAAmB,EACnB,qBAAqB,EACrB,2BAA2B,GAC5B,oBAAgB;AACjB,YAAY,EACV,uBAAuB,EACvB,sBAAsB,EACtB,yBAAyB,EACzB,qBAAqB,EACrB,8BAA8B,EAC9B,mCAAmC,EACnC,2CAA2C,EAC3C,gCAAgC,GACjC,+BAA2B;AAC5B,OAAO,EACL,gBAAgB,EAChB,+BAA+B,GAChC,+BAA2B;AAC5B,OAAO,EAAE,mBAAmB,EAAE,2BAA2B,EAAE,sBAAkB"}

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAeA,OAAO,EACL,gBAAgB,EAChB,+BAA+B,EAChC,+BAA2B;AAC5B,OAAO,EAAE,mBAAmB,EAAE,2BAA2B,EAAE,sBAAkB","sourcesContent":["export type {\n  CoverageStatus,\n  LogSignatureRequest,\n  LogTransactionRequest,\n  NormalizeSignatureRequestFn,\n} from './types';\nexport type {\n  ShieldControllerActions,\n  ShieldControllerEvents,\n  ShieldControllerMessenger,\n  ShieldControllerState,\n  ShieldControllerCheckCoverageAction,\n  ShieldControllerCoverageResultReceivedEvent,\n  ShieldControllerStateChangeEvent,\n} from './ShieldController';\nexport {\n  ShieldController,\n  getDefaultShieldControllerState,\n} from './ShieldController';\nexport { ShieldRemoteBackend, parseSignatureRequestMethod } from './backend';\n"]}
+{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAgBA,OAAO,EACL,gBAAgB,EAChB,+BAA+B,EAChC,+BAA2B;AAC5B,OAAO,EAAE,mBAAmB,EAAE,2BAA2B,EAAE,sBAAkB","sourcesContent":["export type {\n  CoverageStatus,\n  LogSignatureRequest,\n  LogTransactionRequest,\n  NormalizeSignatureRequestFn,\n} from './types';\nexport type {\n  ShieldControllerActions,\n  ShieldControllerEvents,\n  ShieldControllerMessenger,\n  ShieldControllerState,\n  ShieldControllerGetStateAction,\n  ShieldControllerCheckCoverageAction,\n  ShieldControllerCoverageResultReceivedEvent,\n  ShieldControllerStateChangeEvent,\n} from './ShieldController';\nexport {\n  ShieldController,\n  getDefaultShieldControllerState,\n} from './ShieldController';\nexport { ShieldRemoteBackend, parseSignatureRequestMethod } from './backend';\n"]}