veloren/server/src/events/trade.rs

use crate::{events::group_manip::handle_invite, Server};
use common::{
    comp::{
        group::InviteKind,
        inventory::{Inventory, slot::InvSlotId},
    },
    trade::{PendingTrade, TradeActionMsg, Trades},
    uid::Uid,
};
use common_net::{msg::ServerGeneral, sync::WorldSyncExt};
use specs::{world::WorldExt, Entity as EcsEntity};
use tracing::warn;

pub fn handle_initiate_trade(server: &mut Server, interactor: EcsEntity, counterparty: EcsEntity) {
    if let Some(uid) = server.state_mut().ecs().uid_from_entity(counterparty) {
        handle_invite(server, interactor, uid, InviteKind::Trade);
    } else {
        warn!("Entity tried to trade with an entity that lacks an uid");
    }
}

pub fn handle_process_trade_action(
    server: &mut Server,
    entity: EcsEntity,
    trade_id: usize,
    msg: TradeActionMsg,
) {
    if let Some(uid) = server.state.ecs().uid_from_entity(entity) {
        let mut trades = server.state.ecs().write_resource::<Trades>();
        if let TradeActionMsg::Decline = msg {
            let to_notify = trades.decline_trade(trade_id, uid);
            to_notify
                .and_then(|u| server.state.ecs().entity_from_uid(u.0))
                .map(|e| server.notify_client(e, ServerGeneral::DeclinedTrade));
        } else {
            if let Some(inv) = server.state.ecs().read_component::<Inventory>().get(entity) {
                trades.process_trade_action(trade_id, uid, msg, inv);
            }
            if let Some(trade) = trades.trades.get(&trade_id) {
                if trade.should_commit() {
                    // TODO: inventory manip
                }
                // send the updated state to both parties
                for party in trade.parties.iter() {
                    server.state.ecs().entity_from_uid(party.0).map(|e| {
                        server.notify_client(
                            e,
                            ServerGeneral::UpdatePendingTrade(trade_id, trade.clone()),
                        )
                    });
                }
            }
        }
    }
}