wmhttp 0.3.1

a http server for rust
Documentation 
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156

// Copyright 2022 - 2023 Wenmeng See the COPYRIGHT
// file at the top-level directory of this distribution.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.
//
// Author: tickbh
// -----
// Created Date: 2023/09/14 09:42:25

use algorithm::buf::{Binary, BinaryMut, Bt};
use webparse::http::http2::frame::PushPromise;

use std::task::Context;
use tokio::sync::mpsc::Sender;
use webparse::{
    http::http2::frame::{Data, Flag, Frame, FrameHeader, Headers, Kind, StreamIdentifier},
    Method,
};
use webparse::{HeaderMap, HeaderName, HeaderValue};

use crate::{ProtResult, RecvResponse};

#[derive(Debug)]
pub struct SendResponse {
    pub stream_id: StreamIdentifier,
    pub push_id: Option<StreamIdentifier>,
    pub response: RecvResponse,
    pub encode_header: bool,
    pub encode_body: bool,
    pub is_end_stream: bool,

    pub method: Method,
}

impl SendResponse {
    pub fn new(
        stream_id: StreamIdentifier,
        push_id: Option<StreamIdentifier>,
        response: RecvResponse,
        method: Method,
        is_end_stream: bool,
    ) -> Self {
        SendResponse {
            stream_id,
            push_id,
            response,
            encode_header: false,
            encode_body: false,
            is_end_stream,
            method,
        }
    }

    pub fn encode_headers(response: &RecvResponse) -> (HeaderMap, bool) {
        let mut headers = HeaderMap::new();
        let mut is_end = false;
        headers.insert(
            ":status",
            HeaderValue::from_static(response.status().as_str()),
        );
        for h in response.headers().iter() {
            // if h.0 != HeaderName::TRANSFER_ENCODING {
            //     headers.insert(h.0.clone(), h.1.clone());
            // }
            headers.insert(h.0.clone(), h.1.clone());
            if h.0 == HeaderName::CONTENT_LENGTH
                && TryInto::<isize>::try_into(&h.1).unwrap_or(0) == 0
            {
                is_end = true;
            }
        }
        (headers, is_end)
    }

    pub fn encode_frames(&mut self, cx: &mut Context) -> (bool, Vec<Frame<Binary>>) {
        let mut result = vec![];
        if !self.encode_header {
            if let Some(push_id) = &self.push_id {
                let header =
                    FrameHeader::new(Kind::PushPromise, Flag::end_headers(), self.stream_id);
                let (fields, is_end) = Self::encode_headers(&self.response);

                let mut push = PushPromise::new(header, push_id.clone(), fields);
                if is_end {
                    push.flags_mut().set_end_stream();
                }
                push.set_status(self.response.status());
                result.push(Frame::PushPromise(push));
                self.stream_id = push_id.clone();
                self.encode_header = true;
            } else {
                let header = FrameHeader::new(Kind::Headers, Flag::end_headers(), self.stream_id);
                let (fields, is_end) = Self::encode_headers(&self.response);
                let mut header = Headers::new(header, fields);
                if is_end {
                    header.flags_mut().set_end_stream();
                }
                header.set_status(self.response.status());
                result.push(Frame::Headers(header));
                self.encode_header = true;
            }
        }

        if !self.response.body().is_end() || !self.encode_body {
            self.encode_body = true;
            let mut binary = BinaryMut::new();
            let _ = self.response.body_mut().poll_encode_write(cx, &mut binary);
            if binary.remaining() > 0 {
                self.is_end_stream = self.response.body().is_end();
                let flag = if self.is_end_stream {
                    Flag::end_stream()
                } else {
                    Flag::zero()
                };
                let header = FrameHeader::new(Kind::Data, flag, self.stream_id);
                let data = Data::new(header, binary.freeze());
                result.push(Frame::Data(data));
            }
        }

        (self.is_end_stream, result)
    }
}

#[derive(Debug, Clone)]
pub struct SendControl {
    pub stream_id: StreamIdentifier,
    pub sender: Sender<(StreamIdentifier, RecvResponse)>,
    pub method: Method,
}

impl SendControl {
    pub fn new(
        stream_id: StreamIdentifier,
        sender: Sender<(StreamIdentifier, RecvResponse)>,
        method: Method,
    ) -> Self {
        SendControl {
            stream_id,
            sender,
            method,
        }
    }

    pub async fn send_response(&mut self, res: RecvResponse) -> ProtResult<()> {
        let _ = self.sender.send((self.stream_id, res)).await;
        Ok(())
    }
}

unsafe impl Sync for SendControl {}

unsafe impl Send for SendControl {}