Backport webrecorder/wabac.js@588954f

Author: benoit74

src/zimscraperlib/rewriting/js.py

@@ -162,7 +162,7 @@ def create_js_rules() -> list[TransformationRule]:
         (re.compile(r"\.postMessage\b\("), add_prefix(".__WB_pmw(self)")),
         # rewriting `location = ` to custom expression `(...).href =` assignement
         (
-            re.compile(r"(?:^|[^$.+*/%^-])\s?\blocation\b\s*[=]\s*(?![\s\d=])"),
+            re.compile(r"(?:^|[^$.+*/%^-])\s?\blocation\b\s*[=]\s*(?![\s\d=>])"),
             add_suffix_non_prop(check_loc),
         ),
         # rewriting `return this`

tests/rewriting/test_js_rewriting.py

@@ -186,6 +186,10 @@ def wrap_script(text: str) -> str:
             "self.__WB_check_loc(location, argument"
             "s)) || {}).href = http://example.com/",
         ),
+        WrappedTestContent(
+            input_='location => "http://example.com/"',
+            expected='location => "http://example.com/"',
+        ),
         WrappedTestContent(
             input_=" location = http://example.com/2",
             expected=" location = ((self.__WB_check_loc && "
@@ -220,6 +224,24 @@ def wrap_script(text: str) -> str:
             expected="if (self.foo) { console.log('blah') }",
         ),
         WrappedTestContent(input_="window.x = 5", expected="window.x = 5"),
+        WrappedTestContent(
+            input_="""
+            class A {}
+            const B = 5;
+            let C = 4;
+            var D = 3;
+
+            location = "http://example.com/2" """,
+            expected="""
+            class A {}
+            const B = 5;
+            let C = 4;
+            var D = 3;
+
+            location = ((self.__WB_check_loc && """
+            "self.__WB_check_loc(location, arguments)) || {}).href "
+            """= "http://example.com/2" """,
+        ),
         WrappedTestContent(input_=" var    self  ", expected=" let    self  "),
     ]
 )