proxmox-backup/src/main.rs

124 lines
3.5 KiB
Rust
Raw Normal View History

extern crate apitest;
2018-10-30 09:04:30 +00:00
2018-11-03 09:42:48 +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-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 url::form_urlencoded;
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 parse_query(query: &str) -> Value {
println!("PARSE QUERY {}", query);
2018-10-31 09:42:14 +00:00
2018-11-01 10:30:49 +00:00
// fixme: what about repeated parameters (arrays?)
let mut raw_param = HashMap::new();
for (k, v) in form_urlencoded::parse(query.as_bytes()) {
println!("QUERY PARAM {} value {}", k, v);
raw_param.insert(k, v);
}
println!("QUERY HASH {:?}", raw_param);
return json!(null);
}
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))
}
if let Some(info) = API_ROOT.find_method(&components[2..]) {
2018-11-01 10:30:49 +00:00
println!("FOUND INFO");
let api_method_opt = match method {
&Method::GET => info.get,
&Method::PUT => info.put,
&Method::POST => info.post,
&Method::DELETE => info.delete,
_ => None,
};
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-01 10:30:49 +00:00
Some(data) => parse_query(data),
None => json!({}),
};
2018-11-01 14:41:08 +00:00
match (api_method.handler)(param) {
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!{
static ref API_ROOT: MethodInfo = apitest::api3::get_api_definition();
}
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
}