idsmith 0.5.4

use rand::Rng;

use super::IdResult;

static WEIGHTS: [u32; 7] = [2, 9, 8, 7, 6, 3, 4];

fn compute_check(digits: &[u8]) -> u8 {
    let s: u32 = digits
        .iter()
        .zip(WEIGHTS.iter())
        .map(|(&d, &w)| {
            let p = d as u32 * w;
            if p > 9 {
                p % 10
            } else {
                p
            }
        })
        .sum();
    ((10 - s % 10) % 10) as u8
}

pub fn generate(_opts: &super::GenOptions, rng: &mut impl Rng) -> String {
    let mut digits: Vec<u8> = Vec::with_capacity(8);
    digits.push(rng.gen_range(1..=9));
    for _ in 0..6 {
        digits.push(rng.gen_range(0..=9));
    }
    let check = compute_check(&digits);
    digits.push(check);
    digits.iter().map(|d| (b'0' + d) as char).collect()
}

pub fn validate(code: &str) -> bool {
    let clean: String = code.chars().filter(|c| c.is_ascii_digit()).collect();
    if clean.len() != 8 {
        return false;
    }
    let digits: Vec<u8> = clean.bytes().map(|b| b - b'0').collect();
    compute_check(&digits[..7]) == digits[7]
}

pub fn parse(code: &str) -> IdResult {
    let clean: String = code.chars().filter(|c| c.is_ascii_digit()).collect();
    IdResult {
        country_code: "".to_string(),
        code: clean,
        gender: None,
        dob: None,
        valid: validate(code),
    }
}