313 lines
12 KiB
Bash
313 lines
12 KiB
Bash
#!/bin/bash
|
|
|
|
#=======================================================================#
|
|
# Copyright (C) 2020 - 2022 Dominik Willner <th33xitus@gmail.com> #
|
|
# #
|
|
# This file is part of KIAUH - Klipper Installation And Update Helper #
|
|
# https://github.com/th33xitus/kiauh #
|
|
# #
|
|
# This file may be distributed under the terms of the GNU GPLv3 license #
|
|
#=======================================================================#
|
|
|
|
set -e
|
|
|
|
### global variables
|
|
NGINX_SA="/etc/nginx/sites-available"
|
|
NGINX_SE="/etc/nginx/sites-enabled"
|
|
NGINX_CONFD="/etc/nginx/conf.d"
|
|
|
|
#===================================================#
|
|
#=================== REMOVE NGINX ==================#
|
|
#===================================================#
|
|
|
|
function remove_nginx(){
|
|
if [ -f "${SYSTEMD}/nginx.service" ]; then
|
|
status_msg "Stopping Nginx service ..."
|
|
sudo systemctl stop nginx && sudo systemctl disable nginx
|
|
ok_msg "Service stopped and disabled!"
|
|
status_msg "Purging Nginx from system ..."
|
|
sudo apt-get purge nginx nginx-common -y
|
|
sudo update-rc.d -f nginx remove
|
|
print_confirm "Nginx successfully removed!"
|
|
else
|
|
print_error "Looks like Nginx was already removed!\n Skipping..."
|
|
fi
|
|
}
|
|
|
|
#===================================================#
|
|
#===================== HELPERS =====================#
|
|
#===================================================#
|
|
|
|
function set_upstream_nginx_cfg(){
|
|
local current_date
|
|
current_date=$(get_date)
|
|
local upstreams="${NGINX_CONFD}/upstreams.conf"
|
|
local common_vars="${NGINX_CONFD}/common_vars.conf"
|
|
### backup existing nginx configs
|
|
[ ! -d "${BACKUP_DIR}/nginx_cfg" ] && mkdir -p "${BACKUP_DIR}/nginx_cfg"
|
|
if [ -f "${upstreams}" ]; then
|
|
sudo mv "${upstreams}" "${BACKUP_DIR}/nginx_cfg/${current_date}_upstreams.conf"
|
|
fi
|
|
if [ -f "${common_vars}" ]; then
|
|
sudo mv "${common_vars}" "${BACKUP_DIR}/nginx_cfg/${current_date}_common_vars.conf"
|
|
fi
|
|
### transfer ownership of backed up files from root to ${USER}
|
|
local files
|
|
files=$(find "${BACKUP_DIR}/nginx_cfg")
|
|
for file in ${files}; do
|
|
if [ "$(stat -c "%U" "${file}" )" != "${USER}" ]; then
|
|
log_info "chown for user: ${USER} on file: ${file}"
|
|
sudo chown "${USER}" "${file}"
|
|
fi
|
|
done
|
|
### copy nginx configs to target destination
|
|
[ ! -f "${upstreams}" ] && sudo cp "${RESOURCES}/upstreams.conf" "${upstreams}"
|
|
[ ! -f "${common_vars}" ] && sudo cp "${RESOURCES}/common_vars.conf" "${common_vars}"
|
|
}
|
|
|
|
function symlink_webui_nginx_log(){
|
|
local interface=${1} path="${HOME}/klipper_logs"
|
|
local access_log="/var/log/nginx/${interface}-access.log"
|
|
local error_log="/var/log/nginx/${interface}-error.log"
|
|
[ ! -d "${path}" ] && mkdir -p "${path}"
|
|
if [ -f "${access_log}" ] && [ ! -L "${path}/${interface}-access.log" ]; then
|
|
status_msg "Creating symlink for ${access_log} ..."
|
|
ln -s "${access_log}" "${path}"
|
|
ok_msg "Done!"
|
|
fi
|
|
if [ -f "${error_log}" ] && [ ! -L "${path}/${interface}-error.log" ]; then
|
|
status_msg "Creating symlink for ${error_log} ..."
|
|
ln -s "${error_log}" "${path}"
|
|
ok_msg "Done!"
|
|
fi
|
|
}
|
|
|
|
function match_nginx_configs(){
|
|
read_kiauh_ini "${FUNCNAME[0]}"
|
|
local require_service_restart="false"
|
|
local upstreams="${NGINX_CONFD}/upstreams.conf"
|
|
local common_vars="${NGINX_CONFD}/common_vars.conf"
|
|
local mainsail_nginx_cfg="/etc/nginx/sites-available/mainsail"
|
|
local fluidd_nginx_cfg="/etc/nginx/sites-available/fluidd"
|
|
local upstreams_webcams mainsail_webcams fluidd_webcams
|
|
|
|
### reinstall nginx configs if the amount of upstreams don't match anymore
|
|
upstreams_webcams=$(grep -Ec "mjpgstreamer" "/etc/nginx/conf.d/upstreams.conf")
|
|
mainsail_webcams=$(grep -Ec "mjpgstreamer" "${mainsail_nginx_cfg}" 2>/dev/null || echo "0")
|
|
fluidd_webcams=$(grep -Ec "mjpgstreamer" "${fluidd_nginx_cfg}" 2>/dev/null || echo "0")
|
|
status_msg "Checking validity of NGINX configurations ..."
|
|
### check for outdated upstreams.conf
|
|
if ((upstreams_webcams < mainsail_webcams || upstreams_webcams < fluidd_webcams)); then
|
|
status_msg "Outdated upstreams.conf found! Updating ..."
|
|
sudo rm -f "${upstreams}" "${common_vars}"
|
|
set_upstream_nginx_cfg
|
|
require_service_restart="true"
|
|
ok_msg "Done!"
|
|
fi
|
|
### check for outdated mainsail config
|
|
if [ -e "${mainsail_nginx_cfg}" ] && ((upstreams_webcams > mainsail_webcams)); then
|
|
status_msg "Outdated Mainsail config found! Updating ..."
|
|
sudo rm -f "${mainsail_nginx_cfg}"
|
|
sudo cp "${RESOURCES}/klipper_webui_nginx.cfg" "${mainsail_nginx_cfg}"
|
|
sudo sed -i "s/<<UI>>/mainsail/g" "${mainsail_nginx_cfg}"
|
|
sudo sed -i "/root/s/pi/${USER}/" "${mainsail_nginx_cfg}"
|
|
sudo sed -i "s/listen\s[0-9]*;/listen ${mainsail_port};/" "${mainsail_nginx_cfg}"
|
|
sudo sed -i "s/listen\s\[\:*\]\:[0-9]*;/listen \[::\]\:${mainsail_port};/" "${mainsail_nginx_cfg}"
|
|
require_service_restart="true"
|
|
ok_msg "Done!"
|
|
fi
|
|
### check for outdated fluidd config
|
|
if [ -e "${fluidd_nginx_cfg}" ] && ((upstreams_webcams > fluidd_webcams)); then
|
|
status_msg "Outdated Fluidd config found! Updating ..."
|
|
sudo rm -f "${fluidd_nginx_cfg}"
|
|
sudo cp "${RESOURCES}/klipper_webui_nginx.cfg" "${fluidd_nginx_cfg}"
|
|
sudo sed -i "s/<<UI>>/fluidd/g" "${fluidd_nginx_cfg}"
|
|
sudo sed -i "/root/s/pi/${USER}/" "${fluidd_nginx_cfg}"
|
|
sudo sed -i "s/listen\s[0-9]*;/listen ${fluidd_port};/" "${fluidd_nginx_cfg}"
|
|
sudo sed -i "s/listen\s\[\:*\]\:[0-9]*;/listen \[::\]\:${fluidd_port};/" "${fluidd_nginx_cfg}"
|
|
require_service_restart="true"
|
|
ok_msg "Done!"
|
|
fi
|
|
### only restart nginx if configs were updated
|
|
[ "${require_service_restart}" == "true" ] && sudo systemctl restart nginx.service
|
|
}
|
|
|
|
function process_disruptive_services(){
|
|
#handle haproxy service
|
|
if [ "${DISABLE_HAPROXY}" = "true" ] || [ "${REMOVE_HAPROXY}" = "true" ]; then
|
|
if systemctl is-active haproxy -q; then
|
|
status_msg "Stopping haproxy service ..."
|
|
sudo systemctl stop haproxy && ok_msg "Service stopped!"
|
|
fi
|
|
|
|
### disable haproxy
|
|
if [ "${DISABLE_HAPROXY}" = "true" ]; then
|
|
status_msg "Disabling haproxy ..."
|
|
sudo systemctl disable haproxy && ok_msg "Haproxy service disabled!"
|
|
|
|
### remove haproxy
|
|
if [ "${REMOVE_HAPROXY}" = "true" ]; then
|
|
status_msg "Removing haproxy ..."
|
|
sudo apt-get remove haproxy -y && sudo update-rc.d -f haproxy remove && ok_msg "Haproxy removed!"
|
|
fi
|
|
fi
|
|
fi
|
|
|
|
### handle lighttpd service
|
|
if [ "${DISABLE_LIGHTTPD}" = "true" ] || [ "${REMOVE_LIGHTTPD}" = "true" ]; then
|
|
if systemctl is-active lighttpd -q; then
|
|
status_msg "Stopping lighttpd service ..."
|
|
sudo systemctl stop lighttpd && ok_msg "Service stopped!"
|
|
fi
|
|
|
|
### disable lighttpd
|
|
if [ "${DISABLE_LIGHTTPD}" = "true" ]; then
|
|
status_msg "Disabling lighttpd ..."
|
|
sudo systemctl disable lighttpd && ok_msg "Lighttpd service disabled!"
|
|
|
|
### remove lighttpd
|
|
if [ "${REMOVE_LIGHTTPD}" = "true" ]; then
|
|
status_msg "Removing lighttpd ..."
|
|
sudo apt-get remove lighttpd -y && sudo update-rc.d -f lighttpd remove && ok_msg "Lighttpd removed!"
|
|
fi
|
|
fi
|
|
fi
|
|
|
|
### handle apache2 service
|
|
if [ "${DISABLE_APACHE2}" = "true" ] || [ "${REMOVE_APACHE2}" = "true" ]; then
|
|
if systemctl is-active apache2 -q; then
|
|
status_msg "Stopping apache2 service ..."
|
|
sudo systemctl stop apache2 && ok_msg "Service stopped!"
|
|
fi
|
|
|
|
### disable lighttpd
|
|
if [ "${DISABLE_APACHE2}" = "true" ]; then
|
|
status_msg "Disabling lighttpd ..."
|
|
sudo systemctl disable apache2 && ok_msg "Apache2 service disabled!"
|
|
|
|
### remove lighttpd
|
|
if [ "${REMOVE_APACHE2}" = "true" ]; then
|
|
status_msg "Removing apache2 ..."
|
|
sudo apt-get remove apache2 -y && sudo update-rc.d -f apache2 remove && ok_msg "Apache2 removed!"
|
|
fi
|
|
fi
|
|
fi
|
|
}
|
|
|
|
function process_services_dialog(){
|
|
#notify user about haproxy or lighttpd services found and possible issues
|
|
if [ "${HAPROXY_FOUND}" = "true" ] || [ "${LIGHTTPD_FOUND}" = "true" ] || [ "${APACHE2_FOUND}" = "true" ]; then
|
|
while true; do
|
|
echo
|
|
top_border
|
|
echo -e "| ${red}Possibly disruptive/incompatible services found!${white} |"
|
|
hr
|
|
if [ "${HAPROXY_FOUND}" = "true" ]; then
|
|
echo -e "| ● haproxy |"
|
|
fi
|
|
if [ "${LIGHTTPD_FOUND}" = "true" ]; then
|
|
echo -e "| ● lighttpd |"
|
|
fi
|
|
if [ "${APACHE2_FOUND}" = "true" ]; then
|
|
echo -e "| ● apache2 |"
|
|
fi
|
|
hr
|
|
echo -e "| Having those packages installed can lead to unwanted |"
|
|
echo -e "| behaviour. It is recommend to remove those packages. |"
|
|
echo -e "| |"
|
|
echo -e "| 1) Remove packages (recommend) |"
|
|
echo -e "| 2) Disable only (may cause issues) |"
|
|
echo -e "| ${red}3) Skip this step (not recommended)${white} |"
|
|
bottom_border
|
|
read -p "${cyan}###### Please choose:${white} " action
|
|
case "${action}" in
|
|
1)
|
|
echo -e "###### > Remove packages"
|
|
REMOVE_HAPROXY="true"
|
|
REMOVE_LIGHTTPD="true"
|
|
REMOVE_APACHE2="true"
|
|
break;;
|
|
2)
|
|
echo -e "###### > Disable only"
|
|
DISABLE_HAPROXY="true"
|
|
DISABLE_LIGHTTPD="true"
|
|
DISABLE_APACHE2="true"
|
|
break;;
|
|
3)
|
|
echo -e "###### > Skip"
|
|
break;;
|
|
*)
|
|
print_unkown_cmd
|
|
print_msg && clear_msg;;
|
|
esac
|
|
done
|
|
fi
|
|
}
|
|
|
|
function set_nginx_cfg(){
|
|
local interface=${1}
|
|
if [ "${SET_NGINX_CFG}" = "true" ]; then
|
|
local cfg="${RESOURCES}/${interface}"
|
|
#check for dependencies
|
|
local dep=(nginx)
|
|
dependency_check "${dep[@]}"
|
|
#execute operations
|
|
status_msg "Creating Nginx configuration for ${interface^} ..."
|
|
#copy content from resources to the respective nginx config file
|
|
cat "${RESOURCES}/klipper_webui_nginx.cfg" > "${cfg}"
|
|
##edit the nginx config file before moving it
|
|
sed -i "s/<<UI>>/${interface}/g" "${cfg}"
|
|
if [ "${SET_LISTEN_PORT}" != "${DEFAULT_PORT}" ]; then
|
|
status_msg "Configuring port for ${interface^} ..."
|
|
#set listen port ipv4
|
|
sed -i "s/listen\s[0-9]*;/listen ${SET_LISTEN_PORT};/" "${cfg}"
|
|
#set listen port ipv6
|
|
sed -i "s/listen\s\[\:*\]\:[0-9]*;/listen \[::\]\:${SET_LISTEN_PORT};/" "${cfg}"
|
|
fi
|
|
#set correct user
|
|
if [ "${interface}" = "mainsail" ] || [ "${interface}" = "fluidd" ]; then
|
|
sudo sed -i "/root/s/pi/${USER}/" "${cfg}"
|
|
fi
|
|
#moving the config file into correct directory
|
|
sudo mv "${cfg}" "/etc/nginx/sites-available/${interface}"
|
|
ok_msg "Nginx configuration for ${interface^} was set!"
|
|
if [ -n "${SET_LISTEN_PORT}" ]; then
|
|
ok_msg "${interface^} configured for port ${SET_LISTEN_PORT}!"
|
|
else
|
|
ok_msg "${interface^} configured for default port ${DEFAULT_PORT}!"
|
|
fi
|
|
#remove nginx default config
|
|
if [ -e "/etc/nginx/sites-enabled/default" ]; then
|
|
sudo rm "/etc/nginx/sites-enabled/default"
|
|
fi
|
|
#create symlink for own sites
|
|
if [ ! -e "/etc/nginx/sites-enabled/${interface}" ]; then
|
|
sudo ln -s "/etc/nginx/sites-available/${interface}" "/etc/nginx/sites-enabled/"
|
|
fi
|
|
sudo systemctl restart nginx.service
|
|
fi
|
|
}
|
|
|
|
function read_listen_port(){
|
|
local port interface=${1}
|
|
port=$(grep listen "/etc/nginx/sites-enabled/${interface}" | head -1 | sed 's/^\s*//' | cut -d" " -f2 | cut -d";" -f1)
|
|
echo "${port}"
|
|
}
|
|
|
|
function detect_enabled_sites(){
|
|
MAINSAIL_ENABLED="false" FLUIDD_ENABLED="false" OCTOPRINT_ENABLED="false"
|
|
#check if there is another UI config already installed and reads the port they are listening on
|
|
if [ -e "/etc/nginx/sites-enabled/mainsail" ]; then
|
|
SITE_ENABLED="true" && MAINSAIL_ENABLED="true"
|
|
MAINSAIL_PORT=$(read_listen_port "mainsail")
|
|
fi
|
|
if [ -e "/etc/nginx/sites-enabled/fluidd" ]; then
|
|
SITE_ENABLED="true" && FLUIDD_ENABLED="true"
|
|
FLUIDD_PORT=$(read_listen_port "fluidd")
|
|
|
|
fi
|
|
if [ -e "/etc/nginx/sites-enabled/octoprint" ]; then
|
|
SITE_ENABLED="true" && OCTOPRINT_ENABLED="true"
|
|
OCTOPRINT_PORT=$(read_listen_port "octoprint")
|
|
fi
|
|
} |