ptth/src/bin/server.rs

122 lines
2.5 KiB
Rust

use std::{
convert::Infallible,
error::Error,
sync::Arc,
time::Duration,
};
use hyper::{
StatusCode,
Uri,
};
use reqwest::{
Body,
Client,
};
use tokio::{
fs::File,
io::AsyncReadExt,
sync::mpsc::{
channel,
Receiver,
},
time::delay_for,
};
#[tokio::main]
async fn main () -> Result <(), Box <dyn Error>> {
let client = Arc::new (Client::new ());
let mut backoff_delay = 0;
loop {
if backoff_delay > 0 {
delay_for (Duration::from_millis (backoff_delay)).await;
}
let _uri = Uri::builder ()
.scheme ("http")
.authority ("127.0.0.1:4000")
.path_and_query ("/http_listen/alien_wildlands")
.build ().unwrap ();
let req_req = client.get ("http://127.0.0.1:4000/http_listen/alien_wildlands");
println! ("Step 1");
let req_resp = match req_req.send ().await {
Err (e) => {
println! ("Err: {:?}", e);
backoff_delay = backoff_delay * 2 + 500;
continue;
},
Ok (r) => {
backoff_delay = 0;
r
},
};
if req_resp.status () != StatusCode::OK {
continue;
}
println! ("Step 3");
let body = req_resp.bytes ().await?;
let body = String::from (std::str::from_utf8 (&body)?);
println! ("Client requested {}", body);
println! ("Step 4/5");
let payload = String::from ("Ha ha hue hue it worked.\n");
println! ("Step 6");
let client = client.clone ();
tokio::spawn (async move {
let resp_req = Uri::builder ()
.scheme ("http")
.authority ("127.0.0.1:4000")
.path_and_query ("/listen/alien_wildlands")
.build ().unwrap ();
let (tx, rx) = channel (2);
//let rx: Receiver <Vec <u8>> = rx;
tokio::spawn (async move {
let path = "/home/user/pictures/bzqcChY.jpg";
let path = "/home/user/videos/Decearing Egg.webm";
let mut f = File::open (path).await.unwrap ();
let mut tx = tx;
let mut bytes_sent = 0;
loop {
let mut buffer = vec! [0u8; 4096];
let bytes_read = f.read (&mut buffer).await.unwrap ();
buffer.truncate (bytes_read);
if bytes_read == 0 {
break;
}
tx.send (Ok::<_, Infallible> (buffer)).await;
bytes_sent += bytes_read;
println! ("Sent {} bytes", bytes_sent);
delay_for (Duration::from_millis (50)).await;
}
});
let resp_req = client.post (&format! ("http://127.0.0.1:4000/http_response/{}", body)).body (Body::wrap_stream (rx));
println! ("Step 6");
match resp_req.send ().await {
Err (e) => {
println! ("Err: {:?}", e);
},
Ok (_) => (),
}
});
}
}