hatter/local-mini-kms
1
1
Fork 0
Code Issues 2 Pull Requests Projects Releases Wiki Activity

feat: update versions

Browse Source
This commit is contained in:
Hatter Jiang
2022-10-01 18:01:04 +08:00
parent 487739515b
commit 0351735531
4 changed files with 222 additions and 104 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/serve.rs
View File

@@ -83,6 +83,7 @@ async fn response_requests(
(&Method::POST, "/encrypt") => encrypt(req).await,
(&Method::GET, "/status") => status().await,
(&Method::GET, "/version") => get_version().await,
(&Method::GET, "/") => get_root().await,
_ => Ok(Response::builder()
.status(StatusCode::NOT_FOUND)
.body(format!("{}\n", serde_json::to_string_pretty(&json!({
@@ -94,7 +95,7 @@ async fn response_requests(
macro_rules! do_response {
($ex: expr) => (
match $ex {
Ok((status_code, body)) => Ok(Response::builder().status(status_code).body(serde_json::to_string_pretty(&body)?.into())?),
Ok((status_code, body)) => Ok(Response::builder().status(status_code).body((serde_json::to_string_pretty(&body)? + "\n").into())?),
Err(e) => Ok(Response::builder().status(StatusCode::INTERNAL_SERVER_ERROR).body(
format!("{}\n", serde_json::to_string_pretty(&json!({
"error": "internal_error",
@@ -347,6 +348,24 @@ async fn get_version() -> Result<Response<Body>> {
).into())?)
}
async fn get_root() -> Result<Response<Body>> {
if std::env::var("LOCAL_MINI_KMS_HELP").is_ok() {
Ok(Response::builder().body(format!(
r##"{} - {}
Supports commands:
- GET /version
- GET /status
- POST /init
- POST /update
- POST /encrypt
- POST /decrypt
"##, env!("CARGO_PKG_NAME"), env!("CARGO_PKG_VERSION")
).into())?)
} else {
Ok(Response::builder().body("Root Not Found\n".into())?)
}
}
fn get_master_key() -> Option<SecBytes> {
let startup_rw_lock = STATUP_RW_LOCK.lock().expect("Lock read startup rw lock error");
match &*startup_rw_lock {