Backport more changes from wabac.js

Author: benoit74

src/zimscraperlib/rewriting/js.py

@@ -154,13 +154,15 @@ def create_js_rules() -> list[TransformationRule]:
     return [
         # rewriting `eval(...)` - invocation
         (re.compile(r"(?:^|\s)\beval\s*\("), replace_prefix_from(eval_str, "eval")),
+        (re.compile(r"\([\w]+,\s*eval\)\("), m2str(lambda _: f" {eval_str}")),
         # rewriting `x = eval` - no invocation
         (re.compile(r"[=]\s*\beval\b(?![(:.$])"), replace("eval", "self.eval")),
+        (re.compile(r"var\s+self"), replace("var", "let")),
         # rewriting `.postMessage` -> `__WB_pmw(self).postMessage`
         (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

@@ -101,6 +101,40 @@ def test_js_rewrite_post_message(simple_js_rewriter: JsRewriter):
     )
 
 
+@pytest.mark.parametrize(
+    "raw_js,expected",
+    [
+        pytest.param("x = eval; x(a);", "x = self.eval; x(a);", id="case1"),
+        pytest.param(
+            " eval(a)",
+            " WB_wombat_runEval2((_______eval_arg, isGlobal) => { var ge = eval; "
+            "return isGlobal ? ge(_______eval_arg) : eval(_______eval_arg); })"
+            ".eval(this, (function() { return arguments })(),a)",
+            id="case2",
+        ),
+        pytest.param(
+            "$eval = eval; $eval(a);", "$eval = self.eval; $eval(a);", id="case3"
+        ),
+        pytest.param(
+            "foo(a, eval(data));",
+            "foo(a, WB_wombat_runEval2((_______eval_arg, isGlobal) => { var ge = eval; "
+            "return isGlobal ? ge(_______eval_arg) : eval(_______eval_arg); })"
+            ".eval(this, (function() { return arguments })(),data));",
+            id="case4",
+        ),
+        pytest.param(
+            "return(1, eval)(data);",
+            "return WB_wombat_runEval2((_______eval_arg, isGlobal) => { var ge = eval; "
+            "return isGlobal ? ge(_______eval_arg) : eval(_______eval_arg); })"
+            ".eval(this, (function() { return arguments })(),data);",
+            id="case5",
+        ),
+    ],
+)
+def test_js_rewrite_evals(simple_js_rewriter: JsRewriter, raw_js: str, expected: str):
+    assert simple_js_rewriter.rewrite(raw_js) == expected
+
+
 class WrappedTestContent(ContentForTests):
 
     def __init__(
@@ -173,11 +207,20 @@ def wrap_script(text: str) -> str:
             input_="this. location = 'http://example.com/'",
             expected="this. location = 'http://example.com/'",
         ),
+        WrappedTestContent(
+            input_="abc-location = http://example.com/",
+            expected="abc-location = http://example.com/",
+        ),
+        WrappedTestContent(
+            input_="func(location = 0)",
+            expected="func(location = 0)",
+        ),
         WrappedTestContent(
             input_="if (self.foo) { console.log('blah') }",
             expected="if (self.foo) { console.log('blah') }",
         ),
         WrappedTestContent(input_="window.x = 5", expected="window.x = 5"),
+        WrappedTestContent(input_=" var    self  ", expected=" let    self  "),
     ]
 )
 def rewrite_wrapped_content(request: pytest.FixtureRequest):