1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

use bevy::prelude::*;
use de_core::nested_state;
use de_lobby_client::{
    CreateGameRequest, GetGameRequest, JoinGameRequest, SignInRequest, SignUpRequest,
};
use de_multiplayer::MultiplayerShuttingDownEvent;

use self::{
    create::CreateGamePlugin, current::CurrentGamePlugin, gamelisting::GameListingPlugin,
    joined::JoinedGamePlugin, joining::JoiningGamePlugin, requests::RequestsPlugin,
    setup::SetupGamePlugin, signin::SignInPlugin,
};
use crate::{menu::ScreenStatePlugin, MenuState};

mod create;
mod current;
mod gamelisting;
mod joined;
mod joining;
mod requests;
mod setup;
mod signin;

pub(super) struct MultiplayerPlugin;

impl Plugin for MultiplayerPlugin {
    fn build(&self, app: &mut App) {
        app.add_plugins((
            RequestsPlugin::<SignInRequest>::new(),
            RequestsPlugin::<SignUpRequest>::new(),
            RequestsPlugin::<GetGameRequest>::new(),
            RequestsPlugin::<CreateGameRequest>::new(),
            RequestsPlugin::<JoinGameRequest>::new(),
            MultiplayerStatePlugin,
            ScreenStatePlugin::<MultiplayerState>::default(),
            CurrentGamePlugin,
            SignInPlugin,
            GameListingPlugin,
            CreateGamePlugin,
            SetupGamePlugin,
            JoiningGamePlugin,
            JoinedGamePlugin,
        ))
        .add_systems(
            PostUpdate,
            go_to_sign_in
                .run_if(in_state(MenuState::Multiplayer))
                .run_if(on_event::<MultiplayerShuttingDownEvent>()),
        );
    }
}

nested_state!(
    MenuState::Multiplayer -> MultiplayerState,
    doc = "Each state corresponds to an individual multiplayer related menu screen.",
    enter = go_to_sign_in,
    variants = {
        SignIn,
        GameListing,
        GameCreation,
        GameSetup,
        GameJoining,
        GameJoined,
    }
);

fn go_to_sign_in(mut next_state: ResMut<NextState<MultiplayerState>>) {
    next_state.set(MultiplayerState::SignIn);
}