tools/paste
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use environment variables to configure settings

Browse Source 
Signed-off-by: Gunwant Jain <mail@wantguns.dev>
This commit is contained in:
Gunwant Jain
2020-12-30 04:19:46 +05:30
parent 1bc532d237
commit 5fc487c705
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/main.rs
View File

@@ -3,6 +3,7 @@
extern crate tree_magic; extern crate tree_magic;
use std::env;
use std::io::prelude::*; use std::io::prelude::*;
use std::collections::HashMap; use std::collections::HashMap;
use std::path::Path; use std::path::Path;
@@ -23,7 +24,7 @@ fn pretty_retrieve(id: PasteId) -> Option<Template> {
let mut contents = String::new(); let mut contents = String::new();
file.read_to_string(&mut contents).unwrap(); file.read_to_string(&mut contents).unwrap();
let theme = String::from("."); let theme = env::var("THEME").unwrap_or(".".to_string());
let mut map = HashMap::new(); let mut map = HashMap::new();
map.insert("title", id.to_string()); map.insert("title", id.to_string());
@@ -56,8 +57,11 @@ fn upload(paste: Data) -> Result<String, std::io::Error> {
paste.stream_to_file(filepath)?; paste.stream_to_file(filepath)?;
let url = match tree_magic::from_filepath(filepath).as_str().contains("text") { let url = match tree_magic::from_filepath(filepath).as_str().contains("text") {
true => format!("{host}/p/{id}\n", host = "http://localhost:8000", id = id), true => format!("https://{host}/p/{id}\n", host = env::var("HOST")
false => format!("{host}/{id}\n", host = "http://localhost:8000", id = id), .unwrap_or("<no_host_provided>".to_string()), id = id),
false => format!("https://{host}/{id}\n", host = env::var("HOST")
.unwrap_or("<no_host_provided>".to_string()), id = id)
}; };
Ok(url) Ok(url)