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

#![deny(missing_docs, missing_debug_implementations)]

//! Middleware and friends that help deal with protobuf messages in tower-web.
//!
//! <TODO>

#[macro_use(try_ready)]
extern crate futures;

pub(crate) mod common {
    pub use futures::future::{Err as FutErr, Future, FutureResult, Ok as FutOk};
    pub use futures::Poll;
    pub use http::{header::HeaderName, Request as HttpRequest, Response as HttpResponse};
    pub use tower_service::Service;

    #[derive(Debug)]
    pub struct ResponseFuture<T> {
        pub response: T,
    }

    impl<F, RespBody> Future for ResponseFuture<F>
    where
        F: Future<Item = HttpResponse<RespBody>>,
    {
        type Item = F::Item;
        type Error = F::Error;

        // Just pass the response through unmodified:
        fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
            self.response.poll()
        }
    }
}

pub(crate) mod errors;
pub(crate) mod extensions;
pub(crate) mod extractors;
pub(crate) mod middleware;
pub(crate) mod response;
pub(crate) mod types;

pub use middleware::ProtobufMiddleware;
pub use types::{MessagePlus, Proto};

// TODO: deny missing docs
// TODO: check protobuf message name with type_info (feature gated, perhaps)
// TODO: with fork
// TODO: fix Errors in Tower Web