use crate::ship::items::ClientItemId; use std::collections::{HashMap, BTreeMap, BinaryHeap}; use std::cmp::{Ordering, PartialOrd, PartialEq}; use thiserror::Error; use futures::future::join_all; use libpso::character::character;//::InventoryItem; use crate::entity::gateway::EntityGateway; use crate::entity::character::{CharacterEntity, CharacterEntityId}; use crate::entity::item::{ItemEntityId, ItemEntity, ItemDetail, ItemLocation, BankName}; use crate::entity::item::{Meseta, NewItemEntity}; use crate::entity::item::tool::{Tool, ToolType}; use crate::ship::map::MapArea; use crate::ship::ship::ItemDropLocation; use crate::ship::drops::{ItemDrop, ItemDropType}; use crate::ship::location::{AreaClient, RoomId}; #[derive(Debug, Clone)] pub struct IndividualBankItem { pub entity_id: ItemEntityId, pub item_id: ClientItemId, pub item: ItemDetail, } #[derive(Debug, Clone)] pub struct StackedBankItem { pub entity_ids: Vec, pub item_id: ClientItemId, pub tool: Tool, } impl StackedBankItem { pub fn count(&self) -> usize { self.entity_ids.len() } } #[derive(Debug, Clone)] pub enum BankItem { Individual(IndividualBankItem), Stacked(StackedBankItem), } impl std::cmp::PartialEq for BankItem { fn eq(&self, other: &BankItem) -> bool { let mut self_bytes = [0u8; 4]; let mut other_bytes = [0u8; 4]; self_bytes.copy_from_slice(&self.as_client_bytes()[0..4]); other_bytes.copy_from_slice(&other.as_client_bytes()[0..4]); let self_value = u32::from_be_bytes(self_bytes); let other_value = u32::from_be_bytes(other_bytes); self_value.eq(&other_value) } } impl std::cmp::Eq for BankItem {} impl std::cmp::PartialOrd for BankItem { fn partial_cmp(&self, other: &BankItem) -> Option { //let self_bytes = self.as_client_bytes(); //let other_bytes = other.as_client_bytes(); let mut self_bytes = [0u8; 4]; let mut other_bytes = [0u8; 4]; self_bytes.copy_from_slice(&self.as_client_bytes()[0..4]); other_bytes.copy_from_slice(&other.as_client_bytes()[0..4]); let self_value = u32::from_be_bytes(self_bytes); let other_value = u32::from_be_bytes(other_bytes); self_value.partial_cmp(&other_value) } } impl std::cmp::Ord for BankItem { fn cmp(&self, other: &BankItem) -> std::cmp::Ordering { //let self_bytes = self.as_client_bytes(); //let other_bytes = other.as_client_bytes(); let mut self_bytes = [0u8; 4]; let mut other_bytes = [0u8; 4]; self_bytes.copy_from_slice(&self.as_client_bytes()[0..4]); other_bytes.copy_from_slice(&other.as_client_bytes()[0..4]); let self_value = u32::from_le_bytes(self_bytes); let other_value = u32::from_le_bytes(other_bytes); self_value.cmp(&other_value) } } impl BankItem { pub fn set_item_id(&mut self, item_id: ClientItemId) { match self { BankItem::Individual(individual_bank_item) => { individual_bank_item.item_id = item_id }, BankItem::Stacked(stacked_bank_item) => { stacked_bank_item.item_id = item_id } } } pub fn item_id(&self) -> ClientItemId { match self { BankItem::Individual(individual_bank_item) => { individual_bank_item.item_id }, BankItem::Stacked(stacked_bank_item) => { stacked_bank_item.item_id } } } pub fn as_client_bytes(&self) -> [u8; 16] { match self { BankItem::Individual(item) => { match &item.item { ItemDetail::Weapon(w) => w.as_bytes(), ItemDetail::Armor(a) => a.as_bytes(), ItemDetail::Shield(s) => s.as_bytes(), ItemDetail::Unit(u) => u.as_bytes(), ItemDetail::Tool(t) => t.as_individual_bytes(), ItemDetail::TechniqueDisk(d) => d.as_bytes(), ItemDetail::Mag(m) => m.as_bytes(), } }, BankItem::Stacked(item) => { item.tool.as_stacked_bytes(item.entity_ids.len()) }, } } } pub struct CharacterBank(Vec); impl CharacterBank { pub fn new(mut items: Vec) -> CharacterBank { items.sort(); CharacterBank(items) } pub fn initialize_item_ids(&mut self, base_item_id: u32) { for (i, item) in self.0.iter_mut().enumerate() { item.set_item_id(ClientItemId(base_item_id + i as u32)); } } pub fn as_client_bank_items(&self) -> character::Bank { self.0.iter() .enumerate() .fold(character::Bank::default(), |mut bank, (slot, item)| { bank.item_count = (slot + 1) as u32; let bytes = item.as_client_bytes(); bank.items[slot].data1.copy_from_slice(&bytes[0..12]); bank.items[slot].data2.copy_from_slice(&bytes[12..16]); bank.items[slot].item_id = item.item_id().0; bank }) } pub fn as_client_bank_request(&self) -> Vec { self.0.iter() .map(|item| { let bytes = item.as_client_bytes(); let mut data1 = [0; 12]; let mut data2 = [0; 4]; data1.copy_from_slice(&bytes[0..12]); data2.copy_from_slice(&bytes[12..16]); let amount = match item { BankItem::Individual(_individual_bank_item) => { 1 }, BankItem::Stacked(stacked_bank_item) => { stacked_bank_item.count() }, }; character::BankItem { data1: data1, data2: data2, item_id: item.item_id().0, amount: amount as u16, flags: 1, } }) .collect() } pub fn count(&self) -> usize { self.0.len() } }