import base64
from cryptography.hazmat.primitives.asymmetric import x25519
from app.core.config import config

# Вставь сюда свои base64-строки из config.py / .env
private_key_b64 = config.SYSTEM_CHAT_PRIVATE_KEY
public_key_b64 = config.SYSTEM_CHAT_PUBLIC_KEY

try:
    priv_bytes = base64.b64decode(private_key_b64)
    priv_key = x25519.X25519PrivateKey.from_private_bytes(priv_bytes)
    
    # Вычисляем правильный публичный ключ из этого приватного
    correct_pub_bytes = priv_key.public_key().public_bytes_raw()
    correct_pub_b64 = base64.b64encode(correct_pub_bytes).decode('utf-8')
    
    if correct_pub_b64 == public_key_b64:
        print("✅ Ключи сервера идеальны и подходят друг к другу!")
    else:
        print("❌ ОШИБКА: Публичный ключ из config.py НЕ принадлежит этому приватному ключу!")
        print(f"Текущий в config: {public_key_b64}")
        print(f"Должен быть:      {correct_pub_b64} <-- Пропиши его в конфиг")
except Exception as e:
    print(f"Критическая ошибка при чтении байт: {e}")