jeff/trade_bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

Add price checking feature

Browse Source
This commit is contained in:
Jeff 2024-07-08 20:16:10 -04:00
parent 612451142d
commit 27bb52153b
1 changed files with 43 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

68
src/bot.rs
View File

@ -127,8 +127,6 @@ impl Bot {
TradeMode::Take => self.handle_take(trade.clone())?, TradeMode::Take => self.handle_take(trade.clone())?,
} }
} else if self.client.pending_invites().is_empty() { } else if self.client.pending_invites().is_empty() {
self.is_player_notified = false;
self.client.accept_invite(); self.client.accept_invite();
} }
@ -153,22 +151,40 @@ impl Bot {
fn handle_veloren_event(&mut self, event: VelorenEvent) -> Result<(), String> { fn handle_veloren_event(&mut self, event: VelorenEvent) -> Result<(), String> {
match event { match event {
VelorenEvent::Chat(message) => { VelorenEvent::Chat(message) => {
let sender = match message.chat_type {
ChatType::Tell(uid, _) => uid,
ChatType::Say(uid) => uid,
_ => return Ok(()),
};
let content = message.content().as_plain().unwrap_or_default(); let content = message.content().as_plain().unwrap_or_default();
let (command, item_name) = content.split_once(' ').unwrap_or((content, ""));
match message.chat_type { match command {
ChatType::Tell(sender_uid, _) | ChatType::Say(sender_uid) => { "price" => {
match content.trim() { for (item_id, price) in &self.buy_prices {
"prices" => self.send_price_info(&sender_uid)?, if item_id.contains(item_name) {
"take" => { let player_name = self
if !self.client.is_trading() { .find_name(&sender)
self.trade_mode = TradeMode::Take; .ok_or("Failed to find player name")?
self.client.send_invite(sender_uid, InviteKind::Trade); .to_string();
}
self.client.send_command(
"tell".to_string(),
vec![
player_name.clone(),
format!("{item_id} costs {price} coins"),
],
);
} }
_ => {}
} }
} }
"prices" => self.send_all_price_info(&sender)?,
"take" => {
if !self.client.is_trading() {
self.trade_mode = TradeMode::Take;
self.client.send_invite(sender, InviteKind::Trade);
}
}
_ => {} _ => {}
} }
} }
@ -202,7 +218,8 @@ impl Bot {
log::info!("Completed trade: {result:?}"); log::info!("Completed trade: {result:?}");
if let TradeMode::Take = self.trade_mode { if let TradeMode::Take = self.trade_mode {
self.trade_mode = TradeMode::Trade self.is_player_notified = false;
self.trade_mode = TradeMode::Trade;
} }
if let TradeResult::Completed = result { if let TradeResult::Completed = result {
@ -236,14 +253,21 @@ impl Bot {
} }
fn handle_trade(&mut self, trade: PendingTrade) -> Result<(), String> { fn handle_trade(&mut self, trade: PendingTrade) -> Result<(), String> {
if trade.is_empty_trade() {
return Ok(());
}
let my_offer_index = trade let my_offer_index = trade
.which_party(self.client.uid().ok_or("Failed to get uid")?) .which_party(self.client.uid().ok_or("Failed to get uid")?)
.ok_or("Failed to get offer index")?; .ok_or("Failed to get offer index")?;
let their_offer_index = if my_offer_index == 0 { 1 } else { 0 }; let their_offer_index = if my_offer_index == 0 { 1 } else { 0 };
if !self.is_player_notified {
self.send_all_price_info(&trade.parties[their_offer_index])?;
self.is_player_notified = true;
}
if trade.is_empty_trade() {
return Ok(());
}
let (my_offer, their_offer, them) = { let (my_offer, their_offer, them) = {
( (
&trade.offers[my_offer_index], &trade.offers[my_offer_index],
@ -356,12 +380,6 @@ impl Bot {
drop(inventories); drop(inventories);
if !self.is_player_notified {
self.send_price_info(&trade.parties[their_offer_index])?;
self.is_player_notified = true;
}
if their_offered_items_value == 0 && my_offered_items_value == 0 { if their_offered_items_value == 0 && my_offered_items_value == 0 {
return Ok(()); return Ok(());
} }
@ -462,7 +480,7 @@ impl Bot {
Ok(()) Ok(())
} }
fn send_price_info(&mut self, target: &Uid) -> Result<(), String> { fn send_all_price_info(&mut self, target: &Uid) -> Result<(), String> {
let player_name = self let player_name = self
.find_name(target) .find_name(target)
.ok_or("Failed to find player name")? .ok_or("Failed to find player name")?