proxmox-backup/src/main.rs

121 lines
3.6 KiB
Rust
Raw Normal View History

extern crate apitest;
2018-10-30 09:04:30 +00:00
2018-11-05 14:20:27 +00:00
use failure::*;
2018-11-01 12:05:45 +00:00
2018-11-01 10:30:49 +00:00
use std::collections::HashMap;
2018-11-03 09:42:48 +00:00
use lazy_static::lazy_static;
2018-11-01 10:30:49 +00:00
2018-11-03 09:42:48 +00:00
//use apitest::json_schema::*;
2018-10-31 09:42:14 +00:00
use apitest::api_info::*;
2018-11-05 14:20:27 +00:00
use apitest::json_schema::*;
2018-10-31 09:42:14 +00:00
2018-11-03 09:42:48 +00:00
//use serde_derive::{Serialize, Deserialize};
2018-10-30 13:06:37 +00:00
use serde_json::{json, Value};
2018-11-01 10:30:49 +00:00
use hyper::{Method, Body, Request, Response, Server, StatusCode};
2018-10-31 09:42:14 +00:00
use hyper::rt::Future;
use hyper::service::service_fn_ok;
2018-10-30 09:04:30 +00:00
2018-11-01 10:30:49 +00:00
macro_rules! http_error {
($status:ident, $msg:expr) => {{
let mut resp = Response::new(Body::from($msg));
*resp.status_mut() = StatusCode::$status;
return resp;
}}
}
fn handle_request(req: Request<Body>) -> Response<Body> {
2018-10-31 09:42:14 +00:00
let method = req.method();
let path = req.uri().path();
2018-11-01 10:30:49 +00:00
let query = req.uri().query();
let components: Vec<&str> = path.split('/').filter(|x| !x.is_empty()).collect();
let comp_len = components.len();
2018-10-31 09:42:14 +00:00
println!("REQUEST {} {}", method, path);
2018-11-01 10:30:49 +00:00
println!("COMPO {:?}", components);
if comp_len >= 1 && components[0] == "api3" {
println!("GOT API REQUEST");
if comp_len >= 2 {
let format = components[1];
if format != "json" {
http_error!(NOT_FOUND, format!("Unsupported format '{}'\n", format))
}
2018-11-03 14:10:21 +00:00
if let Some(info) = ROUTER.find_method(&components[2..]) {
2018-11-01 10:30:49 +00:00
println!("FOUND INFO");
let api_method_opt = match method {
2018-11-03 14:10:21 +00:00
&Method::GET => &info.get,
2018-11-05 14:20:27 +00:00
&Method::PUT => &info.put,
&Method::POST => &info.post,
&Method::DELETE => &info.delete,
2018-11-03 14:10:21 +00:00
_ => &None,
2018-11-01 10:30:49 +00:00
};
2018-11-01 14:41:08 +00:00
let api_method = match api_method_opt {
2018-11-01 10:30:49 +00:00
Some(m) => m,
_ => http_error!(NOT_FOUND, format!("No such method '{} {}'\n", method, path)),
};
// handle auth
// extract param
2018-11-01 14:41:08 +00:00
let param = match query {
2018-11-05 14:20:27 +00:00
Some(data) => {
2018-11-06 13:18:13 +00:00
match parse_query_string(data, &api_method.parameters, true) {
2018-11-05 14:20:27 +00:00
Ok(query) => query,
2018-11-06 12:10:10 +00:00
Err(ref error_list) => {
let mut msg = String::from("");
for item in error_list {
msg = msg + &item.to_string() + "\n";
}
http_error!(BAD_REQUEST, msg);
}
2018-11-05 14:20:27 +00:00
}
}
2018-11-01 10:30:49 +00:00
None => json!({}),
};
2018-11-07 10:06:37 +00:00
match (api_method.handler)(param, &api_method) {
2018-11-01 14:41:08 +00:00
Ok(res) => {
let json_str = res.to_string();
return Response::new(Body::from(json_str));
}
Err(err) => {
http_error!(NOT_FOUND, format!("Method returned error '{}'\n", err));
}
}
2018-11-01 10:30:49 +00:00
} else {
http_error!(NOT_FOUND, format!("No such path '{}'\n", path));
}
}
}
2018-10-31 09:42:14 +00:00
2018-11-01 10:30:49 +00:00
Response::new(Body::from("RETURN WEB GUI\n"))
2018-10-31 09:42:14 +00:00
}
2018-11-03 09:42:48 +00:00
lazy_static!{
2018-11-03 14:10:21 +00:00
static ref ROUTER: MethodInfo = apitest::api3::router();
2018-11-03 09:42:48 +00:00
}
2018-10-30 09:04:30 +00:00
fn main() {
println!("Fast Static Type Definitions 1");
2018-10-31 09:42:14 +00:00
let addr = ([127, 0, 0, 1], 8007).into();
let new_svc = || {
// service_fn_ok converts our function into a `Service`
2018-11-01 10:30:49 +00:00
service_fn_ok(handle_request)
2018-10-31 09:42:14 +00:00
};
let server = Server::bind(&addr)
.serve(new_svc)
.map_err(|e| eprintln!("server error: {}", e));
// Run this server for... forever!
hyper::rt::run(server);
2018-10-30 09:04:30 +00:00
}