1 files changed, 24 insertions, 0 deletions
diff --git a/Minecraft.World/DefendVillageTargetGoal.cpp b/Minecraft.World/DefendVillageTargetGoal.cpp
new file mode 100644
index 00000000..ad6b3fd8
--- /dev/null
+++ b/Minecraft.World/DefendVillageTargetGoal.cpp
@@ -0,0 +1,24 @@
+#include "stdafx.h"
+#include "net.minecraft.world.entity.ai.village.h"
+#include "net.minecraft.world.entity.animal.h"
+#include "DefendVillageTargetGoal.h"
+
+DefendVillageTargetGoal::DefendVillageTargetGoal(VillagerGolem *golem) : TargetGoal(golem, 16, false, true)
+{
+	this->golem = golem;
+	setRequiredControlFlags(TargetGoal::TargetFlag);
+}
+
+bool DefendVillageTargetGoal::canUse()
+{
+	shared_ptr<Village> village = golem->getVillage();
+	if (village == NULL) return false;
+	potentialTarget = weak_ptr<Mob>(village->getClosestAggressor(dynamic_pointer_cast<Mob>(golem->shared_from_this())));
+	return canAttack(potentialTarget.lock(), false);
+}
+
+void DefendVillageTargetGoal::start()
+{
+	golem->setTarget(potentialTarget.lock());
+	TargetGoal::start();
+}
+\ No newline at end of file