use std::fmt::{self, Display};
use std::str::FromStr;
use compact_str::CompactString;
use crate::InvalidData;
#[derive(Debug, Copy, Clone, Eq, PartialEq, PartialOrd, Ord, Hash)]
pub struct IdCode(pub u64);
const ID_CHAR_MIN: u8 = b'!';
const ID_CHAR_MAX: u8 = b'~';
const NUM_ID_CHARS: u64 = (ID_CHAR_MAX - ID_CHAR_MIN + 1) as u64;
impl IdCode {
#[inline]
pub fn new(v: &[u8]) -> Result<IdCode, InvalidData> {
if v.is_empty() {
return Err(InvalidData("ID cannot be empty"));
}
let mut result = 0u64;
for &i in v.iter() {
if i < ID_CHAR_MIN || i > ID_CHAR_MAX {
return Err(InvalidData("invalid characters in ID"));
}
let c = ((i - ID_CHAR_MIN) as u64) + 1;
result = result
.checked_mul(NUM_ID_CHARS)
.and_then(|x| x.checked_add(c))
.ok_or(InvalidData("ID too long"))?;
}
Ok(IdCode(result - 1))
}
pub const FIRST: IdCode = IdCode(0);
#[inline]
pub fn next(&self) -> IdCode {
IdCode(self.0 + 1)
}
pub fn to_compact_string(self) -> CompactString {
let mut i = self.0;
let mut revname = CompactString::new("");
loop {
let r = i % NUM_ID_CHARS;
revname.push((r as u8 + ID_CHAR_MIN) as char);
if i < NUM_ID_CHARS {
break;
}
i = i / NUM_ID_CHARS - 1;
}
revname.chars().rev().collect()
}
}
impl FromStr for IdCode {
type Err = InvalidData;
#[inline]
fn from_str(s: &str) -> Result<Self, Self::Err> {
IdCode::new(s.as_bytes())
}
}
impl From<u32> for IdCode {
#[inline]
fn from(i: u32) -> IdCode {
IdCode(i as u64)
}
}
impl From<u64> for IdCode {
#[inline]
fn from(i: u64) -> IdCode {
IdCode(i)
}
}
impl Display for IdCode {
#[inline]
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}", self.to_compact_string())
}
}
#[test]
fn test_id_code() {
let mut id = IdCode::FIRST;
for i in 0..10000 {
println!("{} {}", i, id);
assert_eq!(id.to_string().parse::<IdCode>().unwrap(), id);
id = id.next();
}
assert_eq!("!".parse::<IdCode>().unwrap().to_string(), "!");
assert_eq!(
"!!!!!!!!!!".parse::<IdCode>().unwrap().to_string(),
"!!!!!!!!!!"
);
assert_eq!("~".parse::<IdCode>().unwrap().to_string(), "~");
assert_eq!(
"~~~~~~~~~".parse::<IdCode>().unwrap().to_string(),
"~~~~~~~~~"
);
assert_eq!(
"999999999n".parse::<IdCode>().unwrap().to_string(),
"999999999n"
);
assert!("9999999999n".parse::<IdCode>().is_err());
}