2020-04-17 12:11:25 +00:00
|
|
|
use anyhow::{Error};
|
2018-12-05 11:42:25 +00:00
|
|
|
use serde_json::{json, Value};
|
|
|
|
|
|
2018-12-05 17:22:56 +00:00
|
|
|
use hyper::{Body, Response, StatusCode};
|
|
|
|
|
use hyper::header;
|
|
|
|
|
|
2019-11-21 13:36:28 +00:00
|
|
|
use proxmox::api::{HttpError, RpcEnvironment};
|
|
|
|
|
|
2019-02-15 08:55:12 +00:00
|
|
|
/// Extension to set error message for server side logging
|
|
|
|
|
pub struct ErrorMessageExtension(pub String);
|
|
|
|
|
|
2018-12-05 11:42:25 +00:00
|
|
|
pub struct OutputFormatter {
|
|
|
|
|
|
2019-06-07 11:10:56 +00:00
|
|
|
pub format_data: fn(data: Value, rpcenv: &dyn RpcEnvironment) -> Response<Body>,
|
2019-01-26 13:50:37 +00:00
|
|
|
|
|
|
|
|
pub format_error: fn(err: Error) -> Response<Body>,
|
2018-12-05 11:42:25 +00:00
|
|
|
}
|
|
|
|
|
|
2019-01-26 14:08:02 +00:00
|
|
|
static JSON_CONTENT_TYPE: &str = "application/json;charset=UTF-8";
|
2019-01-26 13:50:37 +00:00
|
|
|
|
2019-05-09 11:28:26 +00:00
|
|
|
pub fn json_response(result: Result<Value, Error>) -> Response<Body> {
|
|
|
|
|
match result {
|
|
|
|
|
Ok(data) => json_data_response(data),
|
|
|
|
|
Err(err) => json_error_response(err),
|
|
|
|
|
}
|
|
|
|
|
}
|
2019-01-26 13:50:37 +00:00
|
|
|
|
2019-05-09 11:28:26 +00:00
|
|
|
pub fn json_data_response(data: Value) -> Response<Body> {
|
2019-01-26 13:50:37 +00:00
|
|
|
|
2019-05-09 11:28:26 +00:00
|
|
|
let json_str = data.to_string();
|
2019-01-26 13:50:37 +00:00
|
|
|
|
|
|
|
|
let raw = json_str.into_bytes();
|
2018-12-05 11:42:25 +00:00
|
|
|
|
2019-01-26 13:50:37 +00:00
|
|
|
let mut response = Response::new(raw.into());
|
|
|
|
|
response.headers_mut().insert(
|
|
|
|
|
header::CONTENT_TYPE,
|
2019-01-26 14:08:02 +00:00
|
|
|
header::HeaderValue::from_static(JSON_CONTENT_TYPE));
|
2018-12-05 11:42:25 +00:00
|
|
|
|
2019-01-26 13:50:37 +00:00
|
|
|
response
|
|
|
|
|
}
|
2018-12-05 17:22:56 +00:00
|
|
|
|
2019-12-07 14:29:42 +00:00
|
|
|
fn add_result_attributes(result: &mut Value, rpcenv: &dyn RpcEnvironment)
|
|
|
|
|
{
|
2020-05-18 07:57:35 +00:00
|
|
|
let attributes = match rpcenv.result_attrib().as_object() {
|
|
|
|
|
Some(attr) => attr,
|
|
|
|
|
None => return,
|
|
|
|
|
};
|
2019-12-07 14:29:42 +00:00
|
|
|
|
2020-05-18 07:57:35 +00:00
|
|
|
for (key, value) in attributes {
|
|
|
|
|
result[key] = value.clone();
|
2018-12-05 17:22:56 +00:00
|
|
|
}
|
2019-12-07 14:29:42 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn json_format_data(data: Value, rpcenv: &dyn RpcEnvironment) -> Response<Body> {
|
|
|
|
|
|
|
|
|
|
let mut result = json!({
|
|
|
|
|
"data": data
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
add_result_attributes(&mut result, rpcenv);
|
2019-01-26 13:50:37 +00:00
|
|
|
|
2019-05-09 11:28:26 +00:00
|
|
|
json_data_response(result)
|
2019-01-26 13:50:37 +00:00
|
|
|
}
|
|
|
|
|
|
2019-05-09 11:28:26 +00:00
|
|
|
pub fn json_error_response(err: Error) -> Response<Body> {
|
2019-01-26 13:50:37 +00:00
|
|
|
|
2019-02-17 16:18:44 +00:00
|
|
|
let mut response = if let Some(apierr) = err.downcast_ref::<HttpError>() {
|
|
|
|
|
let mut resp = Response::new(Body::from(apierr.message.clone()));
|
|
|
|
|
*resp.status_mut() = apierr.code;
|
|
|
|
|
resp
|
|
|
|
|
} else {
|
|
|
|
|
let mut resp = Response::new(Body::from(err.to_string()));
|
|
|
|
|
*resp.status_mut() = StatusCode::BAD_REQUEST;
|
|
|
|
|
resp
|
|
|
|
|
};
|
|
|
|
|
|
2019-01-26 13:50:37 +00:00
|
|
|
response.headers_mut().insert(
|
|
|
|
|
header::CONTENT_TYPE,
|
2019-01-26 14:08:02 +00:00
|
|
|
header::HeaderValue::from_static(JSON_CONTENT_TYPE));
|
2019-01-26 13:50:37 +00:00
|
|
|
|
2019-02-15 08:55:12 +00:00
|
|
|
response.extensions_mut().insert(ErrorMessageExtension(err.to_string()));
|
|
|
|
|
|
2019-01-26 13:50:37 +00:00
|
|
|
response
|
2018-12-05 11:42:25 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
pub static JSON_FORMATTER: OutputFormatter = OutputFormatter {
|
2019-05-09 11:15:15 +00:00
|
|
|
format_data: json_format_data,
|
2019-05-09 11:28:26 +00:00
|
|
|
format_error: json_error_response,
|
2018-12-05 11:42:25 +00:00
|
|
|
};
|
|
|
|
|
|
2019-06-07 11:10:56 +00:00
|
|
|
fn extjs_format_data(data: Value, rpcenv: &dyn RpcEnvironment) -> Response<Body> {
|
2018-12-05 11:42:25 +00:00
|
|
|
|
2019-01-26 13:50:37 +00:00
|
|
|
let mut result = json!({
|
|
|
|
|
"data": data,
|
|
|
|
|
"success": true
|
|
|
|
|
});
|
2018-12-05 11:42:25 +00:00
|
|
|
|
2019-12-07 14:29:42 +00:00
|
|
|
add_result_attributes(&mut result, rpcenv);
|
2018-12-05 17:22:56 +00:00
|
|
|
|
2019-05-09 11:28:26 +00:00
|
|
|
json_data_response(result)
|
2019-01-26 13:50:37 +00:00
|
|
|
}
|
2018-12-05 11:42:25 +00:00
|
|
|
|
2019-01-26 13:50:37 +00:00
|
|
|
fn extjs_format_error(err: Error) -> Response<Body> {
|
2018-12-05 11:42:25 +00:00
|
|
|
|
2019-01-26 13:50:37 +00:00
|
|
|
let mut errors = vec![];
|
2019-01-28 12:44:48 +00:00
|
|
|
|
|
|
|
|
let message = err.to_string();
|
|
|
|
|
errors.push(&message);
|
2018-12-05 11:42:25 +00:00
|
|
|
|
2019-01-26 13:50:37 +00:00
|
|
|
let result = json!({
|
2019-01-28 12:44:48 +00:00
|
|
|
"message": message,
|
2019-01-26 13:50:37 +00:00
|
|
|
"errors": errors,
|
|
|
|
|
"success": false
|
|
|
|
|
});
|
2018-12-05 11:42:25 +00:00
|
|
|
|
2019-05-09 11:28:26 +00:00
|
|
|
let mut response = json_data_response(result);
|
2019-02-15 08:55:12 +00:00
|
|
|
|
|
|
|
|
response.extensions_mut().insert(ErrorMessageExtension(message));
|
|
|
|
|
|
|
|
|
|
response
|
2018-12-05 11:42:25 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
pub static EXTJS_FORMATTER: OutputFormatter = OutputFormatter {
|
2019-05-09 11:15:15 +00:00
|
|
|
format_data: extjs_format_data,
|
2019-01-26 13:50:37 +00:00
|
|
|
format_error: extjs_format_error,
|
2018-12-05 11:42:25 +00:00
|
|
|
};
|
