#include <agb_debug.h>
#include <constants/pkmns.h>
#include <pokeagb/pokeagb.h>
#include <pokedex/pdexArrow.h>
#include <pokedex/pdexBall.h>
#include <pokedex/pdexBg.h>
#include <pokedex/pdexScrollBar.h>
#include <pokedex/pdexSelectHalf.h>

//testing ci, no hate plz

#include "pokedex_common.h"

#define TB_TITLE 0
#define TB_PKMN 1
#define TB_SEEN 2
#define TB_CAUGHT 3
#define TB_MAIN 4

#define TB_SEEN_Y (6)
#define TB_CAUGHT_Y 3

#define DEX_SCROLL_MIN 23
#define DEX_SCROLL_MAX 146
#define SCROLL_SPEED_MAX 6

void pdex_load(void);
void pdex_vblank_handler(void);
void pdex_loop(u8 tid);
void pdex_cb_handler(void);

extern const pchar pdex_str_title[];
extern const pchar pdex_str_seen[];
extern const pchar pdex_str_caught[];
extern const pchar pdex_str_empty[];

static const u8 pdex_y_offset[] = {19, 35, 51, 67, 83, 99, 115, 131};
static const u16 scroll_speed_delays[] = {20, 20, 20, 20, 10, 5, 0};

struct TextboxTemplate pdex_boxes[] = {
    {.bg_id = 0, .x = 11, .y = 0, .width = 10, .height = 2, .pal_id = 15, .charbase = 1},
    {.bg_id = 0, .x = 2, .y = 2, .width = 10, .height = 2, .pal_id = 15, .charbase = 21},
    {.bg_id = 0, .x = 3, .y = 13, .width = 9, .height = 3, .pal_id = 15, .charbase = 41},
    {.bg_id = 0, .x = 3, .y = 16, .width = 9, .height = 2, .pal_id = 15, .charbase = 59},

    {.bg_id = 1, .x = 16, .y = 0, .width = 11, .height = 32, .pal_id = 15, .charbase = 77},

    {.bg_id = 0xFF},
};

s16 pdex_get_y_offset(s8 n) {
    s8 modOffset = n + pokedex_context->hardware_scroll_amount;
    if (modOffset < -1) {
        modOffset += 16;
    }

    if (modOffset > 14) {
        modOffset -= 16;
    }
    return (19 + (16 * modOffset));
}

void pdex_main_box_species_fill(s8 n, u16 species, bool seen, bool caught) {
    seen = true;
    s16 y = pdex_get_y_offset(n);
    rboxid_fill_rectangle(TB_MAIN, 0, 0, y, 11 * 8, 16);
    const pchar *stringToPrint = (seen || caught) ? &pokemon_names[species][0] : &pdex_str_empty[0];
    const pchar stringWhitespace[] = {0x0, 0xFF};
    fmt_int_10(string_buffer, pokedex_context->cursor_position_top + n, 2, 3);
    pstrcat(string_buffer, &stringWhitespace[0]);
    pstrcat(string_buffer, stringToPrint);
    rboxid_print(TB_MAIN, FONT_DEX_STD, 4, y, &pdex_text_color, 0, string_buffer);
}

void pdex_update_balls(void) {
    for (u8 i = 0; i < 8; ++i) {
        if (pdex_lazy_lookup_entry(i + pokedex_context->cursor_position_top)->caught)
            OBJID_SHOW(pokedex_context->ball_oams[i]);
        else
            OBJID_HIDE(pokedex_context->ball_oams[i]);
    }
}

void pdex_update_page_full() {
    rboxid_clear_pixels(TB_MAIN, 0);
    for (u16 i = pokedex_context->cursor_position_top; i < pokedex_context->cursor_position_top + 8; ++i) {
        pdex_main_box_species_fill(i - pokedex_context->cursor_position_top, pdex_lazy_lookup_entry(i)->species,
                                   pdex_lazy_lookup_entry(i)->seen, pdex_lazy_lookup_entry(i)->caught);
    }
    pdex_update_balls();
    rboxid_update_tilemap_and_tileset(TB_MAIN);
}

void pdex_load_sc(void) {
    rboxid_clear_pixels(TB_SEEN, 0);
    rboxid_clear_pixels(TB_CAUGHT, 0);
    u32 seen = pokedex_count(false) + 10;
    u32 caught = pokedex_count(true);
    pchar seenBuffer[4];
    pchar caughtBuffer[4];
    fmt_int_10(seenBuffer, seen, 0, MAX3_COUNT_DIGITS(seen));
    fmt_int_10(caughtBuffer, caught, 0, MAX3_COUNT_DIGITS(caught));

    u32 twidthSeen = font_get_width_of_string(FONT_DEX_STD, seenBuffer, 0x0000);
    u32 twidthCaught = font_get_width_of_string(FONT_DEX_STD, caughtBuffer, 0x0000);

    rboxid_print(TB_SEEN, FONT_DEX_STD, 0, TB_SEEN_Y, &pdex_text_color, 0, &pdex_str_seen[0]);
    rboxid_print(TB_CAUGHT, FONT_DEX_STD, 0, TB_CAUGHT_Y, &pdex_text_color, 0, &pdex_str_caught[0]);

    rboxid_print(TB_SEEN, FONT_DEX_STD, TB_STD_RIGHT(twidthSeen, TB_BOT_LEN_PX), TB_SEEN_Y + 1, &pdex_text_color, 0,
                 seenBuffer);
    rboxid_print(TB_CAUGHT, FONT_DEX_STD, TB_STD_RIGHT(twidthCaught, TB_BOT_LEN_PX), TB_CAUGHT_Y + 1, &pdex_text_color,
                 0, caughtBuffer);

    rboxid_update_tilemap_and_tileset(TB_SEEN);
    rboxid_update_tilemap_and_tileset(TB_CAUGHT);
}
void pdex_pokemon_oam_update(u16 species, s8 objid) {
    gpu_pal_apply_compressed(pokemon_palette_normal[species].data, 16 * (objects[objid].final_oam.palette_num + 16),
                             32);
    lz77UnCompVram(pokemon_graphics_front[species].data,
                   ((void *)(objects[objid].final_oam.tile_num * 32) + ADDR_VRAM + 0x10000));
}

s8 pdex_pokemon_oam_load(u16 species, u16 tag) {
    struct SpriteTiles pkmnTiles = {pokemon_graphics_front[species].data, 2048, tag};
    struct SpritePalette pkmnPal = {pokemon_palette_normal[species].data, tag};
    const struct Template pkmnTemplate = {
        .tiles_tag = tag,
        .pal_tag = tag,
        .oam = &pdex_oam_pkmn,
        .animation = &anim_image_empty,
        .graphics = &pkmnTiles,
        .rotscale = &rotscale_empty,
        .callback = oac_nullsub,
    };
    gpu_tile_obj_decompress_alloc_tag_and_upload(&pkmnTiles);

    gpu_pal_decompress_alloc_tag_and_upload(&pkmnPal);
    s8 objid = (s8)template_instanciate_forward_search(&pkmnTemplate, 10, 10, 0);

    objects[objid].pos1.x = 55;
    objects[objid].pos1.y = 76;
    OBJID_HIDE(objid);
    return objid;
}

void pdex_pokemon_load(u16 species) {
    /* this is very temporary */
    rboxid_clear_pixels(TB_PKMN, 0);
    u32 twidth = font_get_width_of_string(FONT_DEX_STD, &pokemon_names[species][0], 0x0000);
    rboxid_print(TB_PKMN, FONT_DEX_STD, TB_STD_CENTER(twidth, TB_STD_LEN_PX), 3, &pdex_text_color, 0,
                 &pokemon_names[species][0]);

    // pokedex_context->pokemon_oam = pdex_pokemon_oam_load(species, pokedex_context->pokemon_oam);

    rboxid_update_tilemap_and_tileset(TB_PKMN);
}

struct SpriteTiles pdex_ball_tiles = {pdexBallTiles, 128, DEX_BALL_TAG};
struct SpritePalette pdex_ball_pal = {pdexBallPal, DEX_BALL_TAG};
const struct OamData pdex_ball_oam = {
    .affine_mode = 0,
    .obj_mode = 0,
    .mosaic = false,
    .shape = 0,
    .size = 1,
};

const struct Template pdex_ball_template = {
    .tiles_tag = DEX_BALL_TAG,
    .pal_tag = DEX_BALL_TAG,
    .oam = &pdex_ball_oam,
    .animation = &anim_image_empty,
    .graphics = &pdex_ball_tiles,
    .rotscale = &rotscale_empty,
    .callback = oac_nullsub,
};

void pdex_pokeballs_init(void) {
    gpu_tile_obj_decompress_alloc_tag_and_upload(&pdex_ball_tiles);
    gpu_pal_obj_alloc_tag_and_apply(&pdex_ball_pal);
    for (u8 i = 0; i < 8; ++i) {
        pokedex_context->ball_oams[i] =
            template_instanciate_forward_search(&pdex_ball_template, 124, 8 + pdex_y_offset[i], 1);
    }
}

void pdex_oac_cursor_follow(struct Object *obj) { obj->pos1.y = objects[pokedex_context->cursor_main_oam].pos1.y; }

void pdex_oac_cursor_main(struct Object *obj) {
    obj->pos1.y = 13 + pdex_y_offset[pokedex_context->cursor_position_internal];
}

struct SpriteTiles pdex_cursor_tiles = {pdexSelectHalfTiles, 1024, DEX_CURSOR_TAG};
struct SpritePalette pdex_cursor_pal = {pdexSelectHalfPal, DEX_CURSOR_TAG};
const struct OamData pdex_cursor_oam = {
    .affine_mode = 0,
    .obj_mode = 0,
    .mosaic = false,
    .shape = 1,
    .size = 3,
};

const struct Template pdex_ball_main_template = {
    .tiles_tag = DEX_CURSOR_TAG,
    .pal_tag = DEX_CURSOR_TAG,
    .oam = &pdex_cursor_oam,
    .animation = &anim_image_empty,
    .graphics = &pdex_cursor_tiles,
    .rotscale = &rotscale_empty,
    .callback = pdex_oac_cursor_main,
};

const struct Template pdex_ball_follow_template = {
    .tiles_tag = DEX_CURSOR_TAG,
    .pal_tag = DEX_CURSOR_TAG,
    .oam = &pdex_cursor_oam,
    .animation = &anim_image_empty,
    .graphics = &pdex_cursor_tiles,
    .rotscale = &rotscale_empty,
    .callback = pdex_oac_cursor_follow,
};

void pdex_cursor_init(void) {
    gpu_tile_obj_decompress_alloc_tag_and_upload(&pdex_cursor_tiles);
    gpu_pal_obj_alloc_tag_and_apply(&pdex_cursor_pal);
    pokedex_context->cursor_main_oam = template_instanciate_forward_search(&pdex_ball_main_template, 144, 32, 0);
    pokedex_context->cursor_follow_oam =
        template_instanciate_forward_search(&pdex_ball_follow_template, 144 + 48, 32, 0);
    objects[pokedex_context->cursor_follow_oam].final_oam.h_flip = true;
}

void pdex_oac_scroll_bar(struct Object *obj) {
    obj->pos1.y = 23 + (((pokedex_context->cursor_position_top + pokedex_context->cursor_position_internal) * 123) /
                        PDEX_LAST_SHOWN);
}

void pdex_oac_arrow(struct Object *obj) {
    if (obj->priv[0]) {
        // this is the down facing arrow
        if (pokedex_context->cursor_position_top + 8 > PDEX_LAST_SHOWN) {
            obj->final_oam.palette_num = obj->priv[2];
        } else {
            obj->final_oam.palette_num = obj->priv[1];
        }
    } else {
        // this is the up facing arrow
        if (pokedex_context->cursor_position_top == pokedex_context->first_seen) {
            obj->final_oam.palette_num = obj->priv[2];
        } else {
            obj->final_oam.palette_num = obj->priv[1];
        }
    }
}

struct SpriteTiles pdex_scroll_bar_tiles = {pdexScrollBarTiles, 32, DEX_SCROLL_TAG};
struct SpritePalette pdex_scroll_bar_pal = {pdexScrollBarPal, DEX_SCROLL_TAG};
struct SpriteTiles pdex_arrow_tiles = {pdexArrowTiles, 128, DEX_ARROW_TAG};
struct SpritePalette pdex_arrow_pal = {pdexArrowPal, DEX_ARROW_TAG};
struct SpritePalette pdex_arrow_pal_gray = {pdexArrowPal, DEX_ARROW_TAG_EPAL};

const struct OamData pdex_scroll_bar_oam = {
    .affine_mode = 0,
    .obj_mode = 0,
    .mosaic = false,
    .shape = 0,
    .size = 0,
};

const struct OamData pdex_arrow_oam = {
    .affine_mode = 0,
    .obj_mode = 0,
    .mosaic = false,
    .shape = 0,
    .size = 1,
};

const struct Template pdex_scroll_bar_template = {
    .tiles_tag = DEX_SCROLL_TAG,
    .pal_tag = DEX_SCROLL_TAG,
    .oam = &pdex_scroll_bar_oam,
    .animation = &anim_image_empty,
    .graphics = &pdex_scroll_bar_tiles,
    .rotscale = &rotscale_empty,
    .callback = pdex_oac_scroll_bar,
};

const struct Template pdex_arrow_template = {
    .tiles_tag = DEX_ARROW_TAG,
    .pal_tag = DEX_ARROW_TAG,
    .oam = &pdex_arrow_oam,
    .animation = &anim_image_empty,
    .graphics = &pdex_arrow_tiles,
    .rotscale = &rotscale_empty,
    .callback = pdex_oac_arrow,
};

void pdex_load_scroll_ui(void) {
    gpu_tile_obj_decompress_alloc_tag_and_upload(&pdex_scroll_bar_tiles);
    gpu_tile_obj_decompress_alloc_tag_and_upload(&pdex_arrow_tiles);
    gpu_pal_obj_alloc_tag_and_apply(&pdex_scroll_bar_pal);
    u8 normalPal = gpu_pal_obj_alloc_tag_and_apply(&pdex_arrow_pal);
    u8 grayPal = gpu_pal_obj_alloc_tag_and_apply(&pdex_arrow_pal_gray);
    tint_palette_gray(&palette_bg_unfaded[16 * (16 + grayPal)], 16);
    (void)template_instanciate_forward_search(&pdex_scroll_bar_template, 233, 23, 0);
    u8 upArrow = template_instanciate_forward_search(&pdex_arrow_template, 232, 19, 0);
    u8 downArrow = template_instanciate_forward_search(&pdex_arrow_template, 232, 150, 0);
    objects[downArrow].priv[0] = true;
    objects[downArrow].final_oam.v_flip = true;
    objects[upArrow].priv[1] = normalPal;
    objects[downArrow].priv[1] = normalPal;
    objects[upArrow].priv[2] = grayPal;
    objects[downArrow].priv[2] = grayPal;
}

void pdex_data_setup(void) { pokedex_context->first_seen = 1; }

void pdex_hardware_scroll(bool up) {
    if (up) {
        if (pokedex_context->hardware_scroll_amount > -15) {
            bgid_mod_y_offset(1, -(16 << 8), 1);
            pokedex_context->hardware_scroll_amount--;
        } else {
            pokedex_context->hardware_scroll_amount = 0;
            bgid_mod_y_offset(1, 0, 0);
        }
    } else {

        if (pokedex_context->hardware_scroll_amount < 15) {
            bgid_mod_y_offset(1, 16 << 8, 1);
            pokedex_context->hardware_scroll_amount++;
        } else {
            pokedex_context->hardware_scroll_amount = 0;
            bgid_mod_y_offset(1, 0, 0);
        }
    }
}

void pdex_fill_previous_slot(void) {
    u16 pIndex = pokedex_context->cursor_position_top - 1;
    pdex_main_box_species_fill(-1, pdex_lazy_lookup_entry(pIndex)->species, pdex_lazy_lookup_entry(pIndex)->seen,
                               pdex_lazy_lookup_entry(pIndex)->caught);
}

void pdex_fill_next_slot(void) {
    u16 pIndex = pokedex_context->cursor_position_top + 8;
    pdex_main_box_species_fill(8, pdex_lazy_lookup_entry(pIndex)->species, pdex_lazy_lookup_entry(pIndex)->seen,
                               pdex_lazy_lookup_entry(pIndex)->caught);
}

void pdex_try_advance(u8 reverse) {
    if (reverse) {
        if (pokedex_context->cursor_position_internal > 0) {
            pokedex_context->cursor_position_internal--;
            m4aSongNumStart(600);
        } else if ((pokedex_context->cursor_position_top) > (pokedex_context->first_seen)) {
            pdex_fill_previous_slot();
            pokedex_context->cursor_position_top--;
            pdex_hardware_scroll(true);
            m4aSongNumStart(600);
        } else {
            return;
        }
    } else {
        if (pokedex_context->cursor_position_internal < 7) {
            pokedex_context->cursor_position_internal++;
            m4aSongNumStart(600);
        } else if (pokedex_context->cursor_position_top < (PDEX_LAST_SHOWN - 7)) {
            pdex_fill_next_slot();
            pokedex_context->cursor_position_top++;
            pdex_hardware_scroll(false);
            m4aSongNumStart(600);
        } else {
            return;
        }
    }

    u16 pkIndexToLoad = pokedex_context->cursor_position_internal + pokedex_context->cursor_position_top;
    u16 speciesToLoad = (pdex_lazy_lookup_entry(pkIndexToLoad)->seen || pdex_lazy_lookup_entry(pkIndexToLoad)->caught)
                            ? pdex_lazy_lookup_entry(pkIndexToLoad)->species
                            : pdex_lazy_lookup_entry(pkIndexToLoad)->species; /* debug, just display the mofo */

    pdex_pokemon_load(speciesToLoad);
    if (reverse) {
        s8 oam = pokedex_context->pokemon_oam; // current oam id
        OBJID_HIDE(oam);
        pokedex_context->pokemon_oam = pokedex_context->prev_pokemon_oam;
        OBJID_SHOW(pokedex_context->pokemon_oam);
        pokedex_context->prev_pokemon_oam = pokedex_context->next_pokemon_oam;
        pokedex_context->next_pokemon_oam = oam;

        pdex_pokemon_oam_update(pdex_lazy_lookup_entry(pkIndexToLoad - 1)->species, pokedex_context->prev_pokemon_oam);
    } else {
        s8 oam = pokedex_context->pokemon_oam;
        OBJID_HIDE(oam);
        pokedex_context->pokemon_oam = pokedex_context->next_pokemon_oam;
        OBJID_SHOW(pokedex_context->pokemon_oam);
        pokedex_context->next_pokemon_oam = pokedex_context->prev_pokemon_oam;
        pokedex_context->prev_pokemon_oam = oam;
        pdex_pokemon_oam_update(pdex_lazy_lookup_entry(pkIndexToLoad + 1)->species, pokedex_context->next_pokemon_oam);
    }
    pdex_update_balls();
}

void pdex_loop(u8 tid) {
    switch (pokedex_context->state) {
    case 0:
        pokedex_context->state++;
        break;
    case 1:
        for (u8 i = 0; i < 4; ++i)
            bgid_send_tilemap(i);
        rboxid_clear_pixels(TB_TITLE, 0);
        rboxid_print(TB_TITLE, FONT_DEX_STD, 0, 0, &pdex_text_color, 0, &pdex_str_title[0]);
        rboxid_update_tilemap_and_tileset(TB_TITLE);

        pdex_data_setup();
        pdex_cursor_init();

        pdex_pokeballs_init();
        pdex_load_scroll_ui();
        u16 currentIndex = pokedex_context->cursor_position_top + pokedex_context->cursor_position_internal;
        pdex_pokemon_load(pdex_lazy_lookup_entry(currentIndex)->species);

        pokedex_context->pokemon_oam =
            pdex_pokemon_oam_load(pdex_lazy_lookup_entry(currentIndex)->species, DEX_PKMN_TAG_ONE);
        OBJID_SHOW(pokedex_context->pokemon_oam);

        pokedex_context->next_pokemon_oam =
            pdex_pokemon_oam_load(pdex_lazy_lookup_entry(currentIndex + 1)->species, DEX_PKMN_TAG_TWO);

        if (currentIndex != pokedex_context->first_seen) {
            pokedex_context->prev_pokemon_oam =
                pdex_pokemon_oam_load(pdex_lazy_lookup_entry(currentIndex - 1)->species, DEX_PKMN_TAG_THREE);
        } else {
            pokedex_context->prev_pokemon_oam = pdex_pokemon_oam_load(0, DEX_PKMN_TAG_THREE);
        }
        pdex_load_sc();
        pdex_update_page_full();

        palette_bg_faded_fill_black();
        pokedex_context->state++;
        break;
    case 2:
        gpu_sync_bg_show(0);
        gpu_sync_bg_show(1);
        gpu_sync_bg_hide(3);

        gpu_sync_bg_show(2);
        fade_screen(0xFFFFFFFF, PDEX_FADEIN_SPD, 16, 0, 0x0000);
        pokedex_context->state++;

        break;
    case 3:
        if (!pal_fade_control.active)
            pokedex_context->state++;
        break;
    case 4:
        /* main control */
        if (super.buttons_new & KEY_A) {
            pokedex_context->state = 15;
            m4aSongNumStart(5);
        }
        if (super.buttons_new & KEY_B) {
            pokedex_context->state = 10;
            m4aSongNumStart(601);
        }
        if ((super.buttons_new & KEY_DOWN)) {
            pdex_try_advance(false);
            pokedex_context->scroll_speed = 0;
            pokedex_context->delay_count = 0;
        } else if (super.buttons_held & KEY_DOWN) {
            if (pokedex_context->scroll_speed < SCROLL_SPEED_MAX) {
                if (pokedex_context->delay_count >= scroll_speed_delays[pokedex_context->scroll_speed]) {
                    pdex_try_advance(false);
                    pokedex_context->delay_count = 0;
                    pokedex_context->scroll_speed++;
                } else {
                    pokedex_context->delay_count++;
                }
            } else {
                pdex_try_advance(false);
            }
        }
        if ((super.buttons_new & KEY_UP)) {
            pdex_try_advance(true);
            pokedex_context->scroll_speed = 0;
            pokedex_context->delay_count = 0;
        } else if (super.buttons_held & KEY_UP) {
            if (pokedex_context->scroll_speed < SCROLL_SPEED_MAX) {
                if (pokedex_context->delay_count >= scroll_speed_delays[pokedex_context->scroll_speed]) {
                    pdex_try_advance(true);
                    pokedex_context->delay_count = 0;
                    pokedex_context->scroll_speed++;
                } else {
                    pokedex_context->delay_count++;
                }
            } else {
                pdex_try_advance(true);
            }
        }
        break;
    case 10:
        /* control flow exit */
        fade_screen(0xFFFFFFFF, PDEX_FADEIN_SPD, 0, 16, 0x000);
        pokedex_context->state++;
        break;
    case 11:
        if (!pal_fade_control.active) {
            task_del(tid);
            pdex_vram_free_bgmaps();
            pdex_free_memory();
            void region_select_load(void);
            set_callback2(region_select_load);
        }
        break;
    case 15:
        /* load the detail screen */
        fade_screen(0xFFFFFFFF, PDEX_FADEIN_SPD, 0, 16, 0x0000);
        pokedex_context->state++;
        break;
    case 16:
        if (!pal_fade_control.active) {
            pdex_vram_free_bgmaps();
            task_del(tid);
            void dexdetail_load(void);
            set_callback2(dexdetail_load);
        }
    default:
        break;
    }
}

void pdex_load_gfx(void) {
    /*TODO: setup text boxes here */
    rbox_init_from_templates(&pdex_boxes[0]);

    lz77UnCompVram(pdexBgTiles, (void *)0x0600C000);
    LZ77UnCompWram(pdexBgMap, bgid_get_tilemap(2));
    gpu_pal_apply_compressed(pdexBgPal, 0, 32);
    gpu_pal_apply(pdex_text_pal, 15 * 16, 32);

    /*setup window*/
    lcd_io_set(REG_ID_WIN0H, (128 << 8) | (216));
    lcd_io_set(REG_ID_WIN0V, ((20 << 8) | (148)));
    lcd_io_set(REG_ID_WININ, WIN_BG0 | WIN_BG1 | WIN_BG2 | WIN_BG3 | WIN_OBJ);
    lcd_io_set(REG_ID_WINOUT, WIN_BG0 | WIN_BG2 | WIN_BG3 | WIN_OBJ);
    bgid_mark_for_sync(0);
}

void pdex_load(void) {

    pdex_vram_setup();
    pdex_load_gfx();

    pokedex_context->pokemon_oam = -1;
    pokedex_context->state = 0;
    pokedex_context->hardware_scroll_amount = 0;
    task_add(pdex_loop, 0);
    set_callback2(pdex_cb_handler);
}