---

import os
import sys
import ctypes
import subprocess
import json
import socket
import customtkinter as ctk

# Конфиг
CONFIG_FILE = "dns_config.json"

class FerrariDNS(ctk.CTk):
    def __init__(self):
        super().__init__()
        
        self.title("Ferrari Private DNS")
        self.geometry("450x500")
        ctk.set_appearance_mode("dark")

        # Загрузка конфига
        self.config = self.load_config()

        # Интерфейс
        self.label = ctk.CTkLabel(self, text="PRIVATE DNS SETTINGS", font=("Arial", 20, "bold"))
        self.label.pack(pady=20)

        self.dns_input = ctk.CTkEntry(self, width=350)
        self.dns_input.pack(pady=10)
        self.dns_input.insert(0, self.config.get("dns_url", "dns.comss.one"))

        self.status_label = ctk.CTkLabel(self, text=f"Статус: {self.config.get('status', 'Ожидание')}", text_color="gray")
        self.status_label.pack(pady=5)

        self.btn_on = ctk.CTkButton(self, text="ВКЛЮЧИТЬ", fg_color="green", command=self.activate)
        self.btn_on.pack(pady=10)

        self.btn_off = ctk.CTkButton(self, text="СБРОСИТЬ (АВТО)", fg_color="#d32f2f", command=self.reset)
        self.btn_off.pack(pady=10)

        self.auto_run_var = ctk.BooleanVar(value=self.config.get("auto_run", False))
        self.check_auto = ctk.CTkCheckBox(self, text="Автозагрузка при старте", variable=self.auto_run_var, command=self.toggle_autorun)
        self.check_auto.pack(pady=20)

    def load_config(self):
        if os.path.exists(CONFIG_FILE):
            try:
                with open(CONFIG_FILE, "r") as f:
                    return json.load(f)
            except:
                pass
        return {"dns_url": "dns.comss.one", "status": "Отключено", "auto_run": False}

    def save_config(self, status):
        config_data = {
            "dns_url": self.dns_input.get(),
            "status": status,
            "auto_run": self.auto_run_var.get()
        }
        with open(CONFIG_FILE, "w") as f:
            json.dump(config_data, f)

    def get_adapter(self):
        try:
            # Берем первый активный адаптер
            cmd = 'powershell "(Get-NetAdapter | Where-Object {$_.Status -eq \'Up\'} | Select-Object -First 1).Name"'
            name = subprocess.check_output(cmd, shell=True).decode('cp866').strip()
            return name if name else "Wi-Fi"
        except:
            return "Wi-Fi"

    def activate(self):
        raw_input = self.dns_input.get().strip()
        adapter = self.get_adapter()
        self.status_label.configure(text="Настройка...", text_color="yellow")
        
        try:
            # Формируем хост и ссылку
            host = raw_input.replace("https://", "").split("/")[0]
            ip = socket.gethostbyname(host)
            doh_template = raw_input if "://" in raw_input else f"https://{raw_input}/dns-query"

            # Команды PowerShell с экранированием кавычек
            # 1. Регистрация сервера в базе DoH
            cmd_reg = f'powershell "Add-DnsClientDohServerAddress -ServerAddress {ip} -DohTemplate \'{doh_template}\' -AllowFallbackToUdp $false -ErrorAction SilentlyContinue"'
            # 2. Установка DNS для адаптера
            cmd_set = f'powershell "Set-DnsClientServerAddress -InterfaceAlias \'{adapter}\' -ServerAddresses {ip}"'
            # 3. Принудительный DoH
            cmd_force = f'powershell "Set-DnsClient -InterfaceAlias \'{adapter}\' -UseDnsOverHttps Only"'

            subprocess.run(cmd_reg, shell=True)
            subprocess.run(cmd_set, shell=True)
            subprocess.run(cmd_force, shell=True)
            subprocess.run('ipconfig /flushdns', shell=True)
            
            self.status_label.configure(text="Статус: АКТИВНО", text_color="green")
            self.save_config("Активно")
        except Exception as e:
            self.status_label.configure(text=f"Ошибка: {str(e)}", text_color="red")

    def reset(self):
        adapter = self.get_adapter()
        try:
            subprocess.run(f'powershell "Set-DnsClientServerAddress -InterfaceAlias \'{adapter}\' -ResetServerAddresses"', shell=True)
            subprocess.run(f'powershell "Set-DnsClient -InterfaceAlias \'{adapter}\' -UseDnsOverHttps Prohibit"', shell=True)
            subprocess.run('ipconfig /flushdns', shell=True)
            self.status_label.configure(text="Статус: Сброшено", text_color="white")
            self.save_config("Отключено")
        except:
            self.status_label.configure(text="Ошибка сброса", text_color="red")

    def toggle_autorun(self):
        path = os.path.realpath(sys.argv[0])
        key = r"Software\Microsoft\Windows\CurrentVersion\Run"
        if self.auto_run_var.get():
            subprocess.run(f'reg add "HKEY_CURRENT_USER\\{key}" /v "FerrariDNS" /t REG_SZ /d "{path}" /f', shell=True)
        else:
            subprocess.run(f'reg delete "HKEY_CURRENT_USER\\{key}" /v "FerrariDNS" /f', shell=True)
        self.save_config(self.config.get("status", "Отключено"))

if __name__ == "__main__":
    if not ctypes.windll.shell32.IsUserAnAdmin():
        # Запуск с правами админа
        ctypes.windll.shell32.ShellExecuteW(None, "runas", sys.executable, f'"{__file__}"', None, 1)
    else:
        app = FerrariDNS()
        app.mainloop()