1
0
mirror of https://github.com/fafhrd91/actix-web synced 2024-11-28 01:52:57 +01:00
actix-web/src/extract.rs

1300 lines
35 KiB
Rust
Raw Normal View History

2019-03-02 20:53:05 +01:00
#![allow(dead_code)]
use std::ops::{Deref, DerefMut};
2018-06-12 23:47:47 +02:00
use std::rc::Rc;
2018-06-21 07:47:01 +02:00
use std::{fmt, str};
2018-04-03 01:19:18 +02:00
use bytes::Bytes;
use encoding::all::UTF_8;
2018-04-14 01:02:01 +02:00
use encoding::types::{DecoderTrap, Encoding};
2019-03-02 07:51:32 +01:00
use futures::future::{err, ok, Either, FutureResult};
use futures::{future, Async, Future, IntoFuture, Poll, Stream};
2018-04-14 01:02:01 +02:00
use mime::Mime;
use serde::de::{self, DeserializeOwned};
2019-03-02 07:51:32 +01:00
use serde::Serialize;
use serde_json;
2018-04-14 01:02:01 +02:00
use serde_urlencoded;
2019-03-02 07:51:32 +01:00
use actix_http::dev::{JsonBody, MessageBody, UrlEncoded};
use actix_http::error::{
Error, ErrorBadRequest, ErrorNotFound, JsonPayloadError, PayloadError,
UrlencodedError,
};
use actix_http::http::StatusCode;
use actix_http::{Extensions, HttpMessage, Response};
2019-03-02 07:51:32 +01:00
use actix_router::PathDeserializer;
use crate::request::HttpRequest;
use crate::responder::Responder;
use crate::service::ServiceFromRequest;
/// Trait implemented by types that can be extracted from request.
///
/// Types that implement this trait can be used with `Route` handlers.
pub trait FromRequest<P>: Sized {
/// The associated error which can be returned.
type Error: Into<Error>;
/// Future that resolves to a Self
type Future: IntoFuture<Item = Self, Error = Self::Error>;
/// Configuration for the extractor
type Config: ExtractorConfig;
/// Convert request to a Self
fn from_request(req: &mut ServiceFromRequest<P>) -> Self::Future;
}
/// Storage for extractor configs
#[derive(Default)]
pub struct ConfigStorage {
pub(crate) storage: Option<Rc<Extensions>>,
}
impl ConfigStorage {
pub fn store<C: ExtractorConfig>(&mut self, config: C) {
if self.storage.is_none() {
self.storage = Some(Rc::new(Extensions::new()));
}
if let Some(ref mut ext) = self.storage {
Rc::get_mut(ext).unwrap().insert(config);
}
}
}
pub trait ExtractorConfig: Default + Clone + 'static {
/// Set default configuration to config storage
fn store_default(ext: &mut ConfigStorage) {
ext.store(Self::default())
}
}
impl ExtractorConfig for () {
fn store_default(_: &mut ConfigStorage) {}
}
#[derive(PartialEq, Eq, PartialOrd, Ord)]
2019-03-02 07:51:32 +01:00
/// Extract typed information from the request's path.
///
/// ## Example
///
/// ```rust
/// use actix_web::{web, http, App, extract::Path};
///
2018-04-18 01:22:25 +02:00
/// /// extract path info from "/{username}/{count}/index.html" url
/// /// {username} - deserializes to a String
/// /// {count} - - deserializes to a u32
/// fn index(info: Path<(String, u32)>) -> String {
/// format!("Welcome {}! {}", info.0, info.1)
/// }
///
/// fn main() {
/// let app = App::new().resource(
2018-06-01 18:37:14 +02:00
/// "/{username}/{count}/index.html", // <- define path parameters
/// |r| r.route(web::get().to(index)) // <- register handler with `Path` extractor
/// );
/// }
/// ```
///
2018-04-14 01:02:01 +02:00
/// It is possible to extract path information to a specific type that
/// implements `Deserialize` trait from *serde*.
///
/// ```rust
/// #[macro_use] extern crate serde_derive;
/// use actix_web::{web, App, extract::Path, Error};
///
/// #[derive(Deserialize)]
/// struct Info {
/// username: String,
/// }
///
/// /// extract `Info` from a path using serde
/// fn index(info: Path<Info>) -> Result<String, Error> {
/// Ok(format!("Welcome {}!", info.username))
/// }
///
/// fn main() {
/// let app = App::new().resource(
2018-06-01 18:37:14 +02:00
/// "/{username}/index.html", // <- define path parameters
/// |r| r.route(web::get().to(index)) // <- use handler with Path` extractor
/// );
/// }
/// ```
2018-04-14 01:02:01 +02:00
pub struct Path<T> {
inner: T,
}
impl<T> AsRef<T> for Path<T> {
fn as_ref(&self) -> &T {
&self.inner
}
}
impl<T> Deref for Path<T> {
type Target = T;
fn deref(&self) -> &T {
&self.inner
}
}
impl<T> DerefMut for Path<T> {
fn deref_mut(&mut self) -> &mut T {
&mut self.inner
}
}
impl<T> Path<T> {
/// Deconstruct to an inner value
pub fn into_inner(self) -> T {
self.inner
}
2019-03-02 07:51:32 +01:00
/// Extract path information from a request
pub fn extract(req: &HttpRequest) -> Result<Path<T>, de::value::Error>
2019-03-02 07:51:32 +01:00
where
T: DeserializeOwned,
{
de::Deserialize::deserialize(PathDeserializer::new(req.match_info()))
.map(|inner| Path { inner })
}
}
impl<T> From<T> for Path<T> {
fn from(inner: T) -> Path<T> {
2018-08-23 18:48:01 +02:00
Path { inner }
}
}
2019-03-02 07:51:32 +01:00
impl<T, P> FromRequest<P> for Path<T>
2018-04-14 01:02:01 +02:00
where
T: DeserializeOwned,
{
2019-03-02 07:51:32 +01:00
type Error = Error;
type Future = FutureResult<Self, Error>;
2019-03-03 09:57:48 +01:00
type Config = ();
#[inline]
fn from_request(req: &mut ServiceFromRequest<P>) -> Self::Future {
2019-03-02 07:51:32 +01:00
Self::extract(req).map_err(ErrorNotFound).into_future()
}
}
impl<T: fmt::Debug> fmt::Debug for Path<T> {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
self.inner.fmt(f)
}
}
impl<T: fmt::Display> fmt::Display for Path<T> {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
self.inner.fmt(f)
}
}
#[derive(PartialEq, Eq, PartialOrd, Ord)]
2019-03-02 07:51:32 +01:00
/// Extract typed information from from the request's query.
///
/// ## Example
///
/// ```rust
2018-06-01 19:27:23 +02:00
/// #[macro_use] extern crate serde_derive;
/// use actix_web::{web, extract, App};
///
///#[derive(Debug, Deserialize)]
///pub enum ResponseType {
/// Token,
/// Code
///}
///
///#[derive(Deserialize)]
///pub struct AuthRequest {
/// id: u64,
/// response_type: ResponseType,
///}
///
/// // Use `Query` extractor for query information.
/// // This handler get called only if request's query contains `username` field
/// // The correct request for this handler would be `/index.html?id=64&response_type=Code"`
/// fn index(info: extract::Query<AuthRequest>) -> String {
/// format!("Authorization request for client with id={} and type={:?}!", info.id, info.response_type)
/// }
///
/// fn main() {
/// let app = App::new().resource(
/// "/index.html",
/// |r| r.route(web::get().to(index))); // <- use `Query` extractor
/// }
/// ```
pub struct Query<T>(T);
impl<T> Deref for Query<T> {
type Target = T;
fn deref(&self) -> &T {
&self.0
}
}
impl<T> DerefMut for Query<T> {
fn deref_mut(&mut self) -> &mut T {
&mut self.0
}
}
impl<T> Query<T> {
/// Deconstruct to a inner value
pub fn into_inner(self) -> T {
self.0
}
}
2019-03-02 07:51:32 +01:00
impl<T, P> FromRequest<P> for Query<T>
2018-04-14 01:02:01 +02:00
where
T: de::DeserializeOwned,
{
2019-03-02 07:51:32 +01:00
type Error = Error;
type Future = FutureResult<Self, Error>;
2019-03-03 09:57:48 +01:00
type Config = ();
#[inline]
fn from_request(req: &mut ServiceFromRequest<P>) -> Self::Future {
2018-05-02 02:19:15 +02:00
serde_urlencoded::from_str::<T>(req.query_string())
2019-03-02 07:51:32 +01:00
.map(|val| ok(Query(val)))
.unwrap_or_else(|e| err(e.into()))
}
}
impl<T: fmt::Debug> fmt::Debug for Query<T> {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
self.0.fmt(f)
}
}
impl<T: fmt::Display> fmt::Display for Query<T> {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
self.0.fmt(f)
}
}
#[derive(PartialEq, Eq, PartialOrd, Ord)]
/// Extract typed information from the request's body.
///
2018-04-14 01:02:01 +02:00
/// To extract typed information from request's body, the type `T` must
/// implement the `Deserialize` trait from *serde*.
///
2018-04-06 19:24:57 +02:00
/// [**FormConfig**](dev/struct.FormConfig.html) allows to configure extraction
/// process.
///
/// ## Example
///
/// ```rust
/// # extern crate actix_web;
/// #[macro_use] extern crate serde_derive;
/// use actix_web::{web, App, extract::Form};
///
/// #[derive(Deserialize)]
/// struct FormData {
/// username: String,
/// }
///
/// /// Extract form data using serde.
/// /// This handler get called only if content type is *x-www-form-urlencoded*
/// /// and content of the request could be deserialized to a `FormData` struct
/// fn index(form: Form<FormData>) -> String {
/// format!("Welcome {}!", form.username)
/// }
/// # fn main() {}
/// ```
pub struct Form<T>(pub T);
impl<T> Form<T> {
/// Deconstruct to an inner value
pub fn into_inner(self) -> T {
self.0
}
}
impl<T> Deref for Form<T> {
type Target = T;
fn deref(&self) -> &T {
&self.0
}
}
impl<T> DerefMut for Form<T> {
fn deref_mut(&mut self) -> &mut T {
&mut self.0
}
}
2019-03-02 07:51:32 +01:00
impl<T, P> FromRequest<P> for Form<T>
2018-04-14 01:02:01 +02:00
where
T: DeserializeOwned + 'static,
2019-03-02 07:51:32 +01:00
P: Stream<Item = Bytes, Error = PayloadError> + 'static,
{
2019-03-02 07:51:32 +01:00
type Error = Error;
type Future = Box<Future<Item = Self, Error = Error>>;
2019-03-03 09:57:48 +01:00
type Config = FormConfig;
#[inline]
fn from_request(req: &mut ServiceFromRequest<P>) -> Self::Future {
2018-06-25 06:58:04 +02:00
let req2 = req.clone();
2019-03-03 09:57:48 +01:00
let cfg = req.load_config::<FormConfig>();
let limit = cfg.limit;
2018-06-12 23:47:47 +02:00
let err = Rc::clone(&cfg.ehandler);
2018-04-29 18:09:08 +02:00
Box::new(
2018-06-25 06:58:04 +02:00
UrlEncoded::new(req)
2019-03-03 09:57:48 +01:00
.limit(limit)
2018-06-25 06:58:04 +02:00
.map_err(move |e| (*err)(e, &req2))
2018-04-29 18:09:08 +02:00
.map(Form),
)
2018-04-04 07:06:18 +02:00
}
}
impl<T: fmt::Debug> fmt::Debug for Form<T> {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
self.0.fmt(f)
}
}
impl<T: fmt::Display> fmt::Display for Form<T> {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
self.0.fmt(f)
}
}
2018-04-04 07:06:18 +02:00
/// Form extractor configuration
///
/// ```rust
2018-04-04 07:06:18 +02:00
/// #[macro_use] extern crate serde_derive;
/// use actix_web::{web, extract, App, Result};
2018-04-04 07:06:18 +02:00
///
/// #[derive(Deserialize)]
/// struct FormData {
/// username: String,
/// }
///
/// /// Extract form data using serde.
/// /// Custom configuration is used for this handler, max payload size is 4k
/// fn index(form: extract::Form<FormData>) -> Result<String> {
2018-04-04 07:06:18 +02:00
/// Ok(format!("Welcome {}!", form.username))
/// }
///
/// fn main() {
/// let app = App::new().resource(
2018-06-01 18:37:14 +02:00
/// "/index.html",
/// |r| {
/// r.route(web::get()
/// // change `Form` extractor configuration
/// .config(extract::FormConfig::default().limit(4097))
/// .to(index))
/// });
2018-04-04 07:06:18 +02:00
/// }
/// ```
2019-03-03 09:57:48 +01:00
#[derive(Clone)]
2019-03-02 07:51:32 +01:00
pub struct FormConfig {
2018-04-04 07:06:18 +02:00
limit: usize,
2019-03-02 07:51:32 +01:00
ehandler: Rc<Fn(UrlencodedError, &HttpRequest) -> Error>,
2018-04-04 07:06:18 +02:00
}
2019-03-02 07:51:32 +01:00
impl FormConfig {
2018-04-04 07:06:18 +02:00
/// Change max size of payload. By default max size is 256Kb
2019-03-03 09:57:48 +01:00
pub fn limit(mut self, limit: usize) -> Self {
2018-04-04 07:06:18 +02:00
self.limit = limit;
self
}
2018-06-21 07:47:01 +02:00
2018-06-12 23:47:47 +02:00
/// Set custom error handler
2019-03-03 09:57:48 +01:00
pub fn error_handler<F>(mut self, f: F) -> Self
2018-06-12 23:47:47 +02:00
where
2019-03-02 07:51:32 +01:00
F: Fn(UrlencodedError, &HttpRequest) -> Error + 'static,
2018-06-12 23:47:47 +02:00
{
self.ehandler = Rc::new(f);
self
}
2018-04-04 07:06:18 +02:00
}
2019-03-03 09:57:48 +01:00
impl ExtractorConfig for FormConfig {}
2019-03-02 07:51:32 +01:00
impl Default for FormConfig {
2018-04-04 07:06:18 +02:00
fn default() -> Self {
2018-06-12 23:47:47 +02:00
FormConfig {
limit: 262_144,
ehandler: Rc::new(|e, _| e.into()),
}
}
}
2019-03-02 07:51:32 +01:00
/// Json helper
///
/// Json can be used for two different purpose. First is for json response
/// generation and second is for extracting typed information from request's
/// payload.
///
/// To extract typed information from request's body, the type `T` must
/// implement the `Deserialize` trait from *serde*.
///
/// [**JsonConfig**](dev/struct.JsonConfig.html) allows to configure extraction
/// process.
///
/// ## Example
///
/// ```rust
2019-03-02 07:51:32 +01:00
/// #[macro_use] extern crate serde_derive;
/// use actix_web::{web, extract, App};
2019-03-02 07:51:32 +01:00
///
/// #[derive(Deserialize)]
/// struct Info {
/// username: String,
/// }
///
/// /// deserialize `Info` from request's body
/// fn index(info: extract::Json<Info>) -> String {
/// format!("Welcome {}!", info.username)
2019-03-02 07:51:32 +01:00
/// }
///
/// fn main() {
/// let app = App::new().resource(
/// "/index.html",
/// |r| r.route(web::post().to(index)));
2019-03-02 07:51:32 +01:00
/// }
/// ```
///
/// The `Json` type allows you to respond with well-formed JSON data: simply
/// return a value of type Json<T> where T is the type of a structure
/// to serialize into *JSON*. The type `T` must implement the `Serialize`
/// trait from *serde*.
///
/// ```rust
2019-03-02 07:51:32 +01:00
/// # #[macro_use] extern crate serde_derive;
/// # use actix_web::*;
/// #
/// #[derive(Serialize)]
/// struct MyObj {
/// name: String,
/// }
///
/// fn index(req: HttpRequest) -> Result<Json<MyObj>> {
/// Ok(Json(MyObj {
/// name: req.match_info().get("name").unwrap().to_string(),
2019-03-02 07:51:32 +01:00
/// }))
/// }
/// # fn main() {}
/// ```
pub struct Json<T>(pub T);
impl<T> Json<T> {
/// Deconstruct to an inner value
pub fn into_inner(self) -> T {
self.0
}
}
impl<T> Deref for Json<T> {
type Target = T;
fn deref(&self) -> &T {
&self.0
}
}
impl<T> DerefMut for Json<T> {
fn deref_mut(&mut self) -> &mut T {
&mut self.0
}
}
impl<T> fmt::Debug for Json<T>
where
T: fmt::Debug,
{
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "Json: {:?}", self.0)
}
}
impl<T> fmt::Display for Json<T>
where
T: fmt::Display,
{
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
fmt::Display::fmt(&self.0, f)
}
}
impl<T: Serialize> Responder for Json<T> {
type Error = Error;
type Future = FutureResult<Response, Error>;
fn respond_to(self, _: &HttpRequest) -> Self::Future {
let body = match serde_json::to_string(&self.0) {
Ok(body) => body,
Err(e) => return err(e.into()),
};
ok(Response::build(StatusCode::OK)
.content_type("application/json")
.body(body))
}
}
impl<T, P> FromRequest<P> for Json<T>
where
T: DeserializeOwned + 'static,
P: Stream<Item = Bytes, Error = PayloadError> + 'static,
{
type Error = Error;
type Future = Box<Future<Item = Self, Error = Error>>;
2019-03-03 09:57:48 +01:00
type Config = JsonConfig;
2019-03-02 07:51:32 +01:00
#[inline]
fn from_request(req: &mut ServiceFromRequest<P>) -> Self::Future {
2019-03-02 07:51:32 +01:00
let req2 = req.clone();
2019-03-03 09:57:48 +01:00
let cfg = req.load_config::<JsonConfig>();
let limit = cfg.limit;
2019-03-02 07:51:32 +01:00
let err = Rc::clone(&cfg.ehandler);
Box::new(
JsonBody::new(req)
2019-03-03 09:57:48 +01:00
.limit(limit)
2019-03-02 07:51:32 +01:00
.map_err(move |e| (*err)(e, &req2))
.map(Json),
)
}
}
/// Json extractor configuration
///
/// ```rust
2019-03-02 07:51:32 +01:00
/// #[macro_use] extern crate serde_derive;
/// use actix_web::{error, extract, web, App, HttpResponse, Json};
2019-03-02 07:51:32 +01:00
///
/// #[derive(Deserialize)]
/// struct Info {
/// username: String,
/// }
///
/// /// deserialize `Info` from request's body, max payload size is 4kb
/// fn index(info: Json<Info>) -> String {
/// format!("Welcome {}!", info.username)
2019-03-02 07:51:32 +01:00
/// }
///
/// fn main() {
/// let app = App::new().resource("/index.html", |r| {
/// r.route(web::post().config(
/// // change json extractor configuration
/// extract::JsonConfig::default().limit(4096)
/// .error_handler(|err, req| { // <- create custom error response
/// error::InternalError::from_response(
/// err, HttpResponse::Conflict().finish()).into()
/// }))
/// .to(index))
2019-03-02 07:51:32 +01:00
/// });
/// }
/// ```
2019-03-03 09:57:48 +01:00
#[derive(Clone)]
2019-03-02 07:51:32 +01:00
pub struct JsonConfig {
limit: usize,
ehandler: Rc<Fn(JsonPayloadError, &HttpRequest) -> Error>,
}
impl JsonConfig {
/// Change max size of payload. By default max size is 256Kb
2019-03-03 09:57:48 +01:00
pub fn limit(mut self, limit: usize) -> Self {
2019-03-02 07:51:32 +01:00
self.limit = limit;
self
}
/// Set custom error handler
2019-03-03 09:57:48 +01:00
pub fn error_handler<F>(mut self, f: F) -> Self
2019-03-02 07:51:32 +01:00
where
F: Fn(JsonPayloadError, &HttpRequest) -> Error + 'static,
{
self.ehandler = Rc::new(f);
self
}
}
2019-03-03 09:57:48 +01:00
impl ExtractorConfig for JsonConfig {}
2019-03-02 07:51:32 +01:00
impl Default for JsonConfig {
fn default() -> Self {
JsonConfig {
limit: 262_144,
ehandler: Rc::new(|e, _| e.into()),
}
}
}
/// Request binary data from a request's payload.
2018-04-03 01:19:18 +02:00
///
/// Loads request's payload and construct Bytes instance.
///
2018-04-14 01:02:01 +02:00
/// [**PayloadConfig**](dev/struct.PayloadConfig.html) allows to configure
/// extraction process.
2018-04-06 19:24:57 +02:00
///
2018-04-03 01:19:18 +02:00
/// ## Example
///
/// ```rust
/// use bytes::Bytes;
/// use actix_web::{web, App};
2018-04-03 01:19:18 +02:00
///
/// /// extract binary data from request
/// fn index(body: Bytes) -> String {
/// format!("Body {:?}!", body)
2018-04-03 01:19:18 +02:00
/// }
2018-04-16 18:30:59 +02:00
///
/// fn main() {
2018-06-01 18:37:14 +02:00
/// let app = App::new()
/// .resource("/index.html", |r| r.route(web::get().to(index)));
2018-04-16 18:30:59 +02:00
/// }
2018-04-03 01:19:18 +02:00
/// ```
2019-03-02 07:51:32 +01:00
impl<P> FromRequest<P> for Bytes
where
P: Stream<Item = Bytes, Error = PayloadError> + 'static,
{
type Error = Error;
type Future =
Either<Box<Future<Item = Bytes, Error = Error>>, FutureResult<Bytes, Error>>;
2019-03-03 09:57:48 +01:00
type Config = PayloadConfig;
2018-04-03 01:19:18 +02:00
#[inline]
fn from_request(req: &mut ServiceFromRequest<P>) -> Self::Future {
2019-03-03 09:57:48 +01:00
let cfg = req.load_config::<PayloadConfig>();
2019-03-02 07:51:32 +01:00
if let Err(e) = cfg.check_mimetype(req) {
return Either::B(err(e));
}
2018-04-05 06:13:48 +02:00
2019-03-03 09:57:48 +01:00
let limit = cfg.limit;
Either::A(Box::new(MessageBody::new(req).limit(limit).from_err()))
2018-04-03 01:19:18 +02:00
}
}
/// Extract text information from a request's body.
2018-04-03 01:19:18 +02:00
///
/// Text extractor automatically decode body according to the request's charset.
///
2018-04-06 19:24:57 +02:00
/// [**PayloadConfig**](dev/struct.PayloadConfig.html) allows to configure
/// extraction process.
///
2018-04-03 01:19:18 +02:00
/// ## Example
///
/// ```rust
/// use actix_web::{web, extract, App};
2018-04-03 01:19:18 +02:00
///
/// /// extract text data from request
/// fn index(text: String) -> String {
/// format!("Body {}!", text)
2018-04-03 01:19:18 +02:00
/// }
2018-04-16 18:30:59 +02:00
///
/// fn main() {
2018-06-01 18:37:14 +02:00
/// let app = App::new().resource("/index.html", |r| {
/// r.route(
/// web::get()
/// .config(extract::PayloadConfig::new(4096)) // <- limit size of the payload
/// .to(index)) // <- register handler with extractor params
2018-06-01 18:37:14 +02:00
/// });
2018-04-16 18:30:59 +02:00
/// }
2018-04-03 01:19:18 +02:00
/// ```
2019-03-02 07:51:32 +01:00
impl<P> FromRequest<P> for String
where
P: Stream<Item = Bytes, Error = PayloadError> + 'static,
{
type Error = Error;
type Future =
Either<Box<Future<Item = String, Error = Error>>, FutureResult<String, Error>>;
2019-03-03 09:57:48 +01:00
type Config = PayloadConfig;
2018-04-03 01:19:18 +02:00
#[inline]
fn from_request(req: &mut ServiceFromRequest<P>) -> Self::Future {
2019-03-03 09:57:48 +01:00
let cfg = req.load_config::<PayloadConfig>();
2019-03-02 07:51:32 +01:00
2018-04-05 06:13:48 +02:00
// check content-type
2019-03-02 07:51:32 +01:00
if let Err(e) = cfg.check_mimetype(req) {
return Either::B(err(e));
}
2018-04-05 06:13:48 +02:00
// check charset
2019-03-02 07:51:32 +01:00
let encoding = match req.encoding() {
Ok(enc) => enc,
Err(e) => return Either::B(err(e.into())),
};
2019-03-03 09:57:48 +01:00
let limit = cfg.limit;
2018-04-03 01:19:18 +02:00
2019-03-02 07:51:32 +01:00
Either::A(Box::new(
2018-06-25 06:58:04 +02:00
MessageBody::new(req)
2019-03-03 09:57:48 +01:00
.limit(limit)
2018-04-29 18:09:08 +02:00
.from_err()
.and_then(move |body| {
2018-04-03 01:19:18 +02:00
let enc: *const Encoding = encoding as *const Encoding;
if enc == UTF_8 {
Ok(str::from_utf8(body.as_ref())
2018-04-14 01:02:01 +02:00
.map_err(|_| ErrorBadRequest("Can not decode body"))?
.to_owned())
2018-04-03 01:19:18 +02:00
} else {
2018-04-14 01:02:01 +02:00
Ok(encoding
.decode(&body, DecoderTrap::Strict)
.map_err(|_| ErrorBadRequest("Can not decode body"))?)
2018-04-03 01:19:18 +02:00
}
2018-04-29 18:09:08 +02:00
}),
2018-04-14 01:02:01 +02:00
))
2018-04-03 01:19:18 +02:00
}
}
/// Optionally extract a field from the request
///
/// If the FromRequest for T fails, return None rather than returning an error response
///
/// ## Example
///
/// ```rust
/// # #[macro_use] extern crate serde_derive;
/// use actix_web::{web, App, Error, FromRequest, ServiceFromRequest};
/// use actix_web::error::ErrorBadRequest;
/// use rand;
///
/// #[derive(Debug, Deserialize)]
/// struct Thing {
/// name: String
/// }
///
/// impl<P> FromRequest<P> for Thing {
/// type Error = Error;
/// type Future = Result<Self, Self::Error>;
/// type Config = ();
///
/// fn from_request(req: &mut ServiceFromRequest<P>) -> Self::Future {
/// if rand::random() {
/// Ok(Thing { name: "thingy".into() })
/// } else {
/// Err(ErrorBadRequest("no luck"))
/// }
///
/// }
/// }
///
/// /// extract `Thing` from request
/// fn index(supplied_thing: Option<Thing>) -> String {
/// match supplied_thing {
/// // Puns not intended
/// Some(thing) => format!("Got something: {:?}", thing),
/// None => format!("No thing!")
/// }
/// }
///
/// fn main() {
/// let app = App::new().resource("/users/:first", |r| {
/// r.route(web::post().to(index))
/// });
/// }
/// ```
2019-03-02 07:51:32 +01:00
impl<T: 'static, P> FromRequest<P> for Option<T>
2018-07-24 23:52:56 +02:00
where
2019-03-02 07:51:32 +01:00
T: FromRequest<P>,
T::Future: 'static,
2018-07-24 23:52:56 +02:00
{
2019-03-02 07:51:32 +01:00
type Error = Error;
type Future = Box<Future<Item = Option<T>, Error = Error>>;
2019-03-03 09:57:48 +01:00
type Config = T::Config;
#[inline]
fn from_request(req: &mut ServiceFromRequest<P>) -> Self::Future {
Box::new(T::from_request(req).into_future().then(|r| match r {
2018-07-24 23:52:56 +02:00
Ok(v) => future::ok(Some(v)),
Err(_) => future::ok(None),
}))
}
}
/// Optionally extract a field from the request or extract the Error if unsuccessful
///
/// If the `FromRequest` for T fails, inject Err into handler rather than returning an error response
///
/// ## Example
///
/// ```rust
/// # #[macro_use] extern crate serde_derive;
/// use actix_web::{web, App, Result, Error, FromRequest, ServiceFromRequest};
/// use actix_web::error::ErrorBadRequest;
/// use rand;
///
/// #[derive(Debug, Deserialize)]
/// struct Thing {
/// name: String
/// }
///
/// impl<P> FromRequest<P> for Thing {
/// type Error = Error;
/// type Future = Result<Thing, Error>;
/// type Config = ();
///
/// fn from_request(req: &mut ServiceFromRequest<P>) -> Self::Future {
/// if rand::random() {
/// Ok(Thing { name: "thingy".into() })
/// } else {
/// Err(ErrorBadRequest("no luck"))
/// }
/// }
/// }
///
/// /// extract `Thing` from request
/// fn index(supplied_thing: Result<Thing>) -> String {
/// match supplied_thing {
/// Ok(thing) => format!("Got thing: {:?}", thing),
/// Err(e) => format!("Error extracting thing: {}", e)
/// }
/// }
///
/// fn main() {
/// let app = App::new().resource("/users/:first", |r| {
/// r.route(web::post().to(index))
/// });
/// }
/// ```
2019-03-02 07:51:32 +01:00
impl<T: 'static, P> FromRequest<P> for Result<T, T::Error>
2018-07-24 23:52:56 +02:00
where
2019-03-02 07:51:32 +01:00
T: FromRequest<P>,
T::Future: 'static,
T::Error: 'static,
2018-07-24 23:52:56 +02:00
{
2019-03-02 07:51:32 +01:00
type Error = Error;
type Future = Box<Future<Item = Result<T, T::Error>, Error = Error>>;
2019-03-03 09:57:48 +01:00
type Config = T::Config;
#[inline]
fn from_request(req: &mut ServiceFromRequest<P>) -> Self::Future {
Box::new(T::from_request(req).into_future().then(|res| match res {
2019-03-02 07:51:32 +01:00
Ok(v) => ok(Ok(v)),
Err(e) => ok(Err(e)),
}))
}
}
2018-04-05 06:13:48 +02:00
/// Payload configuration for request's payload.
2019-03-03 09:57:48 +01:00
#[derive(Clone)]
2018-04-05 06:13:48 +02:00
pub struct PayloadConfig {
limit: usize,
mimetype: Option<Mime>,
}
impl PayloadConfig {
2019-03-03 09:57:48 +01:00
/// Create `PayloadConfig` instance and set max size of payload.
pub fn new(limit: usize) -> Self {
Self::default().limit(limit)
}
2018-04-05 06:13:48 +02:00
/// Change max size of payload. By default max size is 256Kb
2019-03-03 09:57:48 +01:00
pub fn limit(mut self, limit: usize) -> Self {
2018-04-05 06:13:48 +02:00
self.limit = limit;
self
}
2018-04-14 01:02:01 +02:00
/// Set required mime-type of the request. By default mime type is not
/// enforced.
2019-03-03 09:57:48 +01:00
pub fn mimetype(mut self, mt: Mime) -> Self {
2018-04-05 06:13:48 +02:00
self.mimetype = Some(mt);
self
}
fn check_mimetype<P>(&self, req: &ServiceFromRequest<P>) -> Result<(), Error> {
2018-04-05 06:13:48 +02:00
// check content-type
if let Some(ref mt) = self.mimetype {
match req.mime_type() {
Ok(Some(ref req_mt)) => {
if mt != req_mt {
return Err(ErrorBadRequest("Unexpected Content-Type"));
}
2018-04-14 01:02:01 +02:00
}
2018-04-05 06:13:48 +02:00
Ok(None) => {
return Err(ErrorBadRequest("Content-Type is expected"));
2018-04-14 01:02:01 +02:00
}
2018-04-05 06:13:48 +02:00
Err(err) => {
return Err(err.into());
2018-04-14 01:02:01 +02:00
}
2018-04-05 06:13:48 +02:00
}
}
Ok(())
}
}
2019-03-03 09:57:48 +01:00
impl ExtractorConfig for PayloadConfig {}
2018-04-05 06:13:48 +02:00
impl Default for PayloadConfig {
fn default() -> Self {
2018-04-14 01:02:01 +02:00
PayloadConfig {
limit: 262_144,
mimetype: None,
}
2018-04-05 06:13:48 +02:00
}
}
2019-03-03 09:57:48 +01:00
macro_rules! tuple_config ({ $($T:ident),+} => {
impl<$($T,)+> ExtractorConfig for ($($T,)+)
where $($T: ExtractorConfig + Clone,)+
{
fn store_default(ext: &mut ConfigStorage) {
$($T::store_default(ext);)+
}
}
});
macro_rules! tuple_from_req ({$fut_type:ident, $(($n:tt, $T:ident)),+} => {
/// FromRequest implementation for tuple
2019-03-02 07:51:32 +01:00
impl<P, $($T: FromRequest<P> + 'static),+> FromRequest<P> for ($($T,)+)
{
2019-03-02 07:51:32 +01:00
type Error = Error;
type Future = $fut_type<P, $($T),+>;
2019-03-03 09:57:48 +01:00
type Config = ($($T::Config,)+);
fn from_request(req: &mut ServiceFromRequest<P>) -> Self::Future {
2019-03-02 07:51:32 +01:00
$fut_type {
items: <($(Option<$T>,)+)>::default(),
futs: ($($T::from_request(req).into_future(),)+),
2019-03-02 07:51:32 +01:00
}
}
}
2019-03-02 07:51:32 +01:00
#[doc(hidden)]
pub struct $fut_type<P, $($T: FromRequest<P>),+> {
items: ($(Option<$T>,)+),
futs: ($(<$T::Future as futures::IntoFuture>::Future,)+),
}
2019-03-02 07:51:32 +01:00
impl<P, $($T: FromRequest<P>),+> Future for $fut_type<P, $($T),+>
{
type Item = ($($T,)+);
type Error = Error;
fn poll(&mut self) -> Poll<Self::Item, Self::Error> {
let mut ready = true;
$(
2019-03-02 07:51:32 +01:00
if self.items.$n.is_none() {
match self.futs.$n.poll() {
Ok(Async::Ready(item)) => {
self.items.$n = Some(item);
}
Ok(Async::NotReady) => ready = false,
2019-03-02 07:51:32 +01:00
Err(e) => return Err(e.into()),
}
}
)+
if ready {
Ok(Async::Ready(
($(self.items.$n.take().unwrap(),)+)
))
} else {
Ok(Async::NotReady)
}
}
}
});
2019-03-02 07:51:32 +01:00
impl<P> FromRequest<P> for () {
type Error = Error;
type Future = FutureResult<(), Error>;
2019-03-03 09:57:48 +01:00
type Config = ();
2019-03-02 07:51:32 +01:00
fn from_request(_req: &mut ServiceFromRequest<P>) -> Self::Future {
2019-03-02 07:51:32 +01:00
ok(())
}
2018-08-06 09:44:08 +02:00
}
2019-03-03 07:11:24 +01:00
#[rustfmt::skip]
mod m {
use super::*;
2019-03-03 09:57:48 +01:00
tuple_config!(A);
tuple_config!(A, B);
tuple_config!(A, B, C);
tuple_config!(A, B, C, D);
tuple_config!(A, B, C, D, E);
tuple_config!(A, B, C, D, E, F);
tuple_config!(A, B, C, D, E, F, G);
tuple_config!(A, B, C, D, E, F, G, H);
tuple_config!(A, B, C, D, E, F, G, H, I);
tuple_config!(A, B, C, D, E, F, G, H, I, J);
tuple_from_req!(TupleFromRequest1, (0, A));
tuple_from_req!(TupleFromRequest2, (0, A), (1, B));
tuple_from_req!(TupleFromRequest3, (0, A), (1, B), (2, C));
tuple_from_req!(TupleFromRequest4, (0, A), (1, B), (2, C), (3, D));
2018-05-17 21:20:20 +02:00
tuple_from_req!(TupleFromRequest5, (0, A), (1, B), (2, C), (3, D), (4, E));
2019-03-03 07:11:24 +01:00
tuple_from_req!(TupleFromRequest6, (0, A), (1, B), (2, C), (3, D), (4, E), (5, F));
tuple_from_req!(TupleFromRequest7, (0, A), (1, B), (2, C), (3, D), (4, E), (5, F), (6, G));
tuple_from_req!(TupleFromRequest8, (0, A), (1, B), (2, C), (3, D), (4, E), (5, F), (6, G), (7, H));
tuple_from_req!(TupleFromRequest9, (0, A), (1, B), (2, C), (3, D), (4, E), (5, F), (6, G), (7, H), (8, I));
tuple_from_req!(TupleFromRequest10, (0, A), (1, B), (2, C), (3, D), (4, E), (5, F), (6, G), (7, H), (8, I), (9, J));
}
2019-03-03 07:03:45 +01:00
#[cfg(test)]
mod tests {
use actix_http::http::header;
use bytes::Bytes;
use serde_derive::Deserialize;
use super::*;
use crate::test::TestRequest;
#[derive(Deserialize, Debug, PartialEq)]
struct Info {
hello: String,
}
2019-03-02 07:51:32 +01:00
2019-03-03 07:03:45 +01:00
#[test]
fn test_bytes() {
let mut rt = actix_rt::Runtime::new().unwrap();
let mut req = TestRequest::with_header(header::CONTENT_LENGTH, "11")
.set_payload(Bytes::from_static(b"hello=world"))
2019-03-03 10:12:06 +01:00
.to_from();
2019-03-02 07:51:32 +01:00
2019-03-03 07:03:45 +01:00
let s = rt.block_on(Bytes::from_request(&mut req)).unwrap();
assert_eq!(s, Bytes::from_static(b"hello=world"));
}
2019-03-02 07:51:32 +01:00
2019-03-03 07:03:45 +01:00
#[test]
fn test_string() {
let mut rt = actix_rt::Runtime::new().unwrap();
let mut req = TestRequest::with_header(header::CONTENT_LENGTH, "11")
.set_payload(Bytes::from_static(b"hello=world"))
2019-03-03 10:12:06 +01:00
.to_from();
2019-03-02 07:51:32 +01:00
2019-03-03 07:03:45 +01:00
let s = rt.block_on(String::from_request(&mut req)).unwrap();
assert_eq!(s, "hello=world");
}
#[test]
fn test_form() {
let mut rt = actix_rt::Runtime::new().unwrap();
let mut req = TestRequest::with_header(
header::CONTENT_TYPE,
"application/x-www-form-urlencoded",
)
.header(header::CONTENT_LENGTH, "11")
.set_payload(Bytes::from_static(b"hello=world"))
2019-03-03 10:12:06 +01:00
.to_from();
2019-03-03 07:03:45 +01:00
let s = rt.block_on(Form::<Info>::from_request(&mut req)).unwrap();
assert_eq!(s.hello, "world");
}
}
2019-03-02 07:51:32 +01:00
// #[test]
// fn test_option() {
// let req = TestRequest::with_header(
// header::CONTENT_TYPE,
// "application/x-www-form-urlencoded",
// )
// .finish();
// let mut cfg = FormConfig::default();
// cfg.limit(4096);
// match Option::<Form<Info>>::from_request(&req, &cfg)
// .poll()
// .unwrap()
// {
// Async::Ready(r) => assert_eq!(r, None),
// _ => unreachable!(),
// }
// let req = TestRequest::with_header(
// header::CONTENT_TYPE,
// "application/x-www-form-urlencoded",
// )
// .header(header::CONTENT_LENGTH, "9")
// .set_payload(Bytes::from_static(b"hello=world"))
// .finish();
// match Option::<Form<Info>>::from_request(&req, &cfg)
// .poll()
// .unwrap()
// {
// Async::Ready(r) => assert_eq!(
// r,
// Some(Form(Info {
// hello: "world".into()
// }))
// ),
// _ => unreachable!(),
// }
// let req = TestRequest::with_header(
// header::CONTENT_TYPE,
// "application/x-www-form-urlencoded",
// )
// .header(header::CONTENT_LENGTH, "9")
// .set_payload(Bytes::from_static(b"bye=world"))
// .finish();
// match Option::<Form<Info>>::from_request(&req, &cfg)
// .poll()
// .unwrap()
// {
// Async::Ready(r) => assert_eq!(r, None),
// _ => unreachable!(),
// }
// }
// #[test]
// fn test_result() {
// let req = TestRequest::with_header(
// header::CONTENT_TYPE,
// "application/x-www-form-urlencoded",
// )
// .header(header::CONTENT_LENGTH, "11")
// .set_payload(Bytes::from_static(b"hello=world"))
// .finish();
// match Result::<Form<Info>, Error>::from_request(&req, &FormConfig::default())
// .poll()
// .unwrap()
// {
// Async::Ready(Ok(r)) => assert_eq!(
// r,
// Form(Info {
// hello: "world".into()
// })
// ),
// _ => unreachable!(),
// }
// let req = TestRequest::with_header(
// header::CONTENT_TYPE,
// "application/x-www-form-urlencoded",
// )
// .header(header::CONTENT_LENGTH, "9")
// .set_payload(Bytes::from_static(b"bye=world"))
// .finish();
// match Result::<Form<Info>, Error>::from_request(&req, &FormConfig::default())
// .poll()
// .unwrap()
// {
// Async::Ready(r) => assert!(r.is_err()),
// _ => unreachable!(),
// }
// }
// #[test]
// fn test_payload_config() {
// let req = TestRequest::default().finish();
// let mut cfg = PayloadConfig::default();
// cfg.mimetype(mime::APPLICATION_JSON);
// assert!(cfg.check_mimetype(&req).is_err());
// let req = TestRequest::with_header(
// header::CONTENT_TYPE,
// "application/x-www-form-urlencoded",
// )
// .finish();
// assert!(cfg.check_mimetype(&req).is_err());
// let req =
// TestRequest::with_header(header::CONTENT_TYPE, "application/json").finish();
// assert!(cfg.check_mimetype(&req).is_ok());
// }
// #[derive(Deserialize)]
// struct MyStruct {
// key: String,
// value: String,
// }
// #[derive(Deserialize)]
// struct Id {
// id: String,
// }
// #[derive(Deserialize)]
// struct Test2 {
// key: String,
// value: u32,
// }
// #[test]
// fn test_request_extract() {
// let req = TestRequest::with_uri("/name/user1/?id=test").finish();
// let mut router = Router::<()>::default();
// router.register_resource(Resource::new(ResourceDef::new("/{key}/{value}/")));
// let info = router.recognize(&req, &(), 0);
// let req = req.with_route_info(info);
// let s = Path::<MyStruct>::from_request(&req, &()).unwrap();
// assert_eq!(s.key, "name");
// assert_eq!(s.value, "user1");
// let s = Path::<(String, String)>::from_request(&req, &()).unwrap();
// assert_eq!(s.0, "name");
// assert_eq!(s.1, "user1");
// let s = Query::<Id>::from_request(&req, &()).unwrap();
// assert_eq!(s.id, "test");
// let mut router = Router::<()>::default();
// router.register_resource(Resource::new(ResourceDef::new("/{key}/{value}/")));
// let req = TestRequest::with_uri("/name/32/").finish();
// let info = router.recognize(&req, &(), 0);
// let req = req.with_route_info(info);
// let s = Path::<Test2>::from_request(&req, &()).unwrap();
// assert_eq!(s.as_ref().key, "name");
// assert_eq!(s.value, 32);
// let s = Path::<(String, u8)>::from_request(&req, &()).unwrap();
// assert_eq!(s.0, "name");
// assert_eq!(s.1, 32);
// let res = Path::<Vec<String>>::extract(&req).unwrap();
// assert_eq!(res[0], "name".to_owned());
// assert_eq!(res[1], "32".to_owned());
// }
// #[test]
// fn test_extract_path_single() {
// let mut router = Router::<()>::default();
// router.register_resource(Resource::new(ResourceDef::new("/{value}/")));
// let req = TestRequest::with_uri("/32/").finish();
// let info = router.recognize(&req, &(), 0);
// let req = req.with_route_info(info);
// assert_eq!(*Path::<i8>::from_request(&req, &()).unwrap(), 32);
// }
// #[test]
// fn test_tuple_extract() {
// let mut router = Router::<()>::default();
// router.register_resource(Resource::new(ResourceDef::new("/{key}/{value}/")));
// let req = TestRequest::with_uri("/name/user1/?id=test").finish();
// let info = router.recognize(&req, &(), 0);
// let req = req.with_route_info(info);
// let res = match <(Path<(String, String)>,)>::extract(&req).poll() {
// Ok(Async::Ready(res)) => res,
// _ => panic!("error"),
// };
// assert_eq!((res.0).0, "name");
// assert_eq!((res.0).1, "user1");
// let res = match <(Path<(String, String)>, Path<(String, String)>)>::extract(&req)
// .poll()
// {
// Ok(Async::Ready(res)) => res,
// _ => panic!("error"),
// };
// assert_eq!((res.0).0, "name");
// assert_eq!((res.0).1, "user1");
// assert_eq!((res.1).0, "name");
// assert_eq!((res.1).1, "user1");
// let () = <()>::extract(&req);
// }
// }