Saving of unit commands

core/src/mindustry/ai/types/CommandAI.java

@@ -78,6 +78,11 @@ public class CommandAI extends AIController{
         }
     }
 
+    @Override
+    public boolean keepState(){
+        return true;
+    }
+
     @Override
     public Teamc findTarget(float x, float y, float range, boolean air, boolean ground){
         return attackTarget == null ? super.findTarget(x, y, range, air, ground) : attackTarget;

core/src/mindustry/entities/comp/UnitComp.java

@@ -334,8 +334,10 @@ abstract class UnitComp implements Healthc, Physicsc, Hitboxc, Statusc, Teamc, I
     public void afterRead(){
         afterSync();
         //reset controller state
+        if(!(controller instanceof AIController ai && ai.keepState())){
             controller(type.createController(self()));
         }
+    }
 
     @Override
     public void add(){

core/src/mindustry/entities/units/AIController.java

@@ -45,6 +45,14 @@ public class AIController implements UnitController{
         updateMovement();
     }
 
+    /**
+     * @return whether controller state should not be reset after reading.
+     * Do not override unless you know exactly what you are doing.
+     * */
+    public boolean keepState(){
+        return false;
+    }
+
     public void stopShooting(){
         for(var mount : unit.mounts){
             //ignore mount controllable stats too, they should not shoot either