gRPC request actions and "copy as gRPCurl" (#232)

src-tauri/yaak-plugins/src/events.rs

@@ -89,11 +89,16 @@ pub enum InternalEventPayload {
     GetCookieValueRequest(GetCookieValueRequest),
     GetCookieValueResponse(GetCookieValueResponse),
 
-    // Request Actions
+    // HTTP Request Actions
     GetHttpRequestActionsRequest(EmptyPayload),
     GetHttpRequestActionsResponse(GetHttpRequestActionsResponse),
     CallHttpRequestActionRequest(CallHttpRequestActionRequest),
 
+    // Grpc Request Actions
+    GetGrpcRequestActionsRequest(EmptyPayload),
+    GetGrpcRequestActionsResponse(GetGrpcRequestActionsResponse),
+    CallGrpcRequestActionRequest(CallGrpcRequestActionRequest),
+
     // Template Functions
     GetTemplateFunctionsRequest,
     GetTemplateFunctionsResponse(GetTemplateFunctionsResponse),
@@ -116,6 +121,9 @@ pub enum InternalEventPayload {
     RenderHttpRequestRequest(RenderHttpRequestRequest),
     RenderHttpRequestResponse(RenderHttpRequestResponse),
 
+    RenderGrpcRequestRequest(RenderGrpcRequestRequest),
+    RenderGrpcRequestResponse(RenderGrpcRequestResponse),
+
     GetKeyValueRequest(GetKeyValueRequest),
     GetKeyValueResponse(GetKeyValueResponse),
     SetKeyValueRequest(SetKeyValueRequest),
@@ -287,6 +295,21 @@ pub struct RenderHttpRequestResponse {
     pub http_request: HttpRequest,
 }
 
+#[derive(Debug, Clone, Default, Serialize, Deserialize, TS)]
+#[serde(default, rename_all = "camelCase")]
+#[ts(export, export_to = "gen_events.ts")]
+pub struct RenderGrpcRequestRequest {
+    pub grpc_request: GrpcRequest,
+    pub purpose: RenderPurpose,
+}
+
+#[derive(Debug, Clone, Default, Serialize, Deserialize, TS)]
+#[serde(default, rename_all = "camelCase")]
+#[ts(export, export_to = "gen_events.ts")]
+pub struct RenderGrpcRequestResponse {
+    pub grpc_request: GrpcRequest,
+}
+
 #[derive(Debug, Clone, Default, Serialize, Deserialize, TS)]
 #[serde(default, rename_all = "camelCase")]
 #[ts(export, export_to = "gen_events.ts")]
@@ -967,11 +990,6 @@ impl Default for RenderPurpose {
     }
 }
 
-#[derive(Debug, Clone, Default, Serialize, Deserialize, TS)]
-#[serde(default)]
-#[ts(export, export_to = "gen_events.ts")]
-pub struct GetHttpRequestActionsRequest {}
-
 #[derive(Debug, Clone, Default, Serialize, Deserialize, TS)]
 #[serde(default, rename_all = "camelCase")]
 #[ts(export, export_to = "gen_events.ts")]
@@ -1005,6 +1023,40 @@ pub struct CallHttpRequestActionArgs {
     pub http_request: HttpRequest,
 }
 
+#[derive(Debug, Clone, Default, Serialize, Deserialize, TS)]
+#[serde(default, rename_all = "camelCase")]
+#[ts(export, export_to = "gen_events.ts")]
+pub struct GetGrpcRequestActionsResponse {
+    pub actions: Vec<GrpcRequestAction>,
+    pub plugin_ref_id: String,
+}
+
+#[derive(Debug, Clone, Default, Serialize, Deserialize, TS)]
+#[serde(default, rename_all = "camelCase")]
+#[ts(export, export_to = "gen_events.ts")]
+pub struct GrpcRequestAction {
+    pub label: String,
+    #[ts(optional)]
+    pub icon: Option<Icon>,
+}
+
+#[derive(Debug, Clone, Default, Serialize, Deserialize, TS)]
+#[serde(default, rename_all = "camelCase")]
+#[ts(export, export_to = "gen_events.ts")]
+pub struct CallGrpcRequestActionRequest {
+    pub index: i32,
+    pub plugin_ref_id: String,
+    pub args: CallGrpcRequestActionArgs,
+}
+
+#[derive(Debug, Clone, Default, Serialize, Deserialize, TS)]
+#[serde(default, rename_all = "camelCase")]
+#[ts(export, export_to = "gen_events.ts")]
+pub struct CallGrpcRequestActionArgs {
+    pub grpc_request: GrpcRequest,
+    pub proto_files: Vec<String>,
+}
+
 #[derive(Debug, Clone, Default, Serialize, Deserialize, TS)]
 #[serde(default, rename_all = "camelCase")]
 #[ts(export, export_to = "gen_events.ts")]

src-tauri/yaak-plugins/src/manager.rs

@@ -3,10 +3,11 @@ use crate::error::Error::{
 };
 use crate::error::Result;
 use crate::events::{
-    BootRequest, CallHttpAuthenticationActionArgs, CallHttpAuthenticationActionRequest,
-    CallHttpAuthenticationRequest, CallHttpAuthenticationResponse, CallHttpRequestActionRequest,
-    CallTemplateFunctionArgs, CallTemplateFunctionRequest, CallTemplateFunctionResponse,
-    EmptyPayload, FilterRequest, FilterResponse, GetHttpAuthenticationConfigRequest,
+    BootRequest, CallGrpcRequestActionRequest, CallHttpAuthenticationActionArgs,
+    CallHttpAuthenticationActionRequest, CallHttpAuthenticationRequest,
+    CallHttpAuthenticationResponse, CallHttpRequestActionRequest, CallTemplateFunctionArgs,
+    CallTemplateFunctionRequest, CallTemplateFunctionResponse, EmptyPayload, FilterRequest,
+    FilterResponse, GetGrpcRequestActionsResponse, GetHttpAuthenticationConfigRequest,
     GetHttpAuthenticationConfigResponse, GetHttpAuthenticationSummaryResponse,
     GetHttpRequestActionsResponse, GetTemplateFunctionsResponse, GetThemesRequest,
     GetThemesResponse, ImportRequest, ImportResponse, InternalEvent, InternalEventPayload,
@@ -426,6 +427,27 @@ impl PluginManager {
         Ok(themes)
     }
 
+    pub async fn get_grpc_request_actions<R: Runtime>(
+        &self,
+        window: &WebviewWindow<R>,
+    ) -> Result<Vec<GetGrpcRequestActionsResponse>> {
+        let reply_events = self
+            .send_and_wait(
+                &PluginWindowContext::new(window),
+                &InternalEventPayload::GetGrpcRequestActionsRequest(EmptyPayload {}),
+            )
+            .await?;
+
+        let mut all_actions = Vec::new();
+        for event in reply_events {
+            if let InternalEventPayload::GetGrpcRequestActionsResponse(resp) = event.payload {
+                all_actions.push(resp.clone());
+            }
+        }
+
+        Ok(all_actions)
+    }
+
     pub async fn get_http_request_actions<R: Runtime>(
         &self,
         window: &WebviewWindow<R>,
@@ -495,6 +517,23 @@ impl PluginManager {
         Ok(())
     }
 
+    pub async fn call_grpc_request_action<R: Runtime>(
+        &self,
+        window: &WebviewWindow<R>,
+        req: CallGrpcRequestActionRequest,
+    ) -> Result<()> {
+        let ref_id = req.plugin_ref_id.clone();
+        let plugin =
+            self.get_plugin_by_ref_id(ref_id.as_str()).await.ok_or(PluginNotFoundErr(ref_id))?;
+        let event = plugin.build_event_to_send(
+            &PluginWindowContext::new(window),
+            &InternalEventPayload::CallGrpcRequestActionRequest(req),
+            None,
+        );
+        plugin.send(&event).await?;
+        Ok(())
+    }
+
     pub async fn get_http_authentication_summaries<R: Runtime>(
         &self,
         window: &WebviewWindow<R>,