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


pub fn join_lobby(id: ClientId,
                  lobby: LobbyId,
                  client_location: &ClientLocation,
                  clients: &Clients,
                  item_manager: &ItemManager,
                  level_table: &CharacterLevelTable)
                  -> 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_manager, level_table)
        });
    
    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_manager: &ItemManager,
                    level_table: &CharacterLevelTable)
                    -> 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_manager, level_table),
    })
}

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,
    })
}