from flask import (
    Blueprint,
    redirect,
    url_for,
    render_template,
    flash,
    abort,
    session,
)
from flask_login import current_user
from application import db
from forms import FoodItemForm, MacroForm
from models import FoodItem, FoodLog
from datetime import datetime
from application.utils import login_required, macro_arr_to_json
from numpy import array
from zoneinfo import ZoneInfo
from sqlalchemy import func

user_bp = Blueprint(
    "user",
    __name__,
    template_folder="templates",
)

user_bp.before_request(login_required)


@user_bp.route("/dashboard", methods=["GET"])
def dashboard():
    items = current_user.food_items.all()
    return render_template("dashboard.html", items=items)


@user_bp.route("/daily_log", methods=["GET"])
def daily_log():
    # Get today's date according to user's timezone
    today = datetime.now(ZoneInfo(current_user.timezone)).date()

    # Save date in session
    session["selected_date"] = today.isoformat()

    # Get logs from today
    logs_today = (
        current_user.food_logs.join(
            FoodItem, FoodItem.id == FoodLog.food_item_id
        )
        .filter(FoodLog.date_ == today)
        .group_by(FoodItem.id)
        .with_entities(
            FoodItem,  # the full object
            func.sum(FoodLog.amount).label("total_amount"),
        )
        .all()
    )

    # calculate macros
    macros = array((0.0, 0.0, 0.0, 0.0))
    for food_item, amount in logs_today:
        macros += array(food_item.macros()) / 100 * amount
    macros = macro_arr_to_json(macros.tolist())

    # Render HTML
    return render_template(
        "daily_log.html",
        macros=macros,
        logs=logs_today,
        today=today.strftime("%d/%m/%Y"),
        min=min,
    )


@user_bp.route("/delete_food_item/<int:id>", methods=["POST"])
def delete_food_item(id: int):
    item = FoodItem.query.get(id)
    if item:
        if item.owner_id == current_user.id:
            db.session.delete(item)
            db.session.commit()
        else:
            flash("You do not own this food item!")
    else:
        flash("This food item does not exist")
    return redirect(url_for("user.dashboard"))


@user_bp.route("/edit_food_item/<int:id>", methods=["GET", "POST"])
def edit_food_item(id: int):
    item = FoodItem.query.get(id)
    if item:
        if item.owner_id == current_user.id:
            form = FoodItemForm()
            if form.validate_on_submit():
                item.updateFromForm(form=form)
                db.session.commit()
                return redirect(url_for("user.dashboard"))
            form.barcode.data = item.barcode
            form.name.data = item.name
            form.energy.data = item.energy_100
            form.protein.data = item.protein_100
            form.carbs.data = item.carbs_100
            form.sugar.data = item.sugar_100
            form.fat.data = item.fat_100
            form.saturated_fat.data = item.saturated_fat_100
            return render_template("edit_food_item.html", form=form)
    return redirect(url_for("user.dashboard"))


@user_bp.route("/remove_log/<int:id>", methods=["POST"])
def remove_log(id: int):
    log = db.session.get(FoodLog, id)
    # Check if log exists and if user owns log
    if log is None or log.user_id != current_user.id:
        abort(404)

    # Delete log
    db.session.delete(log)
    db.session.commit()
    return redirect(url_for("user.daily_log"))


@user_bp.route("/set_macros", methods=["GET", "POST"])
def set_macros():
    form = MacroForm()

    if form.validate_on_submit():
        current_user.set_macros(
            form.protein.data,
            form.carbohydrates.data,
            form.fat.data,
            form.calories.data,
        )
        db.session.commit()
        return redirect(url_for("user.daily_log"))

    return render_template("settings.html", form=form)