AppFlowy/scripts/flowy-tool/src/dart_event/dart_event.rs

use super::event_template::*;

use crate::util::*;
use flowy_ast::{event_ast::*, *};
use syn::Item;
use walkdir::WalkDir;

pub struct DartEventCodeGen {
    pub rust_source: String,
    pub output_dir: String,
}

impl DartEventCodeGen {
    pub fn gen(&self) {
        let event_crates = parse_dart_event_files(self.rust_source.as_ref());
        let event_ast = event_crates
            .iter()
            .map(|event_crate| parse_event_crate(event_crate))
            .flatten()
            .collect::<Vec<_>>();

        let event_render_ctx = ast_to_event_render_ctx(event_ast.as_ref());

        let mut render_result = String::new();
        for (index, render_ctx) in event_render_ctx.into_iter().enumerate() {
            let mut event_template = EventTemplate::new();

            match event_template.render(render_ctx, index) {
                Some(content) => render_result.push_str(content.as_ref()),
                None => {}
            }
        }

        save_content_to_file_with_diff_prompt(
            render_result.as_ref(),
            self.output_dir.as_str(),
            true,
        );
    }
}

pub struct DartEventCrate {
    crate_path: String,
    crate_name: String,
    event_files: Vec<String>,
}

impl DartEventCrate {
    pub fn from_config(config: &CrateConfig) -> Self {
        DartEventCrate {
            crate_path: config.crate_path.clone(),
            crate_name: config.folder_name.clone(),
            event_files: config.flowy_config.event_files.clone(),
        }
    }
}

pub fn parse_dart_event_files(root: &str) -> Vec<DartEventCrate> {
    WalkDir::new(root)
        .into_iter()
        .filter_entry(|e| !is_hidden(e))
        .filter_map(|e| e.ok())
        .filter(|e| is_crate_dir(e))
        .flat_map(|e| parse_crate_config_from(&e))
        .map(|crate_config| DartEventCrate::from_config(&crate_config))
        .collect::<Vec<DartEventCrate>>()
}

pub fn parse_event_crate(event_crate: &DartEventCrate) -> Vec<EventASTContext> {
    event_crate
        .event_files
        .iter()
        .map(|event_file| {
            let file_path = format!("{}/{}", event_crate.crate_path, event_file);
            let file_content = read_file(file_path.as_ref()).unwrap();
            let ast = syn::parse_file(file_content.as_ref()).expect("Unable to parse file");

            ast.items
                .iter()
                .map(|item| match item {
                    Item::Enum(item_enum) => {
                        let ctxt = Ctxt::new();
                        let attrs = flowy_ast::enum_from_ast(&ctxt, &item_enum.variants);
                        ctxt.check().unwrap();
                        attrs
                            .iter()
                            .filter(|attr| attr.attrs.event_attrs.ignore == false)
                            .enumerate()
                            .map(|(_index, attr)| EventASTContext::from(&attr.attrs))
                            .collect::<Vec<_>>()
                    }
                    _ => vec![],
                })
                .flatten()
                .collect::<Vec<_>>()
        })
        .flatten()
        .collect::<Vec<EventASTContext>>()
}

pub fn ast_to_event_render_ctx(ast: &Vec<EventASTContext>) -> Vec<EventRenderContext> {
    ast.iter()
        .filter(|event_ast| event_ast.event_input.is_some() && event_ast.event_output.is_some())
        .map(|event_ast| EventRenderContext {
            input_deserializer: event_ast
                .event_input
                .as_ref()
                .unwrap()
                .get_ident()
                .unwrap()
                .to_string(),
            output_deserializer: event_ast
                .event_output
                .as_ref()
                .unwrap()
                .get_ident()
                .unwrap()
                .to_string(),
            event: event_ast.event.to_string(),
            event_ty: event_ast.event_ty.to_string(),
        })
        .collect::<Vec<EventRenderContext>>()
}