tauri_plugin_screen_wake_lock/
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;
16
17pub use error::{Error, Result};
18
19#[cfg(desktop)]
20use desktop::ScreenWakeLock;
21#[cfg(mobile)]
22use mobile::ScreenWakeLock;
23
24/// Extensions to [`tauri::App`], [`tauri::AppHandle`] and [`tauri::Window`] to access the screen-wake-lock APIs.
25pub trait ScreenWakeLockExt<R: Runtime> {
26  fn screen_wake_lock(&self) -> &ScreenWakeLock<R>;
27}
28
29impl<R: Runtime, T: Manager<R>> crate::ScreenWakeLockExt<R> for T {
30  fn screen_wake_lock(&self) -> &ScreenWakeLock<R> {
31    self.state::<ScreenWakeLock<R>>().inner()
32  }
33}
34
35/// Initializes the plugin.
36pub fn init<R: Runtime>() -> TauriPlugin<R> {
37  Builder::new("screen-wake-lock")
38    .invoke_handler(tauri::generate_handler![
39      commands::set_enabled,
40      commands::is_enabled,
41      commands::is_supported
42    ])
43    .setup(|app, api| {
44      #[cfg(mobile)]
45      let screen_wake_lock = mobile::init(app, api)?;
46      #[cfg(desktop)]
47      let screen_wake_lock = desktop::init(app, api)?;
48      app.manage(screen_wake_lock);
49      Ok(())
50    })
51    .build()
52}