tuftool 0.16.0

Utility for creating and signing The Update Framework (TUF) repositories
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

// Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: MIT OR Apache-2.0

use crate::error::{self, Result};

use jiff::{SignedDuration, Timestamp};
use snafu::{ensure, ResultExt};

/// Parses a user-specified datetime, either in full RFC 3339 format, or a shorthand like "in 7
/// days"
pub(crate) fn parse_datetime(input: &str) -> Result<Timestamp> {
    // If the user gave an absolute date in a standard format, accept it.
    if let Ok(ts) = input.parse::<Timestamp>() {
        return Ok(ts);
    }

    // Otherwise, pull apart a request like "in 5 days" to get an exact datetime.
    let mut parts: Vec<&str> = input.split_whitespace().collect();
    ensure!(
        parts.len() == 3,
        error::DateArgInvalidSnafu {
            input,
            msg: "expected RFC 3339, or 3 space-separated values like 'in 7 days'",
        }
    );
    let unit_str = parts.pop().unwrap();
    let count_str = parts.pop().unwrap();
    let prefix = parts.pop().unwrap();
    ensure!(
        prefix == "in",
        error::DateArgInvalidSnafu {
            input,
            msg: "expected RFC 3339, or prefix 'in', something like 'in 7 days'",
        }
    );

    let count: u32 = count_str
        .parse()
        .context(error::DateArgCountSnafu { input })?;

    let duration = match unit_str {
        "hour" | "hours" => SignedDuration::from_hours(i64::from(count)),
        "day" | "days" => i64::from(count)
            .checked_mul(24)
            .map(SignedDuration::from_hours)
            .ok_or_else(|| {
                error::DateArgInvalidSnafu {
                    input: count.to_string(),
                    msg: format!("unable to convert {count} to a number of days"),
                }
                .build()
            })?,
        "week" | "weeks" => i64::from(count)
            .checked_mul(24 * 7)
            .map(SignedDuration::from_hours)
            .ok_or_else(|| {
                error::DateArgInvalidSnafu {
                    input: count.to_string(),
                    msg: format!("unable to convert {count} to a number of weeks"),
                }
                .build()
            })?,
        _ => {
            return error::DateArgInvalidSnafu {
                input,
                msg: "date argument's unit must be hours/days/weeks",
            }
            .fail();
        }
    };

    let now = Timestamp::now();
    let then = now + duration;
    Ok(then)
}