1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163

/*!
This is a cool-id-generator.

It makes memorable ids.
honest-turbo-tailor-gregory, romantic-robot-chicken-kenneth and happy-ultra-barista-shane would approve.

```
#[macro_use]
use cool_id_generator::{get_id, Size};

let my_id = get_id(Size::Medium);
println!("{:?}", my_id);
let my_long_id = get_id(Size::Long);
println!("{:?}", my_long_id);

```
*/
#![no_std]

#[macro_use]
extern crate alloc;
use alloc::string::String;

pub mod words;
use words::{ADJECTIVES, ANIMALS, ANIMAL_PREFIX, JOBS, JOBS_PREFIX, NAMES};

use rand::seq::SliceRandom;
use rand::thread_rng;

#[inline]
pub fn get_id(size: Size) -> String {
    let mut rng = thread_rng();

    let animal_or_job = {
        if rand::random::<bool>() {
            (
                ANIMAL_PREFIX.choose(&mut rng).unwrap(),
                ANIMALS.choose(&mut rng).unwrap(),
            )
        } else {
            (
                JOBS_PREFIX.choose(&mut rng).unwrap(),
                JOBS.choose(&mut rng).unwrap(),
            )
        }
    };

    match size {
        Size::Medium => {
            let adj1 = ADJECTIVES.choose(&mut rng).unwrap();
            let name = NAMES.choose(&mut rng).unwrap();
            format!("{}{}-{}-{}", adj1, animal_or_job.0, animal_or_job.1, name)
        }
        Size::Long => {
            let name = NAMES.choose(&mut rng).unwrap();
            let adj1 = ADJECTIVES.choose(&mut rng).unwrap();
            let adj2 = ADJECTIVES.choose(&mut rng).unwrap();
            format!(
                "{}-the-{}-and-{}{}-{}",
                name, adj1, adj2, animal_or_job.0, animal_or_job.1
            )
        }
        Size::VeryLong => {
            let name = NAMES.choose(&mut rng).unwrap();
            let name2 = NAMES.choose(&mut rng).unwrap();
            let adj1 = ADJECTIVES.choose(&mut rng).unwrap();
            let adj2 = ADJECTIVES.choose(&mut rng).unwrap();
            format!(
                "{}-{}-the-{}-and-{}{}-{}",
                name, name2, adj1, adj2, animal_or_job.0, animal_or_job.1
            )
        }
    }
}

/// Size which can be cast into `usize` to use as the size of the output byte array e.g.
/// ## Example
/// ```
/// use cool_id_generator::Size;
///
/// const size: usize = Size::Medium as usize;
/// let byte_array: [u8; size] = [0u8; size];
/// ```
pub enum Size {
    /// The largest size (in bytes) of a `Medium` ID; 1 billion combinations
    Medium = get_id_max_len() as isize,
    /// The largest size (in bytes) of a `Long` ID; 115 billion combinations
    Long = get_long_id_max_len() as isize,
    /// The largest size (in bytes) of a `VeryLong` ID; 10^15 combinations (or 2,088,136,477,473,228)
    VeryLong = get_very_long_id_max_len() as isize,
}

const fn max(a: usize, b: usize) -> usize {
    [a, b][(a < b) as usize]
}

// returns maximum byte lengh of the given array
const fn get_max_len(items: &[&str]) -> usize {
    let mut i = 0;
    let mut largest = 0;
    while i < items.len() {
        let len = items[i].len();
        if len > largest {
            largest = len
        };
        i += 1;
    }
    largest
}

const fn get_very_long_id_max_len() -> usize {
    max(get_max_len(&ANIMAL_PREFIX), get_max_len(&JOBS_PREFIX))
        + get_max_len(&NAMES)
        + get_max_len(&NAMES)
        + get_max_len(&ADJECTIVES)
        + get_max_len(&ADJECTIVES)
        + get_max_len(&ANIMALS)
        + get_max_len(&JOBS)
}

const fn get_long_id_max_len() -> usize {
    max(get_max_len(&ANIMAL_PREFIX), get_max_len(&JOBS_PREFIX))
        + get_max_len(&NAMES)
        + get_max_len(&ADJECTIVES)
        + get_max_len(&ADJECTIVES)
        + get_max_len(&ANIMALS)
        + get_max_len(&JOBS)
}

const fn get_id_max_len() -> usize {
    max(get_max_len(&ANIMAL_PREFIX), get_max_len(&JOBS_PREFIX))
        + get_max_len(&NAMES)
        + get_max_len(&ADJECTIVES)
        + get_max_len(&ANIMALS)
        + get_max_len(&JOBS)
}

#[cfg(test)]
mod tests {
    extern crate std;

    use crate::*;
    use std::println;
    #[test]
    fn it_works() {
        // let yo: Vec<String> = (0..1000).map(|_|get_id()).collect();
        // println!("{:?}", yo.join(" "));

        println!("{}", get_id(Size::Medium));
        println!("{}", get_id(Size::Medium));
        println!("{}", get_id(Size::Medium));
        println!("{}", get_id(Size::Long));
        println!("{}", get_id(Size::Long));
        println!("{}", get_id(Size::Long));
        println!("{}", get_id(Size::VeryLong));
        println!("{}", get_id(Size::VeryLong));
        println!("{}", get_id(Size::VeryLong));

        println!("{}", get_id_max_len());
        println!("{}", get_long_id_max_len());
        println!("{}", get_very_long_id_max_len());
    }
}