use std::{ net::SocketAddr, path::PathBuf, sync::Arc, time::Duration, }; use reqwest::Client; use tokio::{ spawn, sync::oneshot, }; use tracing::{ debug, info, }; #[tokio::main] async fn main () -> anyhow::Result <()> { use ptth_relay::key_validity::BlakeHashWrapper; tracing_subscriber::fmt ().try_init ().ok (); let relay_port = 4002; let proxy_port = 11510; // Start relay let server_name = "aliens_wildland"; let api_key = "AnacondaHardcoverGrannyUnlatchLankinessMutate"; let tripcode = BlakeHashWrapper::from_key (api_key.as_bytes ()); debug! ("Relay is expecting tripcode {}", tripcode.encode_base64 ()); let relay_state = ptth_relay::Relay::build () .port (relay_port) .server (ptth_relay::config::file::Server { name: server_name.to_string (), tripcode, display_name: None, }) .build ().expect ("Can't create relay state"); let relay_state = Arc::new (relay_state); let (stop_relay_tx, stop_relay_rx) = oneshot::channel (); let task_relay = spawn ({ let relay_state = relay_state.clone (); async move { ptth_relay::run_relay ( relay_state, &PathBuf::new (), stop_relay_rx, None ).await } }); assert! (relay_state.list_servers ().await.is_empty ()); // Start proxy let (stop_proxy_tx, stop_proxy_rx) = oneshot::channel (); let task_proxy = spawn (async move { debug_proxy::run_proxy (SocketAddr::from (([0, 0, 0, 0], proxy_port)), format! ("127.0.0.1:{}", relay_port), stop_proxy_rx).await }); // Start server let relay_url = format! ("http://127.0.0.1:{}", proxy_port); let config_file = ptth_server::ConfigFile::new ( server_name.into (), api_key.into (), format! ("{}/7ZSFUKGV", relay_url), ); let (stop_server_tx, stop_server_rx) = oneshot::channel (); let task_server = { spawn (async move { ptth_server::run_server (config_file, stop_server_rx, None, None, None).await }) }; tokio::time::sleep (Duration::from_millis (1000)).await; assert_eq! (relay_state.list_servers ().await, vec! [ server_name.to_string (), ]); let client = Client::builder () .timeout (Duration::from_secs (2)) .build ().expect ("Couldn't build HTTP client"); let resp = client.get (&format! ("{}/frontend/relay_up_check", relay_url)) .send ().await.expect ("Couldn't check if relay is up").bytes ().await.expect ("Couldn't check if relay is up"); assert_eq! (resp, "Relay is up\n"); let resp = client.get (&format! ("{}/frontend/servers/{}/files/COPYING", relay_url, server_name)) .send ().await.expect ("Couldn't find license").bytes ().await.expect ("Couldn't find license"); if blake3::hash (&resp) != [ 0xca, 0x02, 0x92, 0x78, 0x9c, 0x0a, 0x0e, 0xcb, 0xa7, 0x06, 0xf4, 0xb3, 0xf3, 0x49, 0x30, 0x07, 0xa9, 0x95, 0x17, 0x31, 0xc1, 0xd4, 0x32, 0xc5, 0x2c, 0x4a, 0xac, 0x1f, 0x1a, 0xbb, 0xa8, 0xef, ] { panic! ("{}", String::from_utf8 (resp.to_vec ()).expect ("???")); } // Requesting a file from a server that isn't registered // will error out let resp = client.get (&format! ("{}/frontend/servers/obviously_this_server_does_not_exist/files/COPYING", relay_url)) .send ().await.expect ("Couldn't send request to bogus server"); assert_eq! (resp.status (), reqwest::StatusCode::NOT_FOUND); tokio::time::sleep (Duration::from_secs (60)).await; info! ("Shutting down end-to-end test"); stop_server_tx.send (()).expect ("Couldn't shut down server"); task_server.await.expect ("Couldn't join server").expect ("Server error"); info! ("Server stopped"); stop_proxy_tx.send (()).expect ("Couldn't shut down proxy"); task_proxy.await.expect ("Couldn't join proxy").expect ("Proxy error"); info! ("Proxy stopped"); stop_relay_tx.send (()).expect ("Couldn't shut down relay"); task_relay.await.expect ("Couldn't join relay").expect ("Relay error"); info! ("Relay stopped"); Ok (()) }