UltimateTrainingModpack/src/lib.rs

#![feature(proc_macro_hygiene)]
#![feature(with_options)]
#![feature(const_mut_refs)]

mod common;
mod hitbox_visualizer;
mod training;

use crate::common::consts::*;
use crate::common::*;

use skyline::c_str;
use skyline::libc::{c_void, mkdir, fclose, fopen, fwrite};
use skyline::nro::{self, NroInfo};
use smash::app::lua_bind::*;
use smash::app::sv_system;
use smash::lib::lua_const::*;
use smash::lib::L2CValue;
use smash::lua2cpp::L2CFighterCommon;

#[skyline::hook(replace = smash::lua2cpp::L2CFighterCommon_sub_guard_cont)]
pub unsafe fn handle_sub_guard_cont(fighter: &mut L2CFighterCommon) -> L2CValue {
    let module_accessor = sv_system::battle_object_module_accessor(fighter.lua_state_agent);
    if is_training_mode() && is_operation_cpu(module_accessor) {
        if MENU.mash_state == Mash::Attack && MENU.mash_attack_state == Attack::Grab {
            if StatusModule::prev_status_kind(module_accessor, 0) == FIGHTER_STATUS_KIND_GUARD_DAMAGE {
                if WorkModule::get_int(
                    module_accessor,
                    *FIGHTER_INSTANCE_WORK_ID_INT_INVALID_CATCH_FRAME,
                ) == 0
                {
                    if WorkModule::is_enable_transition_term(
                        module_accessor,
                        *FIGHTER_STATUS_TRANSITION_TERM_ID_CONT_CATCH,
                    ) {
                        fighter.fighter_base.change_status(
                            L2CValue::new_int(*FIGHTER_STATUS_KIND_CATCH as u64),
                            L2CValue::new_bool(true),
                        );
                    }
                }
            }
        }
        if MENU.mash_state == Mash::Spotdodge {
            if StatusModule::prev_status_kind(module_accessor, 0) == FIGHTER_STATUS_KIND_GUARD_DAMAGE {
                if WorkModule::is_enable_transition_term(
                    module_accessor,
                    *FIGHTER_STATUS_TRANSITION_TERM_ID_CONT_ESCAPE,
                ) {
                    fighter.fighter_base.change_status(
                        L2CValue::new_int(*FIGHTER_STATUS_KIND_ESCAPE as u64),
                        L2CValue::new_bool(true),
                    );
                }
            }
        }

        if MENU.mash_state == Mash::Attack {
            if MENU.mash_attack_state == Attack::UpB {
                if StatusModule::prev_status_kind(module_accessor, 0) == FIGHTER_STATUS_KIND_GUARD_DAMAGE {
                    // if WorkModule::is_enable_transition_term(
                    //     module_accessor,
                    //     *FIGHTER_STATUS_TRANSITION_TERM_ID_CONT_JUMP_SQUAT_BUTTON,
                    // ) {
                        fighter.fighter_base.change_status(
                            L2CValue::new_int(*FIGHTER_STATUS_KIND_SPECIAL_HI as u64),
                            L2CValue::new_bool(false),
                        );
                    // }
                }
            }
            if MENU.mash_attack_state == Attack::UpSmash {
                if StatusModule::prev_status_kind(module_accessor, 0) == FIGHTER_STATUS_KIND_GUARD_DAMAGE {
                    // if WorkModule::is_enable_transition_term(
                    //     module_accessor,
                    //     *FIGHTER_STATUS_TRANSITION_TERM_ID_CONT_JUMP_SQUAT_BUTTON,
                    // ) {
                        fighter.fighter_base.change_status(
                            L2CValue::new_int(*FIGHTER_STATUS_KIND_ATTACK_HI4_START as u64),
                            L2CValue::new_bool(false),
                        );
                    // }
                }
            }
        }
    }

    original!()(fighter)
}

fn nro_main(nro: &NroInfo<'_>) {
    match nro.name {
        "common" => {
            println!("Loaded common NRO!");
            skyline::install_hook!(handle_sub_guard_cont);
        }
        _ => (),
    }
}

#[skyline::main(name = "training_modpack")]
pub fn main() {
    println!("Training modpack initialized.");
    hitbox_visualizer::hitbox_visualization();
    training::training_mods();
    nro::add_hook(nro_main).unwrap();

    unsafe {
        let buffer = format!("{:x}", MENU as *const _ as u64);
        println!("Writing training_modpack.log with {}...\n", buffer);
        mkdir("sd:/TrainingModpack/\u{0}".as_bytes().as_ptr(), 0777);
        let f = fopen(
            "sd:/TrainingModpack/training_modpack.log\u{0}".as_bytes().as_ptr(),
            "w\u{0}".as_bytes().as_ptr(),
        );

        if !f.is_null() {
            fwrite(c_str(&buffer) as *const c_void, 1, buffer.len(), f);
            fclose(f);
        }
    }
}