surrealpatch/src/net/output.rs

use http::header::{HeaderValue, CONTENT_TYPE};
use http::StatusCode;
use serde::Serialize;
use surrealdb::sql::serde::serialize_internal;

pub enum Output {
	None,
	Fail,
	Text(String),
	Json(Vec<u8>), // JSON
	Cbor(Vec<u8>), // CBOR
	Pack(Vec<u8>), // MessagePack
	Cork(Vec<u8>), // Full type serialization
}

pub fn none() -> Output {
	Output::None
}

pub fn text(val: String) -> Output {
	Output::Text(val)
}

pub fn json<T>(val: &T) -> Output
where
	T: Serialize,
{
	match serde_json::to_vec(val) {
		Ok(v) => Output::Json(v),
		Err(_) => Output::Fail,
	}
}

pub fn cbor<T>(val: &T) -> Output
where
	T: Serialize,
{
	match serde_cbor::to_vec(val) {
		Ok(v) => Output::Cbor(v),
		Err(_) => Output::Fail,
	}
}

pub fn pack<T>(val: &T) -> Output
where
	T: Serialize,
{
	match serde_pack::to_vec(val) {
		Ok(v) => Output::Pack(v),
		Err(_) => Output::Fail,
	}
}

pub fn cork<T>(val: &T) -> Output
where
	T: Serialize,
{
	match serialize_internal(|| serde_pack::to_vec(val)) {
		Ok(v) => Output::Cork(v),
		Err(_) => Output::Fail,
	}
}

impl warp::Reply for Output {
	fn into_response(self) -> warp::reply::Response {
		match self {
			Output::Text(v) => {
				let mut res = warp::reply::Response::new(v.into());
				let con = HeaderValue::from_static("text/plain");
				res.headers_mut().insert(CONTENT_TYPE, con);
				res
			}
			Output::Json(v) => {
				let mut res = warp::reply::Response::new(v.into());
				let con = HeaderValue::from_static("application/json");
				res.headers_mut().insert(CONTENT_TYPE, con);
				res
			}
			Output::Cbor(v) => {
				let mut res = warp::reply::Response::new(v.into());
				let con = HeaderValue::from_static("application/cbor");
				res.headers_mut().insert(CONTENT_TYPE, con);
				res
			}
			Output::Pack(v) => {
				let mut res = warp::reply::Response::new(v.into());
				let con = HeaderValue::from_static("application/pack");
				res.headers_mut().insert(CONTENT_TYPE, con);
				res
			}
			Output::Cork(v) => {
				let mut res = warp::reply::Response::new(v.into());
				let con = HeaderValue::from_static("application/cork");
				res.headers_mut().insert(CONTENT_TYPE, con);
				res
			}
			Output::None => StatusCode::OK.into_response(),
			Output::Fail => StatusCode::INTERNAL_SERVER_ERROR.into_response(),
		}
	}
}