mirror of https://github.com/Aidaho12/haproxy-wi
99 lines
3.2 KiB
Python
99 lines
3.2 KiB
Python
|
import distro
|
||
|
|
|
||
|
|
import modules.roxywi.roxy as roxywi_mod
|
||
|
|
import modules.server.server as server_mod
|
||
|
|
|
||
|
|
|
||
|
|
def get_services_status():
|
||
|
|
services = []
|
||
|
|
is_in_docker = roxywi_mod.is_docker()
|
||
|
|
services_name = {
|
||
|
|
'roxy-wi-checker': '',
|
||
|
|
'roxy-wi-keep_alive': '',
|
||
|
|
'roxy-wi-metrics': '',
|
||
|
|
'roxy-wi-portscanner': '',
|
||
|
|
'roxy-wi-smon': '',
|
||
|
|
'roxy-wi-socket': '',
|
||
|
|
'roxy-wi-prometheus-exporter': 'Prometheus exporter',
|
||
|
|
'prometheus': 'Prometheus service',
|
||
|
|
'grafana-server': 'Grafana service',
|
||
|
|
'fail2ban': 'Fail2ban service',
|
||
|
|
'rabbitmq-server': 'Message broker service'
|
||
|
|
}
|
||
|
|
for s, v in services_name.items():
|
||
|
|
if is_in_docker:
|
||
|
|
cmd = f"sudo supervisorctl status {s}|awk '{{print $2}}'"
|
||
|
|
else:
|
||
|
|
cmd = f"systemctl is-active {s}"
|
||
|
|
|
||
|
|
status, stderr = server_mod.subprocess_execute(cmd)
|
||
|
|
|
||
|
|
if s != 'roxy-wi-keep_alive':
|
||
|
|
service_name = s.split('_')[0]
|
||
|
|
if s == 'grafana-server':
|
||
|
|
service_name = 'grafana'
|
||
|
|
elif s == 'roxy-wi-keep_alive' and distro.id() == 'ubuntu':
|
||
|
|
service_name = 'roxy-wi-keep-alive'
|
||
|
|
else:
|
||
|
|
service_name = s
|
||
|
|
|
||
|
|
if service_name == 'prometheus':
|
||
|
|
cmd = "prometheus --version 2>&1 |grep prometheus|awk '{print $3}'"
|
||
|
|
else:
|
||
|
|
if distro.id() == 'ubuntu':
|
||
|
|
cmd = f"apt list --installed 2>&1 |grep {service_name}|awk '{{print $2}}'|sed 's/-/./'"
|
||
|
|
else:
|
||
|
|
cmd = f"rpm -q {service_name}|awk -F\"{service_name}\" '{{print $2}}' |awk -F\".noa\" '{{print $1}}' |sed 's/-//1' |sed 's/-/./'"
|
||
|
|
service_ver, stderr = server_mod.subprocess_execute(cmd)
|
||
|
|
|
||
|
|
try:
|
||
|
|
if service_ver[0] == 'command' or service_ver[0] == 'prometheus:':
|
||
|
|
service_ver[0] = ''
|
||
|
|
except Exception:
|
||
|
|
pass
|
||
|
|
|
||
|
|
try:
|
||
|
|
services.append([s, status, v, service_ver[0]])
|
||
|
|
except Exception:
|
||
|
|
services.append([s, status, v, ''])
|
||
|
|
|
||
|
|
return services
|
||
|
|
|
||
|
|
|
||
|
|
def update_roxy_wi(service: str) -> str:
|
||
|
|
restart_service = ''
|
||
|
|
services = ['roxy-wi-checker',
|
||
|
|
'roxy-wi',
|
||
|
|
'roxy-wi-keep_alive',
|
||
|
|
'roxy-wi-smon',
|
||
|
|
'roxy-wi-metrics',
|
||
|
|
'roxy-wi-portscanner',
|
||
|
|
'roxy-wi-socket',
|
||
|
|
'roxy-wi-prometheus-exporter']
|
||
|
|
|
||
|
|
if service not in services:
|
||
|
|
raise Exception(f'error: {service} is not part of Roxy-WI')
|
||
|
|
|
||
|
|
if distro.id() == 'ubuntu':
|
||
|
|
try:
|
||
|
|
if service == 'roxy-wi-keep_alive':
|
||
|
|
service = 'roxy-wi-keep-alive'
|
||
|
|
except Exception:
|
||
|
|
pass
|
||
|
|
|
||
|
|
if service != 'roxy-wi':
|
||
|
|
restart_service = f'&& sudo systemctl restart {service}'
|
||
|
|
|
||
|
|
cmd = f'sudo -S apt-get update && sudo apt-get install {service} {restart_service}'
|
||
|
|
else:
|
||
|
|
if service != 'roxy-wi':
|
||
|
|
restart_service = f'&& sudo systemctl restart {service}'
|
||
|
|
cmd = f'sudo -S yum -y install {service} {restart_service}'
|
||
|
|
|
||
|
|
output, stderr = server_mod.subprocess_execute(cmd)
|
||
|
|
|
||
|
|
if stderr != '':
|
||
|
|
return str(stderr)
|
||
|
|
else:
|
||
|
|
return str(output)
|