#!/usr/bin/env python3
import os

import funct
import sql
from jinja2 import Environment, FileSystemLoader
env = Environment(loader=FileSystemLoader('templates/'), autoescape=True)
template = env.get_template('delver.html')

print('Content-type: text/html\n')
funct.check_login()
funct.page_for_admin(level=3)

form = funct.form
serv = form.getvalue('serv')
Select = form.getvalue('del')
configver = form.getvalue('configver')
service = form.getvalue('service')
conf_format = 'cfg'
configs_dir = ''
stderr = ""
aftersave = ""
file = set()

if configver:
	template = env.get_template('configver.html')

try:
	user, user_id, role, token, servers, user_services = funct.get_users_params(disable=1)
except Exception:
	pass

if service == 'keepalived':
	if funct.check_login(service=3):
		configs_dir = funct.get_config_var('configs', 'kp_save_configs_dir')
		title = "Working with versions Keepalived configs"
		conf_format = 'conf'
		servers = sql.get_dick_permit(keepalived=1)
		action = 'versions.py?service=keepalived'
elif service == 'nginx':
	if funct.check_login(service=2):
		configs_dir = funct.get_config_var('configs', 'nginx_save_configs_dir')
		title = "Working with versions NGINX configs"
		conf_format = 'conf'
		servers = sql.get_dick_permit(nginx=1)
		action = 'versions.py?service=nginx'
elif service == 'apache':
	if funct.check_login(service=4):
		configs_dir = funct.get_config_var('configs', 'apache_save_configs_dir')
		title = "Working with versions Apache configs"
		conf_format = 'conf'
		servers = sql.get_dick_permit(apache=1)
		action = 'versions.py?service=apache'
else:
	service = 'haproxy'
	if funct.check_login(service=1):
		title = "Working with versions HAProxy configs"
		configs_dir = funct.get_config_var('configs', 'haproxy_save_configs_dir')
		servers = sql.get_dick_permit(haproxy=1)
		action = "versions.py"

if serv is not None and form.getvalue('del') is not None:
	if Select is not None:
		aftersave = 1
		env = Environment(loader=FileSystemLoader('templates/'))
		template = env.get_template('delver.html')
		for get in form:
			if conf_format in get and serv in get:
				try:
					if form.getvalue('style') == 'new':
						if sql.delete_config_version(service, form.getvalue(get)):
							try:
								os.remove(form.getvalue(get))
							except OSError as e:
								if 'No such file or directory' in str(e):
									pass
					else:
						os.remove(os.path.join(configs_dir, form.getvalue(get)))
					try:
						file.add(form.getvalue(get) + "<br />")
						funct.logging(
							serv, "Version of config has been deleted: %s" % form.getvalue(get), login=1, keep_history=1, service=service
						)
					except Exception:
						pass
				except OSError as e:
					stderr = "Error: %s - %s." % (e.filename, e.strerror)

if serv is not None and form.getvalue('config') is not None:
	configver = configs_dir + configver
	save = form.getvalue('save')
	aftersave = 1

	try:
		funct.logging(
			serv, "Version of config has been uploaded %s" % configver, login=1, keep_history=1, service=service
		)
	except Exception:
		pass

	if service == 'keepalived':
		stderr = funct.upload_and_restart(serv, configver, just_save=save, keepalived=1)
	elif service == 'nginx':
		config_file_name = sql.select_remote_path_from_version(server_ip=serv, service=service, local_path=configver)
		stderr = funct.master_slave_upload_and_restart(serv, configver, just_save=save, nginx=1, config_file_name=config_file_name)
	elif service == 'apache':
		config_file_name = sql.select_remote_path_from_version(server_ip=serv, service=service, local_path=configver)
		stderr = funct.master_slave_upload_and_restart(serv, configver, just_save=save, apache=1, config_file_name=config_file_name)
	else:
		stderr = funct.master_slave_upload_and_restart(serv, configver, just_save=save)


rendered_template = template.render(
	h2=1, title=title, role=role, user=user, select_id="serv", serv=serv, aftersave=aftersave, selects=servers,
	stderr=stderr, open=form.getvalue('open'), Select=form.getvalue('del'), file=file, configver=configver,
	service=service, user_services=user_services, action=action, token=token
)
print(rendered_template)