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

add env_title for site title

Browse Source 
Signed-off-by: Amneesh Singh <natto@weirdnatto.in>
This commit is contained in:
Amneesh Singh
2022-11-23 23:43:30 +05:30
parent 1cf5f76030
commit 282149e2a2
1 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/main.rs
View File

@@ -33,7 +33,8 @@ enum RData {
#[post("/", data = "<upload>")] #[post("/", data = "<upload>")]
async fn post_file(mut upload: Form<Upload<'_>>) -> (Status, RData) { async fn post_file(mut upload: Form<Upload<'_>>) -> (Status, RData) {
if env_use_key() && env_key() != upload.key { let key = env_key();
if !key.is_empty() && key != upload.key {
return ( return (
Status::BadRequest, Status::BadRequest,
RData::Raw(String::from("key not found in the header")), RData::Raw(String::from("key not found in the header")),
@@ -96,7 +97,7 @@ fn index() -> RawHtml<String> {
<head> <head>
<title> <title>
nattofiles {title}
</title> </title>
</head> </head>
@@ -126,7 +127,8 @@ fn index() -> RawHtml<String> {
</html> </html>
"#, "#,
user_url = env_user_url() user_url = env_user_url(),
title = env_title()
)) ))
} }
@@ -134,15 +136,8 @@ fn env_root_dir() -> String {
env::var("ROOT_DIR").unwrap_or(String::from("/var/files")) env::var("ROOT_DIR").unwrap_or(String::from("/var/files"))
} }
fn env_use_key() -> bool {
env::var("USE_KEY")
.unwrap_or(String::from("false"))
.parse::<bool>()
.unwrap_or(false)
}
fn env_key() -> String { fn env_key() -> String {
env::var("KEY").expect("KEY not set in the environment") env::var("KEY").unwrap_or(String::new())
} }
fn env_user_url() -> String { fn env_user_url() -> String {
@@ -154,6 +149,10 @@ fn env_user_url() -> String {
)) ))
} }
fn env_title() -> String {
env::var("TITLE").unwrap_or(String::from("Simpler Filehost"))
}
fn env_cors() -> bool { fn env_cors() -> bool {
env::var("USE_CORS") env::var("USE_CORS")
.unwrap_or(String::from("false")) .unwrap_or(String::from("false"))