restructure folders

http-proxy/src/main.rs

@@ -0,0 +1,90 @@
+use std::net::ToSocketAddrs;
+
+use actix_web::{
+    error, middleware, web, App, Error, HttpRequest, HttpResponse, HttpServer,
+};
+use awc::Client;
+use clap::StructOpt;
+use url::Url;
+
+async fn forward(
+    req: HttpRequest,
+    payload: web::Payload,
+    url: web::Data<Url>,
+    client: web::Data<Client>,
+) -> Result<HttpResponse, Error> {
+    let mut new_url = url.get_ref().clone();
+    new_url.set_path(req.uri().path());
+    new_url.set_query(req.uri().query());
+
+    // TODO: This forwarded implementation is incomplete as it only handles the unofficial
+    // X-Forwarded-For header but not the official Forwarded one.
+    let forwarded_req = client
+        .request_from(new_url.as_str(), req.head())
+        .no_decompress();
+    let forwarded_req = match req.head().peer_addr {
+        Some(addr) => {
+            forwarded_req.insert_header(("x-forwarded-for", format!("{}", addr.ip())))
+        }
+        None => forwarded_req,
+    };
+
+    let res = forwarded_req
+        .send_stream(payload)
+        .await
+        .map_err(error::ErrorInternalServerError)?;
+
+    let mut client_resp = HttpResponse::build(res.status());
+    // Remove `Connection` as per
+    // https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Connection#Directives
+    for (header_name, header_value) in
+        res.headers().iter().filter(|(h, _)| *h != "connection")
+    {
+        client_resp.insert_header((header_name.clone(), header_value.clone()));
+    }
+
+    Ok(client_resp.streaming(res))
+}
+
+#[derive(clap::Parser, Debug)]
+struct CliArguments {
+    listen_addr: String,
+    listen_port: u16,
+    forward_addr: String,
+    forward_port: u16,
+}
+
+#[actix_web::main]
+async fn main() -> std::io::Result<()> {
+    env_logger::init_from_env(env_logger::Env::new().default_filter_or("info"));
+
+    let args = CliArguments::parse();
+
+    let forward_socket_addr = (args.forward_addr, args.forward_port)
+        .to_socket_addrs()?
+        .next()
+        .expect("given forwarding address was not valid");
+
+    let forward_url = format!("http://{}", forward_socket_addr);
+    let forward_url = Url::parse(&forward_url).unwrap();
+
+    log::info!(
+        "starting HTTP server at http://{}:{}",
+        &args.listen_addr,
+        args.listen_port
+    );
+
+    log::info!("forwarding to {forward_url}");
+
+    HttpServer::new(move || {
+        App::new()
+            .app_data(web::Data::new(Client::default()))
+            .app_data(web::Data::new(forward_url.clone()))
+            .wrap(middleware::Logger::default())
+            .default_service(web::to(forward))
+    })
+    .bind((args.listen_addr, args.listen_port))?
+    .workers(2)
+    .run()
+    .await
+}