tenebrous-dicebot/src/dice.rs

32 lines
1.1 KiB
Rust
Raw Normal View History

use crate::context::Context;
use crate::db::variables::UserAndRoom;
use crate::error::BotError;
use crate::error::DiceRollingError;
use crate::parser::Amount;
use crate::parser::Element as NewElement;
use futures::stream::{self, StreamExt, TryStreamExt};
2020-04-20 17:19:50 +00:00
//New hotness
pub async fn calculate_dice_amount(amounts: &[Amount], ctx: &Context<'_>) -> Result<i32, BotError> {
let stream = stream::iter(amounts);
let key = UserAndRoom(&ctx.username, &ctx.room_id);
let variables = &ctx.db.variables.get_user_variables(&key)?;
use DiceRollingError::VariableNotFound;
let dice_amount: Result<i32, BotError> = stream
.then(|amount| async move {
match &amount.element {
NewElement::Number(num_dice) => Ok(*num_dice * amount.operator.mult()),
NewElement::Variable(variable) => variables
.get(variable)
.ok_or(VariableNotFound(variable.clone().to_string()))
.map(|i| *i)
.map_err(|e| e.into()),
}
})
.try_fold(0, |total, num_dice| async move { Ok(total + num_dice) })
.await;
dice_amount
}