surrealpatch/src/net/output.rs

use axum::response::{IntoResponse, Response};
use http::header::{HeaderValue, CONTENT_TYPE};
use http::StatusCode;
use serde::Serialize;
use serde_json::Value as Json;
use surrealdb::dbs;
use surrealdb::sql;

use super::headers::Accept;

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 surrealdb::sql::serde::serialize(val) {
		Ok(v) => Output::Full(v),
		Err(_) => Output::Fail,
	}
}

/// Convert and simplify the value into JSON
pub fn simplify(vec: Vec<dbs::Response>) -> Json {
	let responses: Vec<_> = vec.into_iter().map(dbs::ApiResponse::from).collect();
	sql::to_value(responses).unwrap().into()
}

impl IntoResponse for Output {
	fn into_response(self) -> Response {
		match self {
			Output::Text(v) => {
				([(CONTENT_TYPE, HeaderValue::from(Accept::TextPlain))], v).into_response()
			}
			Output::Json(v) => {
				([(CONTENT_TYPE, HeaderValue::from(Accept::ApplicationJson))], v).into_response()
			}
			Output::Cbor(v) => {
				([(CONTENT_TYPE, HeaderValue::from(Accept::ApplicationCbor))], v).into_response()
			}
			Output::Pack(v) => {
				([(CONTENT_TYPE, HeaderValue::from(Accept::ApplicationPack))], v).into_response()
			}
			Output::Full(v) => {
				([(CONTENT_TYPE, HeaderValue::from(Accept::Surrealdb))], v).into_response()
			}
			Output::None => StatusCode::OK.into_response(),
			Output::Fail => StatusCode::INTERNAL_SERVER_ERROR.into_response(),
		}
	}
}