restructuredpython/restructuredpython/cload.py at 5d6297791e46ea42f2fcdca79f7eb2f88457a264 · sharktide/restructuredpython · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
import importlib
import struct
import sys
import os
import ctypes
import tomllib as toml
import time
from textformat import *

def time_opening(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print(f"Loading {func.__name__} module took {end_time - start_time:.2f} seconds.")
        return result
    return wrapper

spec = importlib.util.find_spec("restructuredpython")
if spec and spec.origin:
    package_dir = os.path.dirname(spec.origin)
    io_dll = os.path.join(package_dir, "lib", "windows-libs", "io64.dll")
    io32_dll = os.path.join(package_dir, "lib", "windows-lib", "io32.dll")
    io_so = os.path.join(package_dir, "lib", "linux-libs", "io.so")
    io_dylib = os.path.join(package_dir, "lib", "macos-libs", "io.dylib")
load_s = time.perf_counter()

if sys.platform == "win32":
    if (struct.calcsize("P") * 8) == 32:
        lib = ctypes.WinDLL(io32_dll)
    else:
        lib = ctypes.WinDLL(io_dll)
elif sys.platform == "darwin":
    lib = ctypes.CDLL(io_dylib)
else:
    lib = ctypes.CDLL(io_so)

def io_s():
    lib.check_file_exists.argtypes = [ctypes.c_char_p]
    lib.check_file_exists.restype = ctypes.c_int

    lib.read_file.argtypes = [ctypes.c_char_p]
    lib.read_file.restype = ctypes.c_char_p

    lib.write_file.argtypes = [ctypes.c_char_p, ctypes.c_char_p]
    lib.write_file.restype = ctypes.c_int

    lib.read_binary_file.argtypes = [
        ctypes.c_char_p, ctypes.POINTER(
            ctypes.c_size_t)]
    lib.read_binary_file.restype = ctypes.POINTER(ctypes.c_char)

io_s()

load_e = time.perf_counter()
count = load_e - load_s
print(f"{bcolors.OKBLUE}Loading modules took {count}s{bcolors.ENDC}")


def load_toml_binary(filename):
    filename = str(filename)
    size = ctypes.c_size_t()
    raw_data_ptr = lib.read_binary_file(filename.encode(), ctypes.byref(size))

    if not raw_data_ptr:
        raise FileNotFoundError(f"{bcolors.BOLD}{bcolors.FAIL}Could not read {filename}{bcolors.ENDC}")

    raw_data = ctypes.string_at(raw_data_ptr, size.value)
    return toml.loads(raw_data.decode())


def read_file_utf8(filename: str) -> str:
    size = ctypes.c_size_t()
    filename_bytes = filename.encode('utf-8')

    ptr = lib.read_binary_file(filename_bytes, ctypes.byref(size))
    if not ptr:
        raise FileNotFoundError(f"{bcolors.BOLD}{bcolors.FAIL}File not found: {filename}{bcolors.ENDC}")

    raw_bytes = ctypes.string_at(ptr, size.value)

    try:
        text = raw_bytes.decode('utf-8')
    except UnicodeDecodeError as e:
        raise UnicodeDecodeError(f"{bcolors.BOLD}{bcolors.FAIL}{e}{bcolors.ENDC}")

    return text