From f3e63821015f8d000143261db6484ec4e4b1248a Mon Sep 17 00:00:00 2001
From: julius <julius@0124816.xyz>
Date: Mon, 10 Mar 2025 13:15:41 +0100
Subject: [PATCH] feat: set first password with token

---
 security.py | 71 ++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 67 insertions(+), 4 deletions(-)

diff --git a/security.py b/security.py
index c272726..d3d5e93 100644
--- a/security.py
+++ b/security.py
@@ -1,6 +1,7 @@
 from datetime import timedelta, timezone, datetime
 from typing import Annotated
 from fastapi import Depends, HTTPException, Request, Response, status
+from fastapi.responses import PlainTextResponse
 from pydantic import BaseModel, ValidationError
 import jwt
 from jwt.exceptions import ExpiredSignatureError, InvalidTokenError
@@ -76,7 +77,7 @@ def get_user(username: str | None):
         try:
             with Session(engine) as session:
                 return session.exec(
-                    select(Player).where(User.username == username)
+                    select(Player).where(Player.username == username)
                 ).one_or_none()
         except OperationalError:
             return
@@ -97,7 +98,7 @@ def create_access_token(data: dict, expires_delta: timedelta | None = None):
         expire = datetime.now(timezone.utc) + expires_delta
     else:
         expire = datetime.now(timezone.utc) + timedelta(
-            seconds=config.access_token_expire_minutes
+            minutes=config.access_token_expire_minutes
         )
     to_encode.update({"exp": expire})
     encoded_jwt = jwt.encode(to_encode, config.secret_key, algorithm="HS256")
@@ -154,7 +155,7 @@ async def get_current_active_user(
 ):
     if current_user.disabled:
         raise HTTPException(status_code=400, detail="Inactive user")
-    return current_user.model_dump(exclude={"hashed_password", "disabled"})
+    return current_user
 
 
 async def login_for_access_token(
@@ -187,7 +188,69 @@ async def logout(response: Response):
     return {"message": "Successfully logged out"}
 
 
-async def read_users_me(
+def generate_one_time_token(username):
+    expire = timedelta(days=7)
+    token = create_access_token(data={"sub": username}, expires_delta=expire)
+    return token
+
+
+class FirstPassword(BaseModel):
+    token: str
+    password: str
+
+
+async def set_first_password(req: FirstPassword):
+    credentials_exception = HTTPException(
+        status_code=status.HTTP_401_UNAUTHORIZED,
+        detail="Could not validate token",
+    )
+    try:
+        payload = jwt.decode(req.token, config.secret_key, algorithms=["HS256"])
+        username: str = payload.get("sub")
+        if username is None:
+            raise credentials_exception
+    except ExpiredSignatureError:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail="Access token expired",
+        )
+    except (InvalidTokenError, ValidationError):
+        raise credentials_exception
+
+    user = get_user(username)
+    if user:
+        with Session(engine) as session:
+            user.hashed_password = get_password_hash(req.password)
+            session.add(user)
+            session.commit()
+            return Response("Password set successfully", status_code=status.HTTP_200_OK)
+
+
+async def change_password(
+    current_password: str,
+    new_password: str,
+    user: Annotated[Player, Depends(get_current_active_user)],
+):
+    if (
+        new_password
+        and user.hashed_password
+        and verify_password(current_password, user.hashed_password)
+    ):
+        with Session(engine) as session:
+            user.hashed_password = get_password_hash(new_password)
+            session.add(user)
+            session.commit()
+            return PlainTextResponse(
+                "Password changed successfully", status_code=status.HTTP_200_OK
+            )
+    else:
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail="Wrong password",
+        )
+
+
+async def read_player_me(
     current_user: Annotated[Player, Depends(get_current_active_user)],
 ):
     return current_user