mod config; mod proxies; mod sideload; use poem::{listener::TcpListener, Route, Server}; use poem_openapi::OpenApiService; use tracing::{error, info, Level}; #[tokio::main] async fn main() -> Result<(), std::io::Error> { // Setting up logging if std::env::var_os("RUST_LOG").is_none() { std::env::set_var("RUST_LOG", "poem=debug"); } tracing_subscriber::fmt() .with_max_level(Level::DEBUG) .init(); // Prepare all the stuff let mut instance = proxies::Instance::new(); info!("Loading proxy regions..."); match proxies::loader::scan_regions( config::C .read() .unwrap() .get_string("regions") .unwrap_or("./regions".to_string()), ) { Err(_) => error!("Loading proxy regions... failed"), Ok((regions, count)) => { instance.regions = regions; info!(count, "Loading proxy regions... done") } }; // Proxies // Sideload let sideload = OpenApiService::new(sideload::SideloadApi, "Sideload API", "1.0") .server("http://localhost:3000/cgi"); let sideload_server = Server::new(TcpListener::bind( config::C .read() .unwrap() .get_string("listen.sideload") .unwrap_or("0.0.0.0:81".to_string()), )) .run(Route::new().nest("/cgi", sideload)); tokio::try_join!(sideload_server)?; Ok(()) }