from files.__main__ import app
from files.classes.hats import *
from files.helpers.alerts import *
from files.helpers.wrappers import *
from files.helpers.const import *
from files.helpers.actions import *
from flask import g

@app.get("/hats")
@auth_required
def hats(v):
	if not FEATURES['HATS']: abort(404)

	owned_hat_ids = [x.hat_id for x in v.owned_hats]

	if v.equipped_hat_id:
		owned = [(v.equipped_hat, v.equipped_hat.author)] + g.db.query(HatDef, User).join(HatDef.author).filter(HatDef.id.in_(owned_hat_ids), HatDef.id != v.equipped_hat_id).order_by(HatDef.price, HatDef.name).all()
	else:
		owned = g.db.query(HatDef, User).join(HatDef.author).filter(HatDef.id.in_(owned_hat_ids)).order_by(HatDef.price, HatDef.name).all()

	not_owned = g.db.query(HatDef, User).join(HatDef.author).filter(HatDef.id.notin_(owned_hat_ids)).order_by(HatDef.price, HatDef.name).all()
	hats = owned + not_owned

	return render_template("hats.html", owned_hat_ids=owned_hat_ids, hats=hats, v=v)

@app.post("/buy_hat/<hat_id>")
@auth_required
def buy_hat(v, hat_id):
	if not FEATURES['HATS']: abort(404)

	try: hat_id = int(hat_id)
	except: return {"error": "Hat not found!"}

	hat = g.db.query(HatDef).filter_by(id=hat_id).one_or_none()
	if not hat: return {"error": "Hat not found!"}

	existing = g.db.query(Hat).filter_by(user_id=v.id, hat_id=hat.id).one_or_none()
	if existing: return {"error": "You already own this hat!"}

	if request.values.get("mb"):
		if v.procoins < hat.price: return {"error": "Not enough marseybux."}, 400
		v.procoins -= hat.price
		hat.author.procoins += hat.price * 0.05
		currency = "marseybux"
	else:
		if v.coins < hat.price: return {"error": "Not enough coins."}, 400
		v.coins -= hat.price
		v.coins_spent_on_hats += hat.price
		hat.author.coins += hat.price * 0.05
		currency = "coins"

	new_hat = Hat(user_id=v.id, hat_id=hat.id)
	g.db.add(new_hat)

	g.db.add(v)
	g.db.add(hat.author)

	if v.id != hat.author.id:
		send_repeatable_notification(
			hat.author.id,
			f":marseycapitalistmanlet: @{v.username} has just bought `{hat.name}`, you have received your 5% cut ({int(hat.price * 0.05)} {currency}) :!marseycapitalistmanlet:"
		)

	if v.num_of_owned_hats >= 250:
		badge_grant(user=v, badge_id=154)
	elif v.num_of_owned_hats >= 100:
		badge_grant(user=v, badge_id=153)
	elif v.num_of_owned_hats >= 25:
		badge_grant(user=v, badge_id=152)

	return {"message": "Hat bought!"}


@app.post("/equip_hat/<hat_id>")
@auth_required
def equip_hat(v, hat_id):
	if not FEATURES['HATS']: abort(404)

	try: hat_id = int(hat_id)
	except: return {"error": "Hat not found!"}

	hat = g.db.query(Hat).filter_by(hat_id=hat_id, user_id=v.id).one_or_none()
	if not hat: return {"error": "You don't own this hat!"}

	v.equipped_hat_id = hat_id
	g.db.add(v)

	return {"message": "Hat equipped!"}

@app.post("/unequip_hat")
@auth_required
def unequip_hat(v):
	if not FEATURES['HATS']: abort(404)

	v.equipped_hat_id = None
	g.db.add(v)

	return {"message": "Hat unequipped!"}