use libpso::packet::ship::*; use crate::common::serverstate::ClientId; use crate::common::leveltable::CharacterLevelTable; use crate::ship::ship::{SendShipPacket, ShipError, Clients, Rooms}; use crate::ship::character::{FullCharacterBytesBuilder}; use crate::ship::location::{ClientLocation, LobbyId, RoomLobby, ClientLocationError}; //use crate::ship::items::; use crate::ship::packet; use crate::ship::items::ItemManager; use crate::entity::gateway::EntityGateway; // this function needs a better home pub fn block_selected(id: ClientId, pkt: &MenuSelect, clients: &mut Clients, item_manager: &ItemManager, level_table: &CharacterLevelTable) -> Result, anyhow::Error> { let client = clients.get_mut(&id).ok_or(ShipError::ClientNotFound(id))?; client.block = pkt.item as usize - 1; let (level, stats) = level_table.get_stats_from_exp(client.character.char_class, client.character.exp); let inventory = item_manager.get_character_inventory(&client.character).unwrap(); let bank = item_manager.get_character_bank(&client.character).unwrap(); let fc = FullCharacterBytesBuilder::new() .character(&client.character) .stats(&stats) .level(level) .inventory(inventory) .bank(bank) .key_config(&client.settings.settings.key_config) .joystick_config(&client.settings.settings.joystick_config) .symbol_chat(&client.settings.settings.symbol_chats) .tech_menu(&client.character.tech_menu.as_bytes()) .option_flags(client.character.option_flags) .build(); Ok(vec![ (id, SendShipPacket::FullCharacter(Box::new(FullCharacter { character: fc, }))), (id, SendShipPacket::CharDataRequest(CharDataRequest {})), (id, SendShipPacket::LobbyList(LobbyList::new())), ]) } pub fn send_player_to_lobby(id: ClientId, _pkt: &CharData, client_location: &mut ClientLocation, clients: &Clients, item_manager: &ItemManager, level_table: &CharacterLevelTable) -> Result, anyhow::Error> { let lobby = client_location.add_client_to_next_available_lobby(id, LobbyId(0)).map_err(|_| ShipError::TooManyClients)?; let join_lobby = packet::builder::lobby::join_lobby(id, lobby, client_location, clients, item_manager, level_table)?; let addto = packet::builder::lobby::add_to_lobby(id, lobby, client_location, clients, item_manager, level_table)?; let neighbors = client_location.get_client_neighbors(id).unwrap(); Ok(vec![(id, SendShipPacket::JoinLobby(join_lobby))] .into_iter() .chain(neighbors.into_iter() .map(|c| (c.client, SendShipPacket::AddToLobby(addto.clone())))).collect()) } #[allow(clippy::too_many_arguments)] pub async fn change_lobby(id: ClientId, requested_lobby: u32, client_location: &mut ClientLocation, clients: &Clients, item_manager: &mut ItemManager, level_table: &CharacterLevelTable, ship_rooms: &mut Rooms, entity_gateway: &mut EG) -> Result, anyhow::Error> { let client = clients.get(&id).ok_or(ShipError::ClientNotFound(id))?; let prev_area = client_location.get_area(id).map_err(|err| -> ClientLocationError {err.into()})?; match prev_area { RoomLobby::Lobby(old_lobby) => { if old_lobby.0 == requested_lobby as usize { return Ok(vec![(id, SendShipPacket::SmallDialog(SmallDialog::new("You are already in this Lobby!".into())))]) } }, RoomLobby::Room(old_room) => { if client_location.get_client_neighbors(id)?.is_empty() { ship_rooms[old_room.0] = None; } item_manager.remove_character_from_room(&client.character); }, } let leave_lobby = packet::builder::lobby::remove_from_lobby(id, client_location)?; let old_neighbors = client_location.get_client_neighbors(id).unwrap(); let mut lobby = LobbyId(requested_lobby as usize); if client_location.add_client_to_lobby(id, lobby).is_err() { match prev_area { RoomLobby::Lobby(_lobby) => { let dialog = SmallDialog::new(String::from("Lobby is full.")); return Ok(vec![(id, SendShipPacket::SmallDialog(dialog))]) } RoomLobby::Room(_room) => { lobby = client_location.add_client_to_next_available_lobby(id, lobby).map_err(|_| ShipError::TooManyClients)?; } } } item_manager.load_character(entity_gateway, &client.character).await?; let join_lobby = packet::builder::lobby::join_lobby(id, lobby, client_location, clients, item_manager, level_table)?; let addto = packet::builder::lobby::add_to_lobby(id, lobby, client_location, clients, item_manager, level_table)?; let neighbors = client_location.get_client_neighbors(id).unwrap(); Ok(vec![(id, SendShipPacket::JoinLobby(join_lobby))] .into_iter() .chain(neighbors.into_iter() .map(|c| (c.client, SendShipPacket::AddToLobby(addto.clone())))) .chain(old_neighbors.into_iter() .map(|c| (c.client, SendShipPacket::LeaveLobby(leave_lobby.clone())))) .collect()) } pub fn remove_from_lobby(id: ClientId, client_location: &mut ClientLocation) -> Result, anyhow::Error> { let area_client = client_location.get_local_client(id)?; let neighbors = client_location.get_client_neighbors(id)?; let leader = client_location.get_leader_by_client(id)?; let leave_lobby_pkt = SendShipPacket::LeaveLobby(LeaveLobby::new(area_client.local_client.id(), leader.local_client.id())); client_location.remove_client_from_area(id)?; Ok(neighbors.into_iter().map(|n| { (n.client, leave_lobby_pkt.clone()) }).collect()) }