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}; //New hotness pub async fn calculate_dice_amount(amounts: &[Amount], ctx: &Context<'_>) -> Result { let stream = stream::iter(amounts); let key = UserAndRoom(&ctx.username, ctx.room.id.as_str()); let variables = &ctx.db.variables.get_user_variables(&key)?; use DiceRollingError::VariableNotFound; let dice_amount: Result = 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 }