feat(message): enable usage of the protoc-gen-prost-serde plugin

cunarist · May 21, 2024 · ec271c0 · ec271c0
1 parent e6fd600
commit ec271c0
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/flutter_ffi_plugin/bin/src/config.dart b/flutter_ffi_plugin/bin/src/config.dart
@@ -5,19 +5,21 @@ class RinfConfigMessage {
   final String inputDir;
   final String rustOutputDir;
   final String dartOutputDir;
+  final bool rustSerde;
 
   RinfConfigMessage({
     required this.inputDir,
     required this.rustOutputDir,
     required this.dartOutputDir,
+    required this.rustSerde,
   });
 
   factory RinfConfigMessage.defaultConfig() {
     return RinfConfigMessage(
-      inputDir: DEFAULT_INPUT_DIR,
-      rustOutputDir: DEFAULT_RUST_OUTPUT_DIR,
-      dartOutputDir: DEFAULT_DART_OUTPUT_DIR,
-    );
+        inputDir: DEFAULT_INPUT_DIR,
+        rustOutputDir: DEFAULT_RUST_OUTPUT_DIR,
+        dartOutputDir: DEFAULT_DART_OUTPUT_DIR,
+        rustSerde: DEFAULT_RUST_SERDE);
   }
 
   factory RinfConfigMessage.from(YamlMap yaml) {
@@ -33,6 +35,7 @@ class RinfConfigMessage {
       inputDir: yaml[KEY_INPUT_DIR] ?? DEFAULT_INPUT_DIR,
       rustOutputDir: yaml[KEY_RUST_OUTPUT_DIR] ?? DEFAULT_RUST_OUTPUT_DIR,
       dartOutputDir: yaml[KEY_DART_OUTPUT_DIR] ?? DEFAULT_DART_OUTPUT_DIR,
+      rustSerde: yaml[KEY_RUST_SERDE] ?? DEFAULT_RUST_SERDE,
     );
   }
 
@@ -47,15 +50,18 @@ class RinfConfigMessage {
   static const KEY_INPUT_DIR = "input_dir";
   static const KEY_RUST_OUTPUT_DIR = "rust_output_dir";
   static const KEY_DART_OUTPUT_DIR = "dart_output_dir";
+  static const KEY_RUST_SERDE = "rust_serde";
 
   static const DEFAULT_INPUT_DIR = "messages/";
   static const DEFAULT_RUST_OUTPUT_DIR = "native/hub/src/messages/";
   static const DEFAULT_DART_OUTPUT_DIR = "lib/messages/";
+  static const DEFAULT_RUST_SERDE = false;
 
   static const MESSAGE_CONFIG_KEYS = [
     KEY_INPUT_DIR,
     KEY_RUST_OUTPUT_DIR,
     KEY_DART_OUTPUT_DIR,
+    KEY_RUST_SERDE
   ];
 }
 

diff --git a/flutter_ffi_plugin/bin/src/message.dart b/flutter_ffi_plugin/bin/src/message.dart
@@ -83,6 +83,7 @@ Future<void> generateMessageCode({
   final cargoInstallCommand = await Process.run('cargo', [
     'install',
     'protoc-gen-prost',
+    ...(messageConfig.rustSerde ? ['protoc-gen-prost-serde'] : [])
   ]);
   if (cargoInstallCommand.exitCode != 0) {
     print(cargoInstallCommand.stderr.toString().trim());
@@ -105,6 +106,7 @@ Future<void> generateMessageCode({
     final protocRustResult = await Process.run('protoc', [
       ...protoPaths,
       '--prost_out=$rustFullPath',
+      ...(messageConfig.rustSerde ? ['--prost-serde_out=$rustFullPath'] : []),
       ...resourceNames.map((name) => '$name.proto'),
     ]);
     if (protocRustResult.exitCode != 0) {