tauri-plugin-askit 0.4.0

Tauri plugin for Agent Stream Kit
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

use agent_stream_kit::ASKit;
use tauri::{
  plugin::{Builder, TauriPlugin},
  Manager, Runtime, RunEvent,
};

mod commands;
mod error;
mod models;
mod observer;

pub use error::{Error, Result};
pub use models::*;

use observer::BoardObserver;

/// Extensions to [`tauri::App`], [`tauri::AppHandle`] and [`tauri::Window`] to access the askit APIs.
pub trait ASKitExt<R: Runtime> {
  fn askit(&self) -> &ASKit;
}

impl<R: Runtime, T: Manager<R>> crate::ASKitExt<R> for T {
  fn askit(&self) -> &ASKit {
    self.state::<ASKit>().inner()
  }
}

/// Initializes the plugin.
pub fn init<R: Runtime>() -> TauriPlugin<R> {
  Builder::new("askit")
    .invoke_handler(tauri::generate_handler![
      commands::get_agent_definition,
      commands::get_agent_definitions,
      commands::get_agent_flows,
      commands::new_agent_flow,
      commands::rename_agent_flow,
      commands::unique_flow_name,
      commands::add_agent_flow,
      commands::remove_agent_flow,
      commands::insert_agent_flow,
      commands::copy_sub_flow,
      commands::start_agent_flow,
      commands::stop_agent_flow,
      commands::new_agent_flow_node,
      commands::add_agent_flow_node,
      commands::remove_agent_flow_node,
      commands::add_agent_flow_edge,
      commands::remove_agent_flow_edge,
      commands::start_agent,
      commands::stop_agent,
      commands::write_board,
      commands::set_agent_configs,
      commands::get_global_configs,
      commands::get_global_configs_map,
      commands::set_global_configs,
      commands::set_global_configs_map,
      commands::get_agent_default_configs,
    ])
    .setup(|app, _api| {
      let askit = ASKit::init()?;
      askit.subscribe(Box::new(BoardObserver {
          app: app.clone(),
      }));
      app.manage(askit);
      Ok(())
    })
    .on_event(|app, event| {
      match event {
        RunEvent::Ready => {
          tauri::async_runtime::block_on(async move {
            let askit = app.state::<ASKit>();
            askit.ready().await.unwrap();
          });
        }
        RunEvent::Exit => {
          let askit = app.state::<ASKit>();
          askit.quit();
        }
        _ => {}
      }
    })
    .build()
}