breez-sdk-liquid/lib/bindings/langs/react-native/src/gen_swift/mod.rs

use std::collections::HashSet;

use askama::Template;
use once_cell::sync::Lazy;
use uniffi_bindgen::interface::*;

use crate::generator::RNConfig;

pub use uniffi_bindgen::bindings::swift::gen_swift::*;

static IGNORED_FUNCTIONS: Lazy<HashSet<String>> = Lazy::new(|| {
    let list: Vec<&str> = vec!["connect", "add_event_listener"];
    HashSet::from_iter(list.into_iter().map(|s| s.to_string()))
});

#[derive(Template)]
#[template(syntax = "rn", escape = "none", path = "mapper.swift")]
#[allow(dead_code)]
pub struct MapperGenerator<'a> {
    config: RNConfig,
    ci: &'a ComponentInterface,
}

impl<'a> MapperGenerator<'a> {
    pub fn new(config: RNConfig, ci: &'a ComponentInterface) -> Self {
        Self { config, ci }
    }
}

#[derive(Template)]
#[template(syntax = "rn", escape = "none", path = "extern.m")]
#[allow(dead_code)]
pub struct ExternGenerator<'a> {
    config: RNConfig,
    ci: &'a ComponentInterface,
}

impl<'a> ExternGenerator<'a> {
    pub fn new(config: RNConfig, ci: &'a ComponentInterface) -> Self {
        Self { config, ci }
    }
}

#[derive(Template)]
#[template(syntax = "rn", escape = "none", path = "module.swift")]
#[allow(dead_code)]
pub struct ModuleGenerator<'a> {
    config: RNConfig,
    ci: &'a ComponentInterface,
}

impl<'a> ModuleGenerator<'a> {
    pub fn new(config: RNConfig, ci: &'a ComponentInterface) -> Self {
        Self { config, ci }
    }
}

pub mod filters {

    use heck::*;
    use uniffi_bindgen::backend::CodeOracle;
    use uniffi_bindgen::backend::{CodeType, TypeIdentifier};

    use super::*;

    fn oracle() -> &'static SwiftCodeOracle {
        &SwiftCodeOracle
    }

    pub fn type_name(codetype: &impl CodeType) -> Result<String, askama::Error> {
        Ok(codetype.type_label(oracle()))
    }

    pub fn fn_name(nm: &str) -> Result<String, askama::Error> {
        Ok(oracle().fn_name(nm))
    }

    pub fn render_to_map(
        t: &TypeIdentifier,
        ci: &ComponentInterface,
        obj_name: &str,
        field_name: &str,
        optional: bool,
    ) -> Result<String, askama::Error> {
        let type_name = filters::type_name(t)?;
        let type_name_str = type_name.as_str();
        let var_name = filters::unquote(filters::var_name(type_name_str)?.as_str())?;
        let mut obj_prefix = "".to_string();
        if !obj_name.is_empty() {
            obj_prefix = format!("{obj_name}.");
        }
        let mut optional_suffix = "";
        if optional {
            optional_suffix = "!";
        }
        let res: Result<String, askama::Error> = match t {
            Type::UInt8 => Ok(format!("{obj_prefix}{field_name}")),
            Type::Int8 => Ok(format!("{obj_prefix}{field_name}")),
            Type::UInt16 => Ok(format!("{obj_prefix}{field_name}")),
            Type::Int16 => Ok(format!("{obj_prefix}{field_name}")),
            Type::UInt32 => Ok(format!("{obj_prefix}{field_name}")),
            Type::Int32 => Ok(format!("{obj_prefix}{field_name}")),
            Type::UInt64 => Ok(format!("{obj_prefix}{field_name}")),
            Type::Int64 => Ok(format!("{obj_prefix}{field_name}")),
            Type::Float32 => Ok(format!("{obj_prefix}{field_name}")),
            Type::Float64 => Ok(format!("{obj_prefix}{field_name}")),
            Type::Boolean => Ok(format!("{obj_prefix}{field_name}")),
            Type::String => Ok(format!("{obj_prefix}{field_name}")),
            Type::Timestamp => unimplemented!("render_to_map: Timestamp is not implemented"),
            Type::Duration => unimplemented!("render_to_map: Duration is not implemented"),
            Type::Object(_) => unimplemented!("render_to_map: Object is not implemented"),
            Type::Record(_) => Ok(format!(
                "dictionaryOf({var_name}: {obj_prefix}{field_name}{optional_suffix})"
            )),
            Type::Enum(inner) => {
                let enum_def = ci.get_enum_definition(inner).unwrap();
                match enum_def.is_flat() {
                    true => Ok(format!(
                        "valueOf( {var_name}: {obj_prefix}{field_name}{optional_suffix})"
                    )),
                    false => Ok(format!(
                        "dictionaryOf({var_name}: {obj_prefix}{field_name}{optional_suffix})"
                    )),
                }
            }
            Type::Error(_) => unimplemented!("render_to_map: Error is not implemented"),
            Type::CallbackInterface(_) => {
                unimplemented!("render_to_map: CallbackInterface is not implemented")
            }
            Type::Optional(inner) => {
                let unboxed = inner.as_ref();
                let inner_render = render_to_map(unboxed, ci, obj_name, field_name, true)?;
                Ok(format!(
                    "{obj_prefix}{field_name} == nil ? nil : {inner_render}"
                ))
            }
            Type::Sequence(inner) => {
                let unboxed = inner.as_ref();
                let type_name = filters::type_name(unboxed)?;
                let var_name = filters::var_name(type_name.as_str())?;
                let var_name = filters::unquote(var_name.as_str())?;
                let as_array_statment = match unboxed {
                    Type::Record(_) => format!(
                        "arrayOf({var_name}List: {obj_prefix}{field_name}{optional_suffix})"
                    ),
                    Type::Enum(_) => format!(
                        "arrayOf({var_name}List: {obj_prefix}{field_name}{optional_suffix})"
                    ),
                    _ => format!("{obj_prefix}{field_name}"),
                };
                Ok(as_array_statment)
            }
            Type::Map(_, _) => unimplemented!("render_to_map: Map is not implemented"),
            Type::External { .. } => {
                unimplemented!("render_to_map: External is not implemented")
            }
            Type::Custom { .. } => {
                unimplemented!("render_to_map: Custom is not implemented")
            }
            Type::Unresolved { .. } => {
                unimplemented!("render_to_map: Unresolved is not implemented")
            }
        };
        res
    }

    pub fn rn_convert_type(
        t: &TypeIdentifier,
        converted_var_name: &str,
    ) -> Result<String, askama::Error> {
        match t {
            Type::Optional(inner) => {
                let unboxed = inner.as_ref();
                let optional = match *unboxed {
                    Type::Int8
                    | Type::UInt8
                    | Type::Int16
                    | Type::UInt16
                    | Type::Int32
                    | Type::UInt32
                    | Type::Int64
                    | Type::UInt64 => {
                        format!("{} == 0 ? nil : {}", converted_var_name, converted_var_name)
                    }
                    Type::Float32 | Type::Float64 => format!(
                        "{} == 0.0 ? nil : {}",
                        converted_var_name, converted_var_name
                    ),
                    Type::String => format!(
                        "{}.isEmpty ? nil : {}",
                        converted_var_name, converted_var_name
                    ),
                    _ => "".to_string(),
                };
                Ok(optional.to_string())
            }
            _ => Ok(converted_var_name.to_string()),
        }
    }

    pub fn rn_return_type(
        t: &TypeIdentifier,
        name: &str,
        optional: bool,
    ) -> Result<String, askama::Error> {
        let mut optional_suffix = "";
        if optional {
            optional_suffix = "!";
        }
        match t {
            Type::Enum(_) | Type::Record(_) => Ok(format!(
                "BreezLiquidSDKMapper.dictionaryOf({}: res{})",
                name, optional_suffix
            )),
            Type::Sequence(inner) => {
                let unboxed = inner.as_ref();
                match unboxed {
                    Type::Enum(_) | Type::Record(_) => Ok(format!(
                        "BreezLiquidSDKMapper.arrayOf({}List: res{})",
                        name, optional_suffix
                    )),
                    _ => Ok(format!("res{}", optional_suffix)),
                }
            }
            _ => Ok(format!("res{}", optional_suffix)),
        }
    }

    pub fn rn_type_name(
        t: &TypeIdentifier,
        ci: &ComponentInterface,
        optional: bool,
    ) -> Result<String, askama::Error> {
        let mut optional_suffix = "";
        if optional {
            optional_suffix = "?";
        }
        match t {
            Type::Record(_) => Ok(format!("[String: Any{}]", optional_suffix)),
            Type::Enum(inner) => {
                let enum_def = ci.get_enum_definition(inner).unwrap();
                match enum_def.is_flat() {
                    false => Ok(format!("[String: Any{}]", optional_suffix)),
                    true => Ok("String".into()),
                }
            }
            Type::Optional(inner) => {
                let unboxed = inner.as_ref();
                rn_type_name(unboxed, ci, optional)
            }
            Type::Sequence(inner) => {
                let unboxed = inner.as_ref();
                Ok(format!("[{}]", rn_type_name(unboxed, ci, optional)?))
            }
            t => {
                let name = filters::type_name(t)?;
                Ok(name.to_string())
            }
        }
    }

    pub fn extern_type_name(
        t: &TypeIdentifier,
        ci: &ComponentInterface,
    ) -> Result<String, askama::Error> {
        match t {
            Type::Boolean => Ok("BOOL".to_string()),
            Type::Int8 | Type::Int16 | Type::Int32 | Type::Int64 => Ok("NSInteger*".to_string()),
            Type::UInt8 | Type::UInt16 | Type::UInt32 | Type::UInt64 => {
                Ok("NSUInteger*".to_string())
            }
            Type::Float32 | Type::Float64 => Ok("NSNumber*".to_string()),
            Type::String => Ok("NSString*".to_string()),
            Type::Enum(inner) => {
                let enum_def = ci.get_enum_definition(inner).unwrap();
                match enum_def.is_flat() {
                    false => Ok("NSDictionary*".to_string()),
                    true => Ok("NSString*".to_string()),
                }
            }
            Type::Record(_) => Ok("NSDictionary*".to_string()),
            Type::Optional(inner) => {
                let unboxed = inner.as_ref();
                extern_type_name(unboxed, ci)
            }
            Type::Sequence(_) => Ok("NSArray*".to_string()),
            _ => Ok("".to_string()),
        }
    }

    pub fn inline_optional_field(
        t: &TypeIdentifier,
        ci: &ComponentInterface,
    ) -> Result<bool, askama::Error> {
        match t {
            Type::Optional(inner) => {
                let unboxed = inner.as_ref();
                inline_optional_field(unboxed, ci)
            }
            _ => {
                let mapped_name = filters::rn_type_name(t, ci, true)?;
                let type_name = filters::type_name(t)?;
                Ok(mapped_name == type_name)
            }
        }
    }

    pub fn render_from_map(
        t: &TypeIdentifier,
        ci: &ComponentInterface,
        map_var_name: &str,
    ) -> Result<String, askama::Error> {
        let res: String = match t {
            Type::UInt8 => map_var_name.to_string(),
            Type::Int8 => map_var_name.to_string(),
            Type::UInt16 => map_var_name.to_string(),
            Type::Int16 => map_var_name.to_string(),
            Type::UInt32 => map_var_name.to_string(),
            Type::Int32 => map_var_name.to_string(),
            Type::UInt64 => map_var_name.to_string(),
            Type::Int64 => map_var_name.to_string(),
            Type::Float32 => map_var_name.to_string(),
            Type::Float64 => map_var_name.to_string(),
            Type::Boolean => map_var_name.to_string(),
            Type::String => map_var_name.to_string(),
            Type::Timestamp => "".into(),
            Type::Duration => "".into(),
            Type::Object(_) => "".into(),
            Type::Record(_) => {
                let record_type_name = type_name(t)?;
                let record_var_name = var_name(&record_type_name)?;
                let record_unquoted_name = unquote(&record_var_name)?;
                format!("try as{record_type_name}({record_unquoted_name}: {map_var_name})")
            }
            Type::Enum(inner) => {
                let enum_def = ci.get_enum_definition(inner).unwrap();
                let enum_var_name = var_name(inner)?;
                let enum_unquoted_name = unquote(&enum_var_name)?;
                match enum_def.is_flat() {
                    false => format!("try as{inner}({enum_unquoted_name}: {map_var_name})"),
                    true => format!("try as{inner}({enum_unquoted_name}: {map_var_name})"),
                }
            }
            Type::Error(_) => "".into(),
            Type::CallbackInterface(_) => "".into(),
            Type::Optional(inner) => {
                let unboxed = inner.as_ref();

                render_from_map(unboxed, ci, map_var_name)?
            }
            Type::Sequence(inner) => {
                let unboxed = inner.as_ref();
                let element_type_name = type_name(unboxed)?;
                match unboxed {
                    Type::Enum(_) | Type::Record(_) => {
                        format!("try as{element_type_name}List(arr: {map_var_name})")
                    }
                    _ => map_var_name.to_string(),
                }
            }
            Type::Map(_, _) => "".into(),
            Type::External { .. } => "".into(),
            Type::Custom { .. } => "".into(),
            Type::Unresolved { .. } => "".into(),
        };
        Ok(res.to_string())
    }

    pub fn var_name(nm: &str) -> Result<String, askama::Error> {
        Ok(format!("`{}`", nm.to_string().to_lower_camel_case()))
    }

    pub fn unquote(nm: &str) -> Result<String, askama::Error> {
        Ok(nm.trim_matches('`').to_string())
    }

    pub fn ignored_function(nm: &str) -> Result<bool, askama::Error> {
        Ok(IGNORED_FUNCTIONS.contains(nm))
    }

    pub fn list_arg(nm: &str) -> Result<String, askama::Error> {
        Ok(format!("`{nm}List`"))
    }

    pub fn temporary(nm: &str) -> Result<String, askama::Error> {
        Ok(format!("{nm}Tmp"))
    }
}