use super::common::{in_window, PrimitiveConfig, PrimitiveState, RawSnapshot, Reason};
#[derive(Debug, Default, Clone)]
pub struct BedExit {
in_bed: bool,
}
impl BedExit {
pub fn new() -> Self { Self::default() }
fn in_bed_zone(snap: &RawSnapshot) -> bool {
!snap.bed_zones.is_empty()
&& snap.active_zones.iter().any(|z| snap.bed_zones.contains(z))
}
pub fn tick(&mut self, snap: &RawSnapshot, cfg: &PrimitiveConfig) -> PrimitiveState {
if snap.since_start < cfg.warmup {
return PrimitiveState::Idle;
}
let now_in_bed = snap.presence && Self::in_bed_zone(snap);
let was_in_bed = self.in_bed;
self.in_bed = now_in_bed;
if was_in_bed && !now_in_bed {
let (start, end) = cfg.bed_exit_window;
if in_window(snap.local_seconds_since_midnight, start, end) {
return PrimitiveState::Event {
event_type: "bed_exit",
reason: Reason::new(&[
"left_bed_zone",
"overnight_window",
]),
};
}
}
PrimitiveState::Idle
}
}
#[cfg(test)]
mod tests {
use super::*;
use std::time::Duration;
fn cfg() -> PrimitiveConfig { PrimitiveConfig::default() }
fn in_bed_overnight(t: u64) -> RawSnapshot {
RawSnapshot {
since_start: Duration::from_secs(120 + t),
presence: true,
active_zones: vec!["bedroom".into()],
bed_zones: vec!["bedroom".into()],
local_seconds_since_midnight: 2 * 3600, ..Default::default()
}
}
fn out_of_bed_overnight(t: u64) -> RawSnapshot {
RawSnapshot {
since_start: Duration::from_secs(120 + t),
presence: true,
active_zones: vec!["hall".into()],
bed_zones: vec!["bedroom".into()],
local_seconds_since_midnight: 2 * 3600,
..Default::default()
}
}
#[test]
fn fires_on_bed_to_non_bed_overnight() {
let mut p = BedExit::new();
let _ = p.tick(&in_bed_overnight(10), &cfg());
let state = p.tick(&out_of_bed_overnight(20), &cfg());
assert!(matches!(state, PrimitiveState::Event { event_type: "bed_exit", .. }));
}
#[test]
fn does_not_fire_during_day() {
let mut p = BedExit::new();
let mut s_in = in_bed_overnight(10);
s_in.local_seconds_since_midnight = 14 * 3600; let _ = p.tick(&s_in, &cfg());
let mut s_out = out_of_bed_overnight(20);
s_out.local_seconds_since_midnight = 14 * 3600;
let state = p.tick(&s_out, &cfg());
assert!(matches!(state, PrimitiveState::Idle));
}
#[test]
fn does_not_fire_without_prior_in_bed() {
let mut p = BedExit::new();
let state = p.tick(&out_of_bed_overnight(20), &cfg());
assert!(matches!(state, PrimitiveState::Idle));
}
#[test]
fn warmup_blocks_initial_transitions() {
let mut p = BedExit::new();
let mut s_in = in_bed_overnight(0);
s_in.since_start = Duration::from_secs(30);
assert!(matches!(p.tick(&s_in, &cfg()), PrimitiveState::Idle));
}
#[test]
fn does_not_fire_when_bed_zones_unconfigured() {
let mut p = BedExit::new();
let mut s_in = in_bed_overnight(10);
s_in.bed_zones.clear();
let _ = p.tick(&s_in, &cfg());
let mut s_out = out_of_bed_overnight(20);
s_out.bed_zones.clear();
let state = p.tick(&s_out, &cfg());
assert!(matches!(state, PrimitiveState::Idle));
}
#[test]
fn fires_just_after_midnight_window_start() {
let mut p = BedExit::new();
let mut s_in = in_bed_overnight(10);
s_in.local_seconds_since_midnight = 22 * 3600 + 5; let _ = p.tick(&s_in, &cfg());
let mut s_out = out_of_bed_overnight(20);
s_out.local_seconds_since_midnight = 22 * 3600 + 10;
let state = p.tick(&s_out, &cfg());
assert!(matches!(state, PrimitiveState::Event { .. }));
}
}