h3i/prompts/h3/
wait.rs

1// Copyright (C) 2024, Cloudflare, Inc.
2// All rights reserved.
3//
4// Redistribution and use in source and binary forms, with or without
5// modification, are permitted provided that the following conditions are
6// met:
7//
8//     * Redistributions of source code must retain the above copyright notice,
9//       this list of conditions and the following disclaimer.
10//
11//     * Redistributions in binary form must reproduce the above copyright
12//       notice, this list of conditions and the following disclaimer in the
13//       documentation and/or other materials provided with the distribution.
14//
15// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
16// IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
17// THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
18// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
19// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
20// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
22// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
23// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
24// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
25// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26
27use std::time::Duration;
28
29use inquire::error::InquireResult;
30use inquire::validator::Validation;
31use inquire::Text;
32
33use crate::actions::h3::Action;
34use crate::actions::h3::StreamEvent;
35use crate::actions::h3::StreamEventType;
36use crate::actions::h3::WaitType;
37
38use super::prompt_stream_id;
39use super::squish_suggester;
40use super::validate_wait_period;
41use super::SuggestionResult;
42
43const DURATION: &str = "duration";
44const HEADERS: &str = "headers";
45const DATA: &str = "data";
46const FINISHED: &str = "stream finished";
47
48pub fn prompt_wait() -> InquireResult<Action> {
49    let wait_type = Text::new("wait type:")
50        .with_autocomplete(&wait_type_suggestor)
51        .with_validator(wait_type_validator)
52        .prompt()?;
53
54    let actual = match wait_type.as_str() {
55        DURATION => Some(prompt_wait_period()),
56        t @ (HEADERS | DATA | FINISHED) => Some(prompt_stream_wait(t)),
57        _ => None,
58    };
59
60    let action = Action::Wait {
61        // unwrap should be safe due to validation
62        wait_type: actual.unwrap()?,
63    };
64
65    Ok(action)
66}
67
68fn wait_type_suggestor(val: &str) -> SuggestionResult<Vec<String>> {
69    let suggestions = [DURATION, HEADERS, DATA, FINISHED];
70
71    squish_suggester(&suggestions, val)
72}
73
74fn wait_type_validator(wait_type: &str) -> SuggestionResult<Validation> {
75    match wait_type {
76        DURATION | HEADERS | DATA | FINISHED => Ok(Validation::Valid),
77        _ => Ok(Validation::Invalid(
78            inquire::validator::ErrorMessage::Default,
79        )),
80    }
81}
82
83fn prompt_stream_wait(stream_wait_type: &str) -> InquireResult<WaitType> {
84    let stream_id = prompt_stream_id()?;
85
86    let event_type = if let HEADERS = stream_wait_type {
87        Some(StreamEventType::Headers)
88    } else if let DATA = stream_wait_type {
89        Some(StreamEventType::Data)
90    } else if let FINISHED = stream_wait_type {
91        Some(StreamEventType::Finished)
92    } else {
93        None
94    }
95    // If somehow we've gotten an invalid input, we can panic. This is post validation so that
96    // shouldn't happen
97    .unwrap();
98
99    Ok(WaitType::StreamEvent(StreamEvent {
100        stream_id,
101        event_type,
102    }))
103}
104
105pub fn prompt_wait_period() -> InquireResult<WaitType> {
106    let period = Text::new("wait period (ms):")
107        .with_validator(validate_wait_period)
108        .prompt()?;
109
110    // period is already validated so unwrap always succeeds
111    let period = Duration::from_millis(period.parse::<u64>().unwrap());
112
113    Ok(WaitType::WaitDuration(period))
114}
h3i/prompts/h3/wait.rs

h3i/prompts/h3/
wait.rs
