2022-09-22 23:27:14 +03:00
|
|
|
pub mod authorization;
|
2022-06-05 22:58:47 +03:00
|
|
|
mod errors;
|
|
|
|
|
mod http;
|
2022-08-30 23:01:42 +03:00
|
|
|
mod rpc_proxy_http;
|
|
|
|
|
mod rpc_proxy_ws;
|
2022-07-14 00:49:57 +03:00
|
|
|
mod users;
|
2022-07-07 06:22:09 +03:00
|
|
|
|
2022-08-16 22:29:00 +03:00
|
|
|
use crate::app::Web3ProxyApp;
|
2022-08-21 12:44:53 +03:00
|
|
|
use ::http::Request;
|
2022-06-05 22:58:47 +03:00
|
|
|
use axum::{
|
2022-08-16 03:33:26 +03:00
|
|
|
body::Body,
|
2022-06-05 22:58:47 +03:00
|
|
|
handler::Handler,
|
|
|
|
|
routing::{get, post},
|
|
|
|
|
Extension, Router,
|
|
|
|
|
};
|
2022-08-07 23:44:56 +03:00
|
|
|
use std::net::SocketAddr;
|
2022-06-05 22:58:47 +03:00
|
|
|
use std::sync::Arc;
|
2022-08-16 03:33:26 +03:00
|
|
|
use tower_http::trace::TraceLayer;
|
|
|
|
|
use tower_request_id::{RequestId, RequestIdLayer};
|
|
|
|
|
use tracing::{error_span, info};
|
2022-06-15 04:02:26 +03:00
|
|
|
|
2022-08-16 03:33:26 +03:00
|
|
|
/// http and websocket frontend for customers
|
2022-08-10 09:22:49 +03:00
|
|
|
pub async fn serve(port: u16, proxy_app: Arc<Web3ProxyApp>) -> anyhow::Result<()> {
|
2022-08-19 23:18:12 +03:00
|
|
|
// create a tracing span for each request with a random request id and the method
|
|
|
|
|
// GET: websocket or static pages
|
|
|
|
|
// POST: http rpc or login
|
2022-08-16 03:33:26 +03:00
|
|
|
let request_tracing_layer =
|
|
|
|
|
TraceLayer::new_for_http().make_span_with(|request: &Request<Body>| {
|
|
|
|
|
// We get the request id from the extensions
|
|
|
|
|
let request_id = request
|
|
|
|
|
.extensions()
|
|
|
|
|
.get::<RequestId>()
|
|
|
|
|
.map(ToString::to_string)
|
|
|
|
|
.unwrap_or_else(|| "unknown".into());
|
|
|
|
|
// And then we put it along with other information into the `request` span
|
|
|
|
|
error_span!(
|
|
|
|
|
"http_request",
|
|
|
|
|
id = %request_id,
|
|
|
|
|
// TODO: do we want these?
|
|
|
|
|
method = %request.method(),
|
|
|
|
|
// uri = %request.uri(),
|
|
|
|
|
)
|
|
|
|
|
});
|
|
|
|
|
|
2022-08-21 11:18:57 +03:00
|
|
|
// build our axum Router
|
2022-09-24 00:46:27 +03:00
|
|
|
// TODO: these should probbably all start with /rpc. then / can be the static site
|
2022-06-05 22:58:47 +03:00
|
|
|
let app = Router::new()
|
2022-08-21 11:18:57 +03:00
|
|
|
// routes should be order most to least common
|
2022-09-24 07:31:06 +03:00
|
|
|
.route("/rpc", post(rpc_proxy_http::proxy_web3_rpc))
|
2022-09-24 00:46:27 +03:00
|
|
|
.route("/rpc", get(rpc_proxy_ws::public_websocket_handler))
|
2022-09-24 07:31:06 +03:00
|
|
|
.route(
|
|
|
|
|
"/rpc/:user_key",
|
|
|
|
|
post(rpc_proxy_http::proxy_web3_rpc_with_key),
|
|
|
|
|
)
|
2022-09-24 00:46:27 +03:00
|
|
|
.route("/rpc/:user_key", get(rpc_proxy_ws::user_websocket_handler))
|
|
|
|
|
.route("/rpc/health", get(http::health))
|
|
|
|
|
.route("/rpc/status", get(http::status))
|
2022-09-10 05:59:07 +03:00
|
|
|
// TODO: make this optional or remove it since it is available on another port
|
2022-09-24 00:46:27 +03:00
|
|
|
.route("/rpc/prometheus", get(http::prometheus))
|
|
|
|
|
.route("/rpc/user/login/:user_address", get(users::get_login))
|
|
|
|
|
.route(
|
|
|
|
|
"/rpc/user/login/:user_address/:message_eip",
|
|
|
|
|
get(users::get_login),
|
|
|
|
|
)
|
|
|
|
|
.route("/rpc/user/login", post(users::post_login))
|
|
|
|
|
.route("/rpc/user", post(users::post_user))
|
|
|
|
|
.route("/rpc/user/logout", get(users::get_logout))
|
2022-08-21 11:18:57 +03:00
|
|
|
// layers are ordered bottom up
|
|
|
|
|
// the last layer is first for requests and last for responses
|
2022-08-07 22:35:24 +03:00
|
|
|
.layer(Extension(proxy_app))
|
2022-08-21 11:18:57 +03:00
|
|
|
// add the request id to our tracing logs
|
2022-08-16 03:33:26 +03:00
|
|
|
.layer(request_tracing_layer)
|
2022-08-21 11:18:57 +03:00
|
|
|
// create a unique id for each request
|
2022-08-16 03:33:26 +03:00
|
|
|
.layer(RequestIdLayer)
|
2022-08-07 22:35:24 +03:00
|
|
|
// 404 for any unknown routes
|
|
|
|
|
.fallback(errors::handler_404.into_service());
|
2022-06-05 22:58:47 +03:00
|
|
|
|
|
|
|
|
// run our app with hyper
|
2022-08-16 03:33:26 +03:00
|
|
|
// TODO: allow only listening on localhost? top_config.app.host.parse()?
|
2022-06-05 22:58:47 +03:00
|
|
|
let addr = SocketAddr::from(([0, 0, 0, 0], port));
|
2022-08-06 04:17:25 +03:00
|
|
|
info!("listening on port {}", port);
|
2022-08-11 05:57:01 +03:00
|
|
|
|
2022-08-26 20:26:17 +03:00
|
|
|
// TODO: into_make_service is enough if we always run behind a proxy. make into_make_service_with_connect_info optional?
|
2022-08-11 05:57:01 +03:00
|
|
|
/*
|
|
|
|
|
It sequentially looks for an IP in:
|
|
|
|
|
- x-forwarded-for header (de-facto standard)
|
|
|
|
|
- x-real-ip header
|
|
|
|
|
- forwarded header (new standard)
|
|
|
|
|
- axum::extract::ConnectInfo (if not behind proxy)
|
|
|
|
|
*/
|
|
|
|
|
let service = app.into_make_service_with_connect_info::<SocketAddr>();
|
|
|
|
|
// let service = app.into_make_service();
|
|
|
|
|
|
2022-08-13 00:00:26 +03:00
|
|
|
// `axum::Server` is a re-export of `hyper::Server`
|
2022-06-05 22:58:47 +03:00
|
|
|
axum::Server::bind(&addr)
|
2022-08-07 09:48:57 +03:00
|
|
|
// TODO: option to use with_connect_info. we want it in dev, but not when running behind a proxy, but not
|
2022-08-11 05:57:01 +03:00
|
|
|
.serve(service)
|
2022-08-13 00:00:26 +03:00
|
|
|
.with_graceful_shutdown(signal_shutdown())
|
2022-06-05 22:58:47 +03:00
|
|
|
.await
|
|
|
|
|
.map_err(Into::into)
|
|
|
|
|
}
|
2022-08-07 22:33:16 +03:00
|
|
|
|
|
|
|
|
/// Tokio signal handler that will wait for a user to press CTRL+C.
|
|
|
|
|
/// We use this in our hyper `Server` method `with_graceful_shutdown`.
|
|
|
|
|
async fn signal_shutdown() {
|
2022-08-10 09:22:49 +03:00
|
|
|
info!("ctrl-c to quit");
|
2022-08-07 22:33:16 +03:00
|
|
|
tokio::signal::ctrl_c()
|
|
|
|
|
.await
|
|
|
|
|
.expect("expect tokio signal ctrl-c");
|
|
|
|
|
info!("signal shutdown");
|
|
|
|
|
}
|
