space-crush/space-crush-common/src/dispatcher.rs

use specs::{Dispatcher as Inner, DispatcherBuilder, World, WorldExt};

use crate::{
    components::Player,
    resources::Global,
    systems::{FleetControl, FleetOwnedUpdate, Movement, OrbitOwnedUpdate, Process, Ships},
    Error,
};

pub struct Dispatcher<'a, 'b> {
    dispatcher: Inner<'a, 'b>,
}

impl<'a, 'b> Dispatcher<'a, 'b> {
    pub fn new(world: &mut World) -> Result<Self, Error> {
        world.insert(Global::default());

        world.register::<Player>();

        let mut dispatcher = DispatcherBuilder::new()
            .with(Process::default(), "process", &[])
            .with(Movement::default(), "movement", &[])
            .with(Ships::new(world), "ships", &[])
            .with(FleetControl::new(world)?, "fleet_control", &[])
            .with(FleetOwnedUpdate::new(world), "fleet_owned_update", &[])
            .with(OrbitOwnedUpdate::new(world), "orbit_owned_update", &[])
            .build();
        dispatcher.setup(world);

        Ok(Self { dispatcher })
    }

    pub fn process(&mut self, world: &World) {
        self.dispatcher.dispatch(world);
    }
}