Define a Function class.

Author: PetterS

quickjs/__init__.py

@@ -1,3 +1,5 @@
+import json
+
 import _quickjs
 
 
@@ -8,3 +10,16 @@ def test():
 Context = _quickjs.Context
 Object = _quickjs.Object
 JSException = _quickjs.JSException
+
+
+class Function:
+    def __init__(self, name: str, code: str) -> None:
+        self.context = Context()
+        self.context.eval(code)
+        self.f = self.context.get(name)
+
+    def __call__(self, *args):
+        result = self.f(*args)
+        if isinstance(result, Object):
+            result = json.loads(result.json())
+        return result

test_quickjs.py

@@ -153,3 +153,15 @@ def test_function_call_unsupported_arg(self):
     def test_json(self):
         d = self.context.eval("d = {data: 42};")
         self.assertEqual(json.loads(d.json()), {"data": 42})
+
+
+class FunctionTest(unittest.TestCase):
+    def test_adder(self):
+        f = quickjs.Function("adder", """
+            function adder(x, y) {
+                return x + y;
+            }
+            """)
+        self.assertEqual(f(1, 1), 2)
+        self.assertEqual(f(100, 200), 300)
+        self.assertEqual(f("a", "b"), "ab")