proot.sh

#!/bin/bash

# Get script directory
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

# GitHub branch for downloads
BRANCH="main"

# Language override variable
OVERRIDE_LANG=""

# Global variables
USE_GUM=false
VERBOSE=false
INSTALL_THEME=false
INSTALL_ICONS=false
INSTALL_WALLPAPERS=false
INSTALL_CURSORS=false
SELECTED_THEME=""
SELECTED_ICON_THEME=""
SELECTED_WALLPAPER=""

# PROOT VARIABLES
PROOT_USERNAME=""
PROOT_PASSWORD=""

#------------------------------------------------------------------------------
# PRELIMINARY ARGUMENT PARSING FOR LANGUAGE
#------------------------------------------------------------------------------
for ARG in "$@"; do
    case $ARG in
        --lang|-l)
            shift
            if [ -n "$1" ]; then
                OVERRIDE_LANG="$1"
                shift
            else
                echo "Error: --lang requires an argument (ex: --lang fr)" >&2
                exit 1
            fi
            ;;
        *)
            ;;
    esac
done

#------------------------------------------------------------------------------
# BOOTSTRAP - Load i18n and lib systems
#------------------------------------------------------------------------------
_loader_url="https://raw.githubusercontent.com/devohmycode/OhMyTermux/$BRANCH/lib/i18n_loader.sh"
mkdir -p "$SCRIPT_DIR/lib"
# Always try to refresh i18n_loader.sh; fall back to cached version if download fails
_loader_tmp=$(mktemp 2>/dev/null || echo "$SCRIPT_DIR/lib/i18n_loader.sh.tmp")
if curl -fL -s -o "$_loader_tmp" "$_loader_url" 2>/dev/null && head -1 "$_loader_tmp" 2>/dev/null | grep -q "^#!/bin/bash"; then
    mv "$_loader_tmp" "$SCRIPT_DIR/lib/i18n_loader.sh"
else
    rm -f "$_loader_tmp" 2>/dev/null
    if [ ! -f "$SCRIPT_DIR/lib/i18n_loader.sh" ]; then
        echo "Warning: Could not download i18n_loader.sh and no cached version available" >&2
    fi
fi
source "$SCRIPT_DIR/lib/i18n_loader.sh"

# Configure error handler keys for this script
ERROR_MSG_KEY="MSG_PROOT_ERROR_INSTALL"
ERROR_REFER_KEY="MSG_PROOT_ERROR_REFER"

#------------------------------------------------------------------------------
# DISPLAY HELP
#------------------------------------------------------------------------------
show_help() {
    clear
    echo "$(t "MSG_PROOT_HELP_TITLE")"
    echo
    echo "$(t "MSG_PROOT_HELP_USAGE")"
    echo "$(t "MSG_PROOT_HELP_OPTIONS")"
    echo "  --gum | -g     $(t "MSG_PROOT_HELP_GUM")"
    echo "  --verbose | -v $(t "MSG_PROOT_HELP_VERBOSE")"
    echo "  --help | -h    $(t "MSG_PROOT_HELP_HELP")"
}

#------------------------------------------------------------------------------
# ARGUMENTS MANAGEMENT
#------------------------------------------------------------------------------
for ARG in "$@"; do
    case $ARG in
        --gum|-g)
            USE_GUM=true
            shift
            ;;
        --verbose|-v)
            VERBOSE=true
            REDIRECT=""
            shift
            ;;
        --help|-h)
            show_help
            exit 0
            ;;
        --username=*)
            PROOT_USERNAME="${ARG#*=}"
            shift
            ;;
        --password=*)
            PROOT_PASSWORD="${ARG#*=}"
            shift
            ;;
        --lang|-l)
            # Skip --lang as it's already processed
            shift
            if [ -n "$1" ]; then
                shift
            fi
            ;;
        *)
            break
            ;;
    esac
done

#------------------------------------------------------------------------------
# DEPENDENCIES CHECK
#------------------------------------------------------------------------------
check_dependencies() {
    if [ "$USE_GUM" = true ]; then
        if $USE_GUM && ! command -v gum &> /dev/null; then
            echo -e "${COLOR_BLUE}Installing gum${COLOR_RESET}"
            pkg update -y > /dev/null 2>&1 && pkg install gum -y > /dev/null 2>&1
        fi
    fi

    if ! command -v proot-distro &> /dev/null; then
        error_msg "Please install proot-distro before continuing."
        exit 1
    fi
}

trap finish EXIT

#------------------------------------------------------------------------------
# PROOT PACKAGES INSTALLATION
#------------------------------------------------------------------------------
install_packages_proot() {
    local PKGS_PROOT=('sudo' 'wget' 'nala' 'xfconf')
    #local PKGS_PROOT=('sudo' 'wget' 'nala' 'xfconf' 'gnome-themes-extra')
    for PKG in "${PKGS_PROOT[@]}"; do
        execute_command "proot-distro login debian --shared-tmp -- env DISPLAY=:1.0 apt install $PKG -y" "Installation of $PKG"
    done
}

#------------------------------------------------------------------------------
# PROOT USER CREATION
#------------------------------------------------------------------------------
create_user_proot() {
    execute_command "
        proot-distro login debian --shared-tmp -- env DISPLAY=:1.0 groupadd storage
        proot-distro login debian --shared-tmp -- env DISPLAY=:1.0 groupadd wheel
        proot-distro login debian --shared-tmp -- env DISPLAY=:1.0 useradd -m -g users -G wheel,audio,video,storage -s /bin/bash '$USERNAME'
        proot-distro login debian --shared-tmp -- env DISPLAY=:1.0 chpasswd <<< '$USERNAME:$PASSWORD'
    " "User creation"
}

#------------------------------------------------------------------------------
# USER RIGHTS CONFIGURATION
#------------------------------------------------------------------------------
configure_user_rights() {
    execute_command "
        # Add user to sudo group
        proot-distro login debian --shared-tmp -- env DISPLAY=:1.0 usermod -aG sudo '$USERNAME'

        # Create sudoers.d file for user
        echo '$USERNAME ALL=(ALL) NOPASSWD: ALL' > '$PROOT_DEBIAN_ROOT/etc/sudoers.d/$USERNAME'
        chmod 0440 '$PROOT_DEBIAN_ROOT/etc/sudoers.d/$USERNAME'

        # Main sudoers file configuration
        echo '%sudo ALL=(ALL:ALL) ALL' >> '$PROOT_DEBIAN_ROOT/etc/sudoers'

        # Check permissions
        chmod 440 '$PROOT_DEBIAN_ROOT/etc/sudoers'
        chown root:root '$PROOT_DEBIAN_ROOT/etc/sudoers'
    " "Sudo rights configuration"
}

#------------------------------------------------------------------------------
# GPU DETECTION
#------------------------------------------------------------------------------
detect_gpu() {
    local platform=$(getprop ro.board.platform 2>/dev/null)
    local hardware=$(getprop ro.hardware 2>/dev/null)

    # Qualcomm Adreno
    if [[ "$platform" =~ ^(msm|sdm|sm|lahaina|taro|kalama|crow|pineapple) ]] || \
       [[ "$hardware" == *"qcom"* ]]; then
        GPU_VENDOR="adreno"
        return
    fi

    # MediaTek (Mali GPU)
    if [[ "$platform" =~ ^mt ]]; then
        GPU_VENDOR="mali"
        return
    fi

    # Samsung Exynos (Mali GPU)
    if [[ "$platform" =~ ^exynos ]] || [[ "$hardware" == *"exynos"* ]]; then
        GPU_VENDOR="mali"
        return
    fi

    # Google Tensor (Mali GPU)
    if [[ "$platform" =~ ^gs ]] || [[ "$platform" == "zuma" ]]; then
        GPU_VENDOR="mali"
        return
    fi

    GPU_VENDOR="unknown"
}

#------------------------------------------------------------------------------
# GPU DETECTION, SAVE AND MESA-VULKAN INSTALLATION
#------------------------------------------------------------------------------
install_mesa_vulkan() {
    info_msg "$(t "MSG_PROOT_GPU_DETECTING")"
    detect_gpu

    # Save GPU vendor for other scripts (zrun, zrunhud in utils.sh)
    mkdir -p "$OHMYTERMUX_CONFIG_DIR"
    echo "$GPU_VENDOR" > "$OHMYTERMUX_CONFIG_DIR/gpu_vendor"

    case "$GPU_VENDOR" in
        adreno)
            info_msg "$(printf "$(t "MSG_PROOT_GPU_DETECTED")" "$(t "MSG_PROOT_GPU_ADRENO")")"
            execute_command "proot-distro login debian --shared-tmp -- apt install -y mesa-vulkan-drivers" "$(t "MSG_PROOT_MESA_INSTALLATION")"
            ;;
        mali)
            info_msg "$(printf "$(t "MSG_PROOT_GPU_DETECTED")" "$(t "MSG_PROOT_GPU_MALI")")"
            warning_msg "$(t "MSG_PROOT_GPU_MALI_WARNING")"
            ;;
        *)
            info_msg "$(printf "$(t "MSG_PROOT_GPU_DETECTED")" "$(t "MSG_PROOT_GPU_UNKNOWN")")"
            warning_msg "$(t "MSG_PROOT_GPU_UNKNOWN_WARNING")"
            ;;
    esac
}

#------------------------------------------------------------------------------
# THEMES COPY
#------------------------------------------------------------------------------
copy_theme() {
    local theme_name="$1"
    local theme_path=""

    case $theme_name in
        "WhiteSur")
            theme_path="WhiteSur-Dark"
            ;;
        "Fluent")
            theme_path="Fluent-dark-compact"
            ;;
        "Lavanda")
            theme_path="Lavanda-dark-compact"
            ;;
    esac

    execute_command "cp -r $PREFIX/share/themes/$theme_path $PROOT_DEBIAN_ROOT/usr/share/themes/" "$(t "MSG_PROOT_THEME_CONFIGURATION")"
}

#------------------------------------------------------------------------------
# ICONS COPY
#------------------------------------------------------------------------------
copy_icons() {
    local icon_theme="$1"
    local icon_path=""

    case $icon_theme in
        "WhiteSur")
            icon_path="WhiteSur-dark"
            ;;
        "McMojave-circle")
            icon_path="McMojave-circle-dark"
            ;;
        "Tela")
            icon_path="Tela-dark"
            ;;
        "Fluent")
            icon_path="Fluent-dark"
            ;;
        "Qogir")
            icon_path="Qogir-dark"
            ;;
    esac

    execute_command "cp -r $PREFIX/share/icons/$icon_path $PROOT_DEBIAN_ROOT/usr/share/icons/" "$(t "MSG_PROOT_ICONS_CONFIGURATION")"
}

#------------------------------------------------------------------------------
# THEMES AND ICONS CONFIGURATION
#------------------------------------------------------------------------------
configure_themes_and_icons() {
    # Load configuration from temporary file
    if [ -f "$OHMYTERMUX_CONFIG_DIR/theme_config.tmp" ]; then
        source "$OHMYTERMUX_CONFIG_DIR/theme_config.tmp"
    fi

    # Create necessary directories
    execute_command "
        mkdir -p \"$PROOT_DEBIAN_ROOT/usr/share/themes\"
        mkdir -p \"$PROOT_DEBIAN_ROOT/usr/share/icons\"
        mkdir -p \"$PROOT_DEBIAN_ROOT/usr/share/backgrounds/whitesur\"
        mkdir -p \"$PROOT_DEBIAN_ROOT/home/$USERNAME/.fonts/\"
        mkdir -p \"$PROOT_DEBIAN_ROOT/home/$USERNAME/.themes/\"
    " "$(t "MSG_PROOT_CREATING_DIRECTORIES")"

    # Skip GTK themes for LXQt (uses Qt/Kvantum themes instead)
    if [ "${DESKTOP_ENV}" = "lxqt" ]; then
        INSTALL_THEME=false
    fi

    # Copy themes if installed
    if [ "$INSTALL_THEME" = true ] && [ -n "$SELECTED_THEME" ]; then
        copy_theme "$SELECTED_THEME"
    fi

    # Copy icons if installed
    if [ "$INSTALL_ICONS" = true ] && [ -n "$SELECTED_ICON_THEME" ]; then
        copy_icons "$SELECTED_ICON_THEME"
    fi

    # Copy wallpapers if installed
    if [ "$INSTALL_WALLPAPERS" = true ]; then
        execute_command "cp -r $PREFIX/share/backgrounds/whitesur/* $PROOT_DEBIAN_ROOT/usr/share/backgrounds/whitesur/" "$(t "MSG_PROOT_WALLPAPERS_CONFIG")"
    fi

    # Cursors configuration
    if [ "$INSTALL_CURSORS" = true ]; then
        cd "$PREFIX/share/icons"
        execute_command "find dist-dark | cpio -pdm \"$PROOT_DEBIAN_ROOT/usr/share/icons\"" "$(t "MSG_PROOT_CURSORS_CONFIG")"

        # Xresources configuration
        cat << EOF > "$PROOT_DEBIAN_ROOT/home/$USERNAME/.Xresources"
Xcursor.theme: dist-dark
EOF
    fi

    # Delete the temporary configuration file
    rm -f "$OHMYTERMUX_CONFIG_DIR/theme_config.tmp"
}

#------------------------------------------------------------------------------
# MAIN FUNCTION
#------------------------------------------------------------------------------
check_dependencies
title_msg "$(t "MSG_PROOT_DEBIAN_INSTALLATION")"

if [ $# -eq 0 ] && [ -z "$PROOT_USERNAME" ] && [ -z "$PROOT_PASSWORD" ]; then
    if [ "$USE_GUM" = true ]; then
        PROOT_USERNAME=$(gum input --prompt "$(t "MSG_PROOT_USERNAME_PROMPT")" --placeholder "$(t "MSG_PROOT_USERNAME_PLACEHOLDER")")
        while true; do
            PROOT_PASSWORD=$(gum input --password --prompt "$(t "MSG_PROOT_PASSWORD_PROMPT")" --placeholder "$(t "MSG_PROOT_PASSWORD_PLACEHOLDER")")
            PASSWORD_CONFIRM=$(gum input --password --prompt "$(t "MSG_PROOT_CONFIRM_PROMPT")" --placeholder "$(t "MSG_PROOT_CONFIRM_PLACEHOLDER")")
            if [ "$PROOT_PASSWORD" = "$PASSWORD_CONFIRM" ]; then
                break
            else
                gum style --foreground 196 "$(t "MSG_PROOT_PASSWORDS_NO_MATCH")"
            fi
        done
    else
        echo -e "${COLOR_BLUE}$(t "MSG_PROOT_ENTER_USERNAME")${COLOR_RESET}"
        read -r PROOT_USERNAME
        tput cuu1
        tput el
        while true; do
            echo -e "${COLOR_BLUE}$(t "MSG_PROOT_ENTER_PASSWORD")${COLOR_RESET}"
            read -rs PROOT_PASSWORD
            tput cuu1
            tput el
            echo -e "${COLOR_BLUE}$(t "MSG_PROOT_CONFIRM_PASSWORD")${COLOR_RESET}"
            read -rs PASSWORD_CONFIRM
            tput cuu1
            tput el
            if [ "$PROOT_PASSWORD" = "$PASSWORD_CONFIRM" ]; then
                break
            else
                echo -e "${COLOR_RED}$(t "MSG_PROOT_PASSWORDS_NO_MATCH")${COLOR_RESET}"
                tput cuu1
                tput el
            fi
        done
    fi
elif [ $# -eq 1 ] && [ -z "$PROOT_PASSWORD" ]; then
    PROOT_USERNAME="$1"
    if [ "$USE_GUM" = true ]; then
        while true; do
            PROOT_PASSWORD=$(gum input --password --prompt "$(t "MSG_PROOT_PASSWORD_PROMPT")" --placeholder "$(t "MSG_PROOT_PASSWORD_PLACEHOLDER")")
            PASSWORD_CONFIRM=$(gum input --password --prompt "$(t "MSG_PROOT_CONFIRM_PROMPT")" --placeholder "$(t "MSG_PROOT_CONFIRM_PLACEHOLDER")")
            if [ "$PROOT_PASSWORD" = "$PASSWORD_CONFIRM" ]; then
                break
            else
                gum style --foreground 196 "$(t "MSG_PROOT_PASSWORDS_NO_MATCH")"
            fi
        done
    else
        while true; do
            echo -e "${COLOR_BLUE}$(t "MSG_PROOT_ENTER_PASSWORD")${COLOR_RESET}"
            read -rs PROOT_PASSWORD
            tput cuu1
            tput el
            echo -e "${COLOR_BLUE}$(t "MSG_PROOT_CONFIRM_PASSWORD")${COLOR_RESET}"
            read -rs PASSWORD_CONFIRM
            tput cuu1
            tput el
            if [ "$PROOT_PASSWORD" = "$PASSWORD_CONFIRM" ]; then
                break
            else
                echo -e "${COLOR_RED}$(t "MSG_PROOT_PASSWORDS_NO_MATCH")${COLOR_RESET}"
                tput cuu1
                tput el
            fi
        done
    fi
elif [ $# -eq 2 ] && [ -z "$PROOT_USERNAME" ] && [ -z "$PROOT_PASSWORD" ]; then
    PROOT_USERNAME="$1"
    PROOT_PASSWORD="$2"
fi

execute_command "proot-distro install debian" "$(t "MSG_PROOT_DISTRIBUTION_INSTALLATION")"

#------------------------------------------------------------------------------
# DEBIAN INSTALLATION CHECK
#------------------------------------------------------------------------------
if [ ! -d "$PROOT_DEBIAN_ROOT" ]; then
    error_msg "$(t "MSG_PROOT_DEBIAN_FAILED")"
    exit 1
fi

execute_command "proot-distro login debian --shared-tmp -- env DISPLAY=:1.0 apt update" "$(t "MSG_PROOT_UPDATE_SEARCH")"
execute_command "proot-distro login debian --shared-tmp -- env DISPLAY=:1.0 apt upgrade -y" "$(t "MSG_PROOT_UPDATE_PACKAGES")"

install_packages_proot

subtitle_msg "$(t "MSG_PROOT_DISTRIBUTION_CONFIG")"

# Use PROOT_USERNAME and PROOT_PASSWORD for create_user_proot
USERNAME="$PROOT_USERNAME"
PASSWORD="$PROOT_PASSWORD"
create_user_proot
configure_user_rights

#------------------------------------------------------------------------------
# TIMEZONE CONFIGURATION
#------------------------------------------------------------------------------
TIMEZONE=$(getprop persist.sys.timezone)
execute_command "
    proot-distro login debian -- rm /etc/localtime
    proot-distro login debian -- cp /usr/share/zoneinfo/$TIMEZONE /etc/localtime
" "$(t "MSG_PROOT_TIMEZONE_CONFIG_MSG")"

#------------------------------------------------------------------------------
# GRAPHIC CONFIGURATION
#------------------------------------------------------------------------------
configure_themes_and_icons

#------------------------------------------------------------------------------
# MESA-VULKAN INSTALLATION
#------------------------------------------------------------------------------
install_mesa_vulkan