refakotorálás előtti állapot

backend/app/api/v1/endpoints/billing.py

@@ -12,6 +12,7 @@ from app.models.payment import PaymentIntent, PaymentIntentStatus
 from app.services.config_service import config
 from app.services.payment_router import PaymentRouter
 from app.services.stripe_adapter import stripe_adapter
+from app.services.billing_engine import upgrade_subscription, get_user_balance
 
 router = APIRouter()
 logger = logging.getLogger(__name__)
@@ -19,55 +20,24 @@ logger = logging.getLogger(__name__)
 @router.post("/upgrade")
 async def upgrade_account(target_package: str, db: AsyncSession = Depends(get_db), current_user: User = Depends(get_current_user)):
     """
-    Univerzális csomagváltó. 
+    Univerzális csomagváltó a Billing Engine segítségével.
     Kezeli az 5+ csomagot, a Rank-ugrást és a különleges 'Service Coin' bónuszokat.
     """
-    # 1. Lekérjük a teljes csomagmátrixot az adminból
-    # Példa JSON: {"premium": {"price": 2000, "rank": 5, "type": "credit"}, "service_pro": {"price": 10000, "rank": 30, "type": "coin"}}
-    package_matrix = await config.get_setting(db, "subscription_packages_matrix")
-    
-    if target_package not in package_matrix:
-        raise HTTPException(status_code=400, detail="Érvénytelen csomagválasztás.")
-
-    pkg_info = package_matrix[target_package]
-    price = pkg_info["price"]
-    
-    # 2. Pénztárca ellenőrzése
-    stmt = select(Wallet).where(Wallet.user_id == current_user.id)
-    wallet = (await db.execute(stmt)).scalar_one_or_none()
-
-    total_balance = wallet.purchased_credits + wallet.earned_credits
-    
-    if total_balance < price:
-        raise HTTPException(status_code=402, detail="Nincs elég kredited a csomagváltáshoz.")
-
-    # 3. Levonási logika (Purchased -> Earned sorrend)
-    if wallet.purchased_credits >= price:
-        wallet.purchased_credits -= price
-    else:
-        remaining = price - wallet.purchased_credits
-        wallet.purchased_credits = 0
-        wallet.earned_credits -= remaining
-
-    # 4. Speciális Szerviz Logika (Service Coins)
-    # Ha a csomag típusa 'coin', akkor a szerviz kap egy kezdő Coin csomagot is
-    if pkg_info.get("type") == "coin":
-        initial_coins = pkg_info.get("initial_coin_bonus", 100)
-        wallet.service_coins += initial_coins
-        logger.info(f"User {current_user.id} upgraded to Service Pro, awarded {initial_coins} coins.")
-
-    # 5. Rang frissítése és naplózás
-    current_user.role = target_package # Pl. 'service_pro' vagy 'vip'
-    
-    db.add(FinancialLedger(
-        user_id=current_user.id,
-        amount=-price,
-        transaction_type=f"UPGRADE_{target_package.upper()}",
-        details=pkg_info
-    ))
-    
-    await db.commit()
-    return {"status": "success", "package": target_package, "rank_granted": pkg_info["rank"]}
+    try:
+        result = await upgrade_subscription(db, current_user.id, target_package)
+        return {
+            "status": "success",
+            "package": target_package,
+            "price_paid": result.get("price_paid", 0.0),
+            "new_plan": result.get("new_plan"),
+            "expires_at": result.get("expires_at"),
+            "transaction": result.get("transaction")
+        }
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))
+    except Exception as e:
+        logger.error(f"Upgrade error: {e}")
+        raise HTTPException(status_code=500, detail=f"Belső hiba: {str(e)}")
 
 
 @router.post("/payment-intent/create")
@@ -331,27 +301,14 @@ async def get_wallet_balance(
     current_user: User = Depends(get_current_user)
 ):
     """
-    Felhasználó pénztárca egyenlegének lekérdezése.
+    Felhasználó pénztárca egyenlegének lekérdezése a Billing Engine segítségével.
     """
     try:
-        stmt = select(Wallet).where(Wallet.user_id == current_user.id)
-        result = await db.execute(stmt)
-        wallet = result.scalar_one_or_none()
-        
-        if not wallet:
-            raise HTTPException(status_code=404, detail="Pénztárca nem található")
-        
-        return {
-            "earned": float(wallet.earned_credits),
-            "purchased": float(wallet.purchased_credits),
-            "service_coins": float(wallet.service_coins),
-            "total": float(
-                wallet.earned_credits +
-                wallet.purchased_credits +
-                wallet.service_coins
-            ),
-        }
-        
+        balances = await get_user_balance(db, current_user.id)
+        return balances
+    except ValueError as e:
+        raise HTTPException(status_code=404, detail=str(e))
     except Exception as e:
         logger.error(f"Pénztárca egyenleg lekérdezési hiba: {e}")
+        raise HTTPException(status_code=500, detail=f"Belső hiba: {str(e)}")
         raise HTTPException(status_code=500, detail=f"Belső hiba: {str(e)}")