97 lines
2.1 KiB
Rust
97 lines
2.1 KiB
Rust
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(),
|
|
}
|
|
}
|
|
}
|