//Source file for the Weapon Fire hook module.
//This hook controls how weapons are fired.
#include "weapon_fire.h"
#include <SCBW/scbwdata.h>
#include <SCBW/enumerations.h>
#include <SCBW/api.h>


//-------- Helper function declarations. Do NOT modify! ---------//

namespace {
	typedef void(__stdcall *GetWeaponHitPosFunc)(const CUnit *unit, s32 *x, s32 *y);
	GetWeaponHitPosFunc const getWeaponHitPos = (GetWeaponHitPosFunc)0x004762C0;
	void createBullet(u8 weaponId, CUnit *source, s16 x, s16 y, u8 attackingPlayer, u8 direction);
} //unnamed namespace


//-------- Actual hook functions --------//

namespace hooks {

	//Fires a weapon from the @p unit.
	//This hook affects the following iscript opcodes: attackwith, attack, castspell
	//This also affects CUnit::fireWeapon().
	void fireWeaponHook(CUnit *unit, u8 weaponId) {
		//Default StarCraft behavior

		//Retrieve the spawning position for the bullet.
		s32 x, y;

		if (weapons_dat::Behavior[weaponId] == WeaponBehavior::AppearOnTargetUnit) {
			if (!unit->orderTarget.unit)
				return;

			getWeaponHitPos(unit, &x, &y);
		}

		else if (weapons_dat::Behavior[weaponId] == WeaponBehavior::AppearOnTargetSite) {
			x = unit->orderTarget.pt.x;
			y = unit->orderTarget.pt.y;
		}

		else {
			s32 forwardOffset = weapons_dat::ForwardOffset[weaponId];

			x = unit->getX() + scbw::getPolarX(forwardOffset, unit->currentDirection1);
			y = unit->getY() + scbw::getPolarY(forwardOffset, unit->currentDirection1)
				- weapons_dat::VerticalOffset[weaponId];
		}

		if (weapons_dat::FlingyId[weaponId] != 0)
		{
			createBullet(weaponId, unit, x, y, unit->playerId, unit->currentDirection1);
		}

		//Blink
		if (unit->id == UnitId::dragoon)
		{
			//if (unit->unusedTimer = 0)
			//{
			//if (weapons_dat::Weapon[weaponId] == 117);
			{
				CUnit *target = unit->orderTarget.unit;
				unit->unusedTimer = 40;
				u32 TeleportX = unit->orderTarget.pt.x - 20;
				u32 TeleportY = unit->orderTarget.pt.y;
				unit->position.x = TeleportX;
				unit->position.y = TeleportY;
				
			}
			
		}
		
	}
}





//-------- Helper function definitions. Do NOT modify! --------//

namespace {

	const u32 Helper_CreateBullet = 0x0048C260;
	void createBullet(u8 weaponId, CUnit *source, s16 x, s16 y, u8 attackingPlayer, u8 direction) {
		u32 attackingPlayer_ = attackingPlayer, direction_ = direction;
		s32 x_ = x, y_ = y;

		__asm {
			PUSHAD
				PUSH direction_
				PUSH attackingPlayer_
				PUSH y_
				PUSH x_
				MOV EAX, source
				MOVZX ECX, weaponId
				CALL Helper_CreateBullet
				POPAD
		}
	}

} //unnamed namespace