136 lines
5.1 KiB
Python
136 lines
5.1 KiB
Python
import aiohttp
|
||
import sys
|
||
import os
|
||
|
||
|
||
|
||
# Добавляем родительскую директорию в 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):
|
||
"""Включение прокси"""
|
||
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:
|
||
raise OSError("Failed to enable proxy")
|
||
print(f"Ошибка при включении прокси: {e}")
|
||
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:
|
||
raise OSError("Failed to disable proxy")
|
||
print(f"Ошибка при выключении прокси: {e}")
|
||
return False
|
||
|
||
async def close(self):
|
||
"""Закрытие сессии"""
|
||
if self.session:
|
||
await self.session.close()
|
||
|
||
async def enable_vpn(args):
|
||
# Инициализация контроллера
|
||
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()
|
||
else:
|
||
raise OSError("Config unset")
|
||
async def disable_vpn(args):
|
||
# Инициализация контроллера
|
||
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()
|
||
else:
|
||
raise OSError("Config unset")
|