From 5fc487c70553abef26fdf583cfb2c2da70ff8522 Mon Sep 17 00:00:00 2001
From: Gunwant Jain <mail@wantguns.dev>
Date: Wed, 30 Dec 2020 04:19:46 +0530
Subject: [PATCH] use environment variables to configure settings

Signed-off-by: Gunwant Jain <mail@wantguns.dev>
---
 src/main.rs | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 1cfb05b..11e0037 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -3,6 +3,7 @@
 
 extern crate tree_magic;
 
+use std::env;
 use std::io::prelude::*;
 use std::collections::HashMap;
 use std::path::Path;
@@ -23,7 +24,7 @@ fn pretty_retrieve(id: PasteId) -> Option<Template> {
 
     let mut contents = String::new();
     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();
     map.insert("title", id.to_string());
@@ -56,8 +57,11 @@ fn upload(paste: Data) -> Result<String, std::io::Error> {
     paste.stream_to_file(filepath)?;
 
     let url = match tree_magic::from_filepath(filepath).as_str().contains("text") {
-        true    => format!("{host}/p/{id}\n", host = "http://localhost:8000", id = id),
-        false   => format!("{host}/{id}\n", host = "http://localhost:8000", id = id),
+        true    => format!("https://{host}/p/{id}\n", host = env::var("HOST")
+            .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)