new refresh dep add refresh endpoint
This commit is contained in:
@@ -36,10 +36,34 @@ class AuthService(BaseService):
|
||||
status_code=401,
|
||||
detail="Incorrect username or password",
|
||||
)
|
||||
access_token = AuthManager.create_access_token(
|
||||
user_token.model_dump()
|
||||
)
|
||||
access_token = AuthManager.create_access_token(user_token.model_dump())
|
||||
refresh_token = AuthManager.create_refresh_token()
|
||||
await self.session.auth.create_one({"token": refresh_token, "user_id": user.id})
|
||||
await self.session.commit()
|
||||
return {"access_token": access_token, "token_type": settings.access_token.token_type, "refresh_token": refresh_token}
|
||||
return {
|
||||
"access_token": access_token,
|
||||
"token_type": settings.access_token.token_type,
|
||||
"refresh_token": refresh_token,
|
||||
}
|
||||
|
||||
async def delete_token(self, token: str) -> None:
|
||||
await self.session.auth.delete_one(token=token)
|
||||
await self.session.commit()
|
||||
|
||||
async def refresh_tokens(self, refresh_token: str, user_data: TokenData):
|
||||
token_record = await self.session.auth.get_one_or_none(token=refresh_token)
|
||||
if not token_record or token_record.user_id != user_data.id:
|
||||
raise HTTPException(status_code=401, detail="Invalid refresh token")
|
||||
new_access_token = AuthManager.create_access_token(user_data.model_dump())
|
||||
new_refresh_token = AuthManager.create_refresh_token()
|
||||
await self.session.auth.delete_one(token=refresh_token)
|
||||
await self.session.auth.create_one({
|
||||
"token": new_refresh_token,
|
||||
"user_id": user_data.id,
|
||||
})
|
||||
await self.session.commit()
|
||||
return {
|
||||
"access_token": new_access_token,
|
||||
"token_type": settings.access_token.token_type,
|
||||
"refresh_token": new_refresh_token,
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user