1use crate::{BuckyError, BuckyErrorCode, BuckyResult};
2
3pub use base58::{FromBase58, ToBase58};
4pub trait ToBase36 {
5 fn to_base36(&self) -> String;
6}
7
8pub trait FromBase36 {
9 fn from_base36(&self) -> BuckyResult<Vec<u8>>;
10}
11
12const ALPHABET: &[u8] = b"0123456789abcdefghijklmnopqrstuvwxyz";
13
14impl ToBase36 for [u8] {
15 fn to_base36(&self) -> String {
16 base_x::encode(ALPHABET, self)
17 }
18}
19
20impl FromBase36 for str {
21 fn from_base36(&self) -> BuckyResult<Vec<u8>> {
22 base_x::decode(ALPHABET, &self.to_ascii_lowercase()).map_err(|e| {
23 let msg = format!("convert string to base36 error! {self}, {e}");
24 BuckyError::new(BuckyErrorCode::InvalidFormat, msg)
25 })
26 }
27}
28
29
30#[cfg(test)]
31mod test {
32 use std::str::FromStr;
33
34 use crate::*;
35
36 fn test_convert(s: &str) -> (ObjectId, String) {
37 let id = ObjectId::from_str(s).unwrap();
38 let v = id.to_base36();
39 println!("{}", v);
40 let id2 = ObjectId::from_base36(&v).unwrap();
41 assert_eq!(id, id2);
42 let id3 = ObjectId::from_str(&v).unwrap();
43 assert_eq!(id, id3);
44
45 let id2 = ObjectId::from_base36(&v.to_uppercase()).unwrap();
46 assert_eq!(id, id2);
47
48 (id, v)
49 }
50
51 #[test]
52 fn test() {
53 let id = ObjectId::default();
54 let v = id.as_slice().to_base36();
55 println!("{}", v);
56
57 let s = "9tGpLNna8UVtPYCfV1LbRN2Bqa5G9vRBKhDhZiWjd7wA";
59 let (id1, id11) = test_convert(s);
60 let s = "9tGpLNna8P1hutR3y6i1gSGuosoLXLxa72HMrcEQnrgk";
63 let (id2, id22) = test_convert(s);
64
65 assert_ne!(id11, id22);
66 assert_ne!(id1, id2);
67
68 let s = "95RvaS5eWQsLpCGgkY773EKomgfa73EcmGP9VHWnwig3";
69 test_convert(s);
70 }
71}