L1L2 gen1Gen2 kialakítása

backend/app/services/social_auth_service.py

@@ -5,14 +5,24 @@ from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy import select
 from app.models.identity import User, Person, SocialAccount, UserRole
 from app.services.security_service import security_service
+from app.core.security import generate_secure_slug
 
 logger = logging.getLogger(__name__)
 
 class SocialAuthService:
     @staticmethod
-    async def get_or_create_social_user(db: AsyncSession, provider: str, social_id: str, email: str, first_name: str, last_name: str):
-        """ 
-        LOGIKA MEGŐRIZVE: Step 1 regisztráció slug és flotta nélkül. 
+    async def get_or_create_social_user(
+        db: AsyncSession,
+        provider: str,
+        social_id: str,
+        email: str,
+        first_name: str,
+        last_name: str,
+        referred_by_code: str = None
+    ):
+        """
+        LOGIKA MEGŐRIZVE: Step 1 regisztráció slug és flotta nélkül.
+        Támogatja a meghívó kód (referral code) átvételét Google SSO regisztrációnál.
         """
         # 1. Meglévő fiók ellenőrzése
         stmt = select(SocialAccount).where(SocialAccount.provider == provider, SocialAccount.social_id == social_id)
@@ -26,11 +36,37 @@ class SocialAuthService:
         user = (await db.execute(stmt_u)).scalar_one_or_none()
 
         if not user:
-            new_person = Person(first_name=first_name or "Social", last_name=last_name or "User", is_active=False)
+            # Meghívó keresése referral kód alapján
+            referred_by_id = None
+            if referred_by_code:
+                referrer_stmt = select(User).where(User.referral_code == referred_by_code)
+                referrer = (await db.execute(referrer_stmt)).scalar_one_or_none()
+                if referrer:
+                    referred_by_id = referrer.id
+                    logger.info(f"Social user {email} referred by {referrer.email} (ID: {referrer.id})")
+                else:
+                    logger.warning(f"Referral code '{referred_by_code}' not found for social registration, ignoring.")
+
+            new_person = Person(
+                first_name=first_name or "Social",
+                last_name=last_name or "User",
+                is_active=False,
+                identity_docs={},
+                ice_contact={}
+            )
             db.add(new_person)
             await db.flush()
 
-            user = User(email=email, person_id=new_person.id, role=UserRole.USER, is_active=False)
+            referral_code = generate_secure_slug(8).upper()
+
+            user = User(
+                email=email,
+                person_id=new_person.id,
+                role=UserRole.USER,
+                is_active=False,
+                referral_code=referral_code,
+                referred_by_id=referred_by_id
+            )
             db.add(user)
             await db.flush()