makepad/old/platform/derive_live/src/generate_cast.rs

use proc_macro::TokenStream;

use makepad_micro_proc_macro::{error, TokenBuilder, TokenParser};

pub fn generate_any_send_trait_api_impl(input: TokenStream) -> TokenStream {
    let mut parser = TokenParser::new(input);
    if let Some(ident) = parser.eat_any_ident() {
        let mut tb = TokenBuilder::new();
        tb.add("impl dyn ").ident(&ident).add(" + Send + Sync{");
        tb.add("    pub fn is<T: ")
            .ident(&ident)
            .add(" + 'static + Send + Sync >(&self) -> bool {");
        tb.add("        let t = std::any::TypeId::of::<T>();");
        tb.add("        let concrete = self.ref_cast_type_id();");
        tb.add("        t == concrete");
        tb.add("    }");
        tb.add("    pub fn downcast_ref<T: ")
            .ident(&ident)
            .add(" + 'static + Send + Sync>(&self) -> Option<&T> {");
        tb.add("        if self.is::<T>() {");
        tb.add("            Some(unsafe {&*(self as *const dyn ")
            .ident(&ident)
            .add(" as *const T)})");
        tb.add("        } else {");
        tb.add("            None");
        tb.add("        }");
        tb.add("    }");
        tb.add("    pub fn downcast_mut<T: ")
            .ident(&ident)
            .add(" + 'static + Send + Sync>(&mut self) -> Option<&mut T> {");
        tb.add("        if self.is::<T>() {");
        tb.add("            Some(unsafe {&mut *(self as *const dyn ")
            .ident(&ident)
            .add(" as *mut T)})");
        tb.add("        } else {");
        tb.add("            None");
        tb.add("        }");
        tb.add("    }");
        tb.add("}");
        tb.end()
    } else {
        error("Expected identifier")
    }
}

pub fn generate_any_trait_api_impl(input: TokenStream) -> TokenStream {
    let mut parser = TokenParser::new(input);
    if let Some(ident) = parser.eat_any_ident() {
        let mut tb = TokenBuilder::new();
        tb.add("impl dyn ").ident(&ident).add(" {");
        tb.add("    pub fn is<T: ")
            .ident(&ident)
            .add(" + 'static >(&self) -> bool {");
        tb.add("        let t = std::any::TypeId::of::<T>();");
        tb.add("        let concrete = self.ref_cast_type_id();");
        tb.add("        t == concrete");
        tb.add("    }");
        tb.add("    pub fn downcast_ref<T: ")
            .ident(&ident)
            .add(" + 'static >(&self) -> Option<&T> {");
        tb.add("        if self.is::<T>() {");
        tb.add("            Some(unsafe {&*(self as *const dyn ")
            .ident(&ident)
            .add(" as *const T)})");
        tb.add("        } else {");
        tb.add("            None");
        tb.add("        }");
        tb.add("    }");
        tb.add("    pub fn downcast_mut<T: ")
            .ident(&ident)
            .add(" + 'static >(&mut self) -> Option<&mut T> {");
        tb.add("        if self.is::<T>() {");
        tb.add("            Some(unsafe {&mut *(self as *const dyn ")
            .ident(&ident)
            .add(" as *mut T)})");
        tb.add("        } else {");
        tb.add("            None");
        tb.add("        }");
        tb.add("    }");
        tb.add("}");
        tb.end()
    } else {
        error("Expected identifier")
    }
}

pub fn derive_default_none_impl(input: TokenStream) -> TokenStream {
    let mut tb = TokenBuilder::new();
    let mut parser = TokenParser::new(input);
    let _main_attribs = parser.eat_attributes();
    parser.eat_ident("pub");
    if parser.eat_ident("enum") {
        if let Some(enum_name) = parser.eat_any_ident() {
            let generic = parser.eat_generic();
            let where_clause = parser.eat_where_clause(None);

            tb.add("impl")
                .ident(&enum_name)
                .stream(generic.clone())
                .stream(where_clause.clone());
            tb.add("{");
            tb.add("   const DEFAULT_NONE_REF:Self = Self::None;");
            tb.add("}");

            tb.add("impl ActionDefaultRef for ")
                .ident(&enum_name)
                .stream(generic.clone())
                .stream(where_clause.clone());
            tb.add("{");
            tb.add("   fn default_ref()->&'static Self{");
            tb.add("      return &Self::DEFAULT_NONE_REF;");
            tb.add("   }");
            tb.add("}");

            /*
            tb.add("impl Into<Box<dyn WidgetAction>> for ").ident(&enum_name).stream(generic.clone()).stream(where_clause.clone());
            tb.add("{");
            tb.add("    fn into(self)->Box<dyn WidgetAction>{");
            tb.add("        Box::new(self)");
            tb.add("    }");
            tb.add("}");*/
            /*
            tb.add("impl ").ident(&enum_name).stream(generic.clone()).stream(where_clause.clone());
            tb.add("{");
            tb.add("    fn into_action(self, uid:WidgetUid)->WidgetActionItem{");
            tb.add("        WidgetActionItem::new(self.into(), uid)");
            tb.add("    }");
            tb.add("}");*/
            tb.add("impl").stream(generic.clone());
            tb.add("Default for")
                .ident(&enum_name)
                .stream(generic)
                .stream(where_clause)
                .add("{");
            tb.add("    fn default()->Self{Self::None}");
            tb.add("}");

            return tb.end();
        }
    }

    parser.unexpected()
}