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

from verify import *
from network import *

print("(1) Создать сайт")
print("(2) Обновить сайт")
print("(3) Сменить тип")
op = input(">> ")

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

	print("\nДомены: .jet")
	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}
	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":
	domain = input("\nДомен сайта: ")
	if not os.path.exists(f"mysites/{domain}"):
		print("Не существует такого сайта.")
		exit()

	# Обновляем версию
	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

	domain = input("\nДомен сайта: ")
	if not os.path.exists(f"mysites/{domain}"):
		print("Не существует такого сайта.")
		exit()

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

	if type == "1":
		conf = read(f"mysites/{domain}/config.json")
		conf["type"] = "static"
		conf.pop("port")
	elif type == "2":
		conf = read(f"mysites/{domain}/config.json")
		port = input("Порт: ")

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

		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}")

	exit()


# Копируем файлы из mysites в cached
rmtree(f"cached/{domain}")
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")



print("Опубликовать сайт?")
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}")