lssa/sequencer_rpc/src/net_utils.rs

use std::io;
use std::sync::Arc;

use actix_cors::Cors;
use actix_web::{App, Error as HttpError, HttpResponse, HttpServer, http, middleware, web};
use futures::Future;
use futures::FutureExt;
use log::info;

use common::rpc_primitives::RpcConfig;
use common::rpc_primitives::message::Message;
use sequencer_core::SequencerCore;
use tokio::sync::Mutex;

use super::JsonHandler;

pub const SHUTDOWN_TIMEOUT_SECS: u64 = 10;

pub const NETWORK: &str = "network";

pub(crate) fn rpc_handler(
    message: web::Json<Message>,
    handler: web::Data<JsonHandler>,
) -> impl Future<Output = Result<HttpResponse, HttpError>> {
    let response = async move {
        let message = handler.process(message.0).await?;
        Ok(HttpResponse::Ok().json(&message))
    };
    response.boxed()
}

fn get_cors(cors_allowed_origins: &[String]) -> Cors {
    let mut cors = Cors::permissive();
    if cors_allowed_origins != ["*".to_string()] {
        for origin in cors_allowed_origins {
            cors = cors.allowed_origin(origin);
        }
    }
    cors.allowed_methods(vec!["GET", "POST"])
        .allowed_headers(vec![http::header::AUTHORIZATION, http::header::ACCEPT])
        .allowed_header(http::header::CONTENT_TYPE)
        .max_age(3600)
}

#[allow(clippy::too_many_arguments)]
pub fn new_http_server(
    config: RpcConfig,
    seuquencer_core: Arc<Mutex<SequencerCore>>,
) -> io::Result<actix_web::dev::Server> {
    let RpcConfig {
        addr,
        cors_allowed_origins,
        polling_config,
        limits_config,
    } = config;
    info!(target:NETWORK, "Starting http server at {addr}");
    let handler = web::Data::new(JsonHandler {
        polling_config,
        sequencer_state: seuquencer_core.clone(),
    });

    // HTTP server
    Ok(HttpServer::new(move || {
        App::new()
            .wrap(get_cors(&cors_allowed_origins))
            .app_data(handler.clone())
            .app_data(web::JsonConfig::default().limit(limits_config.json_payload_max_size))
            .wrap(middleware::Logger::default())
            .service(web::resource("/").route(web::post().to(rpc_handler)))
    })
    .bind(addr)?
    .shutdown_timeout(SHUTDOWN_TIMEOUT_SECS)
    .disable_signals()
    .run())
}