YimMenu/src/native_hooks/all_scripts.hpp

#pragma once
#include "native_hooks.hpp"
#include "natives.hpp"
#include "core/scr_globals.hpp"
#include "fiber_pool.hpp"
#include "util/scripts.hpp"
#include "hooking.hpp"
#include "gta/enums.hpp"

namespace big
{
    namespace all_scripts
    {
        void IS_DLC_PRESENT(rage::scrNativeCallContext* src)
        {
            const auto hash = src->get_arg<rage::joaat_t>(0);

            bool return_value = DLC::IS_DLC_PRESENT(hash);

            if (hash == 0x96F02EE6)
                return_value = return_value || g.settings.dev_dlc;

            src->set_return_value<BOOL>((BOOL)return_value);
        }

        void NETWORK_SET_THIS_SCRIPT_IS_NETWORK_SCRIPT(rage::scrNativeCallContext* src)
        {
            if (rage::scrThread::get() && rage::scrThread::get()->m_handler)
            {
                if (auto hook = g_hooking->m_handler_hooks[(CGameScriptHandler*)rage::scrThread::get()->m_handler].get())
                {
                    hook->disable();
                    g_hooking->m_handler_hooks.erase((CGameScriptHandler*)rage::scrThread::get()->m_handler);
                }
            }

            NETWORK::NETWORK_SET_THIS_SCRIPT_IS_NETWORK_SCRIPT(src->get_arg<int>(0), src->get_arg<BOOL>(1), src->get_arg<int>(2));
        }

        void NETWORK_TRY_TO_SET_THIS_SCRIPT_IS_NETWORK_SCRIPT(rage::scrNativeCallContext* src)
        {
            if (rage::scrThread::get() && rage::scrThread::get()->m_handler)
            {
                if (auto hook = g_hooking->m_handler_hooks[(CGameScriptHandler*)rage::scrThread::get()->m_handler].get())
                {
                    hook->disable();
                    g_hooking->m_handler_hooks.erase((CGameScriptHandler*)rage::scrThread::get()->m_handler);
                }
            }

            src->set_return_value<BOOL>(NETWORK::NETWORK_TRY_TO_SET_THIS_SCRIPT_IS_NETWORK_SCRIPT(src->get_arg<int>(0), src->get_arg<BOOL>(1), src->get_arg<int>(2)));
        }

        void SET_CURRENT_PED_WEAPON(rage::scrNativeCallContext* src)
        {
            const auto ped = src->get_arg<Ped>(0);
            const auto hash = src->get_arg<rage::joaat_t>(1);

            if (g.weapons.interior_weapon && ped == self::ped && hash == RAGE_JOAAT("WEAPON_UNARMED"))
                return;   

            WEAPON::SET_CURRENT_PED_WEAPON(ped, hash, src->get_arg<int>(2));
        }

        void DISABLE_CONTROL_ACTION(rage::scrNativeCallContext* src)
        {
            const auto action = src->get_arg<ControllerInputs>(1);

            if (g.weapons.interior_weapon)
            {
                switch (action)
                {
                case ControllerInputs::INPUT_SELECT_WEAPON:
                case ControllerInputs::INPUT_VEH_SELECT_NEXT_WEAPON:
                case ControllerInputs::INPUT_VEH_SELECT_PREV_WEAPON:
                case ControllerInputs::INPUT_DETONATE:
                case ControllerInputs::INPUT_PICKUP:
                // case ControllerInputs::INPUT_JUMP: TODO: add as separate feature
                case ControllerInputs::INPUT_TALK:
                case ControllerInputs::INPUT_AIM:
                case ControllerInputs::INPUT_MELEE_ATTACK_LIGHT:
                case ControllerInputs::INPUT_MELEE_ATTACK_HEAVY:
                case ControllerInputs::INPUT_MELEE_ATTACK_ALTERNATE:
                case ControllerInputs::INPUT_MELEE_BLOCK:
                case ControllerInputs::INPUT_VEH_ATTACK:
                case ControllerInputs::INPUT_VEH_ATTACK2:
                case ControllerInputs::INPUT_VEH_AIM:
                case ControllerInputs::INPUT_VEH_PASSENGER_ATTACK:
                case ControllerInputs::INPUT_VEH_FLY_SELECT_NEXT_WEAPON:
                case ControllerInputs::INPUT_ATTACK:
                case ControllerInputs::INPUT_NEXT_WEAPON:
                case ControllerInputs::INPUT_PREV_WEAPON:
                case ControllerInputs::INPUT_SELECT_NEXT_WEAPON:
                case ControllerInputs::INPUT_SELECT_PREV_WEAPON:
                    return;
                }
            }

            PAD::DISABLE_CONTROL_ACTION(src->get_arg<int>(0), (int)action, src->get_arg<int>(2));
        }

        void HUD_FORCE_WEAPON_WHEEL(rage::scrNativeCallContext* src)
        {
            if (g.weapons.interior_weapon && src->get_arg<BOOL>(0) == false)
                return;

            HUD::HUD_FORCE_WEAPON_WHEEL(src->get_arg<BOOL>(0));
        }
    }
}