denki 0.1.2

use super::support::{
    make_dimmer_for_hints, make_lightstrip_for_hints, make_plug_for_hints, make_strip_for_hints,
};
use crate::display::hints::{dimmer_hints, lightstrip_hints, plug_hints, strip_hints};

#[test]
fn plug_hints_ene_plug_includes_energy() {
    let p = make_plug_for_hints("KP115", false, true);
    let h = plug_hints(&p, "plug");
    assert!(h.iter().any(|s| s.contains("energy")), "hints: {h:?}");
}

#[test]
fn plug_hints_no_ene_excludes_energy() {
    let p = make_plug_for_hints("HS105", false, false);
    let h = plug_hints(&p, "plug");
    assert!(!h.iter().any(|s| s.contains("energy")), "hints: {h:?}");
}

#[test]
fn plug_hints_on_plug_starts_with_off() {
    let p = make_plug_for_hints("KP115", true, true);
    assert_eq!(plug_hints(&p, "p")[0], "denki off 'p'");
}

#[test]
fn plug_hints_off_plug_starts_with_on() {
    let p = make_plug_for_hints("KP115", false, true);
    assert_eq!(plug_hints(&p, "p")[0], "denki on 'p'");
}

#[test]
fn strip_hints_ene_strip_includes_per_outlet_energy() {
    let s = make_strip_for_hints("HS300", true);
    let h = strip_hints(&s, "strip");
    assert!(
        h.iter().any(|s| s.contains("energy") && s.contains(" 1")),
        "hints: {h:?}"
    );
}

#[test]
fn strip_hints_no_ene_excludes_energy() {
    let s = make_strip_for_hints("KP303", false);
    let h = strip_hints(&s, "strip");
    assert!(!h.iter().any(|s| s.contains("energy")), "hints: {h:?}");
}

#[test]
fn strip_hints_always_includes_per_outlet_on_off() {
    let s = make_strip_for_hints("HS300", true);
    let h = strip_hints(&s, "strip");
    assert!(h.iter().any(|s| s.contains("on") && s.contains(" 1")));
    assert!(h.iter().any(|s| s.contains("off") && s.contains(" 1")));
}

#[test]
fn strip_hints_includes_outlet_rename() {
    let s = make_strip_for_hints("HS300", true);
    let h = strip_hints(&s, "strip");
    assert!(
        h.iter().any(|s| s.contains("outlet-rename")),
        "hints: {h:?}"
    );
}

#[test]
fn lightstrip_hints_for_kl420l5_include_energy_and_monthly_commands() {
    let b = make_lightstrip_for_hints("KL420L5", false);
    let h = lightstrip_hints(&b, "lights");
    assert!(
        h.iter().any(|s| s.contains("energy 'lights'")),
        "hints: {h:?}"
    );
    assert!(
        h.iter().any(|s| s.contains("energy-daily 'lights'")),
        "hints: {h:?}"
    );
    assert!(
        h.iter().any(|s| s.contains("energy-monthly 'lights'")),
        "hints: {h:?}"
    );
    assert!(
        h.iter().any(|s| s == "denki on 'lights'"),
        "the registry advertises light-strip power support: {h:?}"
    );
}

#[test]
fn hints_posix_quote_untrusted_aliases() {
    let p = make_plug_for_hints("KP115", false, true);
    let alias = "lamp'; touch /tmp/pwned; echo '$HOME`id`$(id) next";
    let hints = plug_hints(&p, alias);

    assert_eq!(
        hints[0],
        "denki on 'lamp'\"'\"'; touch /tmp/pwned; echo '\"'\"'$HOME`id`$(id) next'"
    );
}

#[test]
fn dimmer_hints_includes_dim_and_schedules() {
    let d = make_dimmer_for_hints();
    let h = dimmer_hints(&d, "d");
    assert!(h.iter().any(|s| s.contains("dim")), "hints: {h:?}");
    assert!(h.iter().any(|s| s.contains("schedules")), "hints: {h:?}");
}

#[test]
fn all_hint_paths_suppress_unsafe_or_untrusted_targets() {
    use crate::display::hints::{bulb_hints, model_hints, power_hints};
    for alias in [
        "",
        "bad\nline",
        "bad\rline",
        "bad\tline",
        "bad\x1b[2J",
        "bad\u{9b}2J",
        "bad\u{202e}",
        "--help",
    ] {
        assert!(bulb_hints(&make_lightstrip_for_hints("KL135", false), alias).is_empty());
        assert!(lightstrip_hints(&make_lightstrip_for_hints("KL420L5", false), alias).is_empty());
        assert!(plug_hints(&make_plug_for_hints("KP115", false, true), alias).is_empty());
        assert!(dimmer_hints(&make_dimmer_for_hints(), alias).is_empty());
        assert!(strip_hints(&make_strip_for_hints("HS300", true), alias).is_empty());
        assert!(model_hints("unknown", alias, false).is_empty());
        assert!(power_hints(alias, false).is_empty());
        assert!(crate::devices::hint_for("dim", alias).is_none());
        assert!(
            crate::devices::hints(crate::devices::lookup("KP115").unwrap(), alias, false)
                .is_empty()
        );
    }
}

#[cfg(unix)]
#[test]
fn quoted_hints_pass_the_exact_alias_to_a_posix_shell() {
    use crate::display::hints::power_hints;
    // Stub denki; nothing is sent to a device or executed from the hostile alias.
    let alias =
        "lamp'\"; printf INJECTED; # $HOME `printf SUBSTITUTED` $(printf SUBSTITUTED)\\suffix";
    for hint in power_hints(alias, false)
        .into_iter()
        .chain(strip_hints(&make_strip_for_hints("HS300", true), alias))
    {
        let script = format!("denki() {{ printf '%s\\0' \"$@\"; }}; {hint}");
        let result = std::process::Command::new("sh")
            .args(["-c", &script])
            .output()
            .unwrap();
        assert!(result.status.success(), "{hint}");
        assert!(result.stderr.is_empty());
        let words: Vec<_> = result.stdout.split(|b| *b == 0).collect();
        assert_eq!(words[1], alias.as_bytes(), "{hint}");
    }
}