Skip to content

Commit

Permalink
Update install.sh
Browse files Browse the repository at this point in the history
  • Loading branch information
@fabriziosalmi
fabriziosalmi authored Sep 4, 2024
1 parent 82043c9 commit f2221d0
Showing 1 changed file with 45 additions and 100 deletions.
145 changes: 45 additions & 100 deletions install.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -10,10 +10,10 @@ GREEN=$(tput setaf 2)
RED=$(tput setaf 1)
YELLOW=$(tput setaf 3)
BLUE=$(tput setaf 4)
CHECKMARK="\xE2\x9C\x85" # ✔️ Success
CROSSMARK="\xE2\x9D\x8C" # ❌ Error
WARNING="\xE2\x9A\xA0" # ⚠️ Warning
INFO="\xE2\x8F\xB3" # ⏳ Info
CHECKMARK="\xE2\x9C\x85"
CROSSMARK="\xE2\x9D\x8C"
WARNING="\xE2\x9A\xA0"
INFO="\xE2\x8F\xB3"

# Log function with emojis
log() {
Expand All @@ -22,17 +22,7 @@ log() {
local emoji="$3"
local timestamp
timestamp=$(date +"%Y-%m-%d %H:%M:%S")
case $level in
"INFO")
echo -e "${timestamp} [${GREEN}${level}${RESET}] ${emoji} ${message}" | tee -a "$LOGFILE"
;;
"ERROR")
echo -e "${timestamp} [${RED}${level}${RESET}] ${emoji} ${message}" | tee -a "$LOGFILE"
;;
"WARNING")
echo -e "${timestamp} [${YELLOW}${level}${RESET}] ${emoji} ${message}" | tee -a "$LOGFILE"
;;
esac
echo -e "${timestamp} [${level}] ${emoji} ${message}" | tee -a "$LOGFILE"
}

# ASCII Art Header
Expand All @@ -48,15 +38,13 @@ header() {
check_software() {
log "INFO" "Checking for required software..." "$INFO"

# Check for Python3
if ! command -v python3 &> /dev/null; then
log "ERROR" "Python3 is not installed. Please install Python3 and rerun the script." "$CROSSMARK"
exit 1
else
log "INFO" "Python3 is installed." "$CHECKMARK"
fi

# Install required system packages
REQUIRED_SYSTEM_PACKAGES=("git" "python3-flask" "python3-requests" "python3-sklearn" "python3-pandas" "python3-numpy")

for package in "${REQUIRED_SYSTEM_PACKAGES[@]}"; do
Expand Down Expand Up @@ -104,99 +92,50 @@ backup_files() {
install_lxc_autoscale_ml() {
log "INFO" "Installing LXC AutoScale ML..." "$INFO"

# Disable and stop lxc_autoscale if running. Don't use both at the same time
systemctl disable lxc_autoscale 2>/dev/null
systemctl stop lxc_autoscale 2>/dev/null

# Stop lxc_autoscale_ml if running
systemctl stop lxc_autoscale_ml 2>/dev/null
# Disable and stop services if running
systemctl disable lxc_autoscale_ml.service lxc_autoscale_api.service lxc_monitor.service 2>/dev/null
systemctl stop lxc_autoscale_ml.service lxc_autoscale_api.service lxc_monitor.service 2>/dev/null

# Reload systemd
systemctl daemon-reload

# Create necessary directories
mkdir -p /etc/lxc_autoscale_ml /usr/local/bin/lxc_autoscale_api /usr/local/bin/lxc_autoscale_ml

# Create empty __init__.py files to treat directories as Python packages
touch /usr/local/bin/lxc_autoscale_ml/__init__.py /usr/local/bin/lxc_autoscale_api/__init__.py
# Clone the repository
REPO_URL="https://github.com/fabriziosalmi/proxmox-lxc-autoscale-ml.git"
TEMP_DIR=$(mktemp -d)

# Helper function to download files
download_file() {
local url="$1"
local destination="$2"
if curl -sSL -o "$destination" "$url"; then
log "INFO" "Downloaded $url to $destination" "$CHECKMARK"
else
log "ERROR" "Failed to download $url" "$CROSSMARK"
exit 1
fi
}

# Helper function to set up services
setup_service() {
local service_name="$1"
if systemctl enable "$service_name" && systemctl start "$service_name"; then
log "INFO" "${CHECKMARK} Service $service_name started successfully!" "$CHECKMARK"
else
log "ERROR" "${CROSSMARK} Failed to start service $service_name." "$CROSSMARK"
fi
}

# Download and install API application files
BASE_URL="https://raw.githubusercontent.com/fabriziosalmi/proxmox-lxc-autoscale-ml/main"
API_FILES=(
"api/lxc_autoscale_api.py"
"api/cloning_management.py"
"api/config.py"
"api/error_handling.py"
"api/health_check.py"
"api/lxc_management.py"
"api/resource_checking.py"
"api/scaling.py"
"api/snapshot_management.py"
"api/utils.py"
)
log "INFO" "Cloning the repository from $REPO_URL..." "$INFO"
if git clone "$REPO_URL" "$TEMP_DIR"; then
log "INFO" "Successfully cloned the repository." "$CHECKMARK"
else
log "ERROR" "Failed to clone the repository." "$CROSSMARK"
exit 1
fi

for file in "${API_FILES[@]}"; do
download_file "$BASE_URL/$file" "/usr/local/bin/lxc_autoscale_api/$(basename $file)"
done
# Move files to the appropriate locations
log "INFO" "Moving files to their respective directories..." "$INFO"

# Download and install the API configuration file
download_file "$BASE_URL/api/lxc_autoscale_api.yaml" "/etc/lxc_autoscale_ml/lxc_autoscale_api.yaml"

# Download and install the monitor application files
download_file "$BASE_URL/monitor/lxc_monitor.py" "/usr/local/bin/lxc_monitor.py"
download_file "$BASE_URL/api/lxc_autoscale_api.yaml" "/etc/lxc_autoscale_ml/lxc_monitor.yaml"

# Download and install model application files
MODEL_FILES=(
"model/config_manager.py"
"model/data_manager.py"
"model/lock_manager.py"
"model/logger.py"
"model/model.py"
"model/scaling.py"
"model/signal_handler.py"
"model/lxc_autoscale_ml.py"
)
# Move API files
mv "$TEMP_DIR/lxc_autoscale_ml/api/"*.py /usr/local/bin/lxc_autoscale_api/
mv "$TEMP_DIR/lxc_autoscale_ml/api/lxc_autoscale_api.yaml" /etc/lxc_autoscale_ml/

for file in "${MODEL_FILES[@]}"; do
download_file "$BASE_URL/$file" "/usr/local/bin/lxc_autoscale_ml/$(basename $file)"
done
# Move Monitor files
mv "$TEMP_DIR/lxc_autoscale_ml/monitor/lxc_monitor.py" /usr/local/bin/
mv "$TEMP_DIR/lxc_autoscale_ml/monitor/lxc_monitor.yaml" /etc/lxc_autoscale_ml/

# Download and install the model configuration file
download_file "$BASE_URL/model/lxc_autoscale_ml.yaml" "/etc/lxc_autoscale_ml/lxc_autoscale_ml.yaml"
# Move Model files
mv "$TEMP_DIR/lxc_autoscale_ml/model/"*.py /usr/local/bin/lxc_autoscale_ml/
mv "$TEMP_DIR/lxc_autoscale_ml/model/lxc_autoscale_ml.yaml" /etc/lxc_autoscale_ml/

# Corrected URLs for the service files
SERVICE_FILES=(
"lxc_autoscale_ml/api/lxc_autoscale_api.service"
"lxc_autoscale_ml/model/lxc_autoscale_ml.service"
"lxc_autoscale_ml/monitor/lxc_monitor.service"
)
# Move service unit files
mv "$TEMP_DIR/lxc_autoscale_ml/api/lxc_autoscale_api.service" /etc/systemd/system/
mv "$TEMP_DIR/lxc_autoscale_ml/monitor/lxc_monitor.service" /etc/systemd/system/
mv "$TEMP_DIR/lxc_autoscale_ml/model/lxc_autoscale_ml.service" /etc/systemd/system/

for file in "${SERVICE_FILES[@]}"; do
download_file "$BASE_URL/$file" "/etc/systemd/system/$(basename $file)"
done
# Clean up temporary directory
rm -rf "$TEMP_DIR"

# Make the main scripts executable
chmod +x /usr/local/bin/lxc_autoscale_ml/lxc_autoscale_ml.py
Expand All @@ -210,11 +149,17 @@ install_lxc_autoscale_ml() {
setup_service "lxc_autoscale_api.service"
setup_service "lxc_monitor.service"
setup_service "lxc_autoscale_ml.service"
}

# Show status for all installed services
systemctl status lxc_monitor.service --no-pager
systemctl status lxc_autoscale_api.service --no-pager
systemctl status lxc_autoscale_ml.service --no-pager
# Helper function to set up services
setup_service() {
local service_name="$1"
if systemctl enable "$service_name" && systemctl start "$service_name"; then
log "INFO" "${CHECKMARK} Service $service_name started successfully!" "$CHECKMARK"
else
log "ERROR" "${CROSSMARK} Failed to start service $service_name." "$CROSSMARK"
systemctl status "$service_name" --no-pager
fi
}

# Main script execution
Expand Down

0 comments on commit f2221d0

Please sign in to comment.