#include "hooks.h"
#include "base_cheat.h"
#include "ctx.hpp"
#include "util.hpp"
#include "settings.h"
//this is so ugly dont even look at it

//gets called once per frame for every entity
void __fastcall hooks::draw_model_execute( IVModelInfo* ecx_, void* edx_, void* render_ctx, const DrawModelState_t& state, const ModelRenderInfo_t& info, matrix3x4* bone_to_world ) {
	static auto dme_o = cl.m_model_render->get_old_function< decltype( &hooks::draw_model_execute ) >( 19 );

	if( cl.m_panic || g_ctx.m_drawing_postscreenspace || g_settings.misc.hide_from_obs )
		return dme_o( ecx_, edx_, render_ctx, state, info, bone_to_world );

	bool draw = true;
	IMaterial* mat = g_settings.visuals.chams.flat( ) ? g_cheat.chams.m_materials.m_chams_flat : g_cheat.chams.m_materials.m_chams;
	auto model_name = cl.m_modelinfo( )->GetModelName( info.m_model );
	auto local_index = cl.m_engine( )->GetLocalPlayer( );
	int local_team = 0;
	if( g_ctx.m_local )
		local_team = g_ctx.m_local->m_iTeamNum( );
	int team = 0;
	bool is_player = false;

	if( strstr( model_name, xors( "models/player" ) ) && info.m_entity_index == local_index ) {
		if( g_settings.misc.thirdperson( ) && g_ctx.m_local && g_ctx.m_local->is_valid( )
			&& g_ctx.m_local->m_bIsScoped( ) ) {
			float clr[ ] = { 1.0f, 1.0f, 1.0f };
			cl.m_render_view( )->SetColorModulation( clr );
			cl.m_render_view( )->SetBlend( 0.5f );
			return dme_o( ecx_, edx_, render_ctx, state, info, bone_to_world );
		}
	}

	if( strstr( model_name, xors( "models/player" ) ) && info.m_entity_index ) {
		auto player = cl.m_entlist( )->get_client_entity< c_base_player >( info.m_entity_index );

		if( player && player->is_valid( ) ) {
			is_player = true;
			team = player->m_iTeamNum( );

			if( g_settings.visuals.chams.enabled ) {
				if( g_settings.visuals.chams.ignore_z && !g_ctx.m_drawing_screneend && ( team != local_team || g_settings.visuals.chams.friendlies( ) ) )
					draw = false;
				else if( !g_settings.visuals.chams.ignore_z && ( team != local_team || g_settings.visuals.chams.friendlies( ) ) ) {
					fclr_t clr = team != local_team ? g_settings.visuals.chams.color_visible_enemy( ) : g_settings.visuals.chams.color_visible_friendly( );

					mat->SetMaterialVarFlag( MATERIAL_VAR_IGNOREZ, false );
					g_cheat.chams.m_materials.force_material( mat, clr );
					dme_o( ecx_, 0, render_ctx, state, info, bone_to_world );
					cl.m_model_render( )->ForcedMaterialOverride( nullptr );
					cl.m_render_view( )->SetBlend( 1.0f );
					draw = false;
				}
			}

		}
	}

	if( g_settings.misc.transparent_vm && strstr( model_name, xors( "weapon" ) ) ) {
		if( strstr( model_name, xors( "arms" ) ) ) {
			cl.m_render_view( )->SetBlend( 0.6f );
		}
	}

	if( is_player ) {

		if( g_settings.legit.backtracking_visualize( ) && ( team != local_team || g_settings.legit.friendlies( ) ) ) {
			auto ent = cl.m_entlist( )->get_client_entity< c_base_player >( info.m_entity_index );
			auto record = g_cheat.aim.m_lagcomp.find_best_record( info.m_entity_index );
			{ //idk
				auto records = g_cheat.aim.m_lagcomp.get_records( info.m_entity_index );
				if( records.size( ) ) {
					for( auto& it : util::reverse_iterator( records ) ) {
						if( it.is_valid( ) ) {
							record = &it;
							break;
						}
					}
				}
			}

			if( record && record->is_valid( ) && record->m_position.dist_to( ent->get_hitbox_position( 0 ) ) > 5.f ) {
				fclr_t clr_hid = team != local_team ? g_settings.visuals.chams.color_hidden_enemy( ).to_fclr( ) : g_settings.visuals.chams.color_hidden_friendly( ).to_fclr( );

				float backup_modulation[ 3 ]{ };
				bool  backup_ignorez = false;
				float backup_blend = 1.0f;

				backup_blend = cl.m_render_view( )->GetBlend( );
				cl.m_render_view( )->GetColorModulation( backup_modulation );

				//BIG ROFL
				if( backup_modulation[ 0 ] == clr_hid.r( ) && backup_modulation[ 1 ] == clr_hid.g( ) && backup_modulation[ 2 ] == clr_hid.b( )
					&& backup_blend == clr_hid.a( ) ) {
					backup_ignorez = true;
				}

				if( !backup_ignorez ) {
					auto flat_mat = g_cheat.chams.m_materials.m_chams_flat.m_mat;
					flat_mat->SetMaterialVarFlag( MATERIAL_VAR_IGNOREZ, g_settings.visuals.chams.ignore_z && g_settings.visuals.chams.enabled );
					g_cheat.chams.m_materials.force_material( g_cheat.chams.m_materials.m_chams_flat, g_settings.legit.backtracking_col( ) );
					dme_o( ecx_, edx_, render_ctx, state, info, record->m_matrix );
					flat_mat->SetMaterialVarFlag( MATERIAL_VAR_IGNOREZ, false );
				}

				cl.m_render_view( )->SetBlend( backup_blend );
				cl.m_render_view( )->SetColorModulation( backup_modulation );
				cl.m_model_render( )->ForcedMaterialOverride( g_settings.visuals.chams.enabled( ) &&
					( team != local_team || g_settings.visuals.chams.friendlies( ) ) ? mat : nullptr );

				if( draw ) {
					dme_o( ecx_, edx_, render_ctx, state, info, bone_to_world );
					draw = false;
				}
			}
		}
	}

	if( draw ) {
		dme_o( ecx_, 0, render_ctx, state, info, bone_to_world );
	}

	cl.m_model_render( )->ForcedMaterialOverride( nullptr );
}