vrr/eigent
2
0
Fork 0
mirror of https://github.com/eigent-ai/eigent.git synced 2026-04-28 11:40:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

server enhance

Browse source
This commit is contained in:
Wendong-Fan 2025-08-21 00:43:48 +08:00
parent e7a5c5536f
commit c108179fbb
13 changed files with 191 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

23
server/app/component/auth.py
View file

@ -39,17 +39,36 @@ class Auth:
id = payload["id"]
if payload["exp"] < int(datetime.now().timestamp()):
raise TokenException(code.token_expired, _("Validate credentials expired"))
# Accept both old tokens (without type) and new tokens (with type)
# Old tokens are treated as access tokens for backward compatibility
token_type = payload.get("type", "access")
if token_type not in ["access", "refresh"]:
raise TokenException(code.token_invalid, _("Invalid token type"))
except InvalidTokenError:
raise TokenException(code.token_invalid, _("Could not validate credentials"))
return Auth(id, payload["exp"])
@classmethod
def create_access_token(cls, user_id: int, expires_delta: timedelta | None = None):
to_encode: dict = {"id": user_id}
to_encode: dict = {"id": user_id, "type": "access"}
if expires_delta:
expire = datetime.now() + expires_delta
else:
expire = datetime.now() + timedelta(days=30)
# Get expiration from environment or default to 1 hour
expiration_seconds = int(env("JWT_EXPIRATION", "3600"))
expire = datetime.now() + timedelta(seconds=expiration_seconds)
to_encode.update({"exp": expire})
encoded_jwt = jwt.encode(to_encode, Auth.SECRET_KEY, algorithm="HS256")
return encoded_jwt
@classmethod
def create_refresh_token(cls, user_id: int, expires_delta: timedelta | None = None):
to_encode: dict = {"id": user_id, "type": "refresh"}
if expires_delta:
expire = datetime.now() + expires_delta
else:
# Refresh tokens last 7 days by default
expire = datetime.now() + timedelta(days=7)
to_encode.update({"exp": expire})
encoded_jwt = jwt.encode(to_encode, Auth.SECRET_KEY, algorithm="HS256")
return encoded_jwt