from os import system, mkdir, walk
from db import *
from shutil import copyfile, make_archive, rmtree, copytree
from tqdm import tqdm
from time import sleep

from verify import *
from network import *
from domain_check import domain_list

def your_sites():
	sites = {}; num = 1
	for i in next(walk('cached/'), (None, None, []))[1]:
		sites[str(num)] = i
		print(f'{str(num)}. {i}')
		num += 1
	return sites

print('(1) Создать сайт')
print('(2) Обновить сайт')
print('(3) Сменить тип')
print('(4) Авто-раздача сайта')
print('Enter для просто публикации.')

op = input('>> ')

if op == '1':
	from domain_check import *

	print(f'\nДоступные 1 lvl домены: {domain_list()}')
	domain = input('Домен сайта: ')
	if not domain_ok(domain):
		print('Неправильный формат или домен.')
		exit()
	print('\n(1) Статичный / (2) Динамический')
	type = input('Тип: ')

	# Создаём папку и ключи для подписи
	mkdir(f'mysites/{domain}')
	key_gen(f'mysites/{domain}')

	if type == '1':
		conf = {'type': 'static', 'ver': 1}
		print('\nПРИМЕЧАНИЕ: index.html обязателен.')
		# Создаём index.html для загрузки сайта
		with open(f'mysites/{domain}/index.html', 'w') as f:
			f.write('<h1> Hello jetwork! </h1>')
		f.close()
	elif type == '2':
		port = input('Порт сервера: ')
		conf = {'type': 'dynamic', 'ver': 1, 'port': int(port)}
	write(conf, f'mysites/{domain}/config.json')

	# Архивируем и создаём сигнатуру для подтверждения неизменности архива
	make_archive(f'mysites/{domain}', 'zip', f'mysites/{domain}')
	sign(f'mysites/{domain}.zip', f'mysites/{domain}.key', f'mysites/{domain}')

elif op == '2':
	print()
	sites = your_sites()

	domain = sites[input('Домен сайта: ')]

	conf = read(f'mysites/{domain}/config.json')
	type = conf['type']
	if type == 'dynamic':
		port = input('Порт сервера: ')
		conf['port'] = port
		write(conf, f'mysites/{domain}/config.json')

	# Обновляем версию
	conf = read(f'mysites/{domain}/config.json')
	conf['ver'] = conf['ver'] + 1
	write(conf, f'mysites/{domain}/config.json')

	# Архивируем и создаём сигнатуру для подтверждения неизменности архива
	make_archive(f'mysites/{domain}', 'zip', f'mysites/{domain}')
	sign(f'mysites/{domain}.zip', f'mysites/{domain}.key', f'mysites/{domain}')

elif op == '3':
	from os import rmdir

	print()
	sites = your_sites()
	domain = sites[input('Домен сайта: ')]

	print('\n(1) Статичный / (2) Динамический')
	type = input('Тип: ')

	if type == '1':
		conf = read(f'mysites/{domain}/config.json')
		conf['type'] = 'static'
		conf.pop('port')
		with open(f'mysites/{domain}/index.html', 'w') as f:
			f.write('<h1> Hello jetwork! </h1>')
		f.close()
	elif type == '2':
		conf = read(f'mysites/{domain}/config.json')
		port = input('Порт сервера: ')

		clean = input('Удалить все лишние файлы сайта (y/n): ')
		if clean == 'y':
			# Удаляем папку, сохраняем конфиг и копируем публичный ключ
			rmtree(f'mysites/{domain}')
			sleep(0.1)
			system(f'mkdir mysites/{domain}')

		conf['type'] = 'dynamic'
		conf['port'] = int(port)
	write(conf, f'mysites/{domain}/config.json')

	# Обновляем версию
	conf = read(f'mysites/{domain}/config.json')
	conf['ver'] = conf['ver'] + 1
	write(conf, f'mysites/{domain}/config.json')

	# Архивируем и создаём сигнатуру для подтверждения неизменности архива
	make_archive(f'mysites/{domain}', 'zip', f'mysites/{domain}')
	sign(f'mysites/{domain}.zip', f'mysites/{domain}.key', f'mysites/{domain}')

elif op == '4':
	print()
	sites = your_sites()
	domain = sites[input('Домен сайта: ')]

	serv_port = int( read()['our_port'] )
	http_port = client(serv_port, f'is_{domain}')

	print('\nСтарт вечной раздачи...')
	while True:
		ports = port_check(serv_port)
		sleep(1)
		for port in tqdm(ports):
			client(port, f'publish_{domain}<>{http_port}')
		sleep(5)

host = 'bore.pub'
# Проверяем тип сайта
type = read(f'mysites//{domain}/config.json')['type']
# Если динамический - вставляем спец страницу
if type == 'dynamic':
	port = read(f'mysites//{domain}/config.json')['port']
	with open(f'mysites//{domain}/index.html', 'w') as f:
		f.write(f"<iframe src='http://{host}:{port}' style='position:fixed; top:0; left:0; bottom:0; right:0; width:100%; height:100%; border:none; margin:0; padding:0; overflow:hidden; z-index:999999;'></iframe>")
	f.close()


# Копируем файлы из mysites в cached
try:
	rmtree(f'cached/{domain}')
except:
	pass
copytree(f'mysites/{domain}', f'cached/{domain}')
copyfile(f'mysites/{domain}.pem', f'cached/{domain}.pem')
copyfile(f'mysites/{domain}.sig', f'cached/{domain}.sig')
copyfile(f'mysites/{domain}.zip', f'cached/{domain}.zip')


if op != '':
	print('\nОпубликовать сайт?')
	pub = input('y/n >> ')

if pub == 'n':
	exit()

print('Введите ваш порт сервера (при запуске main.py)')
serv_port = int(input('>> '))
http_port = client(serv_port, f'is_{domain}')

print('Получаем все порты...')
ports = port_check(serv_port)

print(ports)

print('Публикуем сайт...')
for port in tqdm(ports):
	client(port, f'publish_{domain}<>{http_port}')