From 87c9ae9030886627ac547e7b4f47caec634fce8e Mon Sep 17 00:00:00 2001
From: Zvonimir Rudinski <zvonki12@gmail.com>
Date: Sun, 16 Nov 2025 17:02:10 +0100
Subject: [PATCH] add token response

---
 database.py | 4 ++--
 main.py     | 4 ++--
 models.py   | 4 ++++
 3 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/database.py b/database.py
index bd03e76..3755552 100644
--- a/database.py
+++ b/database.py
@@ -1,4 +1,4 @@
-from models import User
+from models import User, TokenResponse
 from settings import settings
 from fastapi import HTTPException, status, Request
 import sqlite3
@@ -101,4 +101,4 @@ def login(user: User) -> str:
             detail="Invalid credentials"
         )
 
-    return security.sign_jwt(row)
+    return TokenResponse(token=security.sign_jwt(row))
diff --git a/main.py b/main.py
index 695da3a..21b6dcd 100644
--- a/main.py
+++ b/main.py
@@ -1,4 +1,4 @@
-from fastapi import FastAPI, Depends
+from fastapi import FastAPI, Depends, status
 from contextlib import asynccontextmanager
 import database
 import models
@@ -25,7 +25,7 @@ async def me(user: models.User = Depends(database.get_user_by_token)):
     return user
 
 
-@app.post("/users")
+@app.post("/register", status_code=status.HTTP_201_CREATED)
 async def register(user: models.User):
     database.register(user)
     return user
diff --git a/models.py b/models.py
index bc275ee..2d11557 100644
--- a/models.py
+++ b/models.py
@@ -4,3 +4,7 @@ from pydantic import BaseModel
 class User(BaseModel):
     name: str
     password: str
+
+
+class TokenResponse(BaseModel):
+    token: str