feat: add hooks/autopatch

Signed-off-by: sisungo <[email protected]>

hooks/__init__.py

@@ -1,9 +1,14 @@
 from typing import Callable, TypeAlias
 
-from . import autostrip
+from . import autopatch, autostrip
 
+PreBuild: TypeAlias = Callable[[], None]
 PostBuild: TypeAlias = Callable[[], None]
 
+pre_build: dict[str, PreBuild] = {
+    "autopatch": autopatch.autopatch,
+}
+
 post_build: dict[str, PostBuild] = {
     "autostrip": autostrip.autostrip,
 }

hooks/autopatch.py

@@ -0,0 +1,29 @@
+import os
+import subprocess
+
+import lib.sources as sources
+
+
+def autopatch():
+    if not os.path.exists("patches"):
+        return
+
+    target_dir = os.getcwd() + "/target"
+    source_dir = sources.sources_dir_name()
+    patches_tag_file = f"{target_dir}/patches.tag"
+
+    patches_dir = os.getcwd() + "/patches"
+    patches = os.listdir(patches_dir)
+    patches.sort()
+
+    os.chdir(source_dir)
+    if not os.path.exists(patches_tag_file):
+        open(patches_tag_file, "+w").close()
+    for i in patches:
+        with open(patches_tag_file, "r") as file:
+            if i + "\n" in file.readlines():
+                continue
+        with open(f"{patches_dir}/{i}", "r") as file:
+            subprocess.run(["patch", "-p1"], stdin=file)
+        with open(patches_tag_file, "+a") as file:
+            file.write(i + "\n")

lib/cpp.py

@@ -21,7 +21,12 @@ def make_install(rem: list[str] | None = None, chdir: bool = True):
 
 
 def _do_auto_pack(pc: pkgcomposer.PkgComposer, srcdir: str, dstdir: str, filter):
-    pc.makedir(srcdir)
+    if not os.path.exists(srcdir):
+        return
+    try:
+        pc.makedir(dstdir)
+    except Exception:
+        pass
     with os.scandir(srcdir) as entries:
         for ent in entries:
             if filter(ent):

lib/make.py

@@ -59,3 +59,7 @@ def block_hook(hook: str):
 
 def source_url(url: str):
     sources.source_url(url)
+
+
+def source_git(url: str, branch: str | None = None):
+    sources.source_git(url, branch)

lib/rust.py

@@ -0,0 +1,60 @@
+import os
+import subprocess
+
+from . import arch, make, pkgcomposer
+
+
+def _rust_triple(semios: str) -> str:
+    return semios.replace("-semios-linux", "-unknown-linux-musl")
+
+
+def build(rem: list[str] = []):
+    cmdline = [
+        "cargo",
+        "build",
+        "--release",
+        "--target",
+        _rust_triple(arch.get_target()),
+    ] + rem
+    os.environ["RUSTFLAGS"] = (
+        os.environ.get("RUSTFLAGS", "") + " -C target-feature=-crt-static"
+    )
+    subprocess.run(cmdline)
+
+
+def _do_auto_pack(pc: pkgcomposer.PkgComposer, srcdir: str, dstdir: str, filter):
+    pc.makedir(dstdir)
+    with os.scandir(srcdir) as entries:
+        for ent in entries:
+            if filter(ent):
+                if ent.is_dir(follow_symlinks=False):
+                    pc.makedir(f"{dstdir}/{ent.name}")
+                elif ent.is_file(follow_symlinks=False):
+                    pc.addfile(ent.path, f"{dstdir}/{ent.name}")
+                elif ent.is_symlink() and not os.path.isabs(os.readlink(ent.path)):
+                    pc.addfile(ent.path, f"{dstdir}/{ent.name}")
+
+
+def _autopack_filter_lib(ent) -> bool:
+    if ent.is_dir(follow_symlinks=False):
+        return False
+    if ent.name.endswith(".so"):
+        return True
+    else:
+        return False
+
+
+def _autopack_filter_bin(ent) -> bool:
+    return os.access(ent.path, os.X_OK) and ent.is_file(follow_symlinks=False)
+
+
+def _auto_pack(profiles: list[str], pc: pkgcomposer.PkgComposer):
+    target_dir = f"target/{_rust_triple(arch.get_target())}/release"
+    if "lib" in profiles:
+        _do_auto_pack(pc, target_dir, "lib", _autopack_filter_lib)
+    if "bin" in profiles:
+        _do_auto_pack(pc, target_dir, "bin", _autopack_filter_bin)
+
+
+def use_auto_pack(profiles: list[str]):
+    make.on_pack(lambda pc: _auto_pack(profiles, pc))

lib/sources.py

@@ -1,4 +1,5 @@
 import os
+import subprocess
 import sys
 import tarfile
 import urllib.parse
@@ -109,3 +110,20 @@ def source_url(url: str):
         tar.extractall(extract_dir)
         with open("./target/sources.tag", "wt+") as file:
             file.write(sources_dir)
+
+
+def _git_repo_name(url: str) -> str:
+    return url.split("/")[-1].replace(".git", "")
+
+
+def source_git(url: str, branch: str | None = None):
+    source_path = os.getcwd() + "/target/" + _git_repo_name(url)
+    if os.path.exists("./target/sources.tag"):
+        return
+    cmdline = ["git", "clone", "--depth=1"]
+    if branch:
+        cmdline += ["-b", branch]
+    cmdline += [url, source_path]
+    subprocess.run(cmdline, check=True)
+    with open("./target/sources.tag", "wt+") as file:
+        file.write(source_path)

packages/mksh/pkgbuild.py

@@ -0,0 +1,22 @@
+import os
+
+from lib.make import *
+
+version("1.59.2")
+source_git("https://github.com/mirabilos/mksh-cvs2git.git", "mksh-R59c")
+
+
+def build():
+    os.system("sh ./Build.sh")
+
+
+on_build(build)
+
+
+def pack_mksh(composer):
+    composer.makedir("bin")
+    composer.addfile("mksh", "bin/mksh")
+
+
+package("mksh")
+on_pack(pack_mksh)

x.py

@@ -46,6 +46,13 @@ def run_build_package(package: str):
         print(f"Architecture {lib.arch.get_target()} is not supported for this package")
         exit(1)
 
+    # Run `pre_build` hooks
+    for n in hooks.pre_build.keys():
+        if n in lib.make._blocked_hooks:
+            continue
+        hooks.pre_build[n]()
+        os.chdir(package_dir)
+
     # Build the package
     if lib.make._on_build:
         os.chdir(lib.sources.sources_dir_name())
@@ -54,10 +61,12 @@ def run_build_package(package: str):
         print("on_build(): Not defined")
 
     # Run `post_build` hooks
+    post_build_dir = os.getcwd()
     for n in hooks.post_build.keys():
         if n in lib.make._blocked_hooks:
             continue
         hooks.post_build[n]()
+        os.chdir(post_build_dir)
 
     # Pack packages
     for pkgname in lib.make._pkginfo.keys():