Extract entity main action

1 files changed, 13 insertions, 0 deletions

diff --git a/order.go b/order.go
index f1e89bc..f8ef06d 100644
--- a/order.go
+++ b/order.go
@@ -1,5 +1,7 @@
 package main
 
+import "age_of_empires/physics"
+
 type Order interface {
 	Update(e *Entity, g *Game)
 }
@@ -10,3 +12,14 @@ func (e *Entity) UpdateOrder(g *Game) {
 	}
 	e.Order.Value.Update(e, g)
 }
+
+func (e *Entity) MainAction(g *Game, destination physics.Point, entityAtDestination *Entity, moveMap physics.MoveMap) {
+	if !e.Selection.IsEnabled || !e.Selection.Value.IsSelected {
+		return
+	}
+	if entityAtDestination != nil && entityAtDestination.ResourceSource.IsEnabled {
+		Gather(e, entityAtDestination, g)
+		return
+	}
+	physics.StartMove(&e.Move, e.Position, destination, moveMap)
+}