#!/bin/bash
# BTC-0.1.4.sh - Build Tool Chain (Sovereign Final)
# Identity: dcosnet / dcos.net | Target: Broadwell-HS / Haswell-EP
# Version: 0.1.4 | Persistence: /opt/BTC | Volatile: ramfs
# Copyright (C) 2012-2026 Jeremy Anderson (info@dcos.net)

# --- 1. ARCHITECTURE & RECOVERY ---
set -euo pipefail # Strict exit on error, unset vars, or pipe failure
# Cleanup trap: Ensures ramfs is unmounted even on build failure
trap "echo '>> Interrupt Detected. Cleaning ramfs...'; cd / && umount -l ${SOURCES_ACTIVE} 2>/dev/null || true; exit 1" INT TERM

# --- 2. GLOBAL CONFIG & PERSISTENCE ---
export BTC_DEBUG_LEVEL=1  # 0=Silent, 1=Audit (IW), 2=Trace
export BTC_STRIP_MODE=1   # 1=Lean/Strip (Default), 0=Preserve Symbols
export BTC_ARCHIVE=/opt/BTC # Permanent storage for artifacts/logs

# --- 3. SILICON IDENTITY & DCOSNET NAMESPACE ---
function f_silicon_probe() {
    echo ">> BTC-0.1.4: Interrogating Silicon..."
    # Auto-detect microarch; fallback to HSW for generic/ES chips
    local RAW_ARCH=$(gcc -march=native -Q --help=target | grep -m1 "march=" | awk '{print $2}')
    [[ -z "$RAW_ARCH" || "$RAW_ARCH" == "x86-64" ]] && export TARGET_ARCH="hsw" || export TARGET_ARCH="bdw"
    
    export ISA_TAG="AVX2"
    export OPT_TAG="LTO"
    export SYS_LABEL="DCOSNET-${TARGET_ARCH^^}-${ISA_TAG}-${OPT_TAG}"
    export TARGET="x86_64-dcosnet-linux-gnu"
    # Calculate thread depth: 2GB RAM ceiling per thread to prevent LTO thrashing
    local total_ram=$(free -m | awk '/^Mem:/{print $2}')
    export v_threads="-j$(( (total_ram / 2048) < $(nproc) ? (total_ram / 2048) : $(nproc) ))"

    # Aggressive Forge Profile: O3 + LTO + Hardened Stack
    export GLOBAL_CFLAGS="-O3 -march=native -flto=$(nproc) -fstack-protector-strong -D_FORTIFY_SOURCE=2 --sysroot=${NEWROOT} -pipe"
    export GLOBAL_LDFLAGS="-Wl,-O1 -Wl,--as-needed -flto=$(nproc) --sysroot=${NEWROOT}"
    
    export CCACHE_DIR="${BTC_ARCHIVE}/cache/${SYS_LABEL}"
    mkdir -p "${CCACHE_DIR}"
}

# --- 4. FORENSIC STAMPING (ELF & XATTR) ---
function f_stamp_binary() {
    local target_bin="$1"
    local log_base="$2"
    
    # Immutable ELF Note: The Silicon Birth Certificate
    cat <<EOF > btc_stamp.s
.section .note.BTC,"a"
.align 4
.long 4f - 1f
.long 3f - 2f
.long 1
1: .asciz "DCOSNET"
2: .ascii "Org: dcos.net|K:7.1|I:${ISA_TAG}|O:${OPT_TAG}|T:${DISTRO}|F:${HOSTNAME}"
3: .align 4
4:
EOF
    gcc -c btc_stamp.s -o btc_stamp.o
    objcopy --add-section .note.BTC=btc_stamp.o "${target_bin}"
    rm btc_stamp.s btc_stamp.o

    # Rapid Audit Metadata (Persistent via rsync -X)
    local bin_hash=$(sha256sum "${target_bin}" | awk '{print $1}')
    setfattr -n user.btc.identity -v "BTC-${SYS_LABEL}-${v_linux}-dcosnet" "${target_bin}"
    setfattr -n user.btc.hash -v "${bin_hash}" "${target_bin}"
    
    # External Symbol Extraction: Move debug info to LeadNode archive
    if [[ "${BTC_STRIP_MODE}" -eq 1 ]]; then
        mkdir -p "${BTC_ARCHIVE}/symbols/${SYS_LABEL}"
        objcopy --only-keep-debug "${target_bin}" "${BTC_ARCHIVE}/symbols/${SYS_LABEL}/${log_base}.debug"
        strip --strip-unneeded "${target_bin}"
        objcopy --add-gnu-debuglink="${BTC_ARCHIVE}/symbols/${SYS_LABEL}/${log_base}.debug" "${target_bin}"
    fi
}

# --- 5. THE INVISIBLE GUARD (THERMAL/ENTROPY) ---
function f_guard() {
    local max_temp=85
    local min_mem=800
    while true; do
        local cur_temp=$(cat /sys/class/thermal/thermal_zone*/temp | head -n1 | awk '{print $1/1000}')
        local cur_mem=$(free -m | awk '/^Mem:/{print $7}')
        # Prevent Broadwell thermal runaway or OOM death during link-time-optimization
        if (( cur_temp > max_temp )); then sleep 10; elif (( cur_mem < min_mem )); then sleep 30; else break; fi
    done
}

function f_entropy_shield() {
    # Ensure secure kernel signing and binary sealing via randomness generation
    local cur_ent=$(cat /proc/sys/kernel/random/entropy_avail)
    if (( cur_ent < 250 )); then
        find /bin /sbin -type f -exec ls -l {} + > /dev/null 2>&1 &
        sleep 2 && kill $! 2>/dev/null
    fi
}

# --- 6. EXECUTION ENGINE (FORENSIC LOGGING) ---
function f_exec_log() {
    local cmd="$1"
    local log_base="$2"
    local mode="${3:-build}" 
    
    f_entropy_shield
    f_guard

    if [[ "$mode" == "install" ]]; then
        # Audit FS changes via installwatch; auto-stamp all resulting executables
        stdbuf -oL -eL installwatch -o "${LOGS}/${log_base}.iw" bash -c "$cmd" | \
        pv -t -r -b -N "${log_base}" >> "${LOGS}/${log_base}.log" 2>&1
        
        find ${NEWROOT} -type f -executable -exec bash -c '
            file "$1" | grep -q "ELF" && f_stamp_binary "$1" "'"${log_base}"'"
        ' _ {} \;
    else
        # Standard build: pipe stdout through pv for telemetry; separate stderr
        stdbuf -oL -eL bash -c "${cmd}" | \
        pv -t -r -b -N "${log_base}" | \
        tee -a "${LOGS}/${log_base}.log" > /dev/null \
        2> >(tee -a "${LOGS}/${log_base}.err" >> "${LOGS}/${log_base}.log")
    fi
}

# --- 7. CLEANROOM INFRA (VOLATILE RAMFS) ---
export SOURCES_ACTIVE=/usr/src
export SOURCE_CACHE=${BTC_ARCHIVE}/src
export v_glibc='glibc-2.41'
export v_gcc='gcc-14.2.0'
export v_binutils='binutils-2.46'
export v_linux='linux-7.1'
export NEWROOT=${SOURCES_ACTIVE}/BTC-Forge
export LOGS=${NEWROOT}/LOGS
export DISTRO="DCOSNET"

function f_setup() {
    # Speed-of-light compilation via ramfs; strictly temporary
    local ram_kb=$(grep MemTotal /proc/meminfo | awk '{print $2}')
    mount -t ramfs -o size=$((ram_kb/2/1024))M ramfs ${SOURCES_ACTIVE}
    mkdir -p ${NEWROOT}/{bin,lib,lib64,sbin,etc,usr,boot} ${LOGS}
    ln -sf lib ${NEWROOT}/lib64
}

function f_set_exports() {
    # Pivot to the DCOSNET sovereign cross-compiler
    export CC="ccache ${NEWROOT}/bin/${TARGET}-gcc-${SYS_LABEL}"
    export CXX="ccache ${NEWROOT}/bin/${TARGET}-g++-${SYS_LABEL}"
    export AR="${NEWROOT}/bin/${TARGET}-gcc-ar-${SYS_LABEL}"
    export NM="${NEWROOT}/bin/${TARGET}-gcc-nm-${SYS_LABEL}"
    export RANLIB="${NEWROOT}/bin/${TARGET}-gcc-ranlib-${SYS_LABEL}"
    export CFLAGS="${GLOBAL_CFLAGS}"
    export CXXFLAGS="${GLOBAL_CFLAGS}"
    export LDFLAGS="${GLOBAL_LDFLAGS}"
}

# --- 8. PERSISTENCE BRIDGE (PACKAGING) ---
function f_package() {
    local PKG_NAME="dcosnet-baseline-${SYS_LABEL}-${v_linux}.tar.xz"
    local PKG_PATH="${BTC_ARCHIVE}/completed"
    mkdir -p "${PKG_PATH}"
    
    echo ">> [DCOSNET] Committing Ramfs to Persistent Archive..."
    # Freeze the cleanroom state into a compressed artifact
    tar -cJpf "${PKG_PATH}/${PKG_NAME}" -C "${NEWROOT}" .
    
    # Preserve forensic logs for the LeadNode database
    mkdir -p "${BTC_ARCHIVE}/logs"
    cp -rv "${LOGS}" "${BTC_ARCHIVE}/logs/${SYS_LABEL}_$(date +%Y%m%d)"
}

# --- 9. BUILD STAGES (LINEAR) ---
function f_binutils() {
    cd ${SOURCES_ACTIVE}
    tar -axf ${SOURCE_CACHE}/${v_binutils}*
    cd binutils-* && mkdir -p build && cd build
    f_exec_log "../configure --prefix=${NEWROOT} --target=${TARGET} --with-sysroot=${NEWROOT} --program-suffix=-${SYS_LABEL} --disable-nls --disable-multilib" "binutils-conf"
    f_exec_log "make ${v_threads}" "binutils-make"
    f_exec_log "make install" "binutils-install" "install"
}

function f_kernel_headers() {
    cd ${SOURCES_ACTIVE}
    tar -axf ${SOURCE_CACHE}/${v_linux}*
    cd linux-*
    f_exec_log "make mrproper && make headers" "kernel-headers"
    cp -rv usr/include/* ${NEWROOT}/include
}

function f_gcc_p1() {
    cd ${SOURCES_ACTIVE}
    tar -axf ${SOURCE_CACHE}/${v_gcc}*
    cd gcc-*
    for lib in gmp mpfr mpc; do tar -xf ${SOURCE_CACHE}/${lib}*; mv -v ${lib}-* ${lib}; done
    mkdir -p build && cd build
    f_exec_log "../configure --target=${TARGET} --prefix=${NEWROOT} --with-sysroot=${NEWROOT} --program-suffix=-${SYS_LABEL} --without-headers --disable-shared --disable-threads --enable-languages=c,c++" "gcc1-conf"
    f_exec_log "make ${v_threads}" "gcc1-make"
    f_exec_log "make install" "gcc1-install" "install"
}

function f_glibc() {
    f_set_exports
    cd ${SOURCES_ACTIVE}/glibc-*
    mkdir -p build && cd build
    f_exec_log "../configure --prefix=${NEWROOT} --host=${TARGET} --with-headers=${NEWROOT}/include libc_cv_slibdir=${NEWROOT}/lib" "glibc-conf"
    f_exec_log "make ${v_threads}" "glibc-make"
    f_exec_log "make DESTDIR=${NEWROOT} install" "glibc-install" "install"
}

function f_gcc_p2() {
    f_set_exports
    cd ${SOURCES_ACTIVE}/gcc-*/build && rm -rf *
    f_exec_log "../configure --prefix=${NEWROOT} --target=${TARGET} --program-suffix=-${SYS_LABEL} --enable-languages=c,c++ --with-build-sysroot=${NEWROOT}" "gcc2-conf"
    f_exec_log "make ${v_threads}" "gcc2-make"
    f_exec_log "make install" "gcc2-install" "install"
}

function f_kernel_binary() {
    f_set_exports
    cd ${SOURCES_ACTIVE}/linux-*
    # Sovereign uname/kernel identification
    echo "-dcosnet-${SYS_LABEL}" > .scmversion
    f_exec_log "make ${v_threads} LOCALVERSION=-dcosnet-${SYS_LABEL} bzImage" "kernel-bin-make"
    cp -v arch/x86/boot/bzImage ${NEWROOT}/boot/vmlinuz-${v_linux}-${SYS_LABEL}-dcosnet
}

function f_install_target() {
    local TARGET_PART="${1}"
    local MNT_POINT="/mnt/btc_target"
    mkdir -p ${MNT_POINT} && mount ${TARGET_PART} ${MNT_POINT}
    
    [[ -f "${MNT_POINT}/etc/sorcery/version" ]] && export DISTRO="SourceMage"
    [[ -f "${MNT_POINT}/etc/lunar/version" ]] && export DISTRO="Lunar"

    echo ">> [DCOSNET SWARM] Seeding ${TARGET_PART}..."
    f_exec_log "rsync -avzX ${NEWROOT}/bin/ ${MNT_POINT}/usr/local/bin/" "deploy" "install"
    umount ${MNT_POINT}
}

# --- 10. MAIN (ONE-SHOT) ---
function f_main() {
    [[ $EUID -ne 0 ]] && { echo ">> Root Required."; exit 1; }
    local TARGET_DEV="${1:-}" # Optional physical seed target

    f_silicon_probe
    f_setup
    
    f_binutils
    f_kernel_headers
    f_gcc_p1
    f_glibc
    f_gcc_p2
    f_kernel_binary
    
    #Stage 1: Persistence
    f_package
    
    #Stage 2: Optional Seed
    if [[ -n "${TARGET_DEV}" ]]; then
        if [[ -b "${TARGET_DEV}" ]]; then
            f_install_target "${TARGET_DEV}"
        else
            echo ">> [ERROR] ${TARGET_DEV} is not a valid block device."
            exit 1
        fi
    fi

    #Stage 3: Zero-Footprint Cleanup
    cd / && umount ${SOURCES_ACTIVE}
    echo ">> [SUCCESS] BTC-0.1.4: Sovereign Swarm Artifacts Persistent in ${BTC_ARCHIVE}."
}

f_main "$@"