AppFlowy/frontend/rust-lib/flowy-codegen/src/ast.rs

use flowy_ast::EventEnumAttrs;
use quote::format_ident;

#[allow(dead_code)]
pub struct EventASTContext {
    pub event: syn::Ident,
    pub event_ty: syn::Ident,
    pub event_request_struct: syn::Ident,
    pub event_input: Option<syn::Path>,
    pub event_output: Option<syn::Path>,
    pub event_error: String,
}

impl EventASTContext {
    #[allow(dead_code)]
    pub fn from(enum_attrs: &EventEnumAttrs) -> EventASTContext {
        let command_name = enum_attrs.enum_item_name.clone();
        if command_name.is_empty() {
            panic!("Invalid command name: {}", enum_attrs.enum_item_name);
        }

        let event = format_ident!("{}", &command_name);
        let splits = command_name.split('_').collect::<Vec<&str>>();

        let event_ty = format_ident!("{}", enum_attrs.enum_name);
        let event_request_struct = format_ident!("{}Event", &splits.join(""));

        let event_input = enum_attrs.event_input();
        let event_output = enum_attrs.event_output();
        let event_error = enum_attrs.event_error();

        EventASTContext {
            event,
            event_ty,
            event_request_struct,
            event_input,
            event_output,
            event_error,
        }
    }
}