gauntlet/rust/client/build.rs

use std::env;
use std::fs::File;
use std::io::Write;
use std::path::Path;

use convert_case::{Case, Casing};

use component_model::{Children, Component, ComponentName, create_component_model, Property, PropertyType, SharedType};

fn main() -> anyhow::Result<()> {
    let out_dir = env::var("OUT_DIR")?;
    let dest_path = Path::new(&out_dir).join("components.rs");

    let mut output = String::new();

    let components = create_component_model();

    output.push_str("#[derive(Debug)]\n");
    output.push_str("enum ComponentWidget {\n");

    for component in &components {
        match component {
            Component::Standard { name, props, children, .. } => {
                output.push_str(&format!("    {}", name));

                let has_children = !matches!(children, Children::None);

                if !props.is_empty() || has_children {
                    output.push_str(" {\n");

                    if has_children {
                        let string = match children {
                            Children::StringOrMembers { .. } => "Vec<ComponentWidgetWrapper>".to_owned(),
                            Children::Members { .. } => "Vec<ComponentWidgetWrapper>".to_owned(),
                            Children::String { .. } => "Vec<ComponentWidgetWrapper>".to_owned(),
                            Children::None => panic!("cannot create type for Children::None")
                        };

                        output.push_str(&format!("        children: {},\n", string));
                    }

                    for prop in props {
                        match prop.property_type {
                            PropertyType::Function { .. } => {
                                // client know about functions in properties
                            }
                            PropertyType::Component { .. } => {
                                // component properties are found in children array
                            }
                            _ => {
                                output.push_str(&format!("        {}: {},\n", prop.name, generate_type(&prop, name)));
                            }
                        }
                    }
                    output.push_str("    },\n");
                } else {
                    output.push_str(",\n");
                }
            }
            Component::Root { .. } => {
                output.push_str("    Root {\n");
                output.push_str("        children: Vec<ComponentWidgetWrapper>,\n");
                output.push_str("    },\n");
            }
            Component::TextPart { .. } => {
                output.push_str("    TextPart {\n");
                output.push_str("        value: String,\n");
                output.push_str("    },\n");
            }
        }
    }

    output.push_str("}\n");
    output.push_str("\n");

    for component in &components {
        match component {
            Component::Standard { name: component_name, props, .. } => {
                for prop in props {
                    match &prop.property_type {
                        PropertyType::Union { items } => {
                            output.push_str("#[derive(Debug)]\n");
                            output.push_str(&format!("enum {}{} {{\n", component_name, prop.name.to_case(Case::Pascal)));

                            for (index, property_type) in items.iter().enumerate() {
                                match property_type {
                                    PropertyType::Union { .. } => panic!("nested union should not be used"),
                                    _ => {
                                        output.push_str(&format!("    _{}({}),\n", index, generate_required_type(&property_type, "SHOULD NOT BE USED".to_owned())));
                                    }
                                }
                            }

                            output.push_str("}\n");
                            output.push_str("\n");
                        }
                        _ => {}
                    }
                }
            }
            Component::Root { shared_types, .. } => {
                for (type_name, shared_type) in shared_types {
                    match shared_type {
                        SharedType::Enum { items } => {
                            output.push_str("#[derive(Debug, strum::EnumString)]\n");
                            output.push_str(&format!("enum {} {{\n", type_name));

                            for item in items {
                                output.push_str(&format!("    {},\n", &item));
                            }

                            output.push_str("}\n");
                            output.push_str("\n");
                        }
                        SharedType::Object { items } => {
                            output.push_str("#[derive(Debug)]\n");
                            output.push_str(&format!("struct {} {{\n", type_name));

                            for (property_name, property_type) in items {
                                output.push_str(&format!("    {}: {},\n", &property_name, generate_required_type(&property_type, format!("{}{}", type_name, property_name))));
                            }

                            output.push_str("}\n");
                            output.push_str("\n");
                        }
                    }
                }
            }
            Component::TextPart { .. } => {}
        }
    }


    output.push_str("fn create_component_widget(component_internal_name: &str, properties: HashMap<String, UiPropertyValue>, children: Vec<ComponentWidgetWrapper>) -> anyhow::Result<ComponentWidget> {\n");
    output.push_str("   let widget = match component_internal_name {\n");

    for component in &components {
        match component {
            Component::Standard { internal_name, name, props, children, .. } => {
                output.push_str(&format!("        \"gauntlet:{}\" => ComponentWidget::{}", internal_name, name));

                let has_children = !matches!(children, Children::None);

                if !props.is_empty() || has_children {
                    output.push_str(&" {\n");

                    if has_children {
                        output.push_str("            children,\n");
                    }

                    for prop in props {
                        match prop.property_type {
                            PropertyType::String => {
                                if prop.optional {
                                    output.push_str(&format!("            {}: parse_optional_string(&properties, \"{}\")?,\n", prop.name, prop.name));
                                } else {
                                    output.push_str(&format!("            {}: parse_string(&properties, \"{}\")?,\n", prop.name, prop.name));
                                }
                            },
                            PropertyType::Number => {
                                if prop.optional {
                                    output.push_str(&format!("            {}: parse_optional_number(&properties, \"{}\")?,\n", prop.name, prop.name));
                                } else {
                                    output.push_str(&format!("            {}: parse_number(&properties, \"{}\")?,\n", prop.name, prop.name));
                                }
                            },
                            PropertyType::Boolean => {
                                if prop.optional {
                                    output.push_str(&format!("            {}: parse_optional_boolean(&properties, \"{}\")?,\n", prop.name, prop.name));
                                } else {
                                    output.push_str(&format!("            {}: parse_boolean(&properties, \"{}\")?,\n", prop.name, prop.name));
                                }
                            },
                            PropertyType::Function { .. } => {
                                // client know about functions in properties
                            }
                            PropertyType::Component { .. } => {
                                // component properties are found in a children array
                            }
                            PropertyType::ImageSource => {
                                if prop.optional {
                                    output.push_str(&format!("            {}: parse_bytes_optional(&properties, \"{}\")?,\n", prop.name, prop.name));
                                } else {
                                    output.push_str(&format!("            {}: parse_bytes(&properties, \"{}\")?,\n", prop.name, prop.name));
                                }
                            }
                            PropertyType::Enum { .. } => {
                                if prop.optional {
                                    output.push_str(&format!("            {}: parse_enum_optional(&properties, \"{}\")?,\n", prop.name, prop.name));
                                } else {
                                    output.push_str(&format!("            {}: parse_enum(&properties, \"{}\")?,\n", prop.name, prop.name));
                                }
                            }
                            PropertyType::Union { .. } => {
                                if prop.optional {
                                    output.push_str(&format!("            {}: parse_union_optional(&properties, \"{}\")?,\n", prop.name, prop.name));
                                } else {
                                    output.push_str(&format!("            {}: parse_union(&properties, \"{}\")?,\n", prop.name, prop.name));
                                }
                            }
                            PropertyType::Object { .. } => {
                                if prop.optional {
                                    output.push_str(&format!("            {}: parse_object_optional(&properties, \"{}\")?,\n", prop.name, prop.name));
                                } else {
                                    output.push_str(&format!("            {}: parse_object(&properties, \"{}\")?,\n", prop.name, prop.name));
                                }
                            }
                        };
                    }
                    output.push_str("        },\n");
                } else {
                    output.push_str(",\n");
                }
            }
            Component::Root { .. } => {}
            Component::TextPart { internal_name, props } => {
                let name = match &props[..] {
                    [Property { property_type: PropertyType::String, optional: false, name, .. }] => name,
                    _ => panic!("text_part should have single string not optional prop")
                };
                output.push_str(&format!("        \"gauntlet:{}\" => ComponentWidget::TextPart {{\n", internal_name));
                output.push_str(&format!("            {}: parse_string(&properties, \"{}\")?,\n", name, name));
                output.push_str("        },\n");
            }
        }
    }

    output.push_str("        _ => Err(anyhow::anyhow!(\"cannot create {} type\", component_internal_name))?\n");
    output.push_str("    };\n");
    output.push_str("    Ok(widget)\n");
    output.push_str("}\n");
    output.push_str("\n");


    output.push_str("fn append_component_widget_child(parent: &ComponentWidgetWrapper, child: ComponentWidgetWrapper) -> anyhow::Result<()> {\n");
    output.push_str("    let (ref mut parent, _) = &mut *parent.get_mut();\n");
    output.push_str("    match parent {\n");

    for component in &components {
        match component {
            Component::Standard { name, children, .. } => {
                let has_children = !matches!(children, Children::None);

                if has_children {
                    output.push_str(&format!("        ComponentWidget::{} {{ ref mut children, .. }} => {{\n", name));
                    output.push_str("            match get_component_widget_type(&child) {\n");

                    match children {
                        Children::StringOrMembers { members, text_part_internal_name } => {
                            output.push_str(&format!("                (\"gauntlet:{}\", _) => (),\n", text_part_internal_name));
                            for (_, member) in members {
                                output.push_str(&format!("                (\"gauntlet:{}\", _) => (),\n", member.component_internal_name));
                            }
                        }
                        Children::Members { members } => {
                            for (_, member) in members {
                                output.push_str(&format!("                (\"gauntlet:{}\", _) => (),\n", member.component_internal_name));
                            }
                        }
                        Children::String { text_part_internal_name, .. } => {
                            output.push_str(&format!("                (\"gauntlet:{}\", _) => (),\n", text_part_internal_name));
                        }
                        Children::None => {}
                    }

                    output.push_str(&format!("                (_, name) => Err(anyhow::anyhow!(\"{} cannot have {{}} child\", name))?\n", name));
                    output.push_str("            };\n");
                    output.push_str("            children.push(child)\n");
                    output.push_str("        }\n");
                } else {
                    output.push_str(&format!("        ComponentWidget::{} {{ .. }} => {{\n", name));
                    output.push_str(&format!("            Err(anyhow::anyhow!(\"{} cannot have children\"))?\n", name));
                    output.push_str("        }\n");
                }
            }
            Component::Root { children, .. } => {
                output.push_str("        ComponentWidget::Root { ref mut children, .. } => {\n");
                output.push_str("            match get_component_widget_type(&child) {\n");

                for child in children {
                    output.push_str(&format!("                (\"gauntlet:{}\", _) => (),\n", child.component_internal_name));
                }

                output.push_str("                (_, name) => Err(anyhow::anyhow!(\"root cannot have {} child\", name))?\n");
                output.push_str("            };\n");
                output.push_str("            children.push(child)\n");
                output.push_str("        }\n");

            }
            Component::TextPart { .. } => {
                output.push_str("        ComponentWidget::TextPart { .. } => {\n");
                output.push_str("           Err(anyhow::anyhow!(\"text_part cannot have children\"))?\n");
                output.push_str("        }\n");
            }
        }
    }

    output.push_str("    };\n");
    output.push_str("    Ok(())\n");
    output.push_str("}\n");
    output.push_str("\n");


    output.push_str("fn get_component_widget_children(widget: &ComponentWidgetWrapper) -> anyhow::Result<Vec<ComponentWidgetWrapper>> {\n");
    output.push_str("    let (widget, _) = &*widget.get();\n");
    output.push_str("    let children = match widget {\n");

    for component in &components {
        match component {
            Component::Standard { name, children, .. } => {
                let has_children = !matches!(children, Children::None);
                if has_children {
                    output.push_str(&format!("        ComponentWidget::{} {{ ref children, .. }} => {{\n", name));
                    output.push_str("            children\n");
                    output.push_str("        }\n");
                } else {
                    output.push_str(&format!("        ComponentWidget::{} {{ .. }} => {{\n", name));
                    output.push_str(&format!("            Err(anyhow::anyhow!(\"{} cannot have children\"))?\n", name));
                    output.push_str("        }\n");
                }
            }
            Component::Root { .. } => {
                output.push_str("        ComponentWidget::Root { ref children, .. } => {\n");
                output.push_str("            children\n");
                output.push_str("        }\n");
            }
            Component::TextPart { .. } => {
                output.push_str("        ComponentWidget::TextPart { .. } => {\n");
                output.push_str("            Err(anyhow::anyhow!(\"text part cannot have children\"))?\n");
                output.push_str("        }\n");
            }
        }
    }

    output.push_str("    };\n");
    output.push_str("    Ok(children.iter().cloned().collect())\n");
    output.push_str("}\n");
    output.push_str("\n");


    output.push_str("fn set_component_widget_children(widget: &ComponentWidgetWrapper, new_children: Vec<ComponentWidgetWrapper>) -> anyhow::Result<()> {\n");
    output.push_str("    let (ref mut widget, _) = &mut *widget.get_mut();\n");
    output.push_str("    match widget {\n");

    for component in &components {
        match component {
            Component::Standard { name, children, .. } => {
                let has_children = !matches!(children, Children::None);

                if has_children {
                    output.push_str(&format!("        ComponentWidget::{} {{ ref mut children, .. }} => {{\n", name));
                    output.push_str("            for new_child in &new_children {\n");
                    output.push_str("                match get_component_widget_type(new_child) {\n");

                    match children {
                        Children::StringOrMembers { members, text_part_internal_name } => {
                            output.push_str(&format!("                    (\"gauntlet:{}\", _) => (),\n", text_part_internal_name));
                            for (_, member) in members {
                                output.push_str(&format!("                    (\"gauntlet:{}\", _) => (),\n", member.component_internal_name));
                            }
                        }
                        Children::Members { members } => {
                            for (_, member) in members {
                                output.push_str(&format!("                    (\"gauntlet:{}\", _) => (),\n", member.component_internal_name));
                            }
                        }
                        Children::String { text_part_internal_name, .. } => {
                            output.push_str(&format!("                    (\"gauntlet:{}\", _) => (),\n", text_part_internal_name));
                        }
                        Children::None => {}
                    }

                    output.push_str(&format!("                    (_, name) => Err(anyhow::anyhow!(\"{} cannot have {{}} child\", name))?\n", name));
                    output.push_str("                };\n");
                    output.push_str("            }\n");
                    output.push_str("            *children = new_children\n");
                    output.push_str("        }\n");
                } else {
                    output.push_str(&format!("        ComponentWidget::{} {{ .. }} => {{\n", name));
                    output.push_str(&format!("            Err(anyhow::anyhow!(\"{} cannot have children\"))?\n", name));
                    output.push_str("        }\n");
                }
            }
            Component::Root { children, .. } => {
                output.push_str("        ComponentWidget::Root { ref mut children, .. } => {\n");
                output.push_str("            for new_child in &new_children {\n");
                output.push_str("                match get_component_widget_type(new_child) {\n");

                for child in children {
                    output.push_str(&format!("                    (\"gauntlet:{}\", _) => (),\n", child.component_internal_name));
                }

                output.push_str("                    (_, name) => Err(anyhow::anyhow!(\"root cannot have {} child\", name))?\n");
                output.push_str("                };\n");
                output.push_str("            }\n");
                output.push_str("            *children = new_children\n");
                output.push_str("        }\n");
            }
            Component::TextPart { .. } => {
                output.push_str("        ComponentWidget::TextPart { .. } => {\n");
                output.push_str("           Err(anyhow::anyhow!(\"text_part cannot have children\"))?\n");
                output.push_str("        }\n");
            }
        }
    }

    output.push_str("    };\n");
    output.push_str("    Ok(())\n");
    output.push_str("}\n");
    output.push_str("\n");


    output.push_str("fn get_component_widget_type(widget: &ComponentWidgetWrapper) -> (&str, &str) {\n");
    output.push_str("    let (widget, _) = &*widget.get();\n");
    output.push_str("    match widget {\n");

    for component in &components {
        match component {
            Component::Standard { name, internal_name, .. } => {
                output.push_str(&format!("        ComponentWidget::{} {{ .. }} => (\"gauntlet:{}\", \"{}\"),\n", name, internal_name, name));
            }
            Component::Root { internal_name, .. } => {
                output.push_str(&format!("        ComponentWidget::Root {{ .. }} => (\"gauntlet:{}\", \"Root\"),\n", internal_name));
            }
            Component::TextPart { internal_name, .. } => {
                output.push_str(&format!("        ComponentWidget::TextPart {{ .. }} => (\"gauntlet:{}\", \"TextPart\"),\n", internal_name));
            }
        }
    }

    output.push_str("    }\n");
    output.push_str("}\n");
    output.push_str("\n");

    for component in &components {
        match component {
            Component::Standard { name, props, .. } => {
                for prop in props {
                    let PropertyType::Function { arguments } = &prop.property_type else {
                        continue
                    };

                    output.push_str(&format!("fn create_{}_{}_event(\n", name.to_string().to_case(Case::Snake), prop.name.to_case(Case::Snake)));
                    output.push_str("    widget_id: UiWidgetId,\n");

                    for arg in arguments {
                        output.push_str(&format!("    {}: {}\n", arg.name, generate_type(&arg, name)));
                    }

                    output.push_str(") -> crate::model::UiViewEvent {\n");
                    output.push_str("    crate::model::UiViewEvent::View {\n");
                    output.push_str("        widget_id,\n");
                    output.push_str(&format!("        event_name: \"{}\".to_owned(),\n", prop.name));
                    output.push_str("        event_arguments: vec![\n",);

                    for arg in arguments {
                        match arg.property_type {
                            PropertyType::String => {
                                if arg.optional {
                                    output.push_str(&format!("            {}.map(|{}| common::model::UiPropertyValue::String({})).unwrap_or_else(|| common::model::UiPropertyValue::Undefined),\n", arg.name, arg.name, arg.name));
                                } else {
                                    output.push_str(&format!("            common::model::UiPropertyValue::String({}),\n", arg.name));
                                }
                            }
                            PropertyType::Number => {
                                if arg.optional {
                                    output.push_str(&format!("            {}.map(|{}| common::model::UiPropertyValue::Number({})).unwrap_or_else(|| common::model::UiPropertyValue::Undefined),\n", arg.name, arg.name, arg.name));
                                } else {
                                    output.push_str(&format!("            common::model::UiPropertyValue::Number({}),\n", arg.name));
                                }
                            }
                            PropertyType::Boolean => {
                                if arg.optional {
                                    output.push_str(&format!("            {}.map(|{}| common::model::UiPropertyValue::Bool({})).unwrap_or_else(|| common::model::UiPropertyValue::Undefined),\n", arg.name, arg.name, arg.name));
                                } else {
                                    output.push_str(&format!("            common::model::UiPropertyValue::Bool({}),\n", arg.name));
                                }
                            }
                            _ => {
                                panic!("not yet supported")
                            }
                        }
                    }

                    output.push_str("        ]\n");
                    output.push_str("    }\n");
                    output.push_str("}\n");
                    output.push_str("\n");
                }
            }
            _ => {}
        }
    }

    generate_file(&dest_path, &output)?;

    Ok(())
}

fn generate_file<P: AsRef<Path>>(path: P, text: &str) -> std::io::Result<()> {
    let mut f = File::create(path)?;
    f.write_all(text.as_bytes())
}

fn generate_type(property: &Property, name: &ComponentName) -> String {
    match property.optional {
        true => generate_optional_type(&property.property_type, format!("{}{}", name, &property.name.to_case(Case::Pascal))),
        false => generate_required_type(&property.property_type, format!("{}{}", name, &property.name.to_case(Case::Pascal)))
    }
}

fn generate_optional_type(property_type: &PropertyType, union_name: String) -> String {
    format!("Option<{}>", generate_required_type(property_type, union_name))
}

fn generate_required_type(property_type: &PropertyType, union_name: String) -> String {
    match property_type {
        PropertyType::String => "String".to_owned(),
        PropertyType::Number => "f64".to_owned(),
        PropertyType::Boolean => "bool".to_owned(),
        PropertyType::Function { .. } => panic!("client doesn't know about functions in properties"),
        PropertyType::Component { .. } => panic!("component properties are found in children array"),
        PropertyType::ImageSource => "Vec<u8>".to_owned(),
        PropertyType::Union { .. } => union_name,
        PropertyType::Object { name } => name.to_owned(),
        PropertyType::Enum { name } => name.to_owned()
    }
}