[pbs-devel] [PATCH proxmox-backup 5/9] DailyDuration: implement time_match()

[Dietmar Maurer <dietmar@proxmox.com>]

---
 proxmox-systemd/src/daily_duration.rs | 60 +++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)

diff --git a/proxmox-systemd/src/daily_duration.rs b/proxmox-systemd/src/daily_duration.rs
index bed4eb47..25157b25 100644
--- a/proxmox-systemd/src/daily_duration.rs
+++ b/proxmox-systemd/src/daily_duration.rs
@@ -1,4 +1,9 @@
 use std::cmp::{Ordering, PartialOrd};
+use std::convert::TryInto;
+
+use anyhow::Error;
+
+use proxmox_time::TmEditor;
 
 use super::time::{WeekDays};
 
@@ -29,6 +34,30 @@ pub struct DailyDuration {
     pub end: HmTime,
 }
 
+impl DailyDuration {
+
+    // Test it time is within this frame
+    pub fn time_match(&self, epoch: i64, utc: bool) -> Result<bool, Error> {
+
+        let t = TmEditor::with_epoch(epoch, utc)?;
+
+        let all_days = self.days.is_empty() || self.days.is_all();
+
+        if !all_days { // match day first
+            let day_num: u32 = t.day_num().try_into()?;
+            let day = WeekDays::from_bits(1<<day_num).unwrap();
+            if !self.days.contains(day) { return Ok(false); }
+        }
+
+        let hour = t.hour().try_into()?;
+        let minute = t.min().try_into()?;
+
+        let ctime = HmTime { hour, minute };
+
+        Ok(ctime >= self.start && ctime < self.end)
+    }
+}
+
 #[cfg(test)]
 mod test {
 
@@ -68,6 +97,10 @@ mod test {
         Ok(())
     }
 
+    const fn make_test_time(mday: i32, hour: i32, min: i32) -> i64 {
+        (mday*3600*24 + hour*3600 + min*60) as i64
+    }
+
     #[test]
     fn test_daily_duration_parser() -> Result<(), Error> {
 
@@ -89,4 +122,31 @@ mod test {
 
         Ok(())
     }
+
+    #[test]
+    fn test_time_match() -> Result<(), Error> {
+        const THURSDAY_80_00: i64 = make_test_time(0, 8, 0);
+        const THURSDAY_12_00: i64 = make_test_time(0, 12, 0);
+        const DAY: i64 = 3600*24;
+
+        let duration = parse_daily_duration("thu..fri 8:05-12")?;
+
+        assert!(!duration.time_match(THURSDAY_80_00, true)?);
+        assert!(!duration.time_match(THURSDAY_80_00 + DAY, true)?);
+        assert!(!duration.time_match(THURSDAY_80_00 + 2*DAY, true)?);
+
+        assert!(duration.time_match(THURSDAY_80_00 + 5*60, true)?);
+        assert!(duration.time_match(THURSDAY_80_00 + 5*60 + DAY, true)?);
+        assert!(!duration.time_match(THURSDAY_80_00 + 5*60 + 2*DAY, true)?);
+
+        assert!(duration.time_match(THURSDAY_12_00 - 1, true)?);
+        assert!(duration.time_match(THURSDAY_12_00 - 1 + DAY, true)?);
+        assert!(!duration.time_match(THURSDAY_12_00 - 1 + 2*DAY, true)?);
+
+        assert!(!duration.time_match(THURSDAY_12_00, true)?);
+        assert!(!duration.time_match(THURSDAY_12_00 + DAY, true)?);
+        assert!(!duration.time_match(THURSDAY_12_00 + 2*DAY, true)?);
+
+        Ok(())
+    }
 }
-- 
2.30.2