#include "shop.h"
#include "engine/maths.h"
#include "ppu/ppu.h"

bool hh_show_Shop(){
	static hh_e_ShopStates hh_e_Shop = hh_e_shop_init;
	static hh_e_upgrades upgrades[HH_SHOP_UPG_DISPLAY] = {0};
	static uint8_t selected = 0;
	
	switch (hh_e_Shop)
	{
	case hh_e_shop_init:
		hh_clear_screen();
		hh_clear_sprite();

		//TODO: render shop bg from level file here
		//hh_setup_shop();
		hh_shop_init(&upgrades);
		selected = HH_SHOP_UPG_DISPLAY/2;
		hh_shop_display(selected, &upgrades);
		hh_e_Shop = hh_e_shop_main;
		return false;
		break;
	case hh_e_shop_main:
		if(g_hh_controller_p1.dpad_left || g_hh_controller_p1.dpad_right){
			hh_shift_selected(&selected,(g_hh_controller_p1.dpad_right?1:0),0,HH_SHOP_UPG_DISPLAY);
			hh_shop_display(selected, &upgrades);
		}
		if(g_hh_controller_p1.button_primary){
			//apply selected upgrade
			hh_e_Shop = hh_e_shop_end;
		}
		if(g_hh_controller_p1.button_secondary){
			hh_e_Shop = hh_e_shop_end;
		}
		break;
	case hh_e_shop_end:
		hh_e_Shop = hh_e_shop_init;
		return true;
		break;
	default:
		hh_e_Shop = hh_e_shop_init;
		break;
	}
	return false;
}

void hh_shop_init(hh_e_upgrades* in) {
	for (int i = 0; i < HH_SHOP_UPG_DISPLAY; i++) {
		in[i] = i%HH_SHOP_UPG_COUNT;
	}
}

void hh_shop_display(uint8_t selected, hh_e_upgrades* upgrades) {
	const vec_cor start = {48,16};
	const uint8_t up = 8,
					space = 24+8;

	for (int i = 0; i < HH_SHOP_UPG_DISPLAY; i++) {
		hh_ppu_update_foreground(i+16,
		(hh_s_ppu_loc_fam_entry){
			.horizontal_flip = false, .vertical_flip = false,
			.position_x = i*space+start.x, .position_y = start.y + (i==selected?up:0),
			.palette_index = 7,
			.tilemap_index = i
		});
	}
}

void hh_shift_selected(uint8_t* pos, bool dir, uint8_t min, uint8_t max) {
	if (dir) {
		pos = CLAMP(++pos,min,max);
	} else {
		if (pos > 0) {
			pos--;
		}
	}
}