use await instead of then

AdguardTeam · May 11, 2023 · 5fb4979 · 5fb4979
1 parent 4c4046c
commit 5fb4979
Showing 1 changed file with 9 additions and 11 deletions.
diff --git a/src/scriptlets/prevent-fetch.js b/src/scriptlets/prevent-fetch.js
@@ -132,7 +132,7 @@ export function preventFetch(source, propsToMatch, responseBody = 'emptyObj', re
         return;
     }
 
-    const handlerWrapper = (target, thisArg, args) => {
+    const handlerWrapper = async (target, thisArg, args) => {
         let shouldPrevent = false;
         const fetchData = getFetchData(args);
         if (typeof propsToMatch === 'undefined') {
@@ -145,16 +145,14 @@ export function preventFetch(source, propsToMatch, responseBody = 'emptyObj', re
 
         if (shouldPrevent) {
             hit(source);
-            return Reflect.apply(target, thisArg, args)
-                .then((origResponse) => {
-                    return modifyResponse(
-                        origResponse,
-                        {
-                            body: strResponseBody,
-                            type: responseType,
-                        },
-                    );
-                });
+            const origResponse = await Reflect.apply(target, thisArg, args);
+            return modifyResponse(
+                origResponse,
+                {
+                    body: strResponseBody,
+                    type: responseType,
+                },
+            );
         }
 
         return Reflect.apply(target, thisArg, args);