Use startup system for schedule

DigitalExtinction · Jul 24, 2023 · 0cf46d9 · 0cf46d9
1 parent 1401bb9
commit 0cf46d9
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/crates/core/src/schedule.rs b/crates/core/src/schedule.rs
@@ -6,15 +6,17 @@ pub struct GameSchedulesPlugin;
 
 impl Plugin for GameSchedulesPlugin {
     fn build(&self, app: &mut App) {
-        let mut main = app.world.get_resource_mut::<MainScheduleOrder>().unwrap();
-
-        main.insert_after(First, InputSchedule);
-        main.insert_after(InputSchedule, PreMovement);
-        main.insert_after(PreMovement, Movement);
-        main.insert_after(Movement, PostMovement);
+        app.add_systems(Startup, setup);
     }
 }
 
+fn setup(mut main: ResMut<MainScheduleOrder>) {
+    main.insert_after(First, InputSchedule);
+    main.insert_after(InputSchedule, PreMovement);
+    main.insert_after(PreMovement, Movement);
+    main.insert_after(Movement, PostMovement);
+}
+
 /// All user input is handled during this schedule.
 #[derive(ScheduleLabel, Clone, Debug, PartialEq, Eq, Hash)]
 pub struct InputSchedule;