denki 0.1.1

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
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
110
111
112
113
114
115
116
117
118
119
120
121

use crate::bulb::LightingEffectState;
use colored::Colorize;

use crate::commands::KasaContext;
use crate::devices;
use crate::ops;

pub const BUILTIN_EFFECTS: &[&str] = &[
    "Off",
    "Aurora",
    "Bubbling Cauldron",
    "Candy Cane",
    "Christmas",
    "Flicker",
    "Hanukkah",
    "Haunted Mansion",
    "Icicle",
    "Lightning",
    "Ocean",
    "Rainbow",
    "Raindrop",
    "Spring",
    "Valentines",
];

pub fn resolve(name: &str) -> Option<&'static str> {
    let needle = crate::hosts::normalize(name);
    if needle.is_empty() {
        return None;
    }

    let mut exact = BUILTIN_EFFECTS
        .iter()
        .copied()
        .filter(|effect| crate::hosts::normalize(effect) == needle);
    if let Some(effect) = exact.next()
        && exact.next().is_none()
    {
        return Some(effect);
    }

    let mut fuzzy = BUILTIN_EFFECTS
        .iter()
        .copied()
        .filter(|effect| crate::hosts::normalize(effect).contains(&needle));
    if let Some(effect) = fuzzy.next()
        && fuzzy.next().is_none()
    {
        return Some(effect);
    }

    None
}

pub fn print_catalog(current: &LightingEffectState) {
    let state = if current.enable == 1 { "On" } else { "Off" };
    crate::output::println!("Current effect: {} ({state})", current.name.bold());
    crate::output::println!("{}", "Built-in effects:".bold());
    for effect in BUILTIN_EFFECTS {
        if effect.eq_ignore_ascii_case("off") {
            crate::output::println!("  {effect}");
        } else if effect.eq_ignore_ascii_case(&current.name) {
            crate::output::println!("  {effect}  {}", "(selected)".dimmed());
        } else {
            crate::output::println!("  {effect}");
        }
    }
}

pub async fn handle_effects_command(host: &str) -> anyhow::Result<()> {
    let ctx = KasaContext::load(host, "effects").await?;
    devices::can_get_effects(ctx.kind())?;
    let current = ops::lightstrip_current_effect(ctx.ip()).await?;
    crate::output::record(serde_json::json!({"current":current,"available":BUILTIN_EFFECTS}));
    print_catalog(&current);
    Ok(())
}

pub async fn handle_effect_command(host: &str, name: &str) -> anyhow::Result<()> {
    let ctx = KasaContext::load(host, "effect").await?;
    devices::can_get_effects(ctx.kind())?;

    if name.eq_ignore_ascii_case("off") {
        ops::lightstrip_disable_effect(ctx.ip()).await?;
        crate::output::record(serde_json::json!({"effect":"Off"}));
        crate::output::println!("Effect -> {}", "Off".dimmed());
        return Ok(());
    }

    let resolved = resolve(name).ok_or_else(|| {
        anyhow::anyhow!(
            "No light-strip effect named \"{name}\" found. Available effects: {}",
            BUILTIN_EFFECTS.join(", ")
        )
    })?;
    let current = ops::lightstrip_current_effect(ctx.ip()).await?;
    ops::lightstrip_set_effect(ctx.ip(), &current, resolved).await?;
    crate::output::record(serde_json::json!({"effect":resolved}));
    crate::output::println!("Effect -> {}", resolved.bold());
    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn resolve_matches_exact_and_partial_names() {
        assert_eq!(resolve("Rainbow"), Some("Rainbow"));
        assert_eq!(resolve("bubbl"), Some("Bubbling Cauldron"));
        assert_eq!(resolve("Bubbling Cauldron"), Some("Bubbling Cauldron"));
    }

    #[test]
    fn resolve_rejects_unknown_or_ambiguous_names() {
        assert_eq!(resolve(""), None);
        assert_eq!(resolve("x"), None);
        assert_eq!(resolve("rain"), None);
        assert_eq!(resolve("ca"), None);
    }
}