helpers.py

from datetime import datetime, timezone
from functools import wraps
from flask import session, redirect
from cs50 import SQL

# Create a single shared DB connection (important for performance)
db = SQL("sqlite:///tama.db")


# ---------------------------------------------------
# LOGIN REQUIRED DECORATOR
# ---------------------------------------------------
def login_required(f):
    """
    Decorator that ensures the user is logged in
    before accessing a route.
    """
    @wraps(f)
    def decorated(*args, **kwargs):
        if session.get("user_id") is None:
            return redirect("/login")
        return f(*args, **kwargs)
    return decorated


# ---------------------------------------------------
#  PET STAGE LOGIC
# ---------------------------------------------------
def compute_stage(pet):
    """
    Determine the pet stage based on age in minutes.
    If any stat reaches death conditions, return 'dead'.

    Stages timeline:
    - 0–1 min   → egg
    - 1–2 min   → baby
    - 2–4 min   → child
    - 4–6 min   → teen
    - 6–8 min  → adult
    - 8-10 min   → elder
    """

    # Important: birth stored as TEXT without timezone → parse cleanly
    birth = datetime.fromisoformat(pet["birth"])

    # Use timezone-aware datetime consistently
    now = datetime.now(timezone.utc).replace(tzinfo=None)

    age_minutes = (now - birth).total_seconds() / 60

    # Death conditions
    if (
        age_minutes > 10 or
        pet["health"] <= 0 or
        pet["happiness"] <= 0 or
        pet["hygiene"] <= 0 or
        pet["hunger"] >= 100
    ):
        return "dead"

    if age_minutes < 1:
        return "egg"
    elif age_minutes < 2:
        return "baby"
    elif age_minutes < 4:
        return "child"
    elif age_minutes < 6:
        return "teen"
    elif age_minutes < 8:
        return "adult"
    else:
        return "elder"


# ---------------------------------------------------
#  NEEDS DECAY SYSTEM
# ---------------------------------------------------
def update_needs(pet):
    """
    Applies automatic stat decay based on the number of minutes
    passed since last_update.

    This function:
    - calculates minutes elapsed
    - applies decay formulas
    - updates the database
    - returns updated pet dictionary
    """

    last_update = datetime.fromisoformat(pet["last_update"])
    now = datetime.now(timezone.utc).replace(tzinfo=None)

    seconds = (now - last_update).total_seconds()

    # No decay if less than 1 minute has passed
    if seconds < 5:
        return pet

    stage = compute_stage(pet)
    stage_factor = 1

    if stage == "adult":
        stage_factor = 2
    elif stage == "elder":
        stage_factor = 3

    # --- DECAY RULES ---
    # You can adjust these values to make the game harder/easier.
    hunger = min(100, pet["hunger"] + seconds * 5)
    happiness = max(0, pet["happiness"] - seconds * 2)
    hygiene = max(0, pet["hygiene"] - seconds * 2)
    health = max(0, pet["health"] - seconds * 1 * stage_factor)

    # Update DB only once per request to avoid lag
    db.execute("""
        UPDATE pet
        SET hunger = ?,
            happiness = ?,
            hygiene = ?,
            health = ?,
            last_update = CURRENT_TIMESTAMP
        WHERE id = ?
    """, hunger, happiness, hygiene, health, pet["id"])

    # Update the local dictionary so the caller gets fresh values
    pet["hunger"] = hunger
    pet["happiness"] = happiness
    pet["hygiene"] = hygiene
    pet["health"] = health
    pet["last_update"] = now.isoformat()

    return pet