Skip to main content

deep_time/alloc_parse/
parse_duration.rs

1use crate::{Dt, DtErr, DtErrKind, Lang, Scale, an_err, natural_duration_to_iso};
2use alloc::string::String;
3
4impl Dt {
5    /// Parses duration strings, tries formats in the following order:
6    ///
7    /// 1. Strict ISO 8601 e.g. **`P1DT2H30M`**
8    /// 2. Common natural-language formats e.g. **`2 wks, 3 days, and 2 mins`**
9    /// 3. Media duration format e.g. **`1:07:54:30`**
10    /// 4. Numerical milliseconds, decimals counted as fractional milliseconds
11    ///
12    /// Returns a [`Dt`].
13    pub fn from_str_duration(s: &str, lang: Lang) -> Result<Dt, DtErr> {
14        if s.is_empty() {
15            return Err(an_err!(DtErrKind::Incomplete, "empty"));
16        }
17
18        if Dt::looks_like_iso(s) {
19            return Dt::from_iso_duration(s).map_err(|e| {
20                an_err!(
21                    DtErrKind::InvalidInput,
22                    "iso: {}",
23                    s => e
24                )
25            });
26        }
27
28        let lower = s.to_lowercase();
29        if let Ok(dur) = Dt::from_natural_duration(&lower, lang, true) {
30            return Ok(dur);
31        }
32
33        if let Ok(dur) = Dt::from_str_media_duration(s) {
34            return Ok(dur);
35        }
36
37        if let Ok(ms) = s.parse::<f64>() {
38            if !ms.is_finite() {
39                return Err(an_err!(DtErrKind::OutOfRange, "{}", s));
40            }
41            let nanos = (ms * 1_000_000.0).round() as i128;
42            let span = Dt::from_ns(nanos, Scale::TAI);
43            return Ok(span);
44        }
45
46        Err(an_err!(DtErrKind::InvalidInput, "{}", s))
47    }
48
49    /// Converts a natural language duration into an ISO duration.
50    pub fn natural_to_iso(s: &str, lang: Lang) -> Result<String, DtErr> {
51        let lower = s.to_lowercase();
52        match natural_duration_to_iso(&lower, lang, true) {
53            Ok(iso) => Ok(iso),
54            Err(e) => Err(an_err!(
55                DtErrKind::InvalidInput,
56                "{}",
57                s => e
58            )),
59        }
60    }
61
62    /// Accepts: `P1Y`, `-P2W`, `PT1.5H`, `P1DT2H30M`, `+P3D`, `p1y`, `P1,5S`, `PT0S`, etc.
63    /// Rejects: anything with whitespace, lone "P"/"-P"/"PT", "P123", "Please wait 5m",
64    ///          "1.5h", "P1Yabc", "P1Y!", or **any string longer than 128 bytes**.
65    fn looks_like_iso(s: &str) -> bool {
66        let len = s.len();
67        if matches!(len, 0 | 1) {
68            return false;
69        }
70        let b = s.as_bytes();
71        let mut i = 0usize;
72        // Optional leading sign
73        if matches!(b[0], b'+' | b'-') {
74            i += 1;
75        }
76        // Must start with P/p after optional sign
77        if !matches!(b[i], b'P' | b'p') {
78            return false;
79        }
80        i += 1;
81        let mut has_digit = false;
82        let mut has_designator = false;
83        while i < len {
84            match b[i] {
85                b'0'..=b'9' => has_digit = true,
86                b'.' | b',' => {} // decimal separators allowed by ISO 8601
87                b'Y' | b'y' | b'M' | b'm' | b'W' | b'w' | b'D' | b'd' | b'T' | b't' | b'H'
88                | b'h' | b'S' | b's' => {
89                    has_designator = true;
90                }
91                _ => return false, // any other character = not ISO
92            }
93
94            i += 1;
95        }
96        // Must contain at least one digit *and* one designator after the initial P
97        has_digit && has_designator
98    }
99}