88 lines
2.4 KiB
Python
88 lines
2.4 KiB
Python
|
import os, shutil, glob
|
||
|
|
import lib.cmd_utils as cmd_utils
|
||
|
|
import lib.env as env
|
||
|
|
|
||
|
|
dist_dir = "dist"
|
||
|
|
build_dir = "build"
|
||
|
|
|
||
|
|
|
||
|
|
def package(filename_base, build_distro=True, build_tgz=False, build_stgz=False):
|
||
|
|
|
||
|
|
extension, cmd = get_package_info(build_distro, build_tgz, build_stgz)
|
||
|
|
run_package_cmd(cmd)
|
||
|
|
package_filename = get_package_filename(extension)
|
||
|
|
target_file = f"{filename_base}.{extension}"
|
||
|
|
copy_to_dist_dir(package_filename, target_file)
|
||
|
|
|
||
|
|
|
||
|
|
def get_package_info(build_distro, build_tgz, build_stgz):
|
||
|
|
|
||
|
|
command = None
|
||
|
|
cpack_generator = None
|
||
|
|
file_extension = None
|
||
|
|
|
||
|
|
if build_tgz:
|
||
|
|
cpack_generator = "TGZ"
|
||
|
|
file_extension = "tar.gz"
|
||
|
|
|
||
|
|
elif build_stgz:
|
||
|
|
cpack_generator = "STGZ"
|
||
|
|
file_extension = "sh"
|
||
|
|
|
||
|
|
elif build_distro:
|
||
|
|
|
||
|
|
distro, distro_like, _distro_version = env.get_linux_distro()
|
||
|
|
if not distro_like:
|
||
|
|
distro_like = distro
|
||
|
|
|
||
|
|
if "debian" in distro_like:
|
||
|
|
cpack_generator = "DEB"
|
||
|
|
file_extension = "deb"
|
||
|
|
elif "fedora" in distro_like or "opensuse" in distro_like:
|
||
|
|
cpack_generator = "RPM"
|
||
|
|
file_extension = "rpm"
|
||
|
|
elif "arch" in distro_like:
|
||
|
|
command = ["makepkg", "-s"]
|
||
|
|
file_extension = "pkg.tar.zst"
|
||
|
|
else:
|
||
|
|
raise RuntimeError(f"Linux distro not yet supported: {distro_like}")
|
||
|
|
|
||
|
|
if not cpack_generator and not command:
|
||
|
|
raise RuntimeError("No package generator or command found")
|
||
|
|
|
||
|
|
if cpack_generator:
|
||
|
|
command = ["cpack", "-G", cpack_generator]
|
||
|
|
|
||
|
|
return file_extension, command
|
||
|
|
|
||
|
|
|
||
|
|
def run_package_cmd(command):
|
||
|
|
package_user = env.get_env("LINUX_PACKAGE_USER", required=False)
|
||
|
|
if package_user:
|
||
|
|
cmd_utils.run(["sudo", "chown", "-R", package_user, "build"], check=True)
|
||
|
|
command = ["sudo", "-u", package_user] + command
|
||
|
|
|
||
|
|
cwd = os.getcwd()
|
||
|
|
try:
|
||
|
|
os.chdir("build")
|
||
|
|
cmd_utils.run(command, check=True, print_cmd=True)
|
||
|
|
finally:
|
||
|
|
os.chdir(cwd)
|
||
|
|
|
||
|
|
|
||
|
|
def get_package_filename(extension):
|
||
|
|
files = glob.glob(f"build/*.{extension}")
|
||
|
|
|
||
|
|
if not files:
|
||
|
|
raise ValueError(f"No .{extension} file found in build directory")
|
||
|
|
|
||
|
|
return files[0]
|
||
|
|
|
||
|
|
|
||
|
|
def copy_to_dist_dir(source_file, target_file):
|
||
|
|
os.makedirs(dist_dir, exist_ok=True)
|
||
|
|
|
||
|
|
target_path = f"{dist_dir}/{target_file}"
|
||
|
|
print(f"Copying to: {target_path}")
|
||
|
|
shutil.copy(source_file, target_path)
|