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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

use bevy::prelude::*;
use de_gui::ToastEvent;
use de_lobby_client::{GetGameRequest, JoinGameRequest};
use de_lobby_model::GamePlayerInfo;
use de_multiplayer::{
    ConnectionType, GameJoinedEvent, NetGameConf, ShutdownMultiplayerEvent, StartMultiplayerEvent,
};

use super::{
    current::GameNameRes,
    joined::LocalPlayerRes,
    requests::{Receiver, Sender},
    MultiplayerState,
};

pub(crate) struct JoiningGamePlugin;

impl Plugin for JoiningGamePlugin {
    fn build(&self, app: &mut App) {
        app.add_systems(OnEnter(MultiplayerState::GameJoining), get_game)
            .add_systems(OnExit(MultiplayerState::GameJoining), cleanup)
            .add_systems(
                Update,
                (
                    handle_get_response,
                    handle_joined_event.run_if(on_event::<GameJoinedEvent>()),
                    handle_join_response,
                )
                    .run_if(in_state(MultiplayerState::GameJoining)),
            );
    }
}

fn cleanup(
    mut commands: Commands,
    state: Res<State<MultiplayerState>>,
    mut shutdown: EventWriter<ShutdownMultiplayerEvent>,
) {
    if state.as_ref() != &MultiplayerState::GameJoined {
        commands.remove_resource::<LocalPlayerRes>();
        shutdown.send(ShutdownMultiplayerEvent);
    }
}

fn get_game(game_name: Res<GameNameRes>, mut sender: Sender<GetGameRequest>) {
    sender.send(GetGameRequest::new(game_name.name_owned()));
}

fn handle_get_response(
    mut next_state: ResMut<NextState<MultiplayerState>>,
    mut receiver: Receiver<GetGameRequest>,
    mut multiplayer: EventWriter<StartMultiplayerEvent>,
    mut toasts: EventWriter<ToastEvent>,
) {
    while let Some(result) = receiver.receive() {
        match result {
            Ok(game) => {
                let server = game.setup().server();
                multiplayer.send(StartMultiplayerEvent::new(NetGameConf::new(
                    server.ip(),
                    ConnectionType::JoinGame(server.port()),
                )));
            }
            Err(error) => {
                toasts.send(ToastEvent::new(error));
                next_state.set(MultiplayerState::SignIn);
            }
        }
    }
}

fn handle_joined_event(
    mut commands: Commands,
    game_name: Res<GameNameRes>,
    mut events: EventReader<GameJoinedEvent>,
    mut sender: Sender<JoinGameRequest>,
) {
    let Some(event) = events.read().last() else {
        return;
    };

    commands.insert_resource(LocalPlayerRes::new(event.player()));
    sender.send(JoinGameRequest::new(
        game_name.name_owned(),
        GamePlayerInfo::new(event.player().to_num()),
    ));
}

fn handle_join_response(
    mut next_state: ResMut<NextState<MultiplayerState>>,
    mut receiver: Receiver<JoinGameRequest>,
    mut toasts: EventWriter<ToastEvent>,
) {
    while let Some(result) = receiver.receive() {
        match result {
            Ok(_) => {
                next_state.set(MultiplayerState::GameJoined);
            }
            Err(error) => {
                toasts.send(ToastEvent::new(error));
                next_state.set(MultiplayerState::SignIn);
            }
        }
    }
}