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

use crate::{Constraints, Multipart};
use hyper::{header, Request};

/// An extension trait which extends [`hyper::Request<Body>`](https://docs.rs/hyper/0.14.7/hyper/struct.Request.html)
/// to add some methods to parse request body as `multipart/form-data`.
pub trait RequestMultipartExt {
    /// Convert the request body to [`Multipart`](./struct.Multipart.html) if the `content-type` is `multipart/form-data`.
    ///
    /// # Errors
    ///
    /// This method fails if the request body is not `multipart/form-data` and in this case, you could send a `400 Bad Request` status.
    fn into_multipart(self) -> routerify::Result<Multipart<'static>>;

    /// Convert the request body to [`Multipart`](./struct.Multipart.html) if the `content-type` is `multipart/form-data` with some [constraints](./struct.Constraints.html).
    ///
    /// # Errors
    ///
    /// This method fails if the request body is not `multipart/form-data` and in this case, you could send a `400 Bad Request` status.
    fn into_multipart_with_constraints(self, constraints: Constraints) -> routerify::Result<Multipart<'static>>;
}

impl RequestMultipartExt for Request<hyper::Body> {
    fn into_multipart(self) -> routerify::Result<Multipart<'static>> {
        let boundary = self
            .headers()
            .get(header::CONTENT_TYPE)
            .and_then(|val| val.to_str().ok())
            .and_then(|val| multer::parse_boundary(val).ok());

        if let Some(boundary) = boundary {
            Ok(Multipart::new(self.into_body(), boundary))
        } else {
            Err(Box::new(routerify::Error::new(
                "Content-Type is not multipart/form-data",
            )))
        }
    }

    fn into_multipart_with_constraints(self, constraints: Constraints) -> routerify::Result<Multipart<'static>> {
        let boundary = self
            .headers()
            .get(header::CONTENT_TYPE)
            .and_then(|val| val.to_str().ok())
            .and_then(|val| multer::parse_boundary(val).ok());

        if let Some(boundary) = boundary {
            Ok(Multipart::with_constraints(self.into_body(), boundary, constraints))
        } else {
            Err(Box::new(routerify::Error::new(
                "Content-Type is not multipart/form-data",
            )))
        }
    }
}