diff --git a/src/main.rs b/src/main.rs index b7d29ad..08acb1f 100644 --- a/src/main.rs +++ b/src/main.rs @@ -4,13 +4,7 @@ mod multipart; use actix_files::{Files, NamedFile}; use actix_multipart::Multipart; -use actix_web::{ - error, - http::header::{ContentDisposition, DispositionParam, DispositionType}, - middleware, - web::{self, Bytes}, - App, Error, FromRequest, HttpRequest, HttpResponse, HttpServer, -}; +use actix_web::{error, http::header::{ContentDisposition, DispositionParam, DispositionType}, middleware, web::{self, Bytes}, App, Error, FromRequest, HttpRequest, HttpResponse, HttpServer}; use async_std::{ fs, path::PathBuf, @@ -79,10 +73,10 @@ async fn upload( .finish()) } -async fn uploaded(id: web::Path, config: web::Data) -> Result { +async fn uploaded(id: web::Path, req: web::HttpRequest) -> Result { + let url = req.url_for("file", &[id.as_str()])?; let upload_html = UPLOAD_HTML - .replace("{id}", id.as_ref()) - .replace("{server}", &config.server_url); + .replace("{url}", url.as_str()); Ok(HttpResponse::Ok() .content_type("text/html") .body(upload_html)) @@ -175,7 +169,6 @@ async fn setup_db() -> PgPool { #[derive(Clone)] struct Config { - server_url: String, files_dir: PathBuf, } @@ -188,7 +181,6 @@ async fn main() -> std::io::Result<()> { let pool: PgPool = setup_db().await; let config = Config { - server_url: env::var("SERVER_URL").unwrap_or_else(|_| "http://localhost:8000".to_owned()), files_dir: PathBuf::from(env::var("FILES_DIR").unwrap_or_else(|_| "./files".to_owned())), }; fs::create_dir_all(&config.files_dir) @@ -223,7 +215,7 @@ async fn main() -> std::io::Result<()> { .service(web::resource("/").route(web::get().to(index))) .service(web::resource("/upload").route(web::post().to(upload))) .service(web::resource("/upload/{id}").route(web::get().to(uploaded))) - .service(web::resource("/file/{id}").route(web::get().to(download))) + .service(web::resource("/file/{id}").name("file").route(web::get().to(download))) .service(Files::new("/static", "static").disable_content_disposition()) } }) diff --git a/template/upload.html b/template/upload.html index 7ac8f06..ff1c2a7 100644 --- a/template/upload.html +++ b/template/upload.html @@ -11,8 +11,8 @@

datatrash

Datei-Link: - - {server}/files/{id} + + {url}