use std::ops::Deref;
use tokio_core::reactor::Handle;
use unsafe_any::UnsafeAny;
use util::typemap::TypeMap;
use request::Request;
use state::{FromState, State};
use Data;
pub use state::Key;
pub struct Context {
state: State,
handle: Handle,
request: Request,
body: Data,
}
impl Context {
pub(crate) fn new(handle: Handle, request: Request, state: State, body: Data) -> Self {
Self {
handle,
request,
state,
body,
}
}
#[inline]
pub fn handle(&self) -> &Handle {
&self.handle
}
pub fn data(self) -> Data {
self.body
}
pub fn put<K: Key>(&mut self, value: K::Value) {
self.state.put::<K>(value);
}
pub fn get<T: FromState>(&self) -> &T::Value {
self.state.get::<T>()
}
pub fn try_get<T: FromState>(&self) -> Option<&T::Value> {
self.state.try_get::<T>()
}
pub fn shared(&self) -> &TypeMap<UnsafeAny + Send + Sync> {
self.state.shared()
}
pub fn deconstruct(self) -> (Handle, State, Request, Data) {
(self.handle, self.state, self.request, self.body)
}
}
impl Deref for Context {
type Target = Request;
fn deref(&self) -> &Self::Target {
&self.request
}
}