make sure Startup system runs before any bots join

2 files changed, 20 insertions, 1 deletions

diff --git a/azalea/src/lib.rs b/azalea/src/lib.rs
index bf00cbbe..8f86f97f 100644
--- a/azalea/src/lib.rs
+++ b/azalea/src/lib.rs
@@ -193,9 +193,18 @@ where
         // An event that causes the schedule to run. This is only used internally.
         let (run_schedule_sender, run_schedule_receiver) = mpsc::unbounded_channel();
 
+        let main_schedule_label = self.app.main_schedule_label;
+
         let ecs_lock =
             start_ecs_runner(self.app, run_schedule_receiver, run_schedule_sender.clone());
 
+        // run the main schedule so the startup systems run
+        {
+            let mut ecs = ecs_lock.lock();
+            ecs.run_schedule(main_schedule_label);
+            ecs.clear_trackers();
+        }
+
         let (bot, mut rx) = Client::start_client(
             ecs_lock,
             &account,
diff --git a/azalea/src/swarm/mod.rs b/azalea/src/swarm/mod.rs
index 417ca168..5d1a73df 100644
--- a/azalea/src/swarm/mod.rs
+++ b/azalea/src/swarm/mod.rs
@@ -320,6 +320,9 @@ where
         let (swarm_tx, mut swarm_rx) = mpsc::unbounded_channel();
 
         let (run_schedule_sender, run_schedule_receiver) = mpsc::unbounded_channel();
+
+        let main_schedule_label = self.app.main_schedule_label;
+
         let ecs_lock =
             start_ecs_runner(self.app, run_schedule_receiver, run_schedule_sender.clone());
 
@@ -337,7 +340,14 @@ where
 
             run_schedule_sender,
         };
-        ecs_lock.lock().insert_resource(swarm.clone());
+
+        // run the main schedule so the startup systems run
+        {
+            let mut ecs = ecs_lock.lock();
+            ecs.insert_resource(swarm.clone());
+            ecs.run_schedule(main_schedule_label);
+            ecs.clear_trackers();
+        }
 
         // SwarmBuilder (self) isn't Send so we have to take all the things we need out
         // of it