2019-12-09 01:28:09 +01:00
|
|
|
use std::io::Write;
|
|
|
|
|
|
2019-12-07 15:40:31 +01:00
|
|
|
use actix_multipart::Multipart;
|
|
|
|
|
use actix_web::{middleware, web, App, Error, HttpResponse, HttpServer};
|
2020-03-22 00:31:45 +01:00
|
|
|
use futures::{StreamExt, TryStreamExt};
|
2019-12-07 15:40:31 +01:00
|
|
|
|
|
|
|
|
async fn save_file(mut payload: Multipart) -> Result<HttpResponse, Error> {
|
|
|
|
|
// iterate over multipart stream
|
2020-03-22 00:31:45 +01:00
|
|
|
while let Ok(Some(mut field)) = payload.try_next().await {
|
2019-12-07 15:40:31 +01:00
|
|
|
let content_type = field.content_disposition().unwrap();
|
|
|
|
|
let filename = content_type.get_filename().unwrap();
|
2020-05-19 06:48:49 +02:00
|
|
|
let filepath = format!("./tmp/{}", sanitize_filename::sanitize(&filename));
|
2019-12-09 01:28:09 +01:00
|
|
|
// File::create is blocking operation, use threadpool
|
|
|
|
|
let mut f = web::block(|| std::fs::File::create(filepath))
|
|
|
|
|
.await
|
|
|
|
|
.unwrap();
|
2019-12-07 15:40:31 +01:00
|
|
|
// Field in turn is stream of *Bytes* object
|
|
|
|
|
while let Some(chunk) = field.next().await {
|
|
|
|
|
let data = chunk.unwrap();
|
2019-12-09 01:28:09 +01:00
|
|
|
// filesystem operations are blocking, we have to use threadpool
|
|
|
|
|
f = web::block(move || f.write_all(&data).map(|_| f)).await?;
|
2019-12-07 15:40:31 +01:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
Ok(HttpResponse::Ok().into())
|
2018-06-08 06:34:14 +02:00
|
|
|
}
|
|
|
|
|
|
2019-03-29 21:43:03 +01:00
|
|
|
fn index() -> HttpResponse {
|
2018-06-08 06:34:14 +02:00
|
|
|
let html = r#"<html>
|
|
|
|
|
<head><title>Upload Test</title></head>
|
|
|
|
|
<body>
|
|
|
|
|
<form target="/" method="post" enctype="multipart/form-data">
|
2019-12-07 15:40:31 +01:00
|
|
|
<input type="file" multiple name="file"/>
|
2018-06-08 06:34:14 +02:00
|
|
|
<input type="submit" value="Submit"></button>
|
|
|
|
|
</form>
|
|
|
|
|
</body>
|
|
|
|
|
</html>"#;
|
|
|
|
|
|
2019-03-29 21:43:03 +01:00
|
|
|
HttpResponse::Ok().body(html)
|
2018-04-13 03:18:42 +02:00
|
|
|
}
|
|
|
|
|
|
2019-12-07 18:59:24 +01:00
|
|
|
#[actix_rt::main]
|
|
|
|
|
async fn main() -> std::io::Result<()> {
|
2019-03-29 21:43:03 +01:00
|
|
|
std::env::set_var("RUST_LOG", "actix_server=info,actix_web=info");
|
2019-12-07 15:40:31 +01:00
|
|
|
std::fs::create_dir_all("./tmp").unwrap();
|
2019-12-07 18:59:24 +01:00
|
|
|
|
2019-12-07 15:40:31 +01:00
|
|
|
let ip = "0.0.0.0:3000";
|
2019-12-07 18:59:24 +01:00
|
|
|
|
2019-03-29 21:43:03 +01:00
|
|
|
HttpServer::new(|| {
|
2019-12-07 18:59:24 +01:00
|
|
|
App::new().wrap(middleware::Logger::default()).service(
|
|
|
|
|
web::resource("/")
|
|
|
|
|
.route(web::get().to(index))
|
|
|
|
|
.route(web::post().to(save_file)),
|
2019-12-07 15:40:31 +01:00
|
|
|
)
|
2019-03-10 03:03:09 +01:00
|
|
|
})
|
2019-12-07 15:40:31 +01:00
|
|
|
.bind(ip)?
|
2019-12-25 17:48:33 +01:00
|
|
|
.run()
|
2019-12-07 18:59:24 +01:00
|
|
|
.await
|
2018-04-13 03:18:42 +02:00
|
|
|
}
|
