polywrap_plugin 0.1.11

Core package to build plugins in rust to interact with Polywrap Client
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

use polywrap_core::invoker::Invoker;
use std::{
    collections::HashMap,
    fmt::{Debug, Formatter},
    sync::Arc,
};

use crate::{error::PluginError, method::PluginMethod, module::PluginModule};

#[derive(Clone)]
pub struct PluginModuleWithMethods {
    methods_map: HashMap<String, Arc<PluginMethod>>,
}

impl Default for PluginModuleWithMethods {
    fn default() -> Self {
        Self::new()
    }
}

impl PluginModuleWithMethods {
    pub fn new() -> Self {
        Self {
            methods_map: HashMap::new(),
        }
    }

    pub fn methods(&mut self, methods: HashMap<String, Arc<PluginMethod>>) -> &mut Self {
        self.methods_map = methods;
        self
    }
}

impl PluginModule for PluginModuleWithMethods {
    fn _wrap_invoke(
        &mut self,
        method_name: &str,
        params: &[u8],
        env: Option<&[u8]>,
        invoker: Arc<dyn Invoker>,
    ) -> Result<Vec<u8>, PluginError> {
        if let Some(method) = self.methods_map.get(method_name) {
            (method)(params, env, invoker.as_ref())
        } else {
            Err(PluginError::MethodNotFoundError(method_name.to_string()))
        }
    }
}

impl Debug for PluginModuleWithMethods {
    fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), std::fmt::Error> {
        write!(
            f,
            r#"
      Plugin With Methods
      
      -Methods: {:?}
      "#,
            self.methods_map.keys()
        )
    }
}