setup.sh

#!/bin/bash
# =============================================================================
# DragonBornOS Setup Script
# =============================================================================
# Inicializa el entorno de desarrollo para todos los dispositivos
# =============================================================================

set -e

RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
CYAN='\033[0;36m'
WHITE='\033[1;37m'
NC='\033[0m'

log() { echo -e "${CYAN}[*]${NC} $1"; }
success() { echo -e "${GREEN}[+]${NC} $1"; }
error() { echo -e "${RED}[!]${NC} $1"; exit 1; }
warn() { echo -e "${YELLOW}[-]${NC} $1"; }

echo -e "${WHITE}"
cat << 'BANNER'
  ____  _                       ____             __ _       
 | __ )| | __ _ _ __  _ __  _   _ / ___|___  _ __  / _(_) __ _ 
 |  _ \| |/ _` | '_ \| '_ \| | | | |   / _ \| '_ \| |_| |/ _` |
 | |_) | | (_| | | | | | | | |_| | |__| (_) | | | |  _| | (_| |
 |____/|_|\__,_|_| |_|_| |_|\__,_|\____\___/|_| |_|_| |_|\__, |
                                                         |___/ 
                     ____  _          _     
                    | __ )| | __ _  _| |__  
                    |  _ \| |/ _` || | '_ \ 
                    | |_) | | (_| || | |_) |
                    |____/|_|\__,_||_|_.__/ 
                                             
                SETUP SCRIPT
BANNER
echo -e "${NC}"
echo "============================================================================="
echo "  DragonBornOS Development Environment Setup"
echo "============================================================================="
echo ""

# =============================================================================
# CHECK SYSTEM REQUIREMENTS
# =============================================================================

check_system() {
    log "Checking system requirements..."
    
    # Check OS
    if [ -f /etc/os-release ]; then
        . /etc/os-release
        if [[ ! "$ID" =~ ^(ubuntu|debian|linuxmint)$ ]]; then
            warn "You're not using Ubuntu/Debian. Some commands may fail."
        fi
        log "OS: $NAME $VERSION"
    fi
    
    # Check architecture
    ARCH=$(uname -m)
    if [ "$ARCH" != "x86_64" ]; then
        error "DragonBornOS requires x86_64 architecture. You have: $ARCH"
    fi
    log "Architecture: $ARCH"
    
    # Check disk space
    AVAILABLE=$(df -P . | awk 'NR==2 {print $4}')
    AVAILABLE_GB=$((AVAILABLE / 1024 / 1024))
    if [ $AVAILABLE_GB -lt 100 ]; then
        error "Insufficient disk space. Need 100GB+, have: ${AVAILABLE_GB}GB"
    fi
    log "Available disk space: ${AVAILABLE_GB}GB"
    
    # Check RAM
    RAM=$(free -m | awk 'NR==2 {print $2}')
    RAM_GB=$((RAM / 1024))
    if [ $RAM_GB -lt 16 ]; then
        warn "Low RAM detected. Recommended: 32GB+, have: ${RAM_GB}GB"
    fi
    log "RAM: ${RAM_GB}GB"
    
    success "System check passed!"
}

# =============================================================================
# INSTALL DEPENDENCIES
# =============================================================================

install_dependencies() {
    log "Installing build dependencies..."
    
    if [ -f /etc/debian_version ]; then
        sudo apt update
        sudo apt install -y \
            bc bison build-essential ccache curl flex g++ \
            gcc git gnupg liblz4-tool libncurses5 libncurses5-dev \
            libssl-dev libxml2 libxml2-utils lzop maven \
            pngcrush python3 rsync schedtool squashfs-tools \
            xsltproc zip zlib1g-dev imagemagick \
            openjdk-11-jdk wget libtinfo5 libncurses5 \
            adb fastboot
    
        success "Dependencies installed!"
    else
        warn "Not a Debian-based system. Install dependencies manually."
        echo ""
        echo "Required packages:"
        echo "  bc bison build-essential ccache curl flex g++ gcc git"
        echo "  gnupg liblz4-tool libncurses5 libncurses5-dev libssl-dev"
        echo "  libxml2 libxml2-utils lzop maven pngcrush python3 rsync"
        echo "  schedtool squashfs-tools xsltproc zip zlib1g-dev imagemagick"
        echo "  openjdk-11-jdk wget libtinfo5 libncurses5 adb fastboot"
    fi
}

# =============================================================================
# SETUP REPO TOOL
# =============================================================================

setup_repo() {
    log "Setting up repo tool..."
    
    if [ ! -f ~/bin/repo ]; then
        mkdir -p ~/bin
        curl -s https://storage.googleapis.com/git-repo-downloads/repo > ~/bin/repo
        chmod a+x ~/bin/repo
        success "Repo tool installed"
    else
        success "Repo tool already exists"
    fi
    
    # Add to PATH
    if [[ ":$PATH:" != *":$HOME/bin:"* ]]; then
        echo 'export PATH="$HOME/bin:$PATH"' >> ~/.bashrc
        source ~/.bashrc
        log "Added ~/bin to PATH"
    fi
}

# =============================================================================
# INITIALIZE REPOSITORY
# =============================================================================

init_repo() {
    log "Initializing DragonBornOS repository..."
    
    if [ ! -d ".repo" ]; then
        repo init -u https://github.com/DragonBornOS/platform_manifest.git -b 13.0 --git-lfs
        
        # Add local manifest for devices
        mkdir -p .repo/local_manifests
        
        cat > .repo/local_manifests/devices.xml << 'EOF'
<?xml version="1.0" encoding="UTF-8"?>
<manifest>
    <!-- Moto Edge 30 Neo -->
    <project name="DragonBornOS/device_motorola_neo" path="device/motorola/neo" remote="github" revision="13.0"/>
    <project name="DragonBornOS/kernel_motorola_neo" path="kernel/motorola/neo" remote="github" revision="13.0"/>
    <project name="DragonBornOS/proprietary_vendor_motorola" path="vendor/motorola" remote="github" revision="13.0"/>
    
    <!-- Xiaomi POCO X3 Pro -->
    <project name="DragonBornOS/device_xiaomi_vayu" path="device/xiaomi/vayu" remote="github" revision="13.0"/>
    <project name="DragonBornOS/kernel_xiaomi_sm8250" path="kernel/xiaomi/sm8250" remote="github" revision="13.0"/>
    <project name="DragonBornOS/proprietary_vendor_xiaomi" path="vendor/xiaomi" remote="github" revision="13.0"/>
    
    <!-- Huawei P Smart 2019 -->
    <project name="DragonBornOS/device_huawei_HWSTK-HF" path="device/huawei/HWSTK-HF" remote="github" revision="13.0"/>
    <project name="DragonBornOS/kernel_huawei_kirin710" path="kernel/huawei/kirin710" remote="github" revision="13.0"/>
    <project name="DragonBornOS/proprietary_vendor_huawei" path="vendor/huawei" remote="github" revision="13.0"/>
</manifest>
EOF
        
        success "Repository initialized with local manifest"
    else
        success "Repository already initialized"
    fi
}

# =============================================================================
# SYNC SOURCES
# =============================================================================

sync_sources() {
    log "Syncing sources..."
    
    repo sync -c --force-sync --optimized-fetch --no-tags --no-clone-bundle \
        -j$(nproc --all)
    
    success "Sources synced!"
}

# =============================================================================
# SETUP DEVICE TREES
# =============================================================================

setup_devices() {
    log "Setting up device trees..."
    
    for device in neo vayu HWSTK-HF; do
        if [ -d "device/*/$device" ]; then
            success "Device tree found: $device"
        else
            warn "Device tree not found: $device"
        fi
    done
}

# =============================================================================
# BUILD ENVIRONMENT
# =============================================================================

setup_build_env() {
    log "Setting up build environment..."
    
    if [ -f build/envsetup.sh ]; then
        source build/envsetup.sh
        success "Build environment loaded"
    else
        warn "build/envsetup.sh not found. Run 'repo sync' first."
    fi
    
    # Show lunch options
    echo ""
    log "Available DragonBornOS devices:"
    lunch | grep dragonborn || true
}

# =============================================================================
# MAIN
# =============================================================================

main() {
    check_system
    install_dependencies
    setup_repo
    init_repo
    sync_sources
    setup_devices
    setup_build_env
    
    echo ""
    success "╔═══════════════════════════════════════════════════════════════════╗"
    success "║  DragonBornOS SETUP COMPLETE                                     ║"
    success "╚═══════════════════════════════════════════════════════════════════╝"
    echo ""
    echo "Next steps:"
    echo "  1. Build ROM: ./build.sh {device} build"
    echo "  2. Available devices: neo, vayu, HWSTK-HF"
    echo ""
    echo "Example:"
    echo "  ./build.sh neo build"
    echo ""
}

main "$@"