Start on plugin ctx API (#64)

src-tauri/yaak_plugin_runtime/src/manager.rs

@@ -1,7 +1,7 @@
 use crate::error::Result;
 use crate::events::{
-    ExportHttpRequestRequest, ExportHttpRequestResponse, FilterRequest, FilterResponse,
-    ImportRequest, ImportResponse, InternalEventPayload,
+    ExportHttpRequestRequest, ExportHttpRequestResponse, FilterRequest, FilterResponse
+    , ImportRequest, ImportResponse, InternalEvent, InternalEventPayload,
 };
 
 use crate::error::Error::PluginErr;
@@ -10,6 +10,7 @@ use crate::plugin::start_server;
 use crate::server::PluginRuntimeGrpcServer;
 use std::time::Duration;
 use tauri::{AppHandle, Runtime};
+use tokio::sync::mpsc;
 use tokio::sync::watch::Sender;
 use yaak_models::models::HttpRequest;
 
@@ -35,14 +36,33 @@ impl PluginManager {
         PluginManager { kill_tx, server }
     }
 
-    pub async fn cleanup(&mut self) {
+    pub async fn subscribe(&self) -> (String, mpsc::Receiver<InternalEvent>) {
+        self.server.subscribe().await
+    }
+
+    pub async fn unsubscribe(&self, rx_id: &str) {
+        self.server.unsubscribe(rx_id).await
+    }
+
+    pub async fn cleanup(&self) {
         self.kill_tx.send_replace(true);
 
         // Give it a bit of time to kill
         tokio::time::sleep(Duration::from_millis(500)).await;
     }
 
-    pub async fn run_import(&mut self, content: &str) -> Result<(ImportResponse, String)> {
+    pub async fn reply(
+        &self,
+        source_event: &InternalEvent,
+        payload: &InternalEventPayload,
+    ) -> Result<()> {
+        let reply_id = Some(source_event.clone().id);
+        self.server
+            .send(&payload, source_event.plugin_ref_id.as_str(), reply_id)
+            .await
+    }
+
+    pub async fn run_import(&self, content: &str) -> Result<(ImportResponse, String)> {
         let reply_events = self
             .server
             .send_and_wait(&InternalEventPayload::ImportRequest(ImportRequest {
@@ -67,7 +87,7 @@ impl PluginManager {
     }
 
     pub async fn run_export_curl(
-        &mut self,
+        &self,
         request: &HttpRequest,
     ) -> Result<ExportHttpRequestResponse> {
         let event = self
@@ -90,7 +110,7 @@ impl PluginManager {
     }
 
     pub async fn run_filter(
-        &mut self,
+        &self,
         filter: &str,
         content: &str,
         content_type: &str,