surrealpatch/src/net/output.rs

use http::header::{HeaderValue, CONTENT_TYPE};
use http::StatusCode;
use serde::Serialize;
use serde_json::Value as Json;
use surrealdb::sql;

pub enum Output {
	None,
	Fail,
	Text(String),
	Json(Vec<u8>), // JSON
	Cbor(Vec<u8>), // CBOR
	Pack(Vec<u8>), // MessagePack
	Full(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 full<T>(val: &T) -> Output
where
	T: Serialize,
{
	match bung::to_vec(val) {
		Ok(v) => Output::Full(v),
		Err(_) => Output::Fail,
	}
}

/// Convert and simplify the value into JSON
pub fn simplify<T: Serialize>(v: T) -> Json {
	sql::to_value(v).unwrap().into()
}

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::Full(v) => {
				let mut res = warp::reply::Response::new(v.into());
				let con = HeaderValue::from_static("application/bung");
				res.headers_mut().insert(CONTENT_TYPE, con);
				res
			}
			Output::None => StatusCode::OK.into_response(),
			Output::Fail => StatusCode::INTERNAL_SERVER_ERROR.into_response(),
		}
	}
}