feat: Added support for globals

Author: jonathanharg

binaryen/__global.py

@@ -0,0 +1,19 @@
+from ._binaryen import lib
+from .__expression import Expression
+
+
+class Global:
+    def __init__(self, ref):
+        self.ref = ref
+
+    def get_name(self):
+        return str(lib.BinaryenGlobalGetName(self.ref))
+
+    def get_type(self):
+        return lib.BinaryenGlobalGetType(self.ref)
+
+    def is_mutable(self):
+        return bool(lib.BinaryenGlobalIsMutable(self.ref))
+
+    def get_init_expr(self):
+        return Expression(lib.BinaryenGlobalGetInitExpr(self.ref))

binaryen/__module.py

@@ -4,11 +4,11 @@
 
 from . import literal
 from .__expression import Block, Expression
+from .__global import Global
 from .__feature import Feature
 from .__functionref import FunctionRef
 from ._binaryen import ffi, lib
 from .internals import (
-    BinaryenGlobalRef,
     BinaryenHeapType,
     BinaryenLiteral,
     BinaryenOp,
@@ -385,11 +385,25 @@ def add_function_export(
 
     def add_global(
         self, name: bytes, global_type: BinaryenType, mutable: bool, init: Expression
-    ) -> BinaryenGlobalRef:
+    ):
         ref = lib.BinaryenAddGlobal(self.ref, name, global_type, mutable, init.ref)
-        return ref
+        return Global(ref)
+
+    def get_global(self, name: bytes):
+        ref = lib.BinaryenGetGlobal(self.ref, name)
+        if ref == ffi.NULL:
+            return None
+        return Global(ref)
+
+    def remove_global(self, name: bytes):
+        lib.BinaryenRemoveGlobal(self.ref, name)
+
+    def get_num_globals(self):
+        return int(lib.BinaryenGetNumGlobals(self.ref))
 
-    # TODO: GetGlobal, RemoveGlobal, GetNumGlobals, GetGlobalByIndex
+    def get_global_by_index(self, index: int):
+        ref = lib.BinaryenGetGlobalByIndex(self.ref, index)
+        return Global(ref)
 
     # TODO: AddTag, GetTag, RemoveTag
 

binaryen/internals.py

@@ -22,4 +22,3 @@ class __BaseType:
 BinaryenExternalKind = __NewType("BinaryenExternalKind", __BaseType)
 BinaryenOp = __NewType("BinaryenOp", __BaseType)
 BinaryenLiteral = __NewType("BinaryenLiteral", __BaseType)
-BinaryenGlobalRef = __NewType("BinaryenGlobalRef", __BaseType)