arma-rs-proc 1.13.0

use proc_macro2::TokenStream;
use quote::quote;

use crate::derive::{
    attributes::ContainerAttributes,
    data::{Field, FieldNamed, FieldUnnamed, StructData},
};

pub fn impl_into_arma(attributes: &ContainerAttributes, data: &StructData) -> TokenStream {
    // For simplicity sake we assume that theres no conflicts and everything has already been validated
    match &data {
        StructData::Map(fields) => map_struct(attributes, fields),
        StructData::Tuple(fields) => tuple_struct(attributes, fields),
        StructData::NewType(field) => newtype_struct(attributes, &**field),
    }
}

fn map_struct(attributes: &ContainerAttributes, fields: &[FieldNamed]) -> TokenStream {
    if *attributes.transparent.value() {
        return newtype_struct(attributes, fields.first().unwrap());
    }

    let field_bodies = fields.iter().map(|field| {
        let (ident, name) = (&field.ident, &field.name);

        let (key, value) = if *field.attributes.to_string.value() {
            (quote!(#name.to_string()), quote!(self.#ident.to_string()))
        } else {
            (quote!(#name.to_string()), quote!(self.#ident))
        };

        quote!((#key, arma_rs::IntoArma::to_arma(&#value)))
    });

    quote! {
        std::collections::HashMap::<String, arma_rs::Value>::from([
            #(#field_bodies),*
        ]).to_arma()
    }
}

fn tuple_struct(_attributes: &ContainerAttributes, fields: &[FieldUnnamed]) -> TokenStream {
    let field_bodies = fields.iter().map(|field| {
        let index = &field.index;

        if *field.attributes.to_string.value() {
            quote!(self.#index.to_string())
        } else {
            quote!(self.#index)
        }
    });

    quote! {
        Vec::<arma_rs::Value>::from([
            #(arma_rs::IntoArma::to_arma(&#field_bodies)),*
        ]).to_arma()
    }
}

fn newtype_struct(_attributes: &ContainerAttributes, field: &impl Field) -> TokenStream {
    let token = field.token();

    let field_body = if *field.attributes().to_string.value() {
        quote!(self.#token.to_string())
    } else {
        quote!(self.#token)
    };

    quote! {
        arma_rs::IntoArma::to_arma(&#field_body)
    }
}