rom.sh

#!/usr/bin/env bash

# ==============================================================================
# PROJECT: RPM-OSTree Manager
# AUTHOR: Diogo Pessoa (https://github.com/diogopessoa/rpm-ostree-manager/)
# LICENSE: MIT
# ==============================================================================

# --- Settings and Colors ---
VERSION="0.1.8"
REPO_API_URL="https://api.github.com/repos/diogopessoa/rpm-ostree-manager/releases"
REAL_USER=${SUDO_USER:-$USER}
USER_HOME=$(getent passwd "$REAL_USER" | cut -d: -f6)
DOWNLOADS_DIR="$USER_HOME/Downloads"

# Colors
GREEN='\033[0;32m'
BLUE='\033[0;34m'
RED='\033[0;31m'
PURPLE='\033[0;35m'
CYAN='\033[0;36m'
NC='\033[0m'
BOLD='\033[1m'

# --- Terminal Check (Force open in Ptyxis if launched from Menu) ---
if [[ ! -t 0 ]]; then
    if command -v ptyxis &> /dev/null; then
        ptyxis -- bash -c "$0; echo; echo 'Press any key to exit...'; read -n1"
    else
        gnome-terminal -- bash -c "$0; echo; echo 'Press any key to exit...'; read -n1"
    fi
    exit
fi

check_update() {
    # Fetches the first tag from the releases list to avoid cache issues
    LATEST_VERSION=$(curl -s --connect-timeout 2 "$REPO_API_URL" | jq -r '.[0].tag_name' 2>/dev/null | sed 's/v//' | xargs)

    if [[ -n "$LATEST_VERSION" && "$LATEST_VERSION" != "null" ]]; then
        if [[ "$LATEST_VERSION" != "$VERSION" ]]; then
            echo -e "${PURPLE}${BOLD}New version available: v$LATEST_VERSION${NC}"
            echo -e "${PURPLE}Update at: github.com/diogopessoa/rpm-ostree-manager${NC}\n"
        else
            echo -e "${GREEN}✔ System up to date${NC}\n"
        fi
    fi
}

show_menu() {
    clear
    echo "======================================"
    echo -e "        ${BOLD}RPM-OSTree Manager${NC}"
    echo "            Version $VERSION"
    echo "======================================"
    echo
    echo "[1] 📦 Install Local RPM (Downloads)"
    echo "[2] 🗑️  Remove Layered/Local RPM"
    echo "[3] ↩️  Rollback to Previous State"
    echo "[4] 📌 Pin/Unpin Deployment"
    echo "[5] ⚓ Switch Default Deployment"
    echo "[6] ☑️  Check Status"
    echo "[7] 🛟 Documentation & Help"
    echo "[0] ✖️  Exit"
    
    echo
    echo -e "${BLUE}────────────────────────────────────${NC}"
    echo -ne "${GREEN}Option:[0-7]${NC} "
}

install_rpm() {
    echo -e "\n--- ${BLUE}Install Local RPM${NC} ---"
    echo -e ""
    cd "$DOWNLOADS_DIR" || return
    
    # List .rpm files and store in an array
    mapfile -t files < <(ls *.rpm 2>/dev/null)
    
    if [ ${#files[@]} -eq 0 ]; then
        echo -e "${RED}No .rpm files found in $DOWNLOADS_DIR${NC}"
        echo "Tip: You can also drag and drop an RPM file here from any folder."
    else
        echo "Select a number OR drag and drop an RPM file here:"
        for i in "${!files[@]}"; do
            echo -e "$((i+1))) ${GREEN}${files[$i]}${NC}"
        done
        echo "0) Cancel"
    fi
    
    echo -ne "\nEnter number or drop file: "
    read -rp "" input
    input=$(echo "$input" | tr -d "'\"")

    if [[ -z "$input" ]]; then
        : # Do nothing, will trigger the Return prompt
    elif [[ "$input" == "0" ]]; then
        return
    elif [[ "$input" =~ ^[0-9]+$ ]] && [[ "$input" -le ${#files[@]} ]]; then
        selected_file="${files[$((input-1))]}"
        sudo rpm-ostree install "$selected_file"
        echo -e "${GREEN}Scheduled! Reboot to apply changes.${NC}"
    elif [[ -f "$input" && "$input" == *.rpm ]]; then
        sudo rpm-ostree install "$input"
        echo -e "${GREEN}Scheduled! Reboot to apply changes.${NC}"
    else
        echo -e "${RED}Invalid selection or file!${NC}"
    fi
    read -p "Press Enter to Return..."
}

remove_rpm() {
    echo -e "\n--- ${RED}Remove Layered / Local RPM${NC} ---"
    echo -e ""
        
    # Searches all possible keys store packages
    packages=$(rpm-ostree status --json | jq -r '.deployments[] | select(.booted == true) | 
        (.packages // []) + 
        (."local-packages" // []) + 
        (.local_packages // []) + 
        (."requested-local-packages" // []) | unique | .[]' 2>/dev/null)

    if [ -z "$packages" ]; then
        echo -e "${RED}No layered packages found.${NC}"
    else
        mapfile -t pkg_list < <(echo "$packages")
        echo "Select the package to remove:"
        for i in "${!pkg_list[@]}"; do
            echo -e "$((i+1))) ${RED}${pkg_list[$i]}${NC}"
        done
        echo "0) Cancel"

        read -rp "Enter number: " num
        if [[ -z "$num" ]]; then
            :
        elif [[ "$num" -gt 0 && "$num" -le ${#pkg_list[@]} ]]; then
            selected_pkg="${pkg_list[$((num-1))]}"
            # Clean package name (remove version if present)
            pkg_clean=$(echo "$selected_pkg" | sed 's/-[0-9].*//')
            sudo rpm-ostree uninstall "$pkg_clean"
            echo -e "${GREEN}Removal scheduled! Please reboot the system.${NC}"
        elif [[ "$num" == "0" ]]; then
            return
        else
            echo -e "${RED}Invalid selection!${NC}"
        fi
    fi
    read -p "Press Enter to Return..."
}

rollback() {
    echo -e "\n--- ${GREEN}System Rollback${NC} ---"
    echo -e ""
    read -p "Do you want to revert to the previous state? (y/N): " confirm
    
    if [[ "$confirm" =~ ^[Yy]$ ]]; then
        if sudo rpm-ostree rollback; then
            echo -e "\n${GREEN}Rollback successful!${NC}"
            echo -e "${BLUE}-----------------------------------------------------------------------------------------------------------${NC}"
            echo -e "Restart the system to use the selected deployment as the default."
            echo -e "If the Rollback occurred after updates broke the system, disable them temporarily until fixes are released."
        else
            echo -e "\n${RED}Rollback failed or was cancelled by the system.${NC}"
        fi
    else
        echo -e "\n${RED}Rollback cancelled by user.${NC}"
    fi
    
    echo ""
    read -p "Press Enter to Return..."
}

pin_manager() {
    echo -e "\n--- ${PURPLE}Pin/Unpin Deployment Manager${NC} ---"
    # echo "Fetching deployments..."

    # Extracts version and state from 'pinned'
    mapfile -t deploy_data < <(rpm-ostree status --json | jq -r '.deployments[] | "\(.version)|\(.pinned)"')

    if [ ${#deploy_data[@]} -eq 0 ]; then
        echo -e "${RED}No deployments found.${NC}"
    else
        echo -e "\nSelect a deployment to manage:"
        for i in "${!deploy_data[@]}"; do
            version=$(echo "${deploy_data[$i]}" | cut -d'|' -f1)
            is_pinned=$(echo "${deploy_data[$i]}" | cut -d'|' -f2)

            if [ "$is_pinned" == "true" ]; then
                echo -e "$((i+1))) ${GREEN}${version} (PINNED)${NC}"
            else
                echo -e "$((i+1))) ${BLUE}${version}${NC}"
            fi
        done
        echo "0) Cancel"

        echo -ne "\nEnter number: "
        read -r num
        
        # List deployments
        if [[ "$num" -gt 0 && "$num" -le ${#deploy_data[@]} ]]; then
            idx=$((num-1))
            selected_version=$(echo "${deploy_data[$idx]}" | cut -d'|' -f1)
            selected_pinned=$(echo "${deploy_data[$idx]}" | cut -d'|' -f2)
            
            echo -e "\nSelected: ${BOLD}$selected_version${NC}"
            
            if [ "$selected_pinned" == "true" ]; then
                echo -e "Status: ${GREEN}PINNED${NC}"
                echo -e "1) ${RED}Unpin${NC} (Remove deployment pinned)"
            else
                echo -e "Status: ${BLUE}NOT PINNED${NC}"
                echo -e "1) ${GREEN}Pin${NC} (Pin this version)"
            fi
            echo "0) Cancel"
            echo -ne "\nOption: "
            read -r action

            if [[ "$action" == "1" ]]; then
                if [ "$selected_pinned" == "true" ]; then
                    # Use the index (idx)
                    if sudo ostree admin pin --unpin "$idx"; then
                        echo -e "${GREEN}Version unpinned successfully!${NC}"
                    else
                        echo -e "${RED}Failed to unpin deployment.${NC}"
                    fi
                else
                    if sudo ostree admin pin "$idx"; then
                        echo -e "${GREEN}Version pinned successfully!${NC}"
                    else
                        echo -e "${RED}Failed to pin deployment.${NC}"
                    fi
                fi
            fi
        fi
    fi
    read -p "Press Enter to Return..."
}
        
switch_default_deployment() {
    echo -e "\n--- ${CYAN}Switch Default Deployment${NC} ---"
    echo -e ""
    echo "Select which deployment should be the default for the next boot:"
    
    mapfile -t deployments < <(rpm-ostree status --json | jq -r '.deployments[].version')
    
    if [ ${#deployments[@]} -eq 0 ]; then
        echo -e "${RED}No deployments found.${NC}"
    else
        for i in "${!deployments[@]}"; do
            echo -e "$((i+1))) ${BLUE}${deployments[$i]}${NC}"
        done
        echo "0) Cancel"
        
        read -rp "Enter number: " num
        if [[ "$num" -gt 0 && "$num" -le ${#deployments[@]} ]]; then
            idx=$((num-1))
            selected="${deployments[$idx]}"
            if sudo ostree admin set-default "$idx"; then
                echo -e "${GREEN}Success! Deployment $selected is now the default.${NC}"
                echo "Please reboot to start the selected version."
            else
                echo -e "${RED}Failed to set default deployment.${NC}"
            fi
        fi
    fi
    read -p "Press Enter to Return..."
}
    
show_help() {
    clear
    echo -e "${BLUE}╭──────────────────────────────────────────────────────────╮${NC}"
    echo -e "${BLUE}│${NC}                ${BOLD}QUICK DOCUMENTATION & HELP${NC}                ${BLUE}│${NC}"
    echo -e "${BLUE}╰──────────────────────────────────────────────────────────╯${NC}\n"

    echo -e "1) 📦 Install: Supports ${BOLD}Drag & Drop${NC} of .rpm files or selecting"
    echo -e "   files automatically detected in your Downloads folder."

    echo -e "\n2) 🗑️  Remove: Lists your layered packages numerically."
    echo -e "   The tool automatically cleans version tags for safe removal."

    echo -e "\n3) ↩️  Rollback: Swaps current deployment with previous on next boot,"
    echo -e "   reverting recent changes until stabilized or upgraded."

    echo -e "\n4) 📌 Pin/Unpin: Pinned deployments are ${BOLD}locked${NC} and protected"
    echo -e "   from being automatically deleted by the system."

    echo -e "\n5) ⚓ Switch Default: Sets a specific version as the new Anchor." 
    echo -e "   Future updates will grow from this chosen deployment."

    echo -e "\n6) ☑️  Status: Shows the detailed technical state of your OS."

    echo -e "\n${BLUE}────────────────────────────────────────────────────────────${NC}"
    echo -e "${BOLD}For the complete guide, visit:${NC}"
    echo -e "https://github.com/diogopessoa/rpm-ostree-manager/wiki"
    echo -e "${BLUE}────────────────────────────────────────────────────────────${NC}\n"

    read -p "Press Enter to return to menu..."
}

# --- Main Loop ---
while true; do
    show_menu
    read -r opt
    case "$opt" in
        1) install_rpm ;;
        2) remove_rpm ;;
        3) rollback ;;
        4) pin_manager ;;
        5) switch_default_deployment ;;
        6) clear; rpm-ostree status; echo ""; read -p "Press Enter to Return..." ;; 
        7) show_help ;; 
        0)      
            clear
            echo
            echo -e "    Follow for updates:"
            echo -e "    https://github.com/diogopessoa/rpm-ostree-manager"
            echo -e "    ${BLUE}-------------------------------------------------${NC}"
            echo 
            read -p "    Press Enter to Exit..."
            exit 0 
            ;;
        *)
            echo -e "${RED}Invalid option!${NC}"
            read -p "Press Enter to Return..."
            ;;
    esac
done