1use tauri::{
2 plugin::{Builder, TauriPlugin},
3 Manager, Runtime,
4};
5
6pub use models::*;
7
8#[cfg(desktop)]
9mod desktop;
10#[cfg(mobile)]
11mod mobile;
12
13mod commands;
14mod error;
15mod models;
16mod paths;
17
18pub use error::{Error, Result};
19pub use paths::{
20 get_model_path, get_models_dir, list_available_models, model_exists, validate_path,
21};
22
23#[cfg(desktop)]
24use desktop::Stt;
25#[cfg(mobile)]
26use mobile::Stt;
27
28pub trait SttExt<R: Runtime> {
30 fn stt(&self) -> &Stt<R>;
31}
32
33impl<R: Runtime, T: Manager<R>> crate::SttExt<R> for T {
34 fn stt(&self) -> &Stt<R> {
35 self.state::<Stt<R>>().inner()
36 }
37}
38
39pub fn init<R: Runtime>() -> TauriPlugin<R> {
41 let mut builder = Builder::new("stt");
42
43 #[cfg(desktop)]
44 {
45 builder = builder.invoke_handler(tauri::generate_handler![
46 commands::start_listening,
47 commands::stop_listening,
48 commands::is_available,
49 commands::get_supported_languages,
50 commands::check_permission,
51 commands::request_permission,
52 commands::register_listener,
53 commands::remove_listener,
54 ]);
55 }
56
57 #[cfg(mobile)]
58 {
59 builder = builder.invoke_handler(tauri::generate_handler![
60 commands::start_listening,
61 commands::stop_listening,
62 commands::is_available,
63 commands::get_supported_languages,
64 commands::check_permission,
65 commands::request_permission,
66 ]);
67 }
68
69 builder
70 .setup(|app, api| {
71 #[cfg(mobile)]
72 let stt = mobile::init(app, api)?;
73 #[cfg(desktop)]
74 let stt = desktop::init(app, api)?;
75 app.manage(stt);
76 Ok(())
77 })
78 .build()
79}