mod fields;
mod key;
mod value;
pub use fields::Fields;
pub use key::Key;
pub use value::Value;
use crate::parser::SubtagIterator;
use crate::parser::{parse_language_identifier_from_iter, ParserError, ParserMode};
use crate::subtags::Language;
use crate::LanguageIdentifier;
use alloc::vec;
use litemap::LiteMap;
#[derive(Clone, PartialEq, Eq, Debug, Default, Hash)]
#[allow(clippy::exhaustive_structs)] pub struct Transform {
pub lang: Option<LanguageIdentifier>,
pub fields: Fields,
}
impl Transform {
#[inline]
pub const fn new() -> Self {
Self {
lang: None,
fields: Fields::new(),
}
}
pub fn is_empty(&self) -> bool {
self.lang.is_none() && self.fields.is_empty()
}
pub fn clear(&mut self) {
self.lang = None;
self.fields.clear();
}
pub(crate) fn try_from_iter(iter: &mut SubtagIterator) -> Result<Self, ParserError> {
let mut tlang = None;
let mut tfields = LiteMap::new();
if let Some(subtag) = iter.peek() {
if Language::try_from_bytes(subtag).is_ok() {
tlang = Some(parse_language_identifier_from_iter(
iter,
ParserMode::Partial,
)?);
}
}
let mut current_tkey = None;
let mut current_tvalue = vec![];
while let Some(subtag) = iter.peek() {
if let Some(tkey) = current_tkey {
if let Ok(val) = Value::parse_subtag(subtag) {
current_tvalue.push(val);
} else {
if current_tvalue.is_empty() {
return Err(ParserError::InvalidExtension);
}
tfields.try_insert(
tkey,
Value::from_vec_unchecked(current_tvalue.drain(..).flatten().collect()),
);
current_tkey = None;
continue;
}
} else if let Ok(tkey) = Key::try_from_bytes(subtag) {
current_tkey = Some(tkey);
} else {
break;
}
iter.next();
}
if let Some(tkey) = current_tkey {
if current_tvalue.is_empty() {
return Err(ParserError::InvalidExtension);
}
tfields.try_insert(
tkey,
Value::from_vec_unchecked(current_tvalue.into_iter().flatten().collect()),
);
}
Ok(Self {
lang: tlang,
fields: tfields.into(),
})
}
pub(crate) fn for_each_subtag_str<E, F>(&self, f: &mut F) -> Result<(), E>
where
F: FnMut(&str) -> Result<(), E>,
{
if self.is_empty() {
return Ok(());
}
f("t")?;
if let Some(lang) = &self.lang {
lang.for_each_subtag_str(f)?;
}
self.fields.for_each_subtag_str(f)
}
}
writeable::impl_display_with_writeable!(Transform);
impl writeable::Writeable for Transform {
fn write_to<W: core::fmt::Write + ?Sized>(&self, sink: &mut W) -> core::fmt::Result {
if self.is_empty() {
return Ok(());
}
sink.write_str("t")?;
if let Some(lang) = &self.lang {
sink.write_char('-')?;
writeable::Writeable::write_to(lang, sink)?;
}
if !self.fields.is_empty() {
sink.write_char('-')?;
writeable::Writeable::write_to(&self.fields, sink)?;
}
Ok(())
}
fn writeable_length_hint(&self) -> writeable::LengthHint {
if self.is_empty() {
return writeable::LengthHint::exact(0);
}
let mut result = writeable::LengthHint::exact(1);
if let Some(lang) = &self.lang {
result += writeable::Writeable::writeable_length_hint(lang) + 1;
}
if !self.fields.is_empty() {
result += writeable::Writeable::writeable_length_hint(&self.fields) + 1;
}
result
}
}