Zynh0722/Riven
1
0
Fork 0
forked from mirror/Riven
Code Pull requests Activity
Riven/example/proxy/src/main.rs

176 lines
6.1 KiB
Rust
Raw Normal View History

Adding example API proxy project
2020-06-05 01:51:25 -07:00
// #![deny(warnings)]
use std::convert::Infallible;
Update example proxy for v2, use workspaces. - `example/proxy` is new folder for `example_proxy`. - `riven` is new folder for the main riven lib. - Updated metadata to be an array and include HTTP method.
2021-09-09 14:31:39 -07:00
use hyper::http;
use http::{ Method, Request, Response, StatusCode };
use hyper::{ Body, Server };
Adding example API proxy project
2020-06-05 01:51:25 -07:00
use hyper::service::{ make_service_fn, service_fn };
use hyper::header::HeaderValue;
use lazy_static::lazy_static;
Update example proxy for v2, use workspaces. - `example/proxy` is new folder for `example_proxy`. - `riven` is new folder for the main riven lib. - Updated metadata to be an array and include HTTP method.
2021-09-09 14:31:39 -07:00
use tracing as log;
Adding example API proxy project
2020-06-05 01:51:25 -07:00
use riven::{ RiotApi, RiotApiConfig };
Update example proxy for v2, use workspaces. - `example/proxy` is new folder for `example_proxy`. - `riven` is new folder for the main riven lib. - Updated metadata to be an array and include HTTP method.
2021-09-09 14:31:39 -07:00
use riven::consts::Route;
Adding example API proxy project
2020-06-05 01:51:25 -07:00
lazy_static! {
/// Create lazy static RiotApi instance.
/// Easier than passing it around.
pub static ref RIOT_API: RiotApi = {
let api_key = std::env::var("RGAPI_KEY").ok()
Update example proxy for v2, use workspaces. - `example/proxy` is new folder for `example_proxy`. - `riven` is new folder for the main riven lib. - Updated metadata to be an array and include HTTP method.
2021-09-09 14:31:39 -07:00
.or_else(|| {
let path: std::path::PathBuf = [ env!("CARGO_MANIFEST_DIR"), "../../apikey.txt" ].iter().collect();
std::fs::read_to_string(path).ok()
})
Adding example API proxy project
2020-06-05 01:51:25 -07:00
.expect("Failed to find RGAPI_KEY env var or apikey.txt.");
Add RiotApi::new(impl Into<RiotApiConfig>), deprecate old constructors, update docs - Adds `From<AsRef<[u8]>> for RiotApiConfig` for API keys for the above.
2021-09-19 11:22:09 -07:00
RiotApi::new(RiotApiConfig::with_key(api_key.trim()).preconfig_burst())
Adding example API proxy project
2020-06-05 01:51:25 -07:00
};
}
/// Helper to create JSON error responses.
fn create_json_response(body: &'static str, status: StatusCode) -> Response<Body> {
let mut resp = Response::new(Body::from(body));
*resp.status_mut() = status;
resp.headers_mut().insert(hyper::header::CONTENT_TYPE, HeaderValue::from_static("application/json"));
Cleanup per clippy recommendations
2021-10-29 22:38:48 -07:00
resp
Adding example API proxy project
2020-06-05 01:51:25 -07:00
}
/// Main request handler service.
async fn handle_request(req: Request<Body>) -> Result<Response<Body>, Infallible> {
Update example proxy for v2, use workspaces. - `example/proxy` is new folder for `example_proxy`. - `riven` is new folder for the main riven lib. - Updated metadata to be an array and include HTTP method.
2021-09-09 14:31:39 -07:00
let (parts, body) = req.into_parts();
let http::request::Parts { method, uri, .. } = parts;
Adding example API proxy project
2020-06-05 01:51:25 -07:00
// Handle path.
Update example proxy for v2, use workspaces. - `example/proxy` is new folder for `example_proxy`. - `riven` is new folder for the main riven lib. - Updated metadata to be an array and include HTTP method.
2021-09-09 14:31:39 -07:00
let path_data_opt = parse_path(&method, uri.path());
let ( route, method_id, req_path ) = match path_data_opt {
Adding example API proxy project
2020-06-05 01:51:25 -07:00
None => return Ok(create_json_response(
r#"{"error":"Riot API endpoint method not found."}"#, StatusCode::NOT_FOUND)),
Some(path_data) => path_data,
};
Update example proxy for v2, use workspaces. - `example/proxy` is new folder for `example_proxy`. - `riven` is new folder for the main riven lib. - Updated metadata to be an array and include HTTP method.
2021-09-09 14:31:39 -07:00
log::debug!("Request to route {:?}, method ID {:?}: {} {:?}.", route, method_id, method, req_path);
// Convert http:request::Parts from hyper to reqwest's RequestBuilder.
let body = match hyper::body::to_bytes(body).await {
Err(err) => {
log::info!("Error handling request body: {:#?}", err);
return Ok(create_json_response(
r#"{"error":"Failed to handle request body."}"#, StatusCode::BAD_REQUEST));
},
Ok(bytes) => bytes,
};
let req_builder = RIOT_API.request(method, route.into(), req_path)
.body(body);
Adding example API proxy project
2020-06-05 01:51:25 -07:00
// Send request to Riot API.
Update example proxy for v2, use workspaces. - `example/proxy` is new folder for `example_proxy`. - `riven` is new folder for the main riven lib. - Updated metadata to be an array and include HTTP method.
2021-09-09 14:31:39 -07:00
let resp_result = RIOT_API.execute_raw(method_id, route.into(), req_builder).await;
Adding example API proxy project
2020-06-05 01:51:25 -07:00
let resp_info = match resp_result {
Update example proxy for v2, use workspaces. - `example/proxy` is new folder for `example_proxy`. - `riven` is new folder for the main riven lib. - Updated metadata to be an array and include HTTP method.
2021-09-09 14:31:39 -07:00
Err(err) => {
log::info!("Riot API error: {:#?}", err.source_reqwest_error());
return Ok(create_json_response(
r#"{"error":"Riot API request failed."}"#, StatusCode::INTERNAL_SERVER_ERROR));
},
Adding example API proxy project
2020-06-05 01:51:25 -07:00
Ok(resp_info) => resp_info,
};
// Write output.
let api_response = resp_info.response;
let mut out_response = Response::default();
*out_response.headers_mut() = api_response.headers().clone();
// If None, write "null" to body to be extra nice.
if resp_info.status_none {
*out_response.body_mut() = Body::from("null");
}
// Otherwise copy body.
else {
*out_response.status_mut() = api_response.status();
// Using streams would be faster.
let bytes_result = api_response.bytes().await;
let bytes = match bytes_result {
Err(_err) => return Ok(create_json_response(
r#"{"error":"Failed to get body from Riot API response."}"#, StatusCode::INTERNAL_SERVER_ERROR)),
Ok(bytes) => bytes,
};
*out_response.body_mut() = Body::from((&bytes[..]).to_vec());
}
Cleanup per clippy recommendations
2021-10-29 22:38:48 -07:00
Ok(out_response)
Adding example API proxy project
2020-06-05 01:51:25 -07:00
}
Update example proxy for v2, use workspaces. - `example/proxy` is new folder for `example_proxy`. - `riven` is new folder for the main riven lib. - Updated metadata to be an array and include HTTP method.
2021-09-09 14:31:39 -07:00
/// Gets the region, method_id, and Riot API path based on the given http method and path.
fn parse_path<'a>(http_method: &Method, req_path: &'a str) -> Option<( Route, &'static str, &'a str )> {
Adding example API proxy project
2020-06-05 01:51:25 -07:00
// Split URI into region and rest of path.
let req_path = req_path.trim_start_matches('/');
Update example proxy for v2, use workspaces. - `example/proxy` is new folder for `example_proxy`. - `riven` is new folder for the main riven lib. - Updated metadata to be an array and include HTTP method.
2021-09-09 14:31:39 -07:00
let ( route, req_path ) = req_path.split_at(req_path.find('/')?);
let route: Route = route.to_uppercase().parse().ok()?;
Adding example API proxy project
2020-06-05 01:51:25 -07:00
// Find method_id for given path.
Update example proxy for v2, use workspaces. - `example/proxy` is new folder for `example_proxy`. - `riven` is new folder for the main riven lib. - Updated metadata to be an array and include HTTP method.
2021-09-09 14:31:39 -07:00
let method_id = find_matching_method_id(http_method, req_path)?;
Adding example API proxy project
2020-06-05 01:51:25 -07:00
Cleanup per clippy recommendations
2021-10-29 22:38:48 -07:00
Some(( route, method_id, req_path ))
Adding example API proxy project
2020-06-05 01:51:25 -07:00
}
/// Finds the method_id given the request path.
Update example proxy for v2, use workspaces. - `example/proxy` is new folder for `example_proxy`. - `riven` is new folder for the main riven lib. - Updated metadata to be an array and include HTTP method.
2021-09-09 14:31:39 -07:00
fn find_matching_method_id(http_method: &Method, req_path: &str) -> Option<&'static str> {
for ( endpoint_http_method, ref_path, method_id ) in &riven::meta::ALL_ENDPOINTS {
if http_method == endpoint_http_method && paths_match(ref_path, req_path) {
Adding example API proxy project
2020-06-05 01:51:25 -07:00
return Some(method_id);
}
}
None
}
/// Checks if the request path (req_path) matches the reference path (ref_path).
fn paths_match(ref_path: &str, req_path: &str) -> bool {
let mut ref_iter = ref_path.split('/');
let mut req_iter = req_path.split('/');
loop {
let ref_seg_opt = ref_iter.next();
let req_seg_opt = req_iter.next();
if ref_seg_opt.is_none() != req_seg_opt.is_none() {
return false;
}
if let Some(ref_seg) = ref_seg_opt {
if let Some(req_seg) = req_seg_opt {
if ref_seg.starts_with('{') || ref_seg == req_seg {
continue;
}
return false;
}}
return true;
}
}
#[tokio::main]
pub async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
Update example proxy for v2, use workspaces. - `example/proxy` is new folder for `example_proxy`. - `riven` is new folder for the main riven lib. - Updated metadata to be an array and include HTTP method.
2021-09-09 14:31:39 -07:00
// Setup loggers.
// env_logger::init();
tracing_subscriber::fmt::init();
// Trigger lazy_static.
let _ = &*RIOT_API;
Adding example API proxy project
2020-06-05 01:51:25 -07:00
// For every connection, we must make a `Service` to handle all
// incoming HTTP requests on said connection.
let make_svc = make_service_fn(|_conn| {
// This is the `Service` that will handle the connection.
// `service_fn` is a helper to convert a function that
// returns a Response into a `Service`.
async { Ok::<_, Infallible>(service_fn(handle_request)) }
});
let addr = ([ 127, 0, 0, 1 ], 3000).into();
let server = Server::bind(&addr).serve(make_svc);
Update example proxy for v2, use workspaces. - `example/proxy` is new folder for `example_proxy`. - `riven` is new folder for the main riven lib. - Updated metadata to be an array and include HTTP method.
2021-09-09 14:31:39 -07:00
log::info!("Listening on http://{}", addr);
Adding example API proxy project
2020-06-05 01:51:25 -07:00
server.await?;
Ok(())
}
Copy permalink