#pragma once
#include "acmd_wrapper.h"
#include "useful/const_value_table.h"
#include "../taunt_toggles.h"

using namespace app::lua_bind;

int major, minor, patch;

u64 fighter_manager_addr;
u64 is_training_mode(void) asm("_ZN3app9smashball16is_training_modeEv") LINKABLE;

u8 get_category(u64 module_accessor) {
	return (u8)(*(u32*)(module_accessor + 8) >> 28);
}

bool is_operation_cpu(u64 module_accessor) {
    if (get_category(module_accessor) != BATTLE_OBJECT_CATEGORY_FIGHTER)
        return false;

    int entry_id = WorkModule::get_int(module_accessor, FIGHTER_INSTANCE_WORK_ID_INT_ENTRY_ID);
    u64 fighter_information = FighterManager::get_fighter_information(LOAD64(fighter_manager_addr), entry_id);

    return FighterInformation::is_operation_cpu(fighter_information);
}

bool is_in_hitstun(u64 module_accessor) {
    int status_kind = StatusModule::status_kind(module_accessor);
    return status_kind >= FIGHTER_STATUS_KIND_DAMAGE &&
           status_kind <= FIGHTER_STATUS_KIND_DAMAGE_FALL;
}

bool is_in_shieldstun(u64 module_accessor) {
    int status_kind = StatusModule::status_kind(module_accessor);
    int prev_status = StatusModule::prev_status_kind(module_accessor, 0);
    // If we are taking shield damage or we are droping shield from taking shield damage we are in hitstun
    if(status_kind == FIGHTER_STATUS_KIND_GUARD_DAMAGE || (prev_status == FIGHTER_STATUS_KIND_GUARD_DAMAGE && status_kind == FIGHTER_STATUS_KIND_GUARD_OFF))
    {
        return true;
    }

    return false;
}


bool is_in_landing(u64 module_accessor) {
    int status_kind = StatusModule::status_kind(module_accessor);
    return status_kind >= FIGHTER_STATUS_KIND_LANDING &&
           status_kind <= FIGHTER_STATUS_KIND_LANDING_DAMAGE_LIGHT;
}

bool should_hold_shield(u64 module_accessor)
{
    // We should hold shield if the state requires it
    if (menu.SHIELD_STATE == SHIELD_HOLD || menu.SHIELD_STATE == SHIELD_INFINITE) {
        // If we are not mashing then we will always hold shield
        if(menu.MASH_STATE == NONE)
            return true;

        if(!is_in_shieldstun(module_accessor))
            return true;

        // We will only drop shield if we are in shieldstun and our attack can be performed OOS
        if(menu.MASH_STATE == MASH_ATTACK)
        {
            if(menu.ATTACK_STATE == MASH_NEUTRAL_B || menu.ATTACK_STATE == MASH_SIDE_B || menu.ATTACK_STATE == MASH_DOWN_B)
                return true;
        }
    }

    return false;
}


void perform_defensive_option(u64 module_accessor) {
    if (menu.DEFENSIVE_STATE == RANDOM_DEFENSIVE) {
        const int NUM_GROUND_STATUSES = 3;
        int random_statuses[NUM_GROUND_STATUSES] = {
            FIGHTER_STATUS_KIND_ESCAPE, 
            FIGHTER_STATUS_KIND_ATTACK,
            FIGHTER_STATUS_KIND_GUARD_ON
        };

        int random_status_index = app::sv_math::rand(hash40("fighter"), NUM_GROUND_STATUSES);
        StatusModule::change_status_request_from_script(module_accessor, random_statuses[random_status_index], 1);
    } else if (menu.DEFENSIVE_STATE == DEFENSIVE_SHIELD)
        StatusModule::change_status_request_from_script(module_accessor, FIGHTER_STATUS_KIND_GUARD_ON, 1);
    else if (menu.DEFENSIVE_STATE == DEFENSIVE_SPOTDODGE)
        StatusModule::change_status_request_from_script(module_accessor, FIGHTER_STATUS_KIND_ESCAPE, 1);
    else if (menu.DEFENSIVE_STATE == DEFENSIVE_JAB)
        StatusModule::change_status_request_from_script(module_accessor, FIGHTER_STATUS_KIND_ATTACK, 1);
}