🐐

backend/main.py

@@ -2,6 +2,7 @@ import asyncio
 import logging
 import uuid
 from contextlib import asynccontextmanager
+from datetime import datetime
 
 from sqlalchemy.orm import Session
 from fastapi import FastAPI, Depends, HTTPException, status
@@ -9,11 +10,10 @@ from fastapi.security import OAuth2PasswordBearer, OAuth2PasswordRequestForm
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 
-from database import SessionLocal, get_db
-from database_functions import fill_card_pool
+from database import get_db
+from database_functions import fill_card_pool, check_boosters, BOOSTER_MAX
 from models import Card as CardModel
 from models import User as UserModel
-from card import _get_cards_async
 from auth import hash_password, verify_password, create_access_token, decode_access_token
 
 logger = logging.getLogger("app")
@@ -49,16 +49,6 @@ app.add_middleware(
   allow_headers=["*"],
 )
 
-@app.get("/pack/{size}")
-async def open_pack(size: int = 10, user: UserModel = Depends(get_current_user)):
-  cards = await _get_cards_async(size)
-  return [
-    {**card._asdict(),
-      "card_type": card.card_type.name,
-      "card_rarity": card.card_rarity.name}
-    for card in cards
-  ]
-
 @app.post("/register")
 def register(req: RegisterRequest, db: Session = Depends(get_db)):
   if db.query(UserModel).filter(UserModel.username == req.username).first():
@@ -82,3 +72,43 @@ def login(form: OAuth2PasswordRequestForm = Depends(), db: Session = Depends(get
     raise HTTPException(status_code=400, detail="Invalid username or password")
   token = create_access_token(str(user.id))
   return {"access_token": token, "token_type": "bearer"}
+
+@app.get("/boosters")
+def get_boosters(user: UserModel = Depends(get_current_user), db: Session = Depends(get_db)) -> tuple[int,datetime|None]:
+  return check_boosters(user, db)
+
+@app.post("/open_pack")
+async def open_pack(user: UserModel = Depends(get_current_user), db: Session = Depends(get_db)):
+  check_boosters(user, db)
+
+  if user.boosters == 0:
+    raise HTTPException(status_code=400, detail="No booster packs available")
+
+  cards = (
+    db.query(CardModel)
+    .filter(CardModel.user_id == None)
+    .limit(5)
+    .all()
+  )
+
+  if len(cards) < 5:
+    raise HTTPException(status_code=503, detail="Card pool is low, please try again shortly")
+
+  for card in cards:
+    card.user_id = user.id
+
+  # was_full = user.boosters == BOOSTER_MAX
+  # user.boosters -= 1
+  # if was_full:
+  #   user.boosters_countdown = datetime.now()
+
+  db.commit()
+
+  asyncio.create_task(fill_card_pool())
+
+  return [
+    {**{c.name: getattr(card, c.name) for c in card.__table__.columns},
+      "card_rarity": card.card_rarity,
+      "card_type": card.card_type}
+    for card in cards
+  ]