oauth2_client/
utils.rs

1use core::cmp::{max, min};
2
3use oauth2_core::types::{
4    code_verifier::{CODE_VERIFIER_CHARSET, CODE_VERIFIER_LEN_MAX, CODE_VERIFIER_LEN_MIN},
5    CodeVerifier, Nonce, State,
6};
7use rand::{distributions::Alphanumeric, thread_rng, Rng as _};
8
9pub fn gen_state(length: impl Into<Option<usize>>) -> State {
10    thread_rng()
11        .sample_iter(&Alphanumeric)
12        .take(length.into().unwrap_or(10))
13        .map(char::from)
14        .collect::<String>()
15}
16
17pub fn gen_nonce(length: impl Into<Option<usize>>) -> Nonce {
18    thread_rng()
19        .sample_iter(&Alphanumeric)
20        .take(length.into().unwrap_or(22))
21        .map(char::from)
22        .collect::<String>()
23}
24
25pub fn gen_code_verifier(length: impl Into<Option<usize>>) -> CodeVerifier {
26    let length = length.into().unwrap_or(64);
27    let length = min(CODE_VERIFIER_LEN_MAX, length);
28    let length = max(CODE_VERIFIER_LEN_MIN, length);
29
30    let mut rng = thread_rng();
31
32    (0..length)
33        .map(|_| {
34            let idx = rng.gen_range(0..CODE_VERIFIER_CHARSET.len());
35            CODE_VERIFIER_CHARSET[idx] as char
36        })
37        .collect::<String>()
38}
39
40#[cfg(test)]
41mod tests {
42    use super::*;
43
44    #[test]
45    fn test_gen_code_verifier() {
46        assert_eq!(gen_code_verifier(64).len(), 64);
47    }
48}