fix: added missing signal about success

timemates · Nov 20, 2024 · f628631 · f628631
1 parent 3d12870
commit f628631
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/generator/core/src/commonMain/kotlin/org/timemates/rrpc/codegen/plugin/data/PluginSignal.kt b/generator/core/src/commonMain/kotlin/org/timemates/rrpc/codegen/plugin/data/PluginSignal.kt
@@ -21,6 +21,8 @@ public sealed interface PluginSignal : GPSignal {
 
     @Serializable
     public sealed interface RequestStatusChange : PluginSignal {
+        @Serializable
+        public data class Finished(public val message: String) : RequestStatusChange
         @Serializable
         public data object Accepted : RequestStatusChange
         @Serializable
@@ -93,6 +95,10 @@ public class PluginMessage @PublishedApi internal constructor(
 
                     is PluginSignal.SendOptions -> SendOptionsField(value)
 
+                    is PluginSignal.RequestStatusChange.Finished ->
+                        RequestStatusFinishedField(value)
+
+
                     null -> null
                 }
             }
@@ -127,6 +133,12 @@ public class PluginMessage @PublishedApi internal constructor(
             override val value: PluginSignal.RequestStatusChange.Accepted,
         ) : SignalOneOf
 
+        @JvmInline
+        value class RequestStatusFinishedField(
+            @ProtoNumber(3)
+            override val value: PluginSignal.RequestStatusChange.Finished,
+        ) : SignalOneOf
+
         @JvmInline
         value class RequestStatusFailedField(
             @ProtoNumber(4)

diff --git a/tooling/generator-cli/src/main/kotlin/org/timemates/rrpc/generator/cli/GenerateCommand.kt b/tooling/generator-cli/src/main/kotlin/org/timemates/rrpc/generator/cli/GenerateCommand.kt
@@ -74,6 +74,7 @@ class GenerateCommand(
                                 )
                             }
                             is PluginSignal.SendOptions -> { /* ignored: should've received before */ }
+                            is PluginSignal.RequestStatusChange.Finished -> return@launch
                         }
                     }
                 }