fzyzcjy · fzyzcjy · Apr 8, 2024 · Apr 8, 2024 · Apr 8, 2024 · Apr 8, 2024
diff --git a/frb_codegen/assets/integration_template/REPLACE_ME_RUST_CRATE_DIR/src/frb_generated.rs b/frb_codegen/assets/integration_template/REPLACE_ME_RUST_CRATE_DIR/src/frb_generated.rs
@@ -30,7 +30,7 @@ flutter_rust_bridge::frb_generated_boilerplate!(
     default_rust_opaque = RustOpaqueMoi,
     default_rust_auto_opaque = RustAutoOpaqueMoi,
 );
-const FLUTTER_RUST_BRIDGE_CODEGEN_VERSION: &str = "REPLACE_ME_FRB_VERSION";
+pub(crate) const FLUTTER_RUST_BRIDGE_CODEGEN_VERSION: &str = "REPLACE_ME_FRB_VERSION";
 
 // Section: executor
 

diff --git a/frb_codegen/src/library/codegen/generator/wire/rust/spec_generator/misc/mod.rs b/frb_codegen/src/library/codegen/generator/wire/rust/spec_generator/misc/mod.rs
@@ -40,7 +40,7 @@ pub(crate) fn generate(
         code_header: Acc::new(|_| vec![(generate_code_header() + "\n\n").into()]),
         file_attributes: Acc::new_common(vec![FILE_ATTRIBUTES.to_string().into()]),
         imports: generate_imports(&cache.distinct_types, context),
-        executor: Acc::new_common(vec![generate_executor(context.ir_pack).into()]),
+        executor: Acc::new_common(vec![generate_handler(context.ir_pack).into()]),
         boilerplate: generate_boilerplate(
             context.config.default_stream_sink_codec,
             context.config.default_rust_opaque_codec,
@@ -152,7 +152,7 @@ fn generate_boilerplate(
                     default_rust_opaque = RustOpaque{default_rust_opaque_codec},
                     default_rust_auto_opaque = RustAutoOpaque{default_rust_opaque_codec},
                 );
-                const FLUTTER_RUST_BRIDGE_CODEGEN_VERSION: &str = "{version}";
+                pub(crate) const FLUTTER_RUST_BRIDGE_CODEGEN_VERSION: &str = "{version}";
             "#,
                 version = env!("CARGO_PKG_VERSION"),
             )
@@ -198,9 +198,9 @@ fn generate_boilerplate(
 //     }
 // }
 
-fn generate_executor(ir_pack: &IrPack) -> String {
-    if ir_pack.has_executor {
-        "/* nothing since executor detected */".to_owned()
+fn generate_handler(ir_pack: &IrPack) -> String {
+    if let Some(existing_handler) = &ir_pack.existing_handler {
+        format!("pub use {};", existing_handler.rust_style())
     } else {
         r#"flutter_rust_bridge::frb_generated_default_handler!();"#.to_owned()
     }

diff --git a/frb_codegen/src/library/codegen/generator/wire/rust/spec_generator/misc/ty/dart_fn.rs b/frb_codegen/src/library/codegen/generator/wire/rust/spec_generator/misc/ty/dart_fn.rs
@@ -3,6 +3,7 @@ use crate::codegen::generator::wire::rust::spec_generator::base::*;
 use crate::codegen::generator::wire::rust::spec_generator::misc::ty::WireRustGeneratorMiscTrait;
 use crate::codegen::generator::wire::rust::spec_generator::output_code::WireRustOutputCode;
 use crate::codegen::ir::ty::IrTypeTrait;
+use crate::library::misc::consts::HANDLER_NAME;
 use itertools::Itertools;
 
 impl<'a> WireRustGeneratorMiscTrait for DartFnWireRustGenerator<'a> {
@@ -32,7 +33,7 @@ impl<'a> WireRustGeneratorMiscTrait for DartFnWireRustGenerator<'a> {
 
                     async fn body(dart_opaque: flutter_rust_bridge::DartOpaque, {parameter_names_and_types}) -> {return_type} {{
                         let args = vec![{into_dart_expressions}];
-                        let message = FLUTTER_RUST_BRIDGE_HANDLER.dart_fn_invoke(dart_opaque, args).await;
+                        let message = {HANDLER_NAME}.dart_fn_invoke(dart_opaque, args).await;
                         <{return_type}>::sse_decode_single(message)
                     }}
 

diff --git a/frb_codegen/src/library/codegen/ir/pack.rs b/frb_codegen/src/library/codegen/ir/pack.rs
@@ -17,7 +17,7 @@ pub struct IrPack {
     pub funcs: Vec<IrFunc>,
     pub struct_pool: IrStructPool,
     pub enum_pool: IrEnumPool,
-    pub has_executor: bool,
+    pub existing_handler: Option<NamespacedName>,
     pub unused_types: Vec<NamespacedName>,
 }
 

diff --git a/frb_codegen/src/library/codegen/parser/misc.rs b/frb_codegen/src/library/codegen/parser/misc.rs
@@ -1,3 +1,5 @@
+use crate::library::misc::consts::HANDLER_NAME;
+
 pub(crate) fn parse_has_executor(source_rust_content: &str) -> bool {
-    source_rust_content.contains(&"static ref FLUTTER_RUST_BRIDGE_HANDLER".to_string())
+    source_rust_content.contains(&format!("static {HANDLER_NAME}"))
 }
diff --git a/frb_codegen/src/library/codegen/parser/mod.rs b/frb_codegen/src/library/codegen/parser/mod.rs
@@ -10,6 +10,7 @@ pub(crate) mod type_parser;
 mod unused_checker;
 
 use crate::codegen::dumper::Dumper;
+use crate::codegen::ir::namespace::{Namespace, NamespacedName};
 use crate::codegen::ir::pack::IrPack;
 use crate::codegen::misc::GeneratorProgressBarPack;
 use crate::codegen::parser::function_extractor::extract_generalized_functions_from_file;
@@ -21,6 +22,8 @@ use crate::codegen::parser::type_alias_resolver::resolve_type_aliases;
 use crate::codegen::parser::type_parser::TypeParser;
 use crate::codegen::parser::unused_checker::get_unused_types;
 use crate::codegen::ConfigDumpContent;
+use crate::library::misc::consts::HANDLER_NAME;
+use anyhow::ensure;
 use itertools::Itertools;
 use log::trace;
 use std::path::{Path, PathBuf};
@@ -88,15 +91,30 @@ pub(crate) fn parse(
         .sorted_by_cached_key(|func| func.name.clone())
         .collect_vec();
 
-    let has_executor = (file_data_arr.iter()).any(|file| parse_has_executor(&file.content));
+    let existing_handlers = (file_data_arr.iter())
+        .filter(|file| parse_has_executor(&file.content))
+        .map(|file| {
+            NamespacedName::new(
+                Namespace::new_from_rust_crate_path(&file.path, &config.rust_crate_dir).unwrap(),
+                HANDLER_NAME.to_owned(),
+            )
+        })
+        .collect_vec();
+    ensure!(
+        existing_handlers.len() <= 1,
+        // frb-coverage:ignore-start
+        // This will stop the whole generator and tell the users, so we do not care about testing it
+        "Should have at most one custom handler"
+    );
+    // frb-coverage:ignore-end
 
     let (struct_pool, enum_pool) = type_parser.consume();
 
     let mut ans = IrPack {
         funcs: ir_funcs,
         struct_pool,
         enum_pool,
-        has_executor,
+        existing_handler: existing_handlers.first().cloned(),
         unused_types: vec![],
     };
 

diff --git a/frb_codegen/test_fixtures/library/codegen/parser/mod/generics/expect_ir.json b/frb_codegen/test_fixtures/library/codegen/parser/mod/generics/expect_ir.json
@@ -105,6 +105,7 @@
       "wrapper_name": null
     }
   },
+  "existing_handler": null,
   "funcs": [
     {
       "codec_mode_pack": {
@@ -372,7 +373,6 @@
       "rust_async": false
     }
   ],
-  "has_executor": false,
   "struct_pool": {
     "crate::api/MyGenericStruct": {
       "comments": [],

diff --git a/frb_codegen/test_fixtures/library/codegen/parser/mod/methods/expect_ir.json b/frb_codegen/test_fixtures/library/codegen/parser/mod/methods/expect_ir.json
@@ -21,6 +21,7 @@
       "wrapper_name": null
     }
   },
+  "existing_handler": null,
   "funcs": [
     {
       "codec_mode_pack": {
@@ -221,7 +222,6 @@
       "rust_async": false
     }
   ],
-  "has_executor": false,
   "struct_pool": {
     "crate::api/MyStruct": {
       "comments": [],

diff --git a/frb_codegen/test_fixtures/library/codegen/parser/mod/multi_input_file/expect_ir.json b/frb_codegen/test_fixtures/library/codegen/parser/mod/multi_input_file/expect_ir.json
@@ -1,5 +1,6 @@
 {
   "enum_pool": {},
+  "existing_handler": null,
   "funcs": [
     {
       "codec_mode_pack": {
@@ -44,7 +45,6 @@
       "rust_async": false
     }
   ],
-  "has_executor": false,
   "struct_pool": {},
   "unused_types": []
 }
diff --git a/frb_codegen/test_fixtures/library/codegen/parser/mod/non_qualified_names/expect_ir.json b/frb_codegen/test_fixtures/library/codegen/parser/mod/non_qualified_names/expect_ir.json
@@ -1,5 +1,6 @@
 {
   "enum_pool": {},
+  "existing_handler": null,
   "funcs": [
     {
       "codec_mode_pack": {
@@ -79,7 +80,6 @@
       "rust_async": false
     }
   ],
-  "has_executor": false,
   "struct_pool": {},
   "unused_types": []
 }
diff --git a/frb_codegen/test_fixtures/library/codegen/parser/mod/qualified_names/expect_ir.json b/frb_codegen/test_fixtures/library/codegen/parser/mod/qualified_names/expect_ir.json
@@ -1,5 +1,6 @@
 {
   "enum_pool": {},
+  "existing_handler": null,
   "funcs": [
     {
       "codec_mode_pack": {
@@ -129,7 +130,6 @@
       "rust_async": false
     }
   ],
-  "has_executor": false,
   "struct_pool": {},
   "unused_types": []
 }
diff --git a/frb_codegen/test_fixtures/library/codegen/parser/mod/simple/expect_ir.json b/frb_codegen/test_fixtures/library/codegen/parser/mod/simple/expect_ir.json
@@ -1,5 +1,6 @@
 {
   "enum_pool": {},
+  "existing_handler": null,
   "funcs": [
     {
       "codec_mode_pack": {
@@ -23,7 +24,6 @@
       "rust_async": false
     }
   ],
-  "has_executor": false,
   "struct_pool": {},
   "unused_types": []
 }
diff --git a/frb_codegen/test_fixtures/library/codegen/parser/mod/unused_struct_enum/expect_ir.json b/frb_codegen/test_fixtures/library/codegen/parser/mod/unused_struct_enum/expect_ir.json
@@ -1,7 +1,7 @@
 {
   "enum_pool": {},
+  "existing_handler": null,
   "funcs": [],
-  "has_executor": false,
   "struct_pool": {},
   "unused_types": [
     "crate::api/UnusedStruct",

diff --git a/frb_codegen/test_fixtures/library/codegen/parser/mod/use_type_in_another_file/expect_ir.json b/frb_codegen/test_fixtures/library/codegen/parser/mod/use_type_in_another_file/expect_ir.json
@@ -15,6 +15,7 @@
       "wrapper_name": null
     }
   },
+  "existing_handler": null,
   "funcs": [
     {
       "codec_mode_pack": {
@@ -85,7 +86,6 @@
       "rust_async": false
     }
   ],
-  "has_executor": false,
   "struct_pool": {
     "crate::another_file/StructInAnotherFile": {
       "comments": [],

diff --git a/frb_example/dart_build_rs/rust/src/frb_generated.rs b/frb_example/dart_build_rs/rust/src/frb_generated.rs
@@ -30,7 +30,7 @@ flutter_rust_bridge::frb_generated_boilerplate!(
     default_rust_opaque = RustOpaqueMoi,
     default_rust_auto_opaque = RustAutoOpaqueMoi,
 );
-const FLUTTER_RUST_BRIDGE_CODEGEN_VERSION: &str = "2.0.0-dev.30";
+pub(crate) const FLUTTER_RUST_BRIDGE_CODEGEN_VERSION: &str = "2.0.0-dev.30";
 
 // Section: executor
 

diff --git a/frb_example/dart_minimal/rust/src/frb_generated.rs b/frb_example/dart_minimal/rust/src/frb_generated.rs
@@ -30,7 +30,7 @@ flutter_rust_bridge::frb_generated_boilerplate!(
     default_rust_opaque = RustOpaqueMoi,
     default_rust_auto_opaque = RustAutoOpaqueMoi,
 );
-const FLUTTER_RUST_BRIDGE_CODEGEN_VERSION: &str = "2.0.0-dev.30";
+pub(crate) const FLUTTER_RUST_BRIDGE_CODEGEN_VERSION: &str = "2.0.0-dev.30";
 
 // Section: executor
 

diff --git a/frb_example/deliberate_bad/rust/src/frb_generated.rs b/frb_example/deliberate_bad/rust/src/frb_generated.rs
@@ -30,7 +30,7 @@ flutter_rust_bridge::frb_generated_boilerplate!(
     default_rust_opaque = RustOpaqueMoi,
     default_rust_auto_opaque = RustAutoOpaqueMoi,
 );
-const FLUTTER_RUST_BRIDGE_CODEGEN_VERSION: &str = "2.0.0-dev.30";
+pub(crate) const FLUTTER_RUST_BRIDGE_CODEGEN_VERSION: &str = "2.0.0-dev.30";
 
 // Section: executor
 

diff --git a/frb_example/flutter_via_create/rust/src/frb_generated.rs b/frb_example/flutter_via_create/rust/src/frb_generated.rs
@@ -30,7 +30,7 @@ flutter_rust_bridge::frb_generated_boilerplate!(
     default_rust_opaque = RustOpaqueMoi,
     default_rust_auto_opaque = RustAutoOpaqueMoi,
 );
-const FLUTTER_RUST_BRIDGE_CODEGEN_VERSION: &str = "2.0.0-dev.30";
+pub(crate) const FLUTTER_RUST_BRIDGE_CODEGEN_VERSION: &str = "2.0.0-dev.30";
 
 // Section: executor
 

diff --git a/frb_example/flutter_via_integrate/rust/src/frb_generated.rs b/frb_example/flutter_via_integrate/rust/src/frb_generated.rs
@@ -30,7 +30,7 @@ flutter_rust_bridge::frb_generated_boilerplate!(
     default_rust_opaque = RustOpaqueMoi,
     default_rust_auto_opaque = RustAutoOpaqueMoi,
 );
-const FLUTTER_RUST_BRIDGE_CODEGEN_VERSION: &str = "2.0.0-dev.30";
+pub(crate) const FLUTTER_RUST_BRIDGE_CODEGEN_VERSION: &str = "2.0.0-dev.30";
 
 // Section: executor
 

diff --git a/frb_example/gallery/rust/src/frb_generated.rs b/frb_example/gallery/rust/src/frb_generated.rs
@@ -30,7 +30,7 @@ flutter_rust_bridge::frb_generated_boilerplate!(
     default_rust_opaque = RustOpaqueMoi,
     default_rust_auto_opaque = RustAutoOpaqueMoi,
 );
-const FLUTTER_RUST_BRIDGE_CODEGEN_VERSION: &str = "2.0.0-dev.30";
+pub(crate) const FLUTTER_RUST_BRIDGE_CODEGEN_VERSION: &str = "2.0.0-dev.30";
 
 // Section: executor
 

diff --git a/frb_example/pure_dart/rust/src/api/custom_handler.rs b/frb_example/pure_dart/rust/src/api/custom_handler.rs
@@ -0,0 +1,13 @@
+// FRB_INTERNAL_GENERATOR: {"forbiddenDuplicatorModes": ["sync", "rustAsync", "sse", "sync sse", "rustAsync sse"]}
+
+#[allow(unused_imports)]
+use crate::frb_generated::FLUTTER_RUST_BRIDGE_CODEGEN_VERSION;
+
+// This file demonstrates how to use a custom handler.
+// Usually there is no need for this, and the default handler is used.
+//
+// Here, for simplicity, we still generate code for default handler.
+// But surely you can copy-paste the content of that macro and modify according to your needs.
+flutter_rust_bridge::frb_generated_default_handler!();
+
+// NOTE: For more tests about customizing handler types and contents, please visit `src/auxiliary/custom_handler.rs`
diff --git a/frb_example/pure_dart/rust/src/api/mod.rs b/frb_example/pure_dart/rust/src/api/mod.rs
@@ -9,6 +9,7 @@ pub mod comment;
 #[cfg(target_os = "non_existent_os")]
 pub mod conditionally_compiled_module;
 pub mod constructor;
+pub mod custom_handler;
 pub mod customization;
 pub mod dart_dynamic;
 pub mod dart_fn;