import aiohttp
import sys
import os


V2RAY_VPN_ENABLED = False

# Добавляем родительскую директорию в sys.path
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
if parent_dir not in sys.path:
    sys.path.append(parent_dir)


from utils.windows_toast import show_notification
from utils.config import read_config


class V2rayAController:
    def __init__(self, host='http://localhost:2017', username='admin', password='admin'):
        self.host = host
        self.username = username
        self.password = password
        self.token = None
        self.session = None

    async def authenticate(self):
        """Аутентификация и получение токена"""
        self.session = aiohttp.ClientSession()
        auth_url = f"{self.host}/api/login"
        credentials = {
            'username': self.username,
            'password': self.password
        }
        
        try:
            async with self.session.post(auth_url, json=credentials) as response:
                if response.status == 200:
                    data = await response.json()
                    self.token = data.get('data').get('token')
                    print("V2RayA auth OK")
                    return True
                else:
                    print(f"Ошибка аутентификации: {response.status}")
                    return False
        except Exception as e:
            raise OSError("Failed to authenticate v2ray")
            print(f"Ошибка при аутентификации: {e}")
            return False

    async def enable_proxy(self):
        """Включение прокси"""
        global V2RAY_VPN_ENABLED
        if not self.token:
            print("Не выполнен вход. Пожалуйста, сначала выполните аутентификацию")
            return False

        url = f"{self.host}/api/v2ray"
        headers = {'Authorization': f"Bearer {self.token}"}
        data = {}

        try:
            async with self.session.post(url, headers=headers, json=data) as response:
                if response.status == 200:
                    print("Прокси успешно включен")
                    await show_notification({"title":"✅ V2Ray proxy","message":"Прокси успешно включен"})
                    return True
                else:
                    print(f"Ошибка при включении прокси: {response.status}")
                    return False
        except Exception as e:
            print(f"Ошибка при включении прокси: {e}")
            raise OSError("Failed to enable proxy")
            return False

    async def disable_proxy(self):
        """Выключение прокси"""
       
        if not self.token:
            print("Не выполнен вход. Пожалуйста, сначала выполните аутентификацию")
            return False

        url = f"{self.host}/api/v2ray"
        headers = {'Authorization': f"Bearer {self.token}"}
        data = {'operation': 'stop'}

        try:
            async with self.session.delete(url, headers=headers, json=data) as response:
                if response.status == 200:
                    print("Прокси успешно выключен")
                    await show_notification({"title":"🔴 V2Ray proxy","message":"Прокси успешно выключен"})
                    return True
                else:
                    print(f"Ошибка при выключении прокси: {response.status}")
                    return False
        except Exception as e:
            print(f"Ошибка при выключении прокси: {e}")
            raise OSError("Failed to disable proxy")
            return False

    async def close(self):
        """Закрытие сессии"""
        if self.session:
            await self.session.close()

async def enable_vpn(args):
    global V2RAY_VPN_ENABLED
    # Инициализация контроллера
    config = await read_config("v2ray.yaml")
    if config.get("username") and config.get("password"):
        controller = V2rayAController(
            host='http://localhost:2017',
            username=config.get("username"),
            password=config.get("password")
        )
        # Аутентификация
        if await controller.authenticate():
            await controller.enable_proxy()

        # Закрытие сессии
        await controller.close()
        V2RAY_VPN_ENABLED = True
    else:
         raise OSError("Config unset")
async def disable_vpn(args):
    global V2RAY_VPN_ENABLED
    # Инициализация контроллера
    config = await read_config("v2ray.yaml")
    if config.get("username") and config.get("password"):
        controller = V2rayAController(
            host='http://localhost:2017',
            username=config.get("username"),
            password=config.get("password")
        )
        # Аутентификация
        if await controller.authenticate():
            await controller.disable_proxy()

        # Закрытие сессии
        await controller.close()
        V2RAY_VPN_ENABLED = False
    else:
         raise OSError("Config unset")


async def is_vpn_enabled(args):
    global V2RAY_VPN_ENABLED
    return {
            "value": V2RAY_VPN_ENABLED
        }