install.sh

#!/bin/bash
################################################################################
# System Monitor - Installation Script v1.0
#
# Installs System Monitor to $HOME/.local/bin/ and configures shell alias
#
# Usage:
#   bash install.sh
#
# Environment Variables:
#   INSTALL_DIR: Custom installation directory (default: $HOME/.local/bin)
#   SHELL_RC: Custom shell rc file (default: $HOME/.bashrc for bash)
#
# Privacy Note:
#   By default, this tool fetches your geolocation using your public IP.
#   See PRIVACY.md for details. To disable, use: MONITOR_OFFLINE=1 monitor
#
################################################################################

set -euo pipefail

# Configuration
INSTALL_DIR="${INSTALL_DIR:-$HOME/.local/bin}"
MONITOR_REPO="$(cd "$(dirname "$0")" && pwd)"
MONITOR_SRC="$MONITOR_REPO/src/monitor.py"
MONITOR_BIN="$INSTALL_DIR/monitor"

# Detect shell RC file
if [[ -n "${BASH_VERSION:-}" ]]; then
    SHELL_RC="${SHELL_RC:-$HOME/.bashrc}"
elif [[ -n "${ZSH_VERSION:-}" ]]; then
    SHELL_RC="${SHELL_RC:-$HOME/.zshrc}"
else
    SHELL_RC="${SHELL_RC:-$HOME/.bashrc}"
fi

# Color output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

# Functions
log_info() {
    echo -e "${GREEN}✓${NC} $*"
}

log_warn() {
    echo -e "${YELLOW}⚠${NC} $*"
}

log_error() {
    echo -e "${RED}✗${NC} $*" >&2
}

check_dependencies() {
    local missing=()
    
    for cmd in python3 curl free df uptime ip tput; do
        if ! command -v "$cmd" &> /dev/null; then
            missing+=("$cmd")
        fi
    done
    
    if [[ ${#missing[@]} -gt 0 ]]; then
        log_error "Missing required dependencies: ${missing[*]}"
        log_warn "Install them with your package manager, e.g.:"
        log_warn "  Ubuntu/Debian: sudo apt install ${missing[*]}"
        log_warn "  Fedora: sudo dnf install ${missing[*]}"
        log_warn "  Arch: sudo pacman -S ${missing[*]}"
        return 1
    fi
    return 0
}

check_python_version() {
    local py_version
    py_version=$(python3 --version 2>&1 | grep -oP '\d+\.\d+' | head -1)
    
    if [[ -z "$py_version" ]]; then
        log_error "Could not determine Python version"
        return 1
    fi
    
    # Compare versions: check if 3.6+
    local major minor
    major=$(echo "$py_version" | cut -d. -f1)
    minor=$(echo "$py_version" | cut -d. -f2)
    
    if [[ $major -lt 3 ]] || ( [[ $major -eq 3 ]] && [[ $minor -lt 6 ]] ); then
        log_error "Python 3.6+ required, found $py_version"
        return 1
    fi
    return 0
}

install_monitor() {
    log_info "Creating installation directory: $INSTALL_DIR"
    mkdir -p "$INSTALL_DIR"
    
    if [[ ! -f "$MONITOR_SRC" ]]; then
        log_error "Source file not found: $MONITOR_SRC"
        log_warn "Make sure you're running this script from the project root"
        return 1
    fi
    
    log_info "Installing monitor script..."
    cp "$MONITOR_SRC" "$MONITOR_BIN"
    chmod +x "$MONITOR_BIN"
    
    # Verify installation
    if [[ ! -x "$MONITOR_BIN" ]]; then
        log_error "Failed to install monitor binary"
        return 1
    fi
    
    log_info "Installed: $MONITOR_BIN"
}

configure_alias() {
    if [[ ! -f "$SHELL_RC" ]]; then
        log_warn "Shell RC file not found: $SHELL_RC"
        log_info "You can manually add this alias:  alias monitor='python3 \"$MONITOR_BIN\"'"
        return 0
    fi
    
    log_info "Configuring shell alias in: $SHELL_RC"
    
    # Remove old alias if exists
    sed -i '/# SYSTEM MONITOR INSTALLED/,+1d' "$SHELL_RC" 2>/dev/null || true
    
    # Add new alias
    cat >> "$SHELL_RC" << EOF

# SYSTEM MONITOR INSTALLED
alias monitor='python3 "$MONITOR_BIN"'
EOF
    
    log_info "Alias configured"
}

test_installation() {
    log_info "Testing installation..."
    
    if ! timeout 3 python3 "$MONITOR_BIN" 2>/dev/null || true; then
        log_info "Monitor started successfully (test timeout after 3s)"
    fi
    
    return 0
}

# Main installation flow
main() {
    echo -e "${GREEN}🚀 System Monitor Installer v1.0${NC}\n"
    
    log_info "Checking dependencies..."
    if ! check_dependencies; then
        return 1
    fi
    
    log_info "Checking Python version..."
    if ! check_python_version; then
        return 1
    fi
    
    if ! install_monitor; then
        return 1
    fi
    
    if ! configure_alias; then
        log_warn "Alias configuration incomplete"
    fi
    
    if ! test_installation; then
        log_warn "Installation test failed"
    fi
    
    echo
    echo -e "${GREEN}✓ Installation complete!${NC}"
    echo
    echo "Next steps:"
    echo "  1. Reload your shell: source $SHELL_RC"
    echo "  2. Run the monitor: monitor"
    echo
    echo "Options:"
    echo "  MONITOR_CYCLES=5 monitor         # Run 5 cycles instead of infinite"
    echo "  MONITOR_INTERVAL=3 monitor       # Update every 3 seconds"
    echo "  MONITOR_OFFLINE=1 monitor        # Run without internet/geo-IP"
    echo "  MONITOR_DEBUG=1 monitor          # Enable debug output"
    echo
}

# Run main with error handling
if main; then
    exit 0
else
    log_error "Installation failed"
    exit 1
fi