use crate::error::Result; use crate::events::{InternalEvent, InternalEventPayload, PluginContext}; use crate::plugin_meta::{PluginMetadata, get_plugin_meta}; use crate::util::gen_id; use std::path::Path; use std::sync::Arc; use tokio::sync::{Mutex, mpsc}; #[derive(Clone)] pub struct PluginHandle { pub ref_id: String, pub dir: String, pub(crate) to_plugin_tx: Arc>>, pub(crate) metadata: PluginMetadata, } impl PluginHandle { pub fn new(dir: &str, tx: mpsc::Sender) -> Result { let ref_id = gen_id(); let metadata = get_plugin_meta(&Path::new(dir))?; Ok(PluginHandle { ref_id: ref_id.clone(), dir: dir.to_string(), to_plugin_tx: Arc::new(Mutex::new(tx)), metadata, }) } pub fn info(&self) -> PluginMetadata { self.metadata.clone() } pub fn build_event_to_send( &self, plugin_context: &PluginContext, payload: &InternalEventPayload, reply_id: Option, ) -> InternalEvent { self.build_event_to_send_raw(plugin_context, payload, reply_id) } pub(crate) fn build_event_to_send_raw( &self, plugin_context: &PluginContext, payload: &InternalEventPayload, reply_id: Option, ) -> InternalEvent { let dir = Path::new(&self.dir); InternalEvent { id: gen_id(), plugin_ref_id: self.ref_id.clone(), plugin_name: dir.file_name().unwrap().to_str().unwrap().to_string(), reply_id, payload: payload.clone(), context: plugin_context.clone(), } } pub async fn send(&self, event: &InternalEvent) -> Result<()> { self.to_plugin_tx.lock().await.send(event.to_owned()).await?; Ok(()) } }