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(dstdir)
+    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/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)

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():