salvo-oapi-macros 0.91.1

use proc_macro2::TokenStream;
use quote::{ToTokens, quote};

use crate::feature::{Feature, FeaturesExt, pop_feature};
use crate::{ComponentSchema, ComponentSchemaProps, DiagResult, TryToTokens};

#[derive(Debug)]
pub(crate) struct FlattenedMapSchema {
    tokens: TokenStream,
}

impl FlattenedMapSchema {
    pub(crate) fn new(
        ComponentSchemaProps {
            type_tree,
            features,
            description,
            deprecated,
            object_name,
            compose_context,
        }: ComponentSchemaProps,
    ) -> DiagResult<Self> {
        let mut tokens = TokenStream::new();
        let mut features = features.unwrap_or(Vec::new());
        let deprecated_stream = ComponentSchema::get_deprecated(deprecated);

        let example = features
            .pop_by(|feature| matches!(feature, Feature::Example(_)))
            .map(|f| f.try_to_token_stream())
            .transpose()?;
        let nullable = pop_feature!(features => Feature::Nullable(_))
            .map(|f| f.try_to_token_stream())
            .transpose()?;
        let default = pop_feature!(features => Feature::Default(_))
            .map(|f| f.try_to_token_stream())
            .transpose()?;

        // Maps are treated as generic objects with no named properties and
        // additionalProperties denoting the type
        // maps have 2 child schemas and we are interested the second one of them
        // which is used to determine the additional properties
        let schema_property = ComponentSchema::new(ComponentSchemaProps {
            type_tree: type_tree
                .children
                .as_ref()
                .expect("`ComponentSchema` Map type should have children")
                .get(1)
                .expect("`ComponentSchema` Map type should have 2 child"),
            features: Some(features),
            description: None,
            deprecated: None,
            object_name,
            compose_context,
        })?;

        tokens.extend(quote! {
            #schema_property
                #description
                #deprecated_stream
                #default
        });

        example.to_tokens(&mut tokens);
        nullable.to_tokens(&mut tokens);

        Ok(Self { tokens })
    }
}

impl ToTokens for FlattenedMapSchema {
    fn to_tokens(&self, tokens: &mut TokenStream) {
        self.tokens.to_tokens(tokens);
    }
}