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

use bevy::prelude::*;
use de_conf::Configuration;
use de_core::state::AppState;
use de_lobby_model::Token;
use iyes_progress::prelude::*;

use crate::{
    client::LobbyClient, Authentication, LobbyRequest, ResponseEvent, SignInRequest, SignUpRequest,
};

pub(crate) struct LobbyPlugin;

impl Plugin for LobbyPlugin {
    fn build(&self, app: &mut App) {
        app.init_resource::<Authentication>().add_systems(
            Update,
            (
                setup_client
                    .track_progress()
                    .run_if(in_state(AppState::AppLoading)),
                set_token::<SignInRequest>,
                set_token::<SignUpRequest>,
            ),
        );
    }
}

fn setup_client(
    mut commands: Commands,
    conf: Option<Res<Configuration>>,
    client: Option<Res<LobbyClient>>,
) -> Progress {
    if client.is_some() {
        return true.into();
    }
    let Some(conf) = conf else {
        return false.into();
    };

    let client = LobbyClient::build(conf.multiplayer().lobby().clone());
    commands.insert_resource(client);
    false.into()
}

fn set_token<T>(mut events: EventReader<ResponseEvent<T>>, mut auth: ResMut<Authentication>)
where
    T: LobbyRequest<Response = Token>,
{
    let Some(event) = events.read().last() else {
        return;
    };
    let Ok(token) = event.result() else { return };
    auth.set_token(token.token().to_owned());
}