Skip to main content

tauri_plugin_stt/
lib.rs

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
28/// Extensions to [`tauri::App`], [`tauri::AppHandle`] and [`tauri::Window`] to access the stt APIs.
29pub 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
39/// Initializes the plugin.
40pub 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}