use libpso::packet::ship::*;
use crate::common::serverstate::ClientId;
use crate::ship::ship::{ShipError, Clients};
use crate::ship::location::{ClientLocation, LobbyId, ClientLocationError};
use crate::ship::packet::builder::{player_info};
use crate::ship::items::state::ItemState;


pub fn join_lobby(id: ClientId,
                  lobby: LobbyId,
                  client_location: &ClientLocation,
                  clients: &Clients,
                  item_state: &ItemState)
                  -> Result<JoinLobby, anyhow::Error> {
    let lobby_clients = client_location.get_clients_in_lobby(lobby).map_err(|err| -> ClientLocationError { err.into() })?;
    let playerinfo = lobby_clients.iter()
        .map(|area_client| {
            let client = clients.get(&area_client.client).ok_or(ShipError::ClientNotFound(area_client.client)).unwrap();
            player_info(0x100, client, area_client, item_state)
        });

    let client = clients.get(&id).ok_or(ShipError::ClientNotFound(id)).unwrap();
    let area_client = client_location.get_local_client(id).map_err(|err| -> ClientLocationError { err.into() })?;
    let leader = client_location.get_lobby_leader(lobby).map_err(|err| -> ClientLocationError { err.into() })?;
    Ok(JoinLobby {
        client: area_client.local_client.id(),
        leader: leader.local_client.id(),
        one: 1,
        lobby: lobby.id(),
        block: client.block as u16,
        event: 0,
        padding: 0,
        playerinfo: playerinfo.collect(),
    })
}

pub fn add_to_lobby(id: ClientId,
                    lobby: LobbyId,
                    client_location: &ClientLocation,
                    clients: &Clients,
                    item_state: &ItemState)
                    -> Result<AddToLobby, ShipError> {
    let client = clients.get(&id).ok_or(ShipError::ClientNotFound(id)).unwrap();
    let area_client = client_location.get_local_client(id).map_err(|err| -> ClientLocationError { err.into() })?;
    let leader = client_location.get_lobby_leader(lobby).map_err(|err| -> ClientLocationError { err.into() })?;
    Ok(AddToLobby {
        flag: 1,
        client: area_client.local_client.id(),
        leader: leader.local_client.id(),
        one: 1,
        lobby: lobby.id(),
        block: client.block as u16,
        event: 0,
        padding: 0,
        playerinfo: player_info(0x100, client, &area_client, item_state),
    })
}

pub fn remove_from_lobby(id: ClientId,
                        client_location: &ClientLocation)
                        -> Result<LeaveLobby, ShipError> {
    let prev_area_index = client_location.get_local_client(id).map_err(|err| -> ClientLocationError { err.into() })?.local_client.id();
    let prev_area_leader_index = client_location.get_area_leader(client_location.get_area(id)
        .map_err(|err| -> ClientLocationError { err.into() })?)
        .map_err(|err| -> ClientLocationError { err.into() })?.local_client.id();
    Ok(LeaveLobby {
        client: prev_area_index,
        leader: prev_area_leader_index,
        _padding: 0,
    })
}