denki 0.1.2

CLI for controlling TP-Link smart home devices — denki (電気) means electricity in Japanese
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

//! RTSP stream URLs and snapshots, shared by the camera handlers.

use std::time::Duration;

use crate::tapo_vendor::api::rtsp_snapshot::grab_mjpeg_frame;
use crate::tapo_vendor::error::Error;
use crate::tapo_vendor::responses::{RtspStreamUrl, Snapshot};

/// Returns the RTSP stream URLs of the camera at `ip`, with the credentials
/// URL-encoded into them.
pub(crate) fn rtsp_stream_url(ip: &str, username: &str, password: &str) -> RtspStreamUrl {
    RtspStreamUrl {
        hd: rtsp_url(ip, "stream1", username, password),
        sd: rtsp_url(ip, "stream2", username, password),
        mjpeg: rtsp_url(ip, "stream8", username, password),
    }
}

/// Captures a JPEG snapshot from the MJPEG stream of the camera at `ip`, giving
/// up after `timeout`.
pub(crate) async fn snapshot(
    ip: &str,
    username: &str,
    password: &str,
    timeout: Duration,
) -> Result<Snapshot, Error> {
    // retina rejects URLs with embedded credentials when creds are also
    // passed via SessionOptions, so use the bare base URL here.
    let url = rtsp_url_base(ip, "stream8");
    let creds = retina::client::Credentials {
        username: username.into(),
        password: password.into(),
    };
    let data = grab_mjpeg_frame(&url, creds, timeout).await?;

    Ok(Snapshot {
        data,
        content_type: "image/jpeg".into(),
    })
}

fn rtsp_url_base(ip: &str, stream: &str) -> String {
    format!("rtsp://{ip}:554/{stream}")
}

fn rtsp_url(ip: &str, stream: &str, username: &str, password: &str) -> String {
    // safe: `ip` is the address the handler logged in to, so the URL parses
    let mut url = reqwest::Url::parse(&rtsp_url_base(ip, stream)).expect("valid RTSP base URL");
    // safe: an `rtsp://` URL with a host always accepts credentials
    url.set_username(username).expect("valid username");
    url.set_password(Some(password)).expect("valid password");
    url.to_string()
}