z

#!/bin/bash

#
# Utility for building development tools for Nanvix.
#
# Run './z help' for more information on how to use this utility.
#
# Relevant Environment Variables
#
#   - Z_TARGET: Target architecture.
#   - Z_STAGE: Build stage.
#   - Z_INSTALL_LOCATION: Installation directory.
#   - Z_SYSROOT_LOCATION: Sysroot directory.
#   - Z_RELEASE_NAME: Release name.
#
# shellcheck disable=SC2329
# shellcheck disable=SC2317
# shellcheck disable=SC2034

#==================================================================================================

# Fast fail on errors, unset variables, and pipe failures.
set -euo pipefail

#==================================================================================================
# Import Utils Script
#==================================================================================================

UTILS_SCRIPT_VERSION="c24dedaf76a4833b388597805e2c0625cef0c453"
UTILS_SCRIPT_URL="https://raw.githubusercontent.com/nanvix/scripts/${UTILS_SCRIPT_VERSION}/utils.sh"

utils_script="/tmp/utils-${UTILS_SCRIPT_VERSION}.sh"

# Download only if the file doesn't exist or is empty
if [[ ! -s "$utils_script" ]]; then

    if command -v curl >/dev/null 2>&1; then
        curl -fsSL "$UTILS_SCRIPT_URL" -o "$utils_script" || {
            echo "Failed to download utils.sh"
            rm -f "$utils_script"
            exit 1
        }
    elif command -v wget >/dev/null 2>&1; then
        wget -qO "$utils_script" "$UTILS_SCRIPT_URL" || {
            echo "Failed to download utils.sh"
            rm -f "$utils_script"
            exit 1
        }
    else
        echo "curl or wget required"
        exit 1
    fi
fi

# Source the downloaded script
# shellcheck disable=SC1090
source "${utils_script}"

#==================================================================================================
# Z Constants
#==================================================================================================

# Directory containing this script (resolves symlinks).
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

# Project Name
readonly Z_PROJECT_NAME="LLVM"
# Directory used when building, empty if not required.
readonly Z_PROJECT_BUILD_DIR=""

# Required Build Tools (tool_name => minimum_version)
readonly -A Z_PROJECT_REQUIRED_TOOLS=(
    ["gcc"]="13.3.0"
    ["make"]="4.3.0"
    ["wget"]="1.21.4"
    ["bzip2"]="1.0.8"
    ["tar"]="1.34.0"
    ["cmake"]="3.28.3"
    ["gh"]="2.0.0"
)

# Required System Packages for Setup
readonly Z_PROJECT_REQUIRED_PACKAGES=("build-essential" "wget" "bzip2" "tar" "cmake")

#==================================================================================================
# Project-Specific Constants
#==================================================================================================

# Build Targets
BUILD_TARGETS=("all")

# Install Targets
INSTALL_TARGETS=("install")

# Newlib release (provides C headers and libc)
readonly NEWLIB_RELEASE_TAG="nanvix-v4.4.0-e1c9d0d4c"
readonly NEWLIB_RELEASE_PATTERN="nanvix-newlib-*.tar.bz2"

# Nanvix OS release (used to obtain libposix.a, user.ld, and nanvixd binaries)
readonly NANVIX_RELEASE_TAG="v0.13.16"
readonly NANVIX_RELEASE_PATTERN="nanvix-x86-microvm-multi-process-release-128mb-*.tar.bz2"

#==================================================================================================
# Project-Specific Functions
#==================================================================================================

#
# Description
#
#   Executes project-specific steps for downloading dependencies.
#
# Return Value
#
#   - On success, this function returns zero.
#   - On failure, this function returns non-zero.
#
download_steps() {
    # Stage 0 (compiler only): no sysroot needed.
    if [[ "${Z_STAGE}" == "0" ]]; then
        print_info "Stage 0: no dependencies to download."
        return 0
    fi

    # Stage 1 (runtimes): sysroot must already exist.
    if [[ "${Z_STAGE}" == "1" ]]; then
        local sysroot_location="${Z_SYSROOT_LOCATION:-${Z_INSTALL_LOCATION}}"
        if [[ ! -d "${sysroot_location}" ]]; then
            print_error "Stage 1 requires a sysroot at '${sysroot_location}'."
            return 1
        fi

        # If C headers are missing, download newlib as a bootstrap sysroot.
        if [[ ! -f "${sysroot_location}/usr/include/stdlib.h" ]]; then
            print_info "Stage 1: C headers not found; downloading newlib..."

            local tmp_dir
            tmp_dir="$(mktemp -d)"

            # Download and extract newlib (C headers and libc).
            gh release download "${NEWLIB_RELEASE_TAG}" --repo nanvix/newlib \
                --pattern "${NEWLIB_RELEASE_PATTERN}" \
                --dir "$tmp_dir" || {
                print_error "Failed to download newlib release."
                rm -rf "$tmp_dir"
                return 1
            }
            local newlib_archive
            newlib_archive=$(find "$tmp_dir" -maxdepth 1 -name "${NEWLIB_RELEASE_PATTERN}" | head -1)
            if [[ -z "$newlib_archive" ]]; then
                print_error "No newlib archive found after download."
                rm -rf "$tmp_dir"
                return 1
            fi
            extract_tar_bz2 "$newlib_archive" "${sysroot_location}" || {
                print_error "Failed to extract newlib."
                rm -rf "$tmp_dir"
                return 1
            }
            rm -rf "$tmp_dir"

            if [[ ! -f "${sysroot_location}/usr/include/stdlib.h" ]]; then
                print_error "Stage 1: C headers still missing after extracting newlib."
                return 1
            fi
            print_info "Stage 1: newlib downloaded to '${sysroot_location}'."
        else
            print_info "Stage 1: C headers found at '${sysroot_location}'."
        fi

        # Download Nanvix release for libposix.a and user.ld if not already present.
        local install_location="${Z_INSTALL_LOCATION}"
        if [[ ! -f "${install_location}/lib/libposix.a" ]] || \
        [[ ! -f "${install_location}/lib/user.ld" ]]; then
            echo "Downloading Nanvix release ${NANVIX_RELEASE_TAG}..."
            local nanvix_dir
            nanvix_dir="$(mktemp -d)"
            gh release download "${NANVIX_RELEASE_TAG}" --repo nanvix/nanvix \
                --pattern "${NANVIX_RELEASE_PATTERN}" \
                --dir "$nanvix_dir" || {
                echo "ERROR: Failed to download Nanvix release. Ensure 'gh' is installed and authenticated."
                rm -rf "$nanvix_dir"
                return 1
            }
            local archive
            archive=$(find "$nanvix_dir" -maxdepth 1 -name "${NANVIX_RELEASE_PATTERN}")
            if [[ -z "$archive" ]]; then
                echo "ERROR: No matching archive found in download directory."
                rm -rf "$nanvix_dir"
                return 1
            fi
            if [[ $(echo "$archive" | wc -l) -ne 1 ]]; then
                echo "ERROR: Multiple matching archives found; expected exactly one."
                rm -rf "$nanvix_dir"
                return 1
            fi
            extract_tar_bz2 "$archive" "$nanvix_dir" || {
                echo "ERROR: Failed to extract Nanvix release archive."
                rm -rf "$nanvix_dir"
                return 1
            }
            install -m 644 "$nanvix_dir/lib/libposix.a" "${install_location}/lib/"
            install -m 644 "$nanvix_dir/lib/user.ld" "${install_location}/lib/"
            # Install nanvixd binaries for integration tests.
            mkdir -p "${install_location}/libexec/nanvixd"
            install -m 755 "$nanvix_dir/bin/nanvixd.elf" "${install_location}/libexec/nanvixd/"
            install -m 755 "$nanvix_dir/bin/kernel.elf" "${install_location}/libexec/nanvixd/"
            install -m 755 "$nanvix_dir/bin/linuxd.elf" "${install_location}/libexec/nanvixd/"
            install -m 755 "$nanvix_dir/bin/uservm.elf" "${install_location}/libexec/nanvixd/"
            rm -rf "$nanvix_dir"
        else
            echo "Nanvix OS libraries already present in '${install_location}/lib/'."
        fi

        return 0
    fi

    print_error "No build stage specified. Use --stage=0 or --stage=1."
    return 1
}

#
# Description
#
#   Configures an LLVM stage 0 build (Clang + LLD only, no runtimes).
#
configure_stage0() {
    local sysroot_location="${Z_SYSROOT_LOCATION:-${Z_INSTALL_LOCATION}}"

    # Use sccache as compiler launcher if available.
    local cmake_launcher_options=()
    if command -v sccache >/dev/null 2>&1; then
        cmake_launcher_options+=(
            -DCMAKE_C_COMPILER_LAUNCHER=sccache
            -DCMAKE_CXX_COMPILER_LAUNCHER=sccache
        )
    fi

    cmake -S llvm -B build -G Ninja \
        -DCMAKE_VERBOSE_MAKEFILE=ON \
        -DLLVM_PARALLEL_LINK_JOBS=1 \
        -DLLVM_ENABLE_PROJECTS='clang;lld' \
        -DLLVM_TARGETS_TO_BUILD=X86 \
        -DLLVM_DEFAULT_TARGET_TRIPLE=i686-unknown-nanvix \
        -DDEFAULT_SYSROOT="${sysroot_location}" \
        -DC_INCLUDE_DIRS="${sysroot_location}/usr/include" \
        -DCMAKE_BUILD_TYPE=Release \
        "${cmake_launcher_options[@]}" || {
        print_error "Stage 0 configuration failed."
        return 1
    }

    return 0
}

#
# Description
#
#   Configures an LLVM stage 1 build (runtimes only, using installed Clang).
#
configure_stage1() {
    local sysroot_location="${Z_SYSROOT_LOCATION:-${Z_INSTALL_LOCATION}}"

    # Extract LLVM version for compiler-rt install path.
    local llvm_version_major
    llvm_version_major=$(sed -n 's/.*LLVM_VERSION_MAJOR \([0-9]*\).*/\1/p' cmake/Modules/LLVMVersion.cmake)

    # Config options for compiler-rt.
    local compiler_rt_options=(
        -DCOMPILER_RT_BUILD_BUILTINS=ON
        -DCOMPILER_RT_BUILD_CRT=OFF
        -DCOMPILER_RT_USE_LIBCXX=OFF
        -DCOMPILER_RT_CXX_LIBRARY=none
        -DCOMPILER_RT_CAN_EXECUTE_TESTS=OFF
        -DCOMPILER_RT_BAREMETAL_BUILD=ON
        -DCOMPILER_RT_BUILD_GWP_ASAN=OFF
        -DCOMPILER_RT_BUILD_LIBFUZZER=OFF
        -DCOMPILER_RT_BUILD_MEMPROF=OFF
        -DCOMPILER_RT_BUILD_PROFILE=OFF
        -DCOMPILER_RT_BUILD_CTX_PROFILE=OFF
        -DCOMPILER_RT_BUILD_SANITIZERS=OFF
        -DCOMPILER_RT_BUILD_XRAY=OFF
        -DCOMPILER_RT_BUILD_ORC=OFF
        -DCOMPILER_RT_DEFAULT_TARGET_ONLY=ON
        -DCOMPILER_RT_INSTALL_LIBRARY_DIR="${Z_INSTALL_LOCATION}/lib/clang/${llvm_version_major}/lib/i686-unknown-nanvix"
    )

    # Config options for libc++.
    local libcxx_options=(
        -DLIBCXX_ENABLE_SHARED=OFF
        -DLIBCXX_ENABLE_STATIC=ON
        -DLIBCXX_INCLUDE_TESTS=OFF
        -DLIBCXX_INCLUDE_BENCHMARKS=OFF
        -DLIBCXX_ENABLE_TIME_ZONE_DATABASE=OFF
        -DLIBCXX_HAS_MUSL_LIBC=OFF
    )

    # Config options for libunwind.
    local libunwind_options=(
        -DLIBUNWIND_ENABLE_STATIC=ON
        -DLIBUNWIND_ENABLE_SHARED=OFF
        -DLIBUNWIND_USE_COMPILER_RT=ON
        -DLIBUNWIND_IS_BAREMETAL=ON
        -DLIBUNWIND_ADDITIONAL_COMPILE_FLAGS='-mno-sse;-mno-sse2'
    )

    # Config options for libcxxabi.
    local libcxxabi_options=(
        -DLIBCXXABI_USE_COMPILER_RT=ON
        -DLIBCXXABI_ENABLE_STATIC_UNWINDER=ON
        -DLIBCXXABI_ENABLE_STATIC=ON
        -DLIBCXXABI_ENABLE_SHARED=OFF
    )

    # Use sccache as compiler launcher if available.
    local cmake_launcher_options=()
    if command -v sccache >/dev/null 2>&1; then
        cmake_launcher_options+=(
            -DCMAKE_C_COMPILER_LAUNCHER=sccache
            -DCMAKE_CXX_COMPILER_LAUNCHER=sccache
        )
    fi

    cmake -S runtimes -B build-runtimes -G Ninja \
        -DCMAKE_VERBOSE_MAKEFILE=ON \
        -DCMAKE_MODULE_PATH="${SCRIPT_DIR}/cmake/platforms" \
        -DCMAKE_SYSTEM_NAME=Nanvix \
        -DCMAKE_TRY_COMPILE_TARGET_TYPE=STATIC_LIBRARY \
        -DCMAKE_C_COMPILER="${Z_INSTALL_LOCATION}/bin/clang" \
        -DCMAKE_CXX_COMPILER="${Z_INSTALL_LOCATION}/bin/clang++" \
        -DCMAKE_C_COMPILER_TARGET=i686-unknown-nanvix \
        -DCMAKE_CXX_COMPILER_TARGET=i686-unknown-nanvix \
        -DCMAKE_SYSROOT="${sysroot_location}" \
        -DLLVM_ENABLE_RUNTIMES='compiler-rt;libunwind;libcxx;libcxxabi' \
        -DCMAKE_BUILD_TYPE=Release \
        "${cmake_launcher_options[@]}" \
        "${compiler_rt_options[@]}" \
        "${libcxx_options[@]}" \
        "${libunwind_options[@]}" \
        "${libcxxabi_options[@]}" || {
        print_error "Stage 1 configuration failed."
        return 1
    }

    return 0
}

#
# Description
#
#   Executes project-specific steps for configuring a build.
#
# Return Value
#
#   - On success, this function returns zero.
#   - On failure, this function returns non-zero.
#
# Usage Example
#
#   configure_steps "i686-nanvix" "/usr/local"
#
configure_steps() {
    if [[ "${Z_STAGE}" == "0" ]]; then
        configure_stage0
    elif [[ "${Z_STAGE}" == "1" ]]; then
        configure_stage1
    else
        print_error "No build stage specified. Use --stage=0 or --stage=1."
        return 1
    fi
}

#
# Description
#
#   Project-specific steps for building the project.
#
# Parameters
#
#   - $1: Number of parallel build jobs.
#
# Usage Example
#
#   build_steps 4 "all" "check"
#
build_steps() {
    local parallel_build_option=$1

    local build_dir="build"
    if [[ "${Z_STAGE}" == "1" ]]; then
        build_dir="build-runtimes"
    fi

    cmake --build "${build_dir}" -j "${parallel_build_option}" -v || {
        print_error "Build failed for targets: '${BUILD_TARGETS[*]}'."
        return 1
    }

    return 0
}

#
# Description
#
#   Project-specific steps for installing the project.
#
# Return Value
#
#   - On success, this function returns zero.
#   - On failure, this function returns non-zero.
#
# Usage Example
#
#   install_steps "install" "/usr/local"
#
install_steps() {
    local build_dir="build"
    if [[ "${Z_STAGE}" == "1" ]]; then
        build_dir="build-runtimes"
    fi

    cmake -DCMAKE_INSTALL_PREFIX="${Z_INSTALL_LOCATION}" -P "${build_dir}/cmake_install.cmake" || {
        print_error "Installation failed."
        return 1
    }

    return 0
}

#==================================================================================================
# Main Script
#==================================================================================================

#
# Description
#
#   Verifies that all expected toolchain artifacts are present after installation.
#
# Usage
#
#   ./z verify
#
verify_artifacts() {
    local install_location="${Z_INSTALL_LOCATION:-/opt/nanvix}"

    echo "Checking libc++ artifacts..."
    local failed=0

    for lib in libc++.a libc++abi.a libunwind.a; do
        if [[ ! -f "${install_location}/lib/${lib}" ]]; then
            echo "ERROR: ${install_location}/lib/${lib} not found."
            failed=1
        fi
    done

    local builtins
    builtins=$(find "${install_location}/lib/clang" -path "*/i686-unknown-nanvix/libclang_rt.builtins*.a" 2>/dev/null | head -1)
    if [[ -z "$builtins" ]]; then
        echo "ERROR: compiler-rt builtins library not found."
        failed=1
    fi

    if [[ $failed -ne 0 ]]; then
        echo "ERROR: Artifact verification failed."
        return 1
    fi

    echo "All C++ runtime libraries present."
    return 0
}

#
# Description
#
#   Runs the Nanvix clang driver lit tests to verify the linker invocation
#   produced by the toolchain matches expected behavior.
#
# Usage
#
#   Called automatically by './z test'.
#
test_driver() {
    local lit="${SCRIPT_DIR}/build/bin/llvm-lit"
    if [[ ! -d "${SCRIPT_DIR}/build" ]]; then
        echo "Skipping driver tests: build directory not found."
        return 0
    fi
    if [[ ! -x "$lit" ]]; then
        echo "ERROR: build directory exists but llvm-lit not found at '${lit}'."
        return 1
    fi
    echo "Running Nanvix driver tests..."
    "$lit" "${SCRIPT_DIR}/clang/test/Driver/nanvix.c" \
           "${SCRIPT_DIR}/clang/test/Driver/nanvix.cpp" \
        --no-progress-bar -v || {
        echo "ERROR: Driver tests failed."
        return 1
    }
    echo "Driver tests passed."
    return 0
}

#
# Description
#
#   Runs the C++ smoke test: compiles and links a minimal C++ program using the
#   built clang toolchain and libc++ for the i686-unknown-nanvix target.
#
#   Requires the toolchain to be already built and installed, and the Nanvix OS
#   libraries (libposix.a, user.ld) to be present in the install location
#   (downloaded by download_steps).
#
# Usage
#
#   ./z test
#
test_smoke_cpp() {
    local install_location="${Z_INSTALL_LOCATION:-/opt/nanvix}"

    if [[ ! -x "${install_location}/bin/clang++" ]]; then
        echo "ERROR: clang++ not found at '${install_location}/bin/clang++'. Build and install first."
        return 1
    fi

    # Resolve libc++.a through the driver to handle target-specific sysroot paths.
    local libcxx_archive
    if ! libcxx_archive=$("${install_location}/bin/clang++" \
        --target=i686-unknown-nanvix \
        --sysroot="${install_location}" \
        --print-file-name=libc++.a); then
        echo "ERROR: failed to resolve libc++.a with clang++ --print-file-name."
        return 1
    fi
    if [[ ! -f "$libcxx_archive" ]]; then
        echo "ERROR: libc++ not found (resolved to '$libcxx_archive'). Build and install first."
        return 1
    fi

    if [[ ! -f "${install_location}/lib/libposix.a" ]]; then
        echo "ERROR: libposix.a not found at '${install_location}/lib/libposix.a'. Build and install first."
        return 1
    fi

    if [[ ! -f "${install_location}/lib/user.ld" ]]; then
        echo "ERROR: user.ld not found at '${install_location}/lib/user.ld'. Build and install first."
        return 1
    fi

    echo "Compiling and linking C++ smoke test..."
    local smoke_test_bin="${SCRIPT_DIR}/build/tests/smoke/hello.elf"
    mkdir -p "$(dirname "$smoke_test_bin")"
    "${install_location}/bin/clang++" \
        --target=i686-unknown-nanvix \
        --sysroot="${install_location}" \
        -stdlib=libc++ \
        -lposix \
        -T "${install_location}/lib/user.ld" \
        tests/smoke/hello.cpp \
        -o "$smoke_test_bin" || {
        echo "ERROR: Smoke test failed to compile and link."
        rm -f "$smoke_test_bin"
        return 1
    }

    file "$smoke_test_bin"
    echo "Smoke test passed: C++ program compiled and linked successfully."
    # Export path for integration test.
    SMOKE_TEST_BIN_CPP="$smoke_test_bin"
    return 0
}

#
# Description
#
#   Runs the C smoke test: compiles and links a minimal C program using the
#   built clang toolchain and newlib (libc.a) for the i686-unknown-nanvix target.
#
#   Requires the toolchain to be already built and installed. The Nanvix OS
#   libraries (libposix.a, user.ld) must already be present in the install
#   location (downloaded by download_steps).
#
# Usage
#
#   Called automatically by './z test' before test_smoke_cpp.
#
test_smoke_c() {
    local install_location="${Z_INSTALL_LOCATION:-/opt/nanvix}"

    if [[ ! -x "${install_location}/bin/clang" ]]; then
        echo "ERROR: clang not found at '${install_location}/bin/clang'. Build and install first."
        return 1
    fi

    # Resolve libc.a through the driver to handle target-specific sysroot paths.
    local libc_archive
    if ! libc_archive=$("${install_location}/bin/clang" \
        --target=i686-unknown-nanvix \
        --sysroot="${install_location}" \
        --print-file-name=libc.a); then
        echo "ERROR: failed to resolve libc.a with clang --print-file-name."
        return 1
    fi
    if [[ ! -f "$libc_archive" ]]; then
        echo "ERROR: libc not found (resolved to '$libc_archive'). Build and install first."
        return 1
    fi

    if [[ ! -f "${install_location}/lib/libposix.a" ]]; then
        echo "ERROR: libposix.a not found at '${install_location}/lib/libposix.a'. Build and install first."
        return 1
    fi

    if [[ ! -f "${install_location}/lib/user.ld" ]]; then
        echo "ERROR: user.ld not found at '${install_location}/lib/user.ld'. Build and install first."
        return 1
    fi

    echo "Compiling and linking C smoke test..."
    local smoke_test_bin="${SCRIPT_DIR}/build/tests/smoke/hello-c.elf"
    mkdir -p "$(dirname "$smoke_test_bin")"
    "${install_location}/bin/clang" \
        --target=i686-unknown-nanvix \
        --sysroot="${install_location}" \
        tests/smoke/hello.c \
        -lc \
        -lposix \
        -T "${install_location}/lib/user.ld" \
        -o "$smoke_test_bin" || {
        echo "ERROR: C smoke test failed to compile and link."
        rm -f "$smoke_test_bin"
        return 1
    }

    file "$smoke_test_bin"
    echo "C smoke test passed: program compiled and linked successfully."
    # Export path for integration test.
    SMOKE_TEST_BIN_C="$smoke_test_bin"
    return 0
}

#
# Description
#
#   Runs the compiled C++ hello-world app on nanvixd.elf.
#   Requires /dev/kvm to be available.
#
# Parameters
#
#   $1: Path to the smoke test binary.
#
# Usage
#
#   Called automatically by './z test' when /dev/kvm is available.
#
test_integration() {
    local smoke_test_bin="$1"
    local install_location="${Z_INSTALL_LOCATION:-/opt/nanvix}"

    if [[ ! -f "${smoke_test_bin}" ]]; then
        echo "ERROR: Smoke test binary not found at '${smoke_test_bin}'."
        return 1
    fi

    local nanvixd="${install_location}/libexec/nanvixd/nanvixd.elf"
    if [[ ! -f "${nanvixd}" ]]; then
        echo "ERROR: nanvixd.elf not found at '${nanvixd}'. Run test_smoke first."
        return 1
    fi

    echo "Running integration test: hello.cpp on nanvixd.elf..."
    local timeout_seconds=30
    local output
    output="$(timeout "${timeout_seconds}" \
        "${nanvixd}" \
            -bin-dir "${install_location}/libexec/nanvixd" \
            -console-file /dev/stdout \
            -- "${smoke_test_bin}" </dev/null 2>&1)" || {
        local rc=$?
        if [[ $rc -eq 124 ]]; then
            echo "ERROR: Integration test timed out after ${timeout_seconds}s."
        else
            echo "ERROR: Integration test failed (exit code $rc)."
        fi
        echo "Output:"
        echo "$output"
        return 1
    }

    echo "$output"

    # Verify expected output from the hello-world program.
    if echo "$output" | grep -q "value = 42"; then
        echo "Integration test passed: program ran successfully on nanvixd.elf."
    else
        echo "ERROR: Expected output 'value = 42' not found."
        return 1
    fi

    return 0
}

# Intercept 'verify' and 'test' commands before delegating to zmain.
# NOTE: The z framework (utils.sh) does not support registering custom
# subcommands; its command set is hardcoded. Intercepting $1 before zmain is
# the intended extension pattern. These commands won't appear in './z help'.
if [[ "${1:-}" == "verify" ]]; then
    _load_project_config 2>/dev/null || true
    verify_artifacts
    exit $?
fi

if [[ "${1:-}" == "test" ]]; then
    # Load saved configuration if available.
    _load_project_config 2>/dev/null || true
    test_driver || exit $?
    test_smoke_c || exit $?
    test_smoke_cpp || exit $?

    # Run integration tests if KVM is available.
    if [[ -c /dev/kvm ]]; then
        if [[ -n "${SMOKE_TEST_BIN_C:-}" ]]; then
            test_integration "${SMOKE_TEST_BIN_C}" || exit $?
        fi
        if [[ -n "${SMOKE_TEST_BIN_CPP:-}" ]]; then
            test_integration "${SMOKE_TEST_BIN_CPP}" || exit $?
        fi
    else
        echo "Skipping integration tests: /dev/kvm not available."
    fi
    exit 0
fi

# Run main function with all arguments
zmain "$@" || {
    exit 1
}

exit 0