tauri-plugin-nosleep 2.0.0-beta.1

Tauri plugin to prevent the power save functionality in the OS
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

use nosleep::NoSleep;
use tauri::{
    plugin::{Builder, TauriPlugin},
    Manager, Runtime,
};

use std::sync::Mutex;

#[cfg(desktop)]
mod desktop;

mod commands;
mod error;

pub use error::{Error, Result};

#[cfg(desktop)]
use desktop::Nosleep;

struct NoSleepState {
    no_sleep: Mutex<NoSleep>,
}

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

impl<R: Runtime, T: Manager<R>> crate::NosleepExt<R> for T {
    fn nosleep(&self) -> &Nosleep<R> {
        self.state::<Nosleep<R>>().inner()
    }
}

/// Initializes the plugin.
pub fn init<R: Runtime>() -> TauriPlugin<R> {
    Builder::new("nosleep")
        .invoke_handler(tauri::generate_handler![commands::block, commands::unblock])
        .setup(|app, api| {
            #[cfg(desktop)]
            let nosleep = desktop::init(app, api)?;
            app.manage(nosleep);

            // manage state so it is accessible by the commands
            app.manage(NoSleepState {
                no_sleep: Mutex::new(NoSleep::new()?),
            });
            Ok(())
        })
        .build()
}