/*
 * This file is part of laurelin/client
 * Copyright (C) 2023 Jonni Liljamo <jonni@liljamo.com>
 *
 * Licensed under GPL-3.0-only.
 * See LICENSE for licensing information.
 */

use bevy::prelude::*;
use laurelin_shared::server::{
    channels::{AfterAuthChannel, CookieRefreshChannel, DataRequestChannel},
    messages::{AfterAuth, CookieRefresh, DataRequestResponse, DataRequestType},
};
use naia_bevy_client::{
    events::{ClientTickEvent, ConnectEvent, DisconnectEvent, MessageEvents, RejectEvent},
    Client,
};

use crate::{
    cfg::CfgUser,
    plugins::menu::{
        ui::{
            connect::{ConnectScreenData, ConnectState},
            play::{PlayScreenData, PlayScreenState},
        },
        MenuState,
    },
    Global,
};

use super::send::user::PubUserDetailsEvent;

pub fn connect_events(mut ev: EventReader<ConnectEvent>, client: Client) {
    for _ in ev.iter() {
        if let Ok(server_address) = client.server_address() {
            info!("Laurelin client connected to: '{}'", server_address);
        }
    }
}

pub fn reject_events(mut ev: EventReader<RejectEvent>, mut data: ResMut<ConnectScreenData>) {
    for _ in ev.iter() {
        warn!("Laurelin client rejected from connecting to server!");

        data.error = "rejected from connecting".to_string();
        data.state = if data.state == ConnectState::LoggingIn {
            ConnectState::Login
        } else {
            ConnectState::Register
        }
    }
}

pub fn disconnect_events(mut ev: EventReader<DisconnectEvent>) {
    for _ in ev.iter() {
        info!("Laurelin client disconnected from server.");
    }
}

pub fn message_events(
    mut commands: Commands,
    mut ev: EventReader<MessageEvents>,
    mut pud_ev_w: EventWriter<PubUserDetailsEvent>,
    mut global: ResMut<Global>,
    mut cfg_user: ResMut<CfgUser>,
    mut connect_data: ResMut<ConnectScreenData>,
    mut play_data: ResMut<PlayScreenData>,
) {
    for events in ev.iter() {
        for aa_message in events.read::<AfterAuthChannel, AfterAuth>() {
            // TODO: move handling to a separate function?

            // save details from AfterAuth
            cfg_user.id = aa_message.id;
            cfg_user.username = aa_message.username;
            cfg_user.cookie = aa_message.cookie;

            // reset the connection screen to login
            connect_data.state = ConnectState::Login;

            // take us to the main menu
            commands.insert_resource(NextState(Some(MenuState::Menu)));
        }

        for cr_message in events.read::<CookieRefreshChannel, CookieRefresh>() {
            cfg_user.cookie = cr_message.cookie;
        }

        for response in events.read::<DataRequestChannel, DataRequestResponse>() {
            match DataRequestType::from_u8(&response.r#type) {
                DataRequestType::GameCreate => {
                    // TODO: handle possible error (unwrap,
                    //       and if the response data is an error)
                    play_data.cur_game = response.games.unwrap().get(0).cloned();
                    play_data.state = PlayScreenState::InLobbyHost;
                }
                DataRequestType::GameAllForming => {
                    // TODO: handle possible error
                    let all_forming = response.games.unwrap();

                    for game in &all_forming {
                        pud_ev_w.send(PubUserDetailsEvent {
                            id: game.host_id.clone(),
                        });
                    }

                    play_data.all_forming = all_forming;
                    play_data.waiting_for_all_forming = false;
                }
                DataRequestType::GameMyGames => {
                    // TODO: handle possible error
                    let my_games = response.games.unwrap();

                    for game in &my_games {
                        pud_ev_w.send(PubUserDetailsEvent {
                            id: game.host_id.clone(),
                        });
                        if let Some(guest_id) = &game.guest_id {
                            pud_ev_w.send(PubUserDetailsEvent {
                                id: guest_id.clone(),
                            });
                        }
                    }

                    play_data.my_games = my_games;
                    play_data.waiting_for_my_games = false;
                }
                DataRequestType::PubUserDetails => {
                    // TODO: handle possible error
                    let user = response.users.unwrap().get(0).unwrap().clone();
                    global.users_cache_queue.retain(|id| id != &user.id);
                    global.users_cache.push(user);
                }
            }
        }
    }
}

pub fn tick_events(mut ev: EventReader<ClientTickEvent>, client: Client) {
    if !client.is_connected() {
        return;
    }

    for ClientTickEvent(_tick) in ev.iter() {}
}