librollenspielsache 0.1.2

use super::*;
use crate::modifier::*;
use lazy_static::lazy_static;
use regex::Regex;

/// A Roll represents a single computation, for instance "2d4+5".
///
/// Create with fields:
///
/// ```
/// # use rollenspielsache::dice::Roll;
/// # use rollenspielsache::modifier::*;
/// let new_roll = Roll::new(20, 2, Modifiers::default());
/// ```
///
/// Create by string:
///
/// ```
/// # use rollenspielsache::dice::Roll;
/// # use std::error::Error;
/// # fn main() -> Result<(), Box<dyn Error>> {
/// # use std::str::FromStr;
/// let new_roll = Roll::from_str("2d20-7")?;
/// # Ok(())
/// # }
/// ```
///
/// Execute:
///
/// ```
/// # use rollenspielsache::dice::*;
/// # use rollenspielsache::modifier::*;
/// # use std::error::Error;
/// # use std::str::FromStr;
/// # fn main() -> Result<(), Box<dyn Error>> {
/// # let new_roll = Roll::from_str("2d20-7")?;
///   let result = new_roll.execute();
///   println!("{}", result.total());
/// # assert_eq!(new_roll, Roll::new(20, 2, Modifiers::from(vec![Modifier::new(7, true, ModifierType::Unspecified)])));
/// # Ok(())
/// # }
/// ```
///
/// Roll::default() will build a "1d6+0" roll.
#[derive(Debug, Clone, PartialEq, Serialize)]
pub struct Roll {
    die: Die,
    repeat: usize,
    modifiers: Modifiers,
}

impl Roll {
    pub fn new(die: usize, repeat: usize, modifiers: Modifiers) -> Self {
        Self {
            die: Die::new(die),
            repeat,
            modifiers,
        }
    }
    pub fn get_base(&self) -> usize {
        self.die.sides
    }
    pub fn get_repeat(&self) -> usize {
        self.repeat
    }
    fn get_modifier(&self) -> isize {
        self.modifiers.net()
    }
    pub fn execute(&self) -> RollResult {
        let base = (0..self.repeat).fold(0, |acc, _| acc + self.die.roll() as isize);
        RollResult::new(base, self.get_modifier())
    }
}

impl Default for Roll {
    fn default() -> Self {
        Self {
            die: Die::default(),
            repeat: 1,
            modifiers: Modifiers::default(),
        }
    }
}

impl fmt::Display for Roll {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "{}d{} ({})", self.repeat, self.die, self.modifiers)
    }
}

impl RedisInterface for Roll {
    fn to_redis_hash(&self) -> RedisHash {
        let mut hash = RedisHash::new();
        hash.add_pair("die", &format!("{}", self.die));
        hash.add_pair("repeat", &format!("{}", self.repeat));
        hash.add_pair("modifiers", &format!("{}", self.modifiers)); // TODO this should really become a List
        hash
    }
}

lazy_static! {
    // Minimal match is just the character 'd' - that will roll 1d6+0
    pub static ref ROLL_RE: Regex = Regex::new(r"(?x)
    ^
    (?P<repeat>[[:digit:]]+)?
    d
    (?P<sides>[[:digit:]]+)?
    (
        (?P<sign>\+|-)?
        (?P<offset>[[:digit:]]+)?
    )?$
    ").unwrap();
}

impl FromStr for Roll {
    type Err = std::io::Error;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        if ROLL_RE.is_match(s) {
            let caps = ROLL_RE.captures(s).unwrap();
            let mut negative = false;
            if let Some(sign_s) = caps.name("sign") {
                negative = sign_s.as_str().starts_with('-');
            }
            let mut offset = 0;
            if let Some(offset_s) = caps.name("offset") {
                offset += offset_s.as_str().parse::<isize>().unwrap();
            }

            let modifiers = vec![Modifier::new(offset, negative, ModifierType::Unspecified)];
            let die = match caps.name("sides") {
                Some(sides_s) => sides_s.as_str().parse::<usize>().unwrap(),
                None => 6,
            };
            let mut repeat = 1;
            if let Some(repeat_s) = caps.name("repeat") {
                repeat = repeat_s.as_str().parse::<usize>().unwrap();
            }

            Ok(Self::new(die, repeat, Modifiers::from(modifiers)))
        } else {
            Err(std::io::Error::new(
                std::io::ErrorKind::InvalidInput,
                format!("Bad format: {}", s),
            ))
        }
    }
}

#[cfg(test)]
mod test {
    use super::*;
    use pretty_assertions::assert_eq;
    #[test]
    fn test_roll_from_string() {
        assert_eq!(
            Roll::from_str("2d20").unwrap(),
            Roll {
                die: Die { sides: 20 },
                repeat: 2,
                modifiers: Modifiers::default()
            }
        );
        assert_eq!(
            Roll::from_str("34d100+3").unwrap(),
            Roll {
                die: Die { sides: 100 },
                repeat: 34,
                modifiers: Modifiers::from(vec![Modifier {
                    offset: 3,
                    variant: ModifierType::Unspecified
                }])
            }
        );
        assert_eq!(
            Roll::from_str("d5-20").unwrap(),
            Roll {
                die: Die { sides: 5 },
                repeat: 1,
                modifiers: Modifiers::from(vec![Modifier {
                    offset: -20,
                    variant: ModifierType::Unspecified
                }])
            }
        );
        assert_eq!(Roll::from_str("d").unwrap(), Roll::default());
    }
}