app.py

from cs50 import SQL
from flask import Flask, render_template, request, redirect, session, flash
from flask_session import Session
from werkzeug.security import check_password_hash, generate_password_hash
from datetime import datetime

from helpers import login_required, compute_stage, update_needs

app = Flask(__name__)

# Session configuration
app.config["SESSION_PERMANENT"] = False
app.config["SESSION_TYPE"] = "filesystem"
Session(app)

# Database connection
db = SQL("sqlite:///tama.db")


# ---------------------------------------------------
# HOME
# ---------------------------------------------------
@app.route("/")
@login_required
def index():
    """Main game screen. Loads pet, applies decay, computes stage."""

    pet = db.execute("SELECT * FROM pet WHERE user_id = ?", session["user_id"])

    # No pet yet → redirect to creation
    if not pet:
        return redirect("/create-pet")

    pet = pet[0]

    # Apply time-based decay
    pet = update_needs(pet)

    # Determine stage (egg, baby, teen, adult, elder, dead…)
    stage = compute_stage(pet)

    # Dead pet → show death screen
    if stage == "dead":
        return render_template("dead.html", pet=pet)

    return render_template("index.html", pet=pet, stage=stage)


# ---------------------------------------------------
# LIVE STATUS POLLING ENDPOINT
# (Used by JS to update UI without refreshing)
# ---------------------------------------------------
@app.route("/pet-status")
@login_required
def pet_status():
    """Returns the current stats of the pet as JSON (for AJAX polling)."""

    pet = db.execute("SELECT * FROM pet WHERE user_id = ?",
                     session["user_id"])[0]
    pet = update_needs(pet)
    stage = compute_stage(pet)

    return {
        "health": pet["health"],
        "hunger": pet["hunger"],
        "happiness": pet["happiness"],
        "hygiene": pet["hygiene"],
        "stage": stage
    }


# ---------------------------------------------------
# CREATE PET
# ---------------------------------------------------
@app.route("/create-pet", methods=["GET", "POST"])
@login_required
def create_pet():
    """Allows the user to create a new pet."""

    if request.method == "POST":
        name = request.form.get("name")

        if not name:
            return render_template("create_pet.html", error="Name is required")

        # New pet always starts with default stats
        db.execute("""
            INSERT INTO pet (
                user_id, name, health, hunger, happiness, hygiene, birth, last_update
            )
            VALUES (?, ?, 100, 0, 100, 100, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)
        """, session["user_id"], name)

        return redirect("/")

    return render_template("create_pet.html")


# ---------------------------------------------------
# RESET PET AFTER DEATH
# ---------------------------------------------------
@app.route("/reset-pet")
@login_required
def reset_pet():
    """Deletes current pet after death and redirects to creation page."""
    db.execute("DELETE FROM pet WHERE user_id = ?", session["user_id"])
    return redirect("/create-pet")


# ---------------------------------------------------
# PET ACTIONS
# ---------------------------------------------------
@app.route("/feed")
@login_required
def feed():
    """Feeds the pet (reduces hunger, increases happiness slightly)."""
    db.execute("""
        UPDATE pet
        SET hunger = MAX(hunger - 25, 0),
            happiness = MIN(happiness + 3, 100),
            last_update = CURRENT_TIMESTAMP
        WHERE user_id = ?
    """, session["user_id"])
    return redirect("/")


@app.route("/play")
@login_required
def play():
    """Plays with the pet (boosts happiness but increases hunger)."""
    db.execute("""
        UPDATE pet
        SET happiness = MIN(happiness + 20, 100),
            hunger = MIN(hunger + 10, 100),
            last_update = CURRENT_TIMESTAMP
        WHERE user_id = ?
    """, session["user_id"])
    return redirect("/")


@app.route("/clean")
@login_required
def clean():
    """Cleans the pet (increases hygiene)."""
    db.execute("""
        UPDATE pet
        SET hygiene = MIN(hygiene + 30, 100),
            last_update = CURRENT_TIMESTAMP
        WHERE user_id = ?
    """, session["user_id"])
    return redirect("/")


@app.route("/heal")
@login_required
def heal():
    """Heals the pet (restores health)."""
    db.execute("""
        UPDATE pet
        SET health = MIN(health + 25, 100),
            last_update = CURRENT_TIMESTAMP
        WHERE user_id = ?
    """, session["user_id"])
    return redirect("/")


# ---------------------------------------------------
# AUTHENTICATION
# ---------------------------------------------------
@app.route("/register", methods=["GET", "POST"])
def register():
    if request.method == "POST":

        errors = {}
        username = request.form.get("username")
        password = request.form.get("password")
        confirmation = request.form.get("confirmation")

        # Validation
        if not username:
            errors["username"] = "A username is required."

        if not password:
            errors["password"] = "A password is required."

        if password != confirmation:
            errors["confirmation"] = "Passwords do not match."

        if errors:
            return render_template(
                "register.html",
                errors=errors,
                username=username,
                password=password,
                confirmation=confirmation
            )

        # Try creating user
        try:
            user_id = db.execute(
                "INSERT INTO users (username, hash) VALUES (?, ?)",
                username,
                generate_password_hash(password)
            )
        except Exception:
            errors["username"] = "Username already exists."
            return render_template("register.html", errors=errors)

        session["user_id"] = user_id
        return redirect("/")

    # GET request → send empty errors dict
    return render_template("register.html", errors={})


@app.route("/login", methods=["GET", "POST"])
def login():
    session.clear()

    if request.method == "POST":
        username = request.form.get("username")
        password = request.form.get("password")

        errors = {}

        if not username:
            errors["username"] = "Username is required."

        if not password:
            errors["password"] = "Password is required."

        if errors:
            return render_template("login.html", errors=errors, username=username, password=password)

        row = db.execute("SELECT * FROM users WHERE username = ?", username)

        if len(row) != 1 or not check_password_hash(row[0]["hash"], password):
            errors["password"] = "Invalid credentials."
            return render_template("login.html", errors=errors, username=username)

        session["user_id"] = row[0]["id"]
        return redirect("/")

    # GET REQUEST → return empty errors dict
    return render_template("login.html", errors={})


@app.route("/logout")
def logout():
    """Logs the user out and clears the session."""
    session.clear()
    return redirect("/login")


# ---------------------------------------------------
# MAIN
# ---------------------------------------------------
if __name__ == "__main__":
    app.run(debug=True)