dvrip-rs 0.1.0

A Rust library for interacting with DVR-IP devices ported from python-dvr.
Documentation 
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

use crate::error::Result;
use async_trait::async_trait;
use serde_json::Value;

use crate::constants::{OK_CODES, QCODES};
use crate::dvrip::DVRIPCam;
use std::sync::atomic::Ordering;

pub type AlarmCallback = Box<dyn Fn(Value, u32) + Send + Sync>;

#[async_trait]
pub trait Alarm: Send + Sync {
    /// Set the alarm callback function
    fn set_alarm_callback(&mut self, callback: Option<AlarmCallback>);

    /// Clear the alarm callback
    fn clear_alarm_callback(&mut self);

    /// Start alarm monitoring
    async fn start_alarm_monitoring(&mut self) -> Result<()>;

    /// Stop alarm monitoring
    async fn stop_alarm_monitoring(&mut self) -> Result<()>;

    /// Set remote alarm
    async fn set_remote_alarm(&mut self, state: bool) -> Result<bool>;

    /// Check if monitoring alarms
    fn is_alarm_monitoring(&self) -> bool;
}

#[async_trait]
impl Alarm for DVRIPCam {
    fn set_alarm_callback(&mut self, callback: Option<AlarmCallback>) {
        let alarm_cb = self.alarm_callback.clone();
        if let Ok(handle) = tokio::runtime::Handle::try_current() {
            handle.spawn(async move {
                *alarm_cb.lock().await = callback;
            });
        } else {
            // If not in an async context, create a temporary runtime
            tokio::spawn(async move {
                *alarm_cb.lock().await = callback;
            });
        }
    }

    fn clear_alarm_callback(&mut self) {
        let alarm_cb = self.alarm_callback.clone();
        if let Ok(handle) = tokio::runtime::Handle::try_current() {
            handle.spawn(async move {
                *alarm_cb.lock().await = None;
            });
        } else {
            tokio::spawn(async move {
                *alarm_cb.lock().await = None;
            });
        }
    }

    async fn start_alarm_monitoring(&mut self) -> Result<()> {
        let reply = self
            .get_command(
                "",
                Some(QCODES.get("AlarmSet").copied().unwrap_or(1500) as u32),
            )
            .await?;

        if let Some(ret) = reply.get("Ret").and_then(|r| r.as_u64())
            && !OK_CODES.contains(&(ret as u32))
        {
            return Err(crate::error::DVRIPError::ProtocolError(
                "Failed to start alarm monitoring".to_string(),
            ));
        }

        self.alarm_monitoring.store(true, Ordering::Release);
        self.start_alarm_worker().await;

        Ok(())
    }

    async fn stop_alarm_monitoring(&mut self) -> Result<()> {
        self.alarm_monitoring.store(false, Ordering::Release);

        if let Some(handle) = self.alarm_handle.lock().await.take() {
            handle.abort();
        }

        Ok(())
    }

    async fn set_remote_alarm(&mut self, state: bool) -> Result<bool> {
        let data = serde_json::json!({
            "Event": 0,
            "State": state,
        });

        let reply = self.set_command("OPNetAlarm", data, None).await?;
        if let Some(ret) = reply.get("Ret").and_then(|r| r.as_u64()) {
            return Ok(OK_CODES.contains(&(ret as u32)));
        }
        Ok(false)
    }

    fn is_alarm_monitoring(&self) -> bool {
        self.alarm_monitoring.load(Ordering::Acquire)
    }
}