#include "hooks.hpp"
#include "context.hpp"
#include "base_cheat.hpp"

void __fastcall hooks::update_clientside_animation( void* ecx_, void* edx_ ) {
	static bool first_update = true;

	auto old_func = update_clientside_animation_o;

	if( ecx_ != g_ctx.m_local )
		return old_func( ecx_, 0 );

	auto ent = ( c_base_player* )( ecx_ );

	static float last_update;
	static float last_spawntime = 0.f;

	// Arbitrary number much.
	if( !g_settings.rage.anti_aim || !g_ctx.m_local->is_valid( ) || std::abs( last_update - g_csgo.m_globals->m_curtime ) > TICKS_TO_TIME( 20 ) || last_spawntime != ent->m_flSpawnTime( ) ) {
		last_update = g_csgo.m_globals->m_curtime;
		first_update = true;

		if( last_spawntime != ent->m_flSpawnTime( ) )
			last_spawntime = ent->m_flSpawnTime( );

		return old_func( ecx_, edx_ );
	}

	last_update = g_csgo.m_globals->m_curtime;

	ent->get_animstate( )->m_iLastClientSideAnimationUpdateFramecount = g_csgo.m_globals->m_framecount;
	old_func( ecx_, edx_ );

	if( !first_update ) {

		ent->set_abs_angles( vec3_t( 0, ent->get_animstate( )->m_flGoalFeetYaw, 0 ) );
		bool backup = ent->get_animstate( )->m_bOnGround;
		ent->get_animstate( )->m_bOnGround = false;

		ent->invalidate_bone_cache( );
		ent->ce( )->SetupBones( nullptr, 128, BONE_USED_BY_ANYTHING, 0.f );

		ent->get_animstate( )->m_bOnGround = backup;
	}

	first_update = false;
}