#include "common.hpp"

namespace Ledge {
void enable_transition_term(u64 module_accessor, int transition_id) {
    if (ESCAPE_STATE == ESCAPE_LEDGE && is_training_mode() && is_operation_cpu(module_accessor)) {
        if (StatusModule::status_kind(module_accessor) == FIGHTER_STATUS_KIND_CLIFF_WAIT) {
            if (transition_id == FIGHTER_STATUS_TRANSITION_TERM_ID_CONT_CLIFF_CLIMB) {
                int status = 0;
                int ledge_case = LEDGE_STATE;

                if (LEDGE_STATE == RANDOM_LEDGE)
                    ledge_case = app::sv_math::rand(hash40("fighter"), 4) + 1;

                switch (ledge_case) {
                    case NEUTRAL_LEDGE:
                        status = FIGHTER_STATUS_KIND_CLIFF_CLIMB; break;
                    case ROLL_LEDGE:
                        status = FIGHTER_STATUS_KIND_CLIFF_ESCAPE; break;
                    case JUMP_LEDGE:
                        status = FIGHTER_STATUS_KIND_CLIFF_JUMP1; break;
                    case ATTACK_LEDGE:
                        status = FIGHTER_STATUS_KIND_CLIFF_ATTACK; break;
                }

                StatusModule::change_status_request_from_script(module_accessor, status, 1);
            }
        }
    }
}

void get_command_flag_cat(u64 module_accessor, int category, int& flag) {
    if (ESCAPE_STATE == ESCAPE_LEDGE) {
        int status = StatusModule::status_kind(module_accessor);
        if (status == FIGHTER_STATUS_KIND_CLIFF_JUMP3 ||
            status == FIGHTER_STATUS_KIND_CLIFF_JUMP2 ||
            status == FIGHTER_STATUS_KIND_CLIFF_JUMP1) {
            flag |= FIGHTER_PAD_CMD_CAT1_FLAG_AIR_ESCAPE;
        }

        int prev_status = StatusModule::prev_status_kind(module_accessor, 0);
        if (prev_status == FIGHTER_STATUS_KIND_CLIFF_CLIMB || 
            prev_status == FIGHTER_STATUS_KIND_CLIFF_ATTACK || 
            prev_status == FIGHTER_STATUS_KIND_CLIFF_ESCAPE) {
            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);
        }
    }
}
}