111 lines
3.5 KiB
Rust
111 lines
3.5 KiB
Rust
|
// Copyright (c) 2024 Jonas Schäfer <jonas@zombofant.net>
|
||
|
|
//
|
||
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
||
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||
|
|
|
||
|
|
//! This module concerns the processing of attributes.
|
||
|
|
//!
|
||
|
|
//! In particular, it provides the `#[xml(attribute)]` implementation.
|
||
|
|
|
||
|
|
use quote::{quote, ToTokens};
|
||
|
|
use syn::*;
|
||
|
|
|
||
|
|
use crate::error_message::{self, ParentRef};
|
||
|
|
use crate::meta::{Flag, NameRef, NamespaceRef};
|
||
|
|
use crate::scope::{AsItemsScope, FromEventsScope};
|
||
|
|
use crate::types::{as_optional_xml_text_fn, default_fn, from_xml_text_fn};
|
||
|
|
|
||
|
|
use super::{Field, FieldBuilderPart, FieldIteratorPart, FieldTempInit};
|
||
|
|
|
||
|
|
/// The field maps to an attribute.
|
||
|
|
pub(super) struct AttributeField {
|
||
|
|
/// The optional XML namespace of the attribute.
|
||
|
|
pub(super) xml_namespace: Option<NamespaceRef>,
|
||
|
|
|
||
|
|
/// The XML name of the attribute.
|
||
|
|
pub(super) xml_name: NameRef,
|
||
|
|
|
||
|
|
/// Flag indicating whether the value should be defaulted if the
|
||
|
|
/// attribute is absent.
|
||
|
|
pub(super) default_: Flag,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Field for AttributeField {
|
||
|
|
fn make_builder_part(
|
||
|
|
&self,
|
||
|
|
scope: &FromEventsScope,
|
||
|
|
container_name: &ParentRef,
|
||
|
|
member: &Member,
|
||
|
|
ty: &Type,
|
||
|
|
) -> Result<FieldBuilderPart> {
|
||
|
|
let FromEventsScope { ref attrs, .. } = scope;
|
||
|
|
let ty = ty.clone();
|
||
|
|
let xml_namespace = &self.xml_namespace;
|
||
|
|
let xml_name = &self.xml_name;
|
||
|
|
|
||
|
|
let missing_msg = error_message::on_missing_attribute(container_name, member);
|
||
|
|
|
||
|
|
let xml_namespace = match xml_namespace {
|
||
|
|
Some(v) => v.to_token_stream(),
|
||
|
|
None => quote! {
|
||
|
|
::xso::exports::rxml::Namespace::none()
|
||
|
|
},
|
||
|
|
};
|
||
|
|
|
||
|
|
let from_xml_text = from_xml_text_fn(ty.clone());
|
||
|
|
|
||
|
|
let on_absent = match self.default_ {
|
||
|
|
Flag::Absent => quote! {
|
||
|
|
return ::core::result::Result::Err(::xso::error::Error::Other(#missing_msg).into())
|
||
|
|
},
|
||
|
|
Flag::Present(_) => {
|
||
|
|
let default_ = default_fn(ty.clone());
|
||
|
|
quote! {
|
||
|
|
#default_()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
Ok(FieldBuilderPart::Init {
|
||
|
|
value: FieldTempInit {
|
||
|
|
init: quote! {
|
||
|
|
match #attrs.remove(#xml_namespace, #xml_name).map(#from_xml_text).transpose()? {
|
||
|
|
::core::option::Option::Some(v) => v,
|
||
|
|
::core::option::Option::None => #on_absent,
|
||
|
|
}
|
||
|
|
},
|
||
|
|
ty: ty.clone(),
|
||
|
|
},
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
fn make_iterator_part(
|
||
|
|
&self,
|
||
|
|
_scope: &AsItemsScope,
|
||
|
|
_container_name: &ParentRef,
|
||
|
|
bound_name: &Ident,
|
||
|
|
_member: &Member,
|
||
|
|
ty: &Type,
|
||
|
|
) -> Result<FieldIteratorPart> {
|
||
|
|
let xml_namespace = match self.xml_namespace {
|
||
|
|
Some(ref v) => quote! { ::xso::exports::rxml::Namespace::from(#v) },
|
||
|
|
None => quote! {
|
||
|
|
::xso::exports::rxml::Namespace::NONE
|
||
|
|
},
|
||
|
|
};
|
||
|
|
let xml_name = &self.xml_name;
|
||
|
|
|
||
|
|
let as_optional_xml_text = as_optional_xml_text_fn(ty.clone());
|
||
|
|
|
||
|
|
Ok(FieldIteratorPart::Header {
|
||
|
|
generator: quote! {
|
||
|
|
#as_optional_xml_text(#bound_name)?.map(|#bound_name| ::xso::Item::Attribute(
|
||
|
|
#xml_namespace,
|
||
|
|
::std::borrow::Cow::Borrowed(#xml_name),
|
||
|
|
#bound_name,
|
||
|
|
));
|
||
|
|
},
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|