fix profile

api/controllers/service_api_with_auth/user/profile.py

@@ -4,13 +4,13 @@ from controllers.service_api_with_auth import api
 from controllers.service_api_with_auth.wraps import validate_user_token_and_extract_info
 from flask import Blueprint, request
 from flask_restful import Api, Resource  # type: ignore
-from models.model import EndUser
+from models.model import App, EndUser
 from services.end_user_service import EndUserService
 
 
 class UserProfile(Resource):
     @validate_user_token_and_extract_info
-    def get(self, end_user: EndUser):
+    def get(self, app_model: App, end_user: EndUser):
         """Get user profile.
         ---
         tags:
@@ -39,11 +39,11 @@ class UserProfile(Resource):
             description: Invalid or missing token
         """
         # Use the service to get user profile
-        profile = EndUserService.get_user_profile(end_user.id)
+        profile = EndUserService.get_user_profile(end_user.external_user_id)
         return profile
 
     @validate_user_token_and_extract_info
-    def put(self, end_user: EndUser):
+    def put(self, app_model: App, end_user: EndUser):
         """Update user profile.
         ---
         tags:

api/controllers/service_api_with_auth/wraps.py

@@ -11,7 +11,7 @@ from flask_login import user_logged_in  # type: ignore
 from flask_restful import Resource  # type: ignore
 from libs.login import _get_user
 from libs.passport import PassportService
-from models.account import Account, Tenant, TenantAccountJoin, TenantStatus
+from models.account import Account, AccountStatus, Tenant, TenantAccountJoin, TenantStatus
 from models.model import ApiToken, App, EndUser
 from pydantic import BaseModel  # type: ignore
 from services.account_service import AccountService
@@ -55,16 +55,21 @@ def validate_user_token_and_extract_info(view: Optional[Callable] = None):
             try:
                 decoded = PassportService().verify(auth_token)
                 user_id = decoded.get("user_id")
+                expired_at = decoded.get("exp")
+
             except Exception as e:
                 raise Unauthorized(f"Failed to extract user_id from token: {str(e)}")
 
             if not user_id:
                 raise Unauthorized("Invalid token: missing user_id")
 
+            if expired_at < datetime.now(UTC).timestamp():
+                raise Unauthorized("Token has expired")
+
             account = AccountService.load_user(user_id)
             if account is None:
                 raise Unauthorized("Invalid token: user not found")
-            if account.status != Account.AccountStatus.ACTIVE:
+            if account.status != AccountStatus.ACTIVE:
                 raise Unauthorized("Invalid token: account is not active")
 
             app_id = request.headers.get("X-App-Id")
@@ -90,7 +95,7 @@ def validate_user_token_and_extract_info(view: Optional[Callable] = None):
 
             kwargs["app_model"] = app_model
 
-            kwargs["end_user"] = create_or_update_end_user_for_user_id(app_model, user_id)
+            kwargs["end_user"] = create_or_update_end_user_for_user_id(app_model, account.id)
 
             return view_func(*args, **kwargs)
 
@@ -244,8 +249,8 @@ def create_or_update_end_user_for_user_id(app_model: App, user_id: Optional[str]
         .filter(
             EndUser.tenant_id == app_model.tenant_id,
             EndUser.app_id == app_model.id,
-            EndUser.session_id == user_id,
+            EndUser.external_user_id == user_id,
-            EndUser.type == "service_api",
+            EndUser.type == "service_api_with_auth",
         )
         .first()
     )
@@ -254,9 +259,9 @@ def create_or_update_end_user_for_user_id(app_model: App, user_id: Optional[str]
         end_user = EndUser(
             tenant_id=app_model.tenant_id,
             app_id=app_model.id,
-            type="service_api",
+            type="service_api_with_auth",
-            is_anonymous=user_id == "DEFAULT-USER",
             session_id=user_id,
+            external_user_id=user_id,
         )
         db.session.add(end_user)
         db.session.commit()

api/services/end_user_service.py

@@ -3,6 +3,7 @@ from typing import Any, Dict, Optional, Tuple
 from extensions.ext_database import db
 from models.account import Account
 from models.model import EndUser
+from services.account_service import AccountService
 
 
 class EndUserService:
@@ -18,7 +19,7 @@ class EndUserService:
             Dict containing user profile information
         """
         # Get EndUser information
-        end_user = db.session.query(EndUser).filter(EndUser.id == end_user_id).first()
+        end_user = db.session.query(EndUser).filter(EndUser.external_user_id == end_user_id).first()
 
         if not end_user:
             return {"username": None, "gender": "unknown", "major": None, "email": None}