import os

import paramiko

import modules.db.sql as sql
import modules.common.common as common
from modules.server import ssh_connection
import modules.roxywi.common as roxywi_common
import modules.roxy_wi_tools as roxy_wi_tools

form = common.form
error_mess = common.error_mess
get_config = roxy_wi_tools.GetConfigVar()


def return_ssh_keys_path(server_ip: str, **kwargs) -> dict:
	lib_path = get_config.get_config_var('main', 'lib_path')
	ssh_settings = {}

	if kwargs.get('id'):
		sshs = sql.select_ssh(id=kwargs.get('id'))
	else:
		sshs = sql.select_ssh(serv=server_ip)

	for ssh in sshs:
		ssh_settings.setdefault('enabled', ssh.enable)
		ssh_settings.setdefault('user', ssh.username)
		ssh_settings.setdefault('password', ssh.password)
		ssh_key = f'{lib_path}/keys/{ssh.name}.pem' if ssh.enable == 1 else ''
		ssh_settings.setdefault('key', ssh_key)

	ssh_port = [str(server[10]) for server in sql.select_servers(server=server_ip)]
	ssh_settings.setdefault('port', ssh_port[0])

	return ssh_settings


def ssh_connect(server_ip):
	ssh_settings = return_ssh_keys_path(server_ip)
	ssh = ssh_connection.SshConnection(server_ip, ssh_settings['port'], ssh_settings['user'],
										ssh_settings['password'], ssh_settings['enabled'], ssh_settings['key'])

	return ssh


def create_ssh_cred() -> None:
	from jinja2 import Environment, FileSystemLoader

	user_group = roxywi_common.get_user_group()
	name = common.checkAjaxInput(form.getvalue('new_ssh'))
	name = f'{name}_{user_group}'
	enable = common.checkAjaxInput(form.getvalue('ssh_enable'))
	group = common.checkAjaxInput(form.getvalue('new_group'))
	username = common.checkAjaxInput(form.getvalue('ssh_user'))
	password = common.checkAjaxInput(form.getvalue('ssh_pass'))
	page = common.checkAjaxInput(form.getvalue('page'))
	page = page.split("#")[0]

	if username is None or name is None:
		print(error_mess)
	else:
		if sql.insert_new_ssh(name, enable, group, username, password):
			env = Environment(loader=FileSystemLoader('templates/ajax'), autoescape=True)
			template = env.get_template('/new_ssh.html')
			output_from_parsed_template = template.render(groups=sql.select_groups(), sshs=sql.select_ssh(name=name), page=page)
			print(output_from_parsed_template)
			roxywi_common.logging('Roxy-WI server', f'A new SSH credentials {name} has created', roxywi=1, login=1)


def upload_ssh_key() -> None:
	user_group = roxywi_common.get_user_group()
	name = common.checkAjaxInput(form.getvalue('name'))

	try:
		key = paramiko.pkey.load_private_key(form.getvalue('ssh_cert'))
	except Exception as e:
		print(f'error: Cannot save SSH key file: {e}')
		return

	lib_path = get_config.get_config_var('main', 'lib_path')
	full_dir = f'{lib_path}/keys/'
	ssh_keys = f'{name}.pem'

	try:
		_check_split = name.split('_')[1]
		split_name = True
	except Exception:
		split_name = False

	if not os.path.isfile(ssh_keys) and not split_name:
		name = f'{name}_{user_group}'

	if not os.path.exists(full_dir):
		os.makedirs(full_dir)

	ssh_keys = f'{full_dir}{name}.pem'

	try:
		key.write_private_key_file(ssh_keys)
	except Exception as e:
		print(f'error: Cannot save SSH key file: {e}')
		return
	else:
		print(f'success: SSH key has been saved into: {ssh_keys}')

	try:
		os.chmod(ssh_keys, 0o600)
	except IOError as e:
		roxywi_common.logging('Roxy-WI server', e.args[0], roxywi=1)

	roxywi_common.logging("Roxy-WI server", f"A new SSH cert has been uploaded {ssh_keys}", roxywi=1, login=1)


def update_ssh_key() -> None:
	ssh_id = common.checkAjaxInput(form.getvalue('id'))
	name = common.checkAjaxInput(form.getvalue('name'))
	enable = common.checkAjaxInput(form.getvalue('ssh_enable'))
	group = common.checkAjaxInput(form.getvalue('group'))
	username = common.checkAjaxInput(form.getvalue('ssh_user'))
	password = common.checkAjaxInput(form.getvalue('ssh_pass'))
	new_ssh_key_name = ''

	if username is None:
		print(error_mess)
	else:
		lib_path = get_config.get_config_var('main', 'lib_path')

		for sshs in sql.select_ssh(id=ssh_id):
			ssh_enable = sshs.enable
			ssh_key_name = f'{lib_path}/keys/{sshs.name}.pem'
			new_ssh_key_name = f'{lib_path}/keys/{name}.pem'

		if ssh_enable == 1:
			os.rename(ssh_key_name, new_ssh_key_name)
			os.chmod(new_ssh_key_name, 0o600)

		sql.update_ssh(ssh_id, name, enable, group, username, password)
		roxywi_common.logging('Roxy-WI server', f'The SSH credentials {name} has been updated ', roxywi=1, login=1)


def delete_ssh_key() -> None:
	lib_path = get_config.get_config_var('main', 'lib_path')
	sshdel = common.checkAjaxInput(form.getvalue('sshdel'))
	name = ''
	ssh_enable = 0
	ssh_key_name = ''

	for sshs in sql.select_ssh(id=sshdel):
		ssh_enable = sshs.enable
		name = sshs.name
		ssh_key_name = f'{lib_path}/keys/{sshs.name}.pem'

	if ssh_enable == 1:
		try:
			os.remove(ssh_key_name)
		except Exception:
			pass
	if sql.delete_ssh(sshdel):
		print("Ok")
		roxywi_common.logging('Roxy-WI server', f'The SSH credentials {name} has deleted', roxywi=1, login=1)