Prevent actions on certain units from being removed

source/match/hud/ProductionQueue.gd

@@ -15,6 +15,8 @@ func _ready():
 
 
 func _reset():
+	if not is_inside_tree():
+		return
 	_remove_queue_elements()
 	_clear_observed_production_manager()
 	if _try_observing_production_manager():

source/match/players/simple-clairvoyant-ai/EconomyController.gd

@@ -39,7 +39,7 @@ func provision(resources, metadata):
 		_number_of_pending_worker_resource_requests -= 1
 		if _ccs.is_empty():
 			return
-		_ccs[0].action.produce(WorkerScene)  # TODO: investigate how come CC has no action
+		_ccs[0].action.produce(WorkerScene)
 		_number_of_pending_workers += 1
 	elif metadata == "cc":
 		assert(

source/match/units/AircraftFactory.gd

@@ -8,3 +8,10 @@ func _ready():
 	if not is_constructed():
 		await constructed
 	action = ManagingProduction.new()
+
+
+func _set_action(action_node):
+	if not _action_locked and action == null:
+		super(action_node)
+	elif action_node != null:
+		action_node.queue_free()

source/match/units/AntiAirTurret.gd

@@ -8,3 +8,10 @@ func _ready():
 	if not is_constructed():
 		await constructed
 	action = WaitingForTargets.new()
+
+
+func _set_action(action_node):
+	if not _action_locked and action == null:
+		super(action_node)
+	elif action_node != null:
+		action_node.queue_free()

source/match/units/AntiGroundTurret.gd

@@ -8,3 +8,10 @@ func _ready():
 	if not is_constructed():
 		await constructed
 	action = WaitingForTargets.new()
+
+
+func _set_action(action_node):
+	if not _action_locked and action == null:
+		super(action_node)
+	elif action_node != null:
+		action_node.queue_free()

source/match/units/CommandCenter.gd

@@ -8,3 +8,10 @@ func _ready():
 	if not is_constructed():
 		await constructed
 	action = ManagingProduction.new()
+
+
+func _set_action(action_node):
+	if not _action_locked and action == null:
+		super(action_node)
+	elif action_node != null:
+		action_node.queue_free()

source/match/units/VehicleFactory.gd

@@ -8,3 +8,10 @@ func _ready():
 	if not is_constructed():
 		await constructed
 	action = ManagingProduction.new()
+
+
+func _set_action(action_node):
+	if not _action_locked and action == null:
+		super(action_node)
+	elif action_node != null:
+		action_node.queue_free()