use super::substitution_map::SubstitutionsMap;

pub fn substitutions() -> SubstitutionsMap {
    let mut substitutions = SubstitutionsMap::new();

    add_managed_types(&mut substitutions);
    add_storage_mappers(&mut substitutions);

    substitutions
}

#[rustfmt::skip]
fn add_managed_type(substitutions: &mut SubstitutionsMap, type_name: &proc_macro2::TokenStream) {
    substitutions.add_substitution(
        quote!(#type_name<Self::Api>), 
        quote!(#type_name<Self::Api>));
    substitutions.add_substitution(
        quote!(#type_name::),
        quote!(multiversx_sc::types::#type_name::<Self::Api>::),
    );
    substitutions.add_substitution(
        quote!(#type_name),
        quote!(multiversx_sc::types::#type_name<Self::Api>),
    );
}

fn add_managed_type_with_generics(
    substitutions: &mut SubstitutionsMap,
    type_name: &proc_macro2::TokenStream,
) {
    substitutions.add_substitution(
        quote!(#type_name<Self::Api, ),
        quote!(#type_name<Self::Api, ),
    );
    substitutions.add_substitution(quote!(#type_name<), quote!(#type_name<Self::Api, ));
}

fn add_managed_types(substitutions: &mut SubstitutionsMap) {
    // types::io
    add_managed_type(substitutions, &quote!(ManagedSCError));

    // types::interaction
    add_managed_type(substitutions, &quote!(AsyncCall));

    // types::managed::basic
    add_managed_type(substitutions, &quote!(BigFloat));
    add_managed_type(substitutions, &quote!(BigInt));
    add_managed_type(substitutions, &quote!(BigUint));
    add_managed_type(substitutions, &quote!(EllipticCurveComponents));
    add_managed_type(substitutions, &quote!(EllipticCurve));
    add_managed_type(substitutions, &quote!(ManagedBuffer));
    add_managed_type(substitutions, &quote!(ManagedMap));

    // types::managed::multi-value
    add_managed_type(substitutions, &quote!(ManagedAsyncCallError));
    add_managed_type_with_generics(substitutions, &quote!(ManagedAsyncCallResult));
    add_managed_type(substitutions, &quote!(EsdtTokenPaymentMultiArg));
    add_managed_type(substitutions, &quote!(EsdtTokenPaymentMultiValue));
    add_managed_type_with_generics(substitutions, &quote!(MultiValueEncodedIterator));
    add_managed_type_with_generics(substitutions, &quote!(MultiValueEncoded));
    add_managed_type_with_generics(substitutions, &quote!(ManagedVarArgs));
    add_managed_type_with_generics(substitutions, &quote!(ManagedMultiResultVec));
    add_managed_type_with_generics(substitutions, &quote!(MultiValueManagedVecCounted));
    add_managed_type_with_generics(substitutions, &quote!(ManagedCountedVarArgs));
    add_managed_type_with_generics(substitutions, &quote!(ManagedCountedMultiResultVec));
    add_managed_type_with_generics(substitutions, &quote!(MultiValueManagedVec));

    // types::managed::wrapped
    add_managed_type(substitutions, &quote!(EgldOrEsdtTokenIdentifier));
    add_managed_type(substitutions, &quote!(EgldOrEsdtTokenPayment));
    add_managed_type(substitutions, &quote!(EsdtTokenData));
    add_managed_type(substitutions, &quote!(EsdtTokenPayment));
    add_managed_type(substitutions, &quote!(ManagedAddress));
    add_managed_type(substitutions, &quote!(ManagedBufferBuilder));
    add_managed_type_with_generics(substitutions, &quote!(ManagedByteArray));
    add_managed_type_with_generics(substitutions, &quote!(ManagedOption));
    add_managed_type_with_generics(substitutions, &quote!(ManagedRef));
    add_managed_type_with_generics(substitutions, &quote!(ManagedVecOwnedIterator));
    add_managed_type_with_generics(substitutions, &quote!(ManagedVecRefIterator));
    add_managed_type_with_generics(substitutions, &quote!(ManagedVecRef));
    add_managed_type_with_generics(substitutions, &quote!(ManagedVec));
    add_managed_type_with_generics(substitutions, &quote!(PreloadedManagedBuffer));
    add_managed_type(substitutions, &quote!(RandomnessSource));
    add_managed_type(substitutions, &quote!(TokenIdentifier));
    add_managed_type(substitutions, &quote!(FunctionCall));
}

fn add_storage_mapper_single_generic_arg(
    substitutions: &mut SubstitutionsMap,
    mapper_name: &proc_macro2::TokenStream,
) {
    substitutions.add_substitution(
        quote!(#mapper_name<Self::Api>),
        quote!(#mapper_name<Self::Api>),
    );
    substitutions.add_substitution(quote!(#mapper_name), quote!(#mapper_name<Self::Api>));
}

fn add_storage_mapper(
    substitutions: &mut SubstitutionsMap,
    mapper_name: &proc_macro2::TokenStream,
) {
    add_managed_type_with_generics(substitutions, mapper_name);
}

fn add_storage_mappers(substitutions: &mut SubstitutionsMap) {
    add_storage_mapper_single_generic_arg(substitutions, &quote!(FungibleTokenMapper));
    add_storage_mapper_single_generic_arg(substitutions, &quote!(NonFungibleTokenMapper));
    add_storage_mapper_single_generic_arg(substitutions, &quote!(TokenAttributesMapper));
    add_storage_mapper_single_generic_arg(substitutions, &quote!(UniqueIdMapper));
    add_storage_mapper_single_generic_arg(substitutions, &quote!(UserMapper));
    add_storage_mapper_single_generic_arg(substitutions, &quote!(AddressToIdMapper));

    add_storage_mapper(substitutions, &quote!(BiDiMapper));
    add_storage_mapper(substitutions, &quote!(LinkedListMapper));
    add_storage_mapper(substitutions, &quote!(MapMapper));
    add_storage_mapper(substitutions, &quote!(MapStorageMapper));
    add_storage_mapper(substitutions, &quote!(SetMapper));
    add_storage_mapper(substitutions, &quote!(UnorderedSetMapper));
    add_storage_mapper(substitutions, &quote!(SingleValueMapper));
    add_storage_mapper(substitutions, &quote!(VecMapper));
    add_storage_mapper(substitutions, &quote!(QueueMapper));
    add_storage_mapper(substitutions, &quote!(WhitelistMapper));
}
