llvm.py

import subprocess
import os
import sys
import argparse


def install_llvm(cwd, install_dir=None, release="Debug"):
    # install build dependencies
    subprocess.run(["sudo", "apt", "update"])
    subprocess.run(["sudo", "apt", "install", "-y", "cmake", "python3-yaml", "ninja-build"], check=True)

    # clone the git project
    subprocess.run(["git", "clone", "--depth=1", "--branch", "main", "https://github.com/llvm/llvm-project.git"], check=True)

    # setup and configure
    os.chdir("./llvm-project")
    subprocess.run(["ls", "-al"])
    if install_dir == None:
        subprocess.run(
            [
                "cmake",
                "-G",
                "Ninja",
                "-S",
                "llvm",
                "-B",
                "build",
                f"-DCMAKE_BUILD_TYPE={release.capitalize()}",
            ]
        )
    else:
        subprocess.run(
            [
                "cmake",
                "-G",
                "Ninja",
                "-S",
                "llvm",
                "-B",
                "build",
                f"-DCMAKE_INSTALL_PREFIX={install_dir}",
                "-DCMAKE_BUILD_TYPE=Debug"
            ],
            check=True,
        )

    # build
    # os.chdir("build")
    # subprocess.run(["make", "-j8"])
    # subprocess.run(["make", "install"])
    # Build using Ninja (automatically uses all available cores)
    subprocess.run(["ninja", "-C", "build"], check=True)

    # Install
    subprocess.run(["ninja", "-C", "build", "install"], check=True)

    # resetting the path
    os.chdir(cwd)


if __name__ == "__main__":
    parser = argparse.ArgumentParser("llvm.py", "llvm build script")
    parser.add_argument(
        "-o", "--outpath", type=str, help="the install path", default=None
    )
    parser.add_argument(
        "-t",
        "--type",
        type=str,
        help="the type of release mode by default it is RELEASE [DEBUG, RELEASE options]",
        default="RELEASE",
    )
    args = parser.parse_args()
    cwd = os.getcwd()
    install_llvm(cwd, args.outpath, args.type)