neutralts/bif/
parse_bif_rand.rs

1#![doc = include_str!("../../doc/bif-rand.md")]
2
3use crate::{bif::constants::*, bif::Bif, bif::BifError, constants::*};
4use rand::Rng;
5
6impl<'a> Bif<'a> {
7    /*
8        {:rand;  :}
9        {:rand; 1..100 :}
10    */
11    pub(crate) fn parse_bif_rand(&mut self) -> Result<(), BifError> {
12        if self.mod_filter || self.mod_negate || self.mod_scope {
13            return Err(self.bif_error(BIF_ERROR_MODIFIER_NOT_ALLOWED));
14        }
15
16        let mut rng = rand::rng();
17        self.code = self.src.trim().to_string();
18
19        if self.src.contains(BIF_OPEN) {
20            self.code = new_child_parse!(self, &self.code, self.mod_scope);
21        } else {
22            self.code = self.src.trim().to_string();
23        }
24
25        if self.code.is_empty() {
26            self.out = rng.random_range(100000000..=999999999).to_string();
27        } else {
28            // TODO comprobar rangos
29            self.code = self.code.replace("..", " ");
30            let mut parts = self.code.split_whitespace();
31
32            let from = match parts.next() {
33                Some(value) => match value.parse::<i32>() {
34                    Ok(num) => num,
35                    Err(_) => {
36                        return Err(self.bif_error(BIF_ERROR_ARGUMENT_NOT_NUMBER));
37                    }
38                },
39                None => {
40                    return Err(self.bif_error(BIF_ERROR_ARGUMENTS_NOT_FOUND));
41                }
42            };
43
44            let to = match parts.next() {
45                Some(value) => match value.parse::<i32>() {
46                    Ok(num) => num,
47                    Err(_) => {
48                        return Err(self.bif_error(BIF_ERROR_ARGUMENT_NOT_NUMBER));
49                    }
50                },
51                None => {
52                    return Err(self.bif_error(BIF_ERROR_ARGUMENTS_NOT_FOUND));
53                }
54            };
55
56            if from > to {
57                return Err(self.bif_error(BIF_ERROR_FROM_GREATER_THAN_TO));
58            }
59
60            self.out = rng.random_range(from..=to).to_string();
61        }
62
63        Ok(())
64    }
65}
66
67#[cfg(test)]
68#[path = "parse_bif_rand_tests.rs"]
69mod tests;
neutralts/bif/parse_bif_rand.rs

neutralts/bif/
parse_bif_rand.rs
