feat: back invite system into registrantion

Author: hardusss

backend/NextVibeAPI/user/serializers_pac/google_registration.py

@@ -13,10 +13,16 @@
 class GoogleRegister(serializers.ModelSerializer):
     token = serializers.SerializerMethodField()
     avatar_url = serializers.URLField(write_only=True, required=False)
+    from_invite_code = serializers.CharField(
+        required=False,
+        allow_null=True,
+        allow_blank=True,
+        write_only=True
+    )
 
     class Meta:
         model = User
-        fields = ("user_id", "email", "username", "token", "avatar", "avatar_url")
+        fields = ("user_id", "email", "username", "token", "avatar", "avatar_url", "from_invite_code")
 
     def get_token(self, obj):
         refresh = RefreshToken.for_user(obj)
@@ -27,13 +33,29 @@ def get_token(self, obj):
 
     @transaction.atomic
     def create(self, validated_data):
+        from user.models import InviteUser
+        
         avatar_url = validated_data.pop("avatar_url", None)
+        invite_code = validated_data.pop("from_invite_code", None)
+        
+        invite_obj = None
+        if invite_code:
+            invite_obj = InviteUser.objects.filter(invite_code=invite_code).first()
 
         user, created = User.objects.get_or_create(
             email=validated_data["email"],
-            defaults={"username": validated_data["username"]},
+            defaults={
+                "username": validated_data["username"],
+                "from_invite_code": invite_obj,
+                "auth_provider": "google",
+            },
         )
 
+        if created and invite_obj:
+            from django.db.models import F
+            invite_obj.invited_count = F('invited_count') + 1
+            invite_obj.save(update_fields=['invited_count'])
+
         if created and avatar_url:
             try:
                 response = requests.get(avatar_url, timeout=5)

backend/NextVibeAPI/user/views_pac/google_registration.py

@@ -29,6 +29,9 @@ def post(self, request, *args, **kwargs):
             serializer = GoogleRegister(user)
             return Response(serializer.data, status=200)
 
+        if "from_invite_code" not in request.data:
+            return Response({"error": "invite_code_required"}, status=status.HTTP_400_BAD_REQUEST)
+
         username = request.data.get("username")
         avatar_url = request.data.get("avatar_url")
         if not username:

backend/NextVibeAPI/user/views_pac/wallet_singin.py

@@ -44,9 +44,13 @@ def post(self, request):
                 'username': user.username
             })
         else:
+            if "from_invite_code" not in request.data:
+                return Response({"error": "invite_code_required"}, status=status.HTTP_400_BAD_REQUEST)
+
             serializer = UserWalletSignInSerializer(data={
                 "wallet_address": wallet_address,
-                "username": username
+                "username": username,
+                "from_invite_code": request.data.get("from_invite_code")
             })
             if serializer.is_valid():
                 user = serializer.save()