#!/usr/bin/env python3
"""
MLM Payout E2E teszt script (egyszerűsített).
Szimulál egy előfizetési fizetést (10,000 HUF) és ellenőrzi a CREDIT Wallet jóváírásokat L1, L2, L3 usereknek.
Futtatás: docker compose exec sf_api python3 /app/app/scripts/test_mlm_payout_simple.py
"""

import asyncio
import sys
import os
sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))

from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy import select, update, func
from decimal import Decimal

from app.models.identity import User, Person, Wallet
from app.models.system.system import SystemParameter
from app.core.config import settings

async def test_mlm_payout():
    engine = create_async_engine(settings.DATABASE_URL, echo=True)
    async_session = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
    
    async with async_session() as session:
        print("=== MLM Payout E2E Teszt (Egyszerűsített) ===")
        
        # 1. MLM paraméterek lekérése
        print("\n1. MLM paraméterek lekérése...")
        params = {}
        for key in ["mlm_level1_percent", "mlm_level2_percent", "mlm_level3_percent"]:
            stmt = select(SystemParameter).where(SystemParameter.key == key)
            param = (await session.execute(stmt)).scalar_one_or_none()
            if param:
                params[key] = int(param.value.get("value", 0))
                print(f"  {key}: {params[key]}%")
            else:
                params[key] = 0
                print(f"  {key}: NINCS BEÁLLÍTVA!")
        
        # 2. Meglévő teszt userek keresése vagy létrehozása
        print("\n2. Teszt userek keresése...")
        
        # L1 user keresése (root meghívó)
        l1_stmt = select(User).where(User.email == "test_l1@example.com")
        l1_user = (await session.execute(l1_stmt)).scalar_one_or_none()
        
        if not l1_user:
            print("  L1 user nem létezik, teszt adatok hiányoznak.")
            print("  Futtasd először az init_mlm_parameters_fixed.py-t és regisztrálj néhány teszt usert!")
            return
        
        # L2 user keresése (L1 meghívja)
        l2_stmt = select(User).where(User.email == "test_l2@example.com")
        l2_user = (await session.execute(l2_stmt)).scalar_one_or_none()
        
        # L3 user keresése (L2 meghívja)
        l3_stmt = select(User).where(User.email == "test_l3@example.com")
        l3_user = (await session.execute(l3_stmt)).scalar_one_or_none()
        
        # L4 user keresése (fizető, L3 meghívja)
        l4_stmt = select(User).where(User.email == "test_l4_payer@example.com")
        l4_user = (await session.execute(l4_stmt)).scalar_one_or_none()
        
        users = [l1_user, l2_user, l3_user, l4_user]
        user_names = ["L1", "L2", "L3", "L4 (fizető)"]
        
        for idx, (user, name) in enumerate(zip(users, user_names)):
            if user:
                print(f"  {name}: {user.email} (ID: {user.id}, referred_by: {user.referred_by_id})")
            else:
                print(f"  {name}: NEM LÉTEZIK")
        
        # 3. Szimulált fizetés (10,000 HUF)
        payment_amount = Decimal("10000.00")
        print(f"\n3. Szimulált fizetés: {payment_amount} HUF (L4 user fizet)")
        
        # 4. MLM lánc felépítése
        print("\n4. MLM lánc felépítése...")
        chain = []
        if l4_user and l4_user.referred_by_id:
            # L3 keresése
            if l3_user and l3_user.id == l4_user.referred_by_id:
                chain.append(l3_user)
                # L2 keresése
                if l3_user.referred_by_id:
                    if l2_user and l2_user.id == l3_user.referred_by_id:
                        chain.append(l2_user)
                        # L1 keresése
                        if l2_user.referred_by_id:
                            if l1_user and l1_user.id == l2_user.referred_by_id:
                                chain.append(l1_user)
        
        print(f"  Lánc hossza: {len(chain)}")
        for idx, user in enumerate(chain):
            print(f"  L{idx+1}: {user.email} (ID: {user.id})")
        
        # 5. Jutalékok kiszámítása és CREDIT Wallet jóváírás (szimuláció)
        print("\n5. Jutalékok kiszámítása (szimuláció)...")
        levels = ["mlm_level1_percent", "mlm_level2_percent", "mlm_level3_percent"]
        
        for idx, referrer in enumerate(chain):
            if idx >= 3:
                break
            percent = params[levels[idx]]
            commission = (payment_amount * Decimal(percent) / Decimal(100)).quantize(Decimal("0.01"))
            
            print(f"  L{idx+1} ({referrer.email}): {percent}% -> {commission} HUF")
            
            # CREDIT Wallet keresése
            wallet_stmt = select(Wallet).where(
                Wallet.user_id == referrer.id
            )
            wallet = (await session.execute(wallet_stmt)).scalar_one_or_none()
            
            if wallet:
                print(f"    Wallet megtalálva: {wallet.balance} {wallet.currency}")
                # Szimulált jóváírás
                new_balance = wallet.balance + commission
                print(f"    Új egyenleg: {new_balance} {wallet.currency}")
            else:
                print(f"    NINCS Wallet a usernek!")
        
        # 6. API végpont tesztelése (GET /me/network)
        print("\n6. API végpont tesztelése (GET /me/network)...")
        if l1_user:
            # L1 hálózatának lekérdezése
            network_stmt = select(User).where(User.referred_by_id == l1_user.id)
            network_result = await session.execute(network_stmt)
            network_users = network_result.scalars().all()
            
            print(f"  L1 ({l1_user.email}) közvetlen meghívottai: {len(network_users)}")
            for u in network_users:
                print(f"    - {u.email} (kód: {u.referral_code})")
        
        print("\n✅ MLM Payout teszt (szimuláció) sikeresen lefutott!")
        print("=== TESZT VÉGE ===")
        print("\nJAVASLAT: A teljes teszteléshez:")
        print("1. Regisztrálj 4 teszt usert láncolatban (L1 -> L2 -> L3 -> L4)")
        print("2. Futtasd a valós billing engine-t a payment success eseménnyel")
        print("3. Ellenőrizd a CREDIT Wallet egyenlegeket a pgAdmin-ban")

if __name__ == "__main__":
    asyncio.run(test_mlm_payout())