Change host url to https when it's enabled, should fix some problems downloading attachments

This commit is contained in:
Daniel García 2018-07-11 16:23:39 +02:00
parent ec37004dfe
commit e88d8c856d
1 changed files with 5 additions and 1 deletions

View File

@ -111,7 +111,11 @@ impl<'a, 'r> FromRequest<'a, 'r> for Headers {
// Get host
let host = match headers.get_one("Host") {
Some(host) => format!("http://{}", host), // TODO: Check if HTTPS
Some(host) => {
use std::env;
let protocol = if env::var("ROCKET_TLS").is_ok() {"https"} else {"http"};
format!("{}://{}", protocol, host)
},
_ => String::new()
};