uuinfo 0.6.1

use sqids::Sqids;

use crate::schema::{Args, IDInfo};
use crate::utils::factor_size_hex_bits_color_from_text;

pub fn parse_sqid(args: &Args) -> Option<IDInfo> {
    let mut version: Option<String> = Some("Default alphabet".to_string());
    let sqids = match &args.alphabet {
        Some(alphabet) => match Sqids::builder().alphabet(alphabet.chars().collect()).build() {
            Ok(value) => {
                version = Some("Custom alphabet".to_string());
                value
            }
            Err(_) => {
                println!("Invalid alphabet for Sqids.");
                std::process::exit(1);
            }
        },
        None => Sqids::default(),
    };
    let numbers = sqids.decode(&args.id);
    if numbers.is_empty() {
        return None;
    }
    let (size, hex, bits, _) = factor_size_hex_bits_color_from_text(&args.id);

    Some(IDInfo {
        id_type: "Sqid".to_string(),
        version,
        standard: args.id.clone(),
        parsed: Some("as ASCII".to_string()),
        size,
        node1: Some(numbers.iter().map(|x| x.to_string()).collect::<Vec<_>>().join(", ")),
        hex,
        bits,
        color_map: Some((0..size).map(|_| "4").collect::<String>()),
        ..Default::default()
    })
}

pub fn parse_sqid_maybe(args: &Args) -> Option<IDInfo> {
    let parsed = parse_sqid(args)?;
    let binding = parsed.node1.clone()?;
    let mut segments = binding.split(", ").collect::<Vec<_>>();
    if segments.len() == 1 {
        return None;
    }
    segments.remove(0);
    if segments.iter().all(|x| *x == "0") {
        return None;
    }
    Some(parsed)
}