h3i/prompts/h3/
priority.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 inquire::error::InquireResult;
28use inquire::Select;
29use inquire::Text;
30
31use super::stream::prompt_fin_stream;
32use crate::actions::h3::Action;
33use crate::prompts::h3;
34
35use quiche;
36
37const REQUEST: &str = "request";
38const PUSH: &str = "push";
39
40pub fn prompt_priority() -> InquireResult<Action> {
41    let stream_id = h3::prompt_stream_id()?;
42
43    let ty = prompt_request_or_push()?;
44    let prioritized_element_id = h3::prompt_varint("Prioritized Element ID:")?;
45
46    let priority_field_value = Text::new("priority field value:").prompt()?;
47
48    let frame = if ty.as_str() == REQUEST {
49        quiche::h3::frame::Frame::PriorityUpdateRequest {
50            prioritized_element_id,
51            priority_field_value: priority_field_value.into(),
52        }
53    } else {
54        quiche::h3::frame::Frame::PriorityUpdatePush {
55            prioritized_element_id,
56            priority_field_value: priority_field_value.into(),
57        }
58    };
59
60    let fin_stream = prompt_fin_stream()?;
61
62    let action = Action::SendFrame {
63        stream_id,
64        fin_stream,
65        frame,
66    };
67
68    Ok(action)
69}
70
71fn prompt_request_or_push() -> InquireResult<String> {
72    Ok(Select::new("request or push:", vec![REQUEST, PUSH])
73        .prompt()?
74        .to_string())
75}
h3i/prompts/h3/priority.rs

h3i/prompts/h3/
priority.rs
