feature: riftlib runtime acquisition

Author: AminRezaei0x443

README.md

@@ -5,7 +5,7 @@
 [![PyPI version](https://img.shields.io/badge/rift--framework-0.8.5-informational?style=flat-square&color=FFFF91&labelColor=360825)](https://pypi.org/project/rift-framework/)
 [![Telegram](https://img.shields.io/badge/Telegram-@skyring__org-informational?style=flat-square&color=0088cc&labelColor=360825)](https://t.me/skyring_org)
 [![Telegram](https://img.shields.io/badge/Docs-docs.skyring.io/rift-informational?style=flat-square&color=6A0F49&labelColor=360825)](https://docs.skyring.io/rift/)
-> _A magical **Python3** -> **FunC** portal_
+> _A magical **Python3** -> **TON** portal_
 
 Rift is smart contract development framework in Python for [TON (The Open Network)](https://ton.org). Its purpose is to make the development, testing, and deployment procedures much easier!
 

build.py

@@ -13,6 +13,7 @@
 from rift.fift.types.util import create_entry
 from rift.fift.utils import calc_method_id
 from rift.native import NativeLib, native_call
+from rift.runtime.riftlib import RiftLibSetup
 
 
 class FiftError(RuntimeError):
@@ -32,6 +33,7 @@ def __init__(
         load_utils=False,
     ):
         if not lib_path:
+            lib_path = RiftLibSetup.acquire_lib(ensure=True)
             p = Path(sys.modules["rift"].__file__).parent.parent
             lib_path = os.path.join(
                 p.absolute(),

rift/fift/fift.py

@@ -38,7 +38,8 @@ def __init__(self, res: TVMError | TVMResult | ExecutionError) -> None:
     def expect_ok(self) -> TVMResult:
         if self.state != State.Ok:
             raise TestError(
-                "Expected successful exit but got error!", self.curr_error(),
+                "Expected successful exit but got error!",
+                self.curr_error(),
             )
         return self.result
 

rift/fift/test_result.py

@@ -0,0 +1,108 @@
+import os
+import platform
+import subprocess
+
+import requests
+from tqdm import tqdm
+
+from rift.runtime.config import Config
+
+
+class RiftLibSetup:
+    urls = {
+        "windows": "https://github.com/AminRezaei0x443/ton/releases/download/v0.1.0/rift-lib-win64.dll",
+        "darwin": "https://github.com/AminRezaei0x443/ton/releases/download/v0.1.0/rift-lib-macOS.dylib",
+        "linux-u18": "https://github.com/AminRezaei0x443/ton/releases/download/v0.1.0/rift-lib-ubuntu-18.04.so",
+        "linux-u20": "https://github.com/AminRezaei0x443/ton/releases/download/v0.1.0/rift-lib-ubuntu-20.04.so",
+        "linux-u22": "https://github.com/AminRezaei0x443/ton/releases/download/v0.1.0/rift-lib-ubuntu-22.04.so",
+    }
+
+    @classmethod
+    def acquire_lib(cls, ensure=False) -> str:
+        if ensure:
+            cls.setup_riftlib()
+        return os.path.join(Config.dirs.user_data_dir, "_riftlib.pyd")
+
+    @classmethod
+    def is_setup(cls) -> bool:
+        return os.path.exists(cls.acquire_lib())
+
+    @classmethod
+    def _ubuntu_version(cls) -> int:
+        try:
+            if os.path.exists("/etc/lsb-release"):
+                with open("/etc/lsb-release") as f:
+                    c = f.read().strip()
+                lines = c.split("\n")
+                info = dict(map(lambda x: x.split("="), lines))
+                if info["DISTRIB_ID"].lower() == "ubuntu":
+                    return int(info["DISTRIB_RELEASE"].split(".")[0])
+            return -1
+        except Exception:
+            return -1
+
+    @classmethod
+    def determine_lib(cls) -> str:
+        s = platform.uname().system.lower()
+        if s == "windows" or s == "darwin":
+            return cls.urls[s]
+        uv = cls._ubuntu_version()
+        if uv == -1:
+            raise RuntimeError("Unsupported os")
+        return cls.urls[f"linux-u{uv}"]
+
+    @classmethod
+    def _call_get(cls, command):
+        p = subprocess.Popen(
+            command, stdout=subprocess.PIPE, stderr=subprocess.PIPE
+        )
+        out, err = p.communicate()
+        if p.returncode == 0:
+            return True, out.decode("utf-8")
+        return False, err.decode("utf-8")
+
+    @classmethod
+    def get_shared_libs(cls):
+        ok, res = cls._call_get("ldconfig -p")
+        if not ok:
+            raise RuntimeError("Error fetching shared libraries!")
+        res = filter(lambda x: "=>" in x, res.split("\n"))
+        res = map(
+            lambda x: list(map(lambda f: f.strip(), x.split("=>"))), res
+        )
+        libs = dict(res)
+        return libs
+
+    @classmethod
+    def get_glibc_version(cls):
+        ok, res = cls._call_get("ldconfig --version")
+        if not ok:
+            raise RuntimeError("Error fetching glibc version!")
+        return res.split("\n")[0].split(" ")[-1].strip()
+
+    @classmethod
+    def get_env_info(cls):
+        return {
+            "libs": cls.get_shared_libs(),
+            "glibc": cls.get_glibc_version(),
+        }
+
+    @classmethod
+    def _download(cls, addr: str, to: str, buffer=1024):
+        response = requests.get(addr, stream=True)
+        file_size = int(response.headers.get("Content-Length"))
+        pbar = tqdm(total=file_size, unit="B", unit_scale=True)
+        with open(to, "wb") as f:
+            for chunk in response.iter_content(chunk_size=buffer):
+                if chunk:  # filter out keep-alive new chunks
+                    f.write(chunk)
+                    pbar.update(buffer)
+        pbar.close()
+
+    @classmethod
+    def setup_riftlib(cls):
+        lib_path = cls.acquire_lib()
+        if cls.is_setup():
+            return
+        url = cls.determine_lib()
+        cls._download(url, lib_path, buffer=4096)

rift/runtime/riftlib.py

@@ -1,10 +1,9 @@
-from setuptools import setup, find_packages
-from build import build_kwargs
+from setuptools import find_packages, setup
 
 setup(
     name="rift-framework",
     version="0.9.5",
-    description="A magical Python3 -> FunC portal",
+    description="A magical Python3 -> TON portal",
     license="MIT",
     packages=find_packages(exclude=["test"]),
     author="Amin Rezaei",
@@ -18,5 +17,4 @@
     url="https://github.com/sky-ring/rift",
     install_requires=[],
     extras_require={},
-    **build_kwargs(),
 )

setup.py

@@ -1,6 +1,6 @@
 [flake8]
 ignore=D100, D102, N802, DUO105, DUO110, S102, Q000, F401, F403, E203, I001,
- W503, I005, E731, B010, B009, N804, D401, DAR101, I003, S101, RST301, B016, N801, E721, B023, F405, N805, DAR201, E800, DUO102, E501, S110, D200, D400
+ W503, I005, E731, B010, B009, N804, D401, DAR101, I003, S101, RST301, B016, N801, E721, B023, F405, N805, DAR201, E800, DUO102, E501, S110, D200, D400, S603, S404
 extend-ignore=WPS, D1
 exclude =
     .git,