natto1784/simpler-filehost
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add a form for multipart upload

Browse Source 
Signed-off-by: Amneesh Singh <natto@weirdnatto.in>
This commit is contained in:
Amneesh Singh
2022-11-23 21:19:00 +05:30
parent 64133b78d2
commit b23a5f18c6
2 changed files with 79 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
src/main.rs
View File

@@ -3,9 +3,18 @@
#[macro_use] #[macro_use]
extern crate rocket; extern crate rocket;
use rand::distributions::{Alphanumeric, DistString}; use rand::{
use rand::thread_rng; distributions::{Alphanumeric, DistString},
use rocket::{form::Form, fs::NamedFile, fs::TempFile, http::Status, response::status, Config}; thread_rng,
};
use rocket::{
form::Form,
fs::{NamedFile, TempFile},
http::Status,
response::Redirect,
Config,
};
use rocket_dyn_templates::{context, Template};
use std::{env, fs}; use std::{env, fs};
#[derive(FromForm)] #[derive(FromForm)]
@@ -13,14 +22,22 @@ struct Upload<'r> {
file: TempFile<'r>, file: TempFile<'r>,
#[field(default = "")] #[field(default = "")]
key: String, key: String,
#[field(default = false)]
redirect: bool,
}
#[derive(Responder)]
enum RData {
Raw(String),
Redir(Redirect),
} }
#[post("/", data = "<upload>")] #[post("/", data = "<upload>")]
async fn post_file(mut upload: Form<Upload<'_>>) -> status::Custom<String> { async fn post_file(mut upload: Form<Upload<'_>>) -> (Status, RData) {
if env_use_key() && env_key() != upload.key { if env_use_key() && env_key() != upload.key {
return status::Custom( return (
Status::BadRequest, Status::BadRequest,
"key not found in the header".to_string(), RData::Raw(String::from("key not found in the header")),
); );
} }
@@ -38,15 +55,27 @@ async fn post_file(mut upload: Form<Upload<'_>>) -> status::Custom<String> {
if let Err(error) = uploaded { if let Err(error) = uploaded {
println!("Error while copying from temp file: {:?}", error); println!("Error while copying from temp file: {:?}", error);
return status::Custom( return (
Status::InternalServerError, Status::InternalServerError,
"Some stupid internal error occurred".to_string(), RData::Raw(String::from("Some stupid internal error occurred")),
); );
} }
return status::Custom(Status::Ok, format!("{}/{}", env_user_url(), new_name)); let file_url = format!("{}/{}", env_user_url(), new_name);
if upload.redirect {
return (
Status::SeeOther,
RData::Redir(Redirect::to(file_url)),
);
}
return (Status::Ok, RData::Raw(file_url));
} else { } else {
return status::Custom(Status::BadRequest, "File name invalid".to_string()); return (
Status::BadRequest,
RData::Raw(String::from("File name invalid")),
);
} }
} }
@@ -58,23 +87,17 @@ async fn get_file(filename: String) -> Option<NamedFile> {
} }
#[get("/")] #[get("/")]
fn index() -> String { fn index() -> Template {
format!( Template::render("index", context! {user_url: env_user_url()})
"Use curl to upload:\n\
curl -F file=@\"[file]\" {url}\n\
If key is enabled then a field \"key\" might be required in which case it would be\n\
curl -F file=@\"[file]\" -F \"key=[key]\" {url}",
url = env_user_url()
)
} }
fn env_root_dir() -> String { fn env_root_dir() -> String {
env::var("ROOT_DIR").unwrap_or("/var/files".to_string()) env::var("ROOT_DIR").unwrap_or(String::from("/var/files"))
} }
fn env_use_key() -> bool { fn env_use_key() -> bool {
env::var("USE_KEY") env::var("USE_KEY")
.unwrap_or("false".to_string()) .unwrap_or(String::from("false"))
.parse::<bool>() .parse::<bool>()
.unwrap_or(false) .unwrap_or(false)
} }
@@ -94,7 +117,7 @@ fn env_user_url() -> String {
fn env_cors() -> bool { fn env_cors() -> bool {
env::var("USE_CORS") env::var("USE_CORS")
.unwrap_or("false".to_string()) .unwrap_or(String::from("false"))
.parse::<bool>() .parse::<bool>()
.unwrap_or(false) .unwrap_or(false)
} }
@@ -109,9 +132,11 @@ fn rocket() -> _ {
if env_cors() { if env_cors() {
rocket::build() rocket::build()
.attach(cors) .attach(cors)
.attach(Template::fairing())
.mount("/", routes![post_file, get_file, index]) .mount("/", routes![post_file, get_file, index])
} else { } else {
rocket::build() rocket::build()
.attach(Template::fairing())
.mount("/", routes![post_file, get_file, index]) .mount("/", routes![post_file, get_file, index])
} }
} }

33
templates/index.html.hbs Normal file
View File

@@ -0,0 +1,33 @@
<html>
<head>
<title>
nattofiles
</title>
</head>
<body>
<p>
Use curl to upload:
<br>
<code>
curl -F file=@"[file]" {{user_url}}
</code>
<br>
If key is enabled then a field "key" might be required in which case it would be
<br>
<code>
curl -F file=@"[file]" -F "key=[key]" {{user_url}}
</code>
</p>
<form method="POST" enctype="multipart/form-data">
<label for="key">Key: </label>
<input type="text" id="key"> <br>
<input type="file" name="file" id="file">
<input type="hidden" name="redirect" id="true">
<input type="submit" value="Upload">
</form>
</body>
</html>