1use super::*;
2
3use crate::error::strings::InvalidIso646Character;
4use alloc::{borrow::ToOwned, boxed::Box, string::String, vec::Vec};
5use once_cell::race::OnceBox;
6
7#[derive(Debug, Default, Clone, Hash, PartialEq, Eq, PartialOrd, Ord)]
16#[allow(clippy::module_name_repetitions)]
17pub struct VisibleString(Vec<u8>);
18static CHARACTER_MAP: OnceBox<alloc::collections::BTreeMap<u32, u32>> = OnceBox::new();
19static INDEX_MAP: OnceBox<alloc::collections::BTreeMap<u32, u32>> = OnceBox::new();
20
21impl VisibleString {
22 pub fn from_iso646_bytes(bytes: &[u8]) -> Result<Self, InvalidIso646Character> {
28 bytes.iter().try_for_each(|byte| {
29 if Self::CHARACTER_SET.contains(&(*byte as u32)) {
30 Ok(())
31 } else {
32 Err(InvalidIso646Character { character: *byte })
33 }
34 })?;
35
36 Ok(Self(bytes.to_owned()))
37 }
38 #[must_use]
40 pub fn as_iso646_bytes(&self) -> &[u8] {
41 &self.0
42 }
43}
44
45impl StaticPermittedAlphabet for VisibleString {
46 const CHARACTER_SET: &'static [u32] = &[
48 0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x27, 0x28, 0x29, 0x2A, 0x2B, 0x2C, 0x2D, 0x2E,
49 0x2F, 0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, 0x38, 0x39, 0x3A, 0x3B, 0x3C, 0x3D,
50 0x3E, 0x3F, 0x40, 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48, 0x49, 0x4A, 0x4B, 0x4C,
51 0x4D, 0x4E, 0x4F, 0x50, 0x51, 0x52, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58, 0x59, 0x5A, 0x5B,
52 0x5C, 0x5D, 0x5E, 0x5F, 0x60, 0x61, 0x62, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68, 0x69, 0x6A,
53 0x6B, 0x6C, 0x6D, 0x6E, 0x6F, 0x70, 0x71, 0x72, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78, 0x79,
54 0x7A, 0x7B, 0x7C, 0x7D, 0x7E,
55 ];
56
57 fn chars(&self) -> Box<dyn Iterator<Item = u32> + '_> {
58 Box::from(self.0.iter().map(|byte| *byte as u32))
59 }
60
61 #[track_caller]
62 fn push_char(&mut self, ch: u32) {
63 debug_assert!(
64 Self::CHARACTER_SET.contains(&ch),
65 "{} not in character set",
66 ch
67 );
68 self.0.push(ch as u8);
69 }
70
71 fn index_map() -> &'static alloc::collections::BTreeMap<u32, u32> {
72 INDEX_MAP.get_or_init(Self::build_index_map)
73 }
74
75 fn character_map() -> &'static alloc::collections::BTreeMap<u32, u32> {
76 CHARACTER_MAP.get_or_init(Self::build_character_map)
77 }
78}
79
80impl core::fmt::Display for VisibleString {
81 fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
82 f.write_str(&String::from_utf8(self.as_iso646_bytes().to_owned()).unwrap())
83 }
84}
85
86impl AsnType for VisibleString {
87 const TAG: Tag = Tag::VISIBLE_STRING;
88}
89
90impl Encode for VisibleString {
91 fn encode_with_tag_and_constraints<E: Encoder>(
92 &self,
93 encoder: &mut E,
94 tag: Tag,
95 constraints: Constraints,
96 ) -> Result<(), E::Error> {
97 encoder
98 .encode_visible_string(tag, constraints, self)
99 .map(drop)
100 }
101}
102
103impl Decode for VisibleString {
104 fn decode_with_tag_and_constraints<D: Decoder>(
105 decoder: &mut D,
106 tag: Tag,
107 constraints: Constraints,
108 ) -> Result<Self, D::Error> {
109 decoder.decode_visible_string(tag, constraints)
110 }
111}
112
113impl TryFrom<alloc::string::String> for VisibleString {
114 type Error = InvalidIso646Character;
115
116 fn try_from(value: alloc::string::String) -> Result<Self, Self::Error> {
117 Self::from_iso646_bytes(value.as_bytes())
118 }
119}
120
121impl TryFrom<&'_ str> for VisibleString {
122 type Error = InvalidIso646Character;
123
124 fn try_from(value: &str) -> Result<Self, Self::Error> {
125 Self::from_iso646_bytes(value.as_bytes())
126 }
127}
128
129impl TryFrom<alloc::vec::Vec<u8>> for VisibleString {
130 type Error = InvalidIso646Character;
131
132 fn try_from(value: alloc::vec::Vec<u8>) -> Result<Self, Self::Error> {
133 Self::from_iso646_bytes(&value)
134 }
135}
136
137impl TryFrom<&'_ [u8]> for VisibleString {
138 type Error = InvalidIso646Character;
139
140 fn try_from(value: &[u8]) -> Result<Self, Self::Error> {
141 Self::from_iso646_bytes(value)
142 }
143}
144
145impl TryFrom<bytes::Bytes> for VisibleString {
146 type Error = InvalidIso646Character;
147
148 fn try_from(value: bytes::Bytes) -> Result<Self, Self::Error> {
149 Self::try_from(&*value)
150 }
151}
152
153impl From<VisibleString> for bytes::Bytes {
154 fn from(value: VisibleString) -> Self {
155 value.0.into()
156 }
157}
158
159impl From<VisibleString> for alloc::string::String {
160 fn from(value: VisibleString) -> Self {
161 Self::from_utf8(value.as_iso646_bytes().to_owned()).unwrap()
162 }
163}