feat: save load merkle and gio request

Author: lynoferraz

.dockerignore

@@ -1,2 +1,4 @@
 .*
 *.dockerfile
+wheelhouse
+*.cpp

.gitignore

@@ -1,6 +1,3 @@
-dist
-pycmt.egg-info
-build
 *.cpp
 .*
 !.gitignore

build_wheels.sh

@@ -1,6 +1,8 @@
 #!/bin/bash
 set -e -u -x
 
+project_path=${1:-.}
+
 function repair_wheel {
     wheel="$1"
     if ! auditwheel show "$wheel"; then
@@ -12,7 +14,7 @@ function repair_wheel {
 
 # Compile wheels
 for PYBIN in /opt/python/*/bin; do
-    "${PYBIN}/pip3" wheel /opt/build --no-deps -w wheelhouse/
+    "${PYBIN}/pip3" wheel $project_path --no-deps -w wheelhouse/
 done
 
 # Bundle external shared libraries into the wheels

libcmt.pxd

@@ -1,4 +1,4 @@
-from libc.stdint cimport uint64_t, uint8_t
+from libc.stdint cimport uint64_t, uint32_t, uint16_t, uint8_t
 
 cdef extern from "/usr/include/libcmt/io.h":
     cdef enum:
@@ -34,8 +34,13 @@ cdef extern from "/usr/include/libcmt/rollup.h":
     ctypedef struct cmt_rollup_finish_t:
         bint accept_previous_request
         int next_request_type
-    # ctypedef struct cmt_gio_t:
-    #     pass
+    ctypedef struct cmt_gio_t:
+        uint16_t domain
+        uint32_t id_length
+        void *id
+        uint16_t response_code
+        uint32_t response_data_length
+        void *response_data
 
     int cmt_rollup_init(cmt_rollup_t* me)
     int cmt_rollup_fini(cmt_rollup_t* me)
@@ -50,7 +55,7 @@ cdef extern from "/usr/include/libcmt/rollup.h":
     int cmt_rollup_read_inspect_state(cmt_rollup_t *me, cmt_rollup_inspect_t *inspect)
     int cmt_rollup_finish(cmt_rollup_t *me, cmt_rollup_finish_t *finish)
 
-    # int cmt_gio_request(cmt_rollup_t *me, cmt_gio_t *req)
-    # int cmt_rollup_load_merkle(cmt_rollup_t *me, char *path)
-    # int cmt_rollup_save_merkle(cmt_rollup_t *me, char *path)
-    # void cmt_rollup_reset_merkle(cmt_rollup_t *me)
+    int cmt_gio_request(cmt_rollup_t *me, cmt_gio_t *req)
+    int cmt_rollup_load_merkle(cmt_rollup_t *me, char *path)
+    int cmt_rollup_save_merkle(cmt_rollup_t *me, char *path)
+    void cmt_rollup_reset_merkle(cmt_rollup_t *me)

pycmt.pyx

@@ -4,7 +4,6 @@ cimport libcmt
 
 cdef class Rollup:
     cdef libcmt.cmt_rollup_t _c_rollup[1]
-    cdef libcmt.cmt_abi_address_t *app_address
 
     def __cinit__(self):
         libcmt.cmt_rollup_init(self._c_rollup)
@@ -29,7 +28,6 @@ cdef class Rollup:
         err = libcmt.cmt_rollup_read_advance_state(self._c_rollup, input)
         if err != 0:
             raise Exception(f"Failed to read advance ({err})")
-        self.app_address = &input[0].app_contract
         payload = input[0].payload.data[:input[0].payload.length]
         app_contract = input[0].app_contract.data[:libcmt.CMT_ABI_ADDRESS_LENGTH]
         msg_sender = input[0].msg_sender.data[:libcmt.CMT_ABI_ADDRESS_LENGTH]
@@ -106,3 +104,38 @@ cdef class Rollup:
         err = libcmt.cmt_rollup_emit_exception(self._c_rollup, payload_ptr)
         if err != 0:
             raise Exception(f"Failed to emit exception ({err})")
+
+    cpdef load_merkle(self, str merkle_path):
+        b_merkle_path = merkle_path.encode('UTF-8')
+        cdef char *c_merkle_path = b_merkle_path
+        err = libcmt.cmt_rollup_load_merkle(self._c_rollup, c_merkle_path)
+        if err != 0:
+            raise Exception(f"Failed to load merkle ({err})")
+
+    cpdef save_merkle(self, str merkle_path):
+        b_merkle_path = merkle_path.encode('UTF-8')
+        cdef char *c_merkle_path = b_merkle_path
+        err = libcmt.cmt_rollup_save_merkle(self._c_rollup, c_merkle_path)
+        if err != 0:
+            raise Exception(f"Failed to save merkle ({err})")
+
+    cpdef reset_merkle(self):
+        err = libcmt.cmt_rollup_reset_merkle(self._c_rollup)
+        if err != 0:
+            raise Exception(f"Failed to reset merkle ({err})")
+
+    cpdef object gio_request(self, int domain, bytes id):
+        cdef libcmt.cmt_gio_t req_ptr[1]
+        req_ptr[0].domain = domain
+        req_ptr[0].id = <void *>id
+        req_ptr[0].id_length = len(id)
+        err = libcmt.cmt_gio_request(self._c_rollup, req_ptr)
+        if err != 0:
+            raise Exception(f"Failed to request gio ({err})")
+        response = (<uint8_t *>req_ptr[0].response_data)[:req_ptr[0].response_data_length]
+        ret = {
+            "response_data_length": req_ptr[0].response_data_length,
+            "response_data": response,
+            "response_code": req_ptr[0].response_code
+        }
+        return ret

setup.py

@@ -3,12 +3,12 @@
 
 setup(
     name='pycmt',
-    version=os.getenv("VERSION", '0.0.0-dev'),
+    version=os.getenv("VERSION", '0.0.0'),
     py_modules=['pycmt'],
     ext_modules = [
         Extension("pycmt",
             sources=["pycmt.pyx"],
-            language="c++",
+            # language="c++",
             extra_objects=['/usr/lib/libcmt.a'],
         )
     ],