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}