tauri_plugin_screen_wake_lock/
lib.rs1use 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
24pub 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
35pub 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}