Typed array support

Author: agg23

playwright/_impl/_js_handle.py

@@ -12,9 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import base64
 import collections.abc
 import datetime
 import math
+import struct
 import traceback
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
@@ -260,6 +262,56 @@ def parse_value(value: Any, refs: Optional[Dict[int, Any]] = None) -> Any:
 
         if "b" in value:
             return value["b"]
+
+        if "ta" in value:
+            encoded_bytes = value["ta"]["b"]
+            decoded_bytes = base64.b64decode(encoded_bytes)
+            array_type = value["ta"]["k"]
+            if array_type == "i8":
+                word_size = 1
+                fmt = "b"
+            elif array_type == "ui8" or array_type == "ui8c":
+                word_size = 1
+                fmt = "B"
+            elif array_type == "i16":
+                word_size = 2
+                fmt = "h"
+            elif array_type == "ui16":
+                word_size = 2
+                fmt = "H"
+            elif array_type == "i32":
+                word_size = 4
+                fmt = "i"
+            elif array_type == "ui32":
+                word_size = 4
+                fmt = "I"
+            elif array_type == "f32":
+                word_size = 4
+                fmt = "f"
+            elif array_type == "f64":
+                word_size = 8
+                fmt = "d"
+            elif array_type == "bi64":
+                word_size = 8
+                fmt = "q"
+            elif array_type == "bui64":
+                word_size = 8
+                fmt = "Q"
+            else:
+                raise ValueError(f"Unsupported array type: {array_type}")
+
+            byte_len = len(decoded_bytes)
+            if byte_len % word_size != 0:
+                raise ValueError(
+                    f"Decoded bytes length {byte_len} is not a multiple of word size {word_size}"
+                )
+
+            if byte_len == 0:
+                return []
+            array_len = byte_len // word_size
+            # "<" denotes little-endian
+            format_string = f"<{array_len}{fmt}"
+            return list(struct.unpack(format_string, decoded_bytes))
     return value
 
 

tests/async/test_page_evaluate.py

@@ -65,6 +65,29 @@ async def test_evaluate_transfer_arrays(page: Page) -> None:
     assert result == [1, 2, 3]
 
 
+async def test_evaluate_transfer_typed_arrays(page: Page) -> None:
+    async def test_typed_array(
+        typed_array: str, expected: list[float], value_suffix: Optional[str]
+    ) -> None:
+        value_suffix = "" if value_suffix is None else value_suffix
+        result = await page.evaluate(
+            f"() => new {typed_array}([1{value_suffix}, 2{value_suffix}, 3{value_suffix}])"
+        )
+        assert result == expected
+
+    await test_typed_array("Int8Array", [1, 2, 3], None)
+    await test_typed_array("Uint8Array", [1, 2, 3], None)
+    await test_typed_array("Uint8ClampedArray", [1, 2, 3], None)
+    await test_typed_array("Int16Array", [1, 2, 3], None)
+    await test_typed_array("Uint16Array", [1, 2, 3], None)
+    await test_typed_array("Int32Array", [1, 2, 3], None)
+    await test_typed_array("Uint32Array", [1, 2, 3], None)
+    await test_typed_array("Float32Array", [1.5, 2.5, 3.5], ".5")
+    await test_typed_array("Float64Array", [1.5, 2.5, 3.5], ".5")
+    await test_typed_array("BigInt64Array", [1, 2, 3], "n")
+    await test_typed_array("BigUint64Array", [1, 2, 3], "n")
+
+
 async def test_evaluate_transfer_bigint(page: Page) -> None:
     assert await page.evaluate("() => 42n") == 42
     assert await page.evaluate("a => a", 17) == 17