feat(config): support overriding the server URL

src/config.rs

@@ -30,6 +30,8 @@ pub struct Settings {
 pub struct ServerConfig {
     /// The socket address to bind.
     pub address: String,
+    /// URL that can be used to access the server externally.
+    pub url: Option<String>,
     /// Number of workers to start.
     pub workers: Option<usize>,
     /// Maximum content length.

src/server.rs

@@ -137,6 +137,18 @@ async fn upload(
 ) -> Result<HttpResponse, Error> {
     let connection = request.connection_info().clone();
     let host = connection.peer_addr().unwrap_or("unknown host");
+    let server_url = match config
+        .read()
+        .map_err(|_| error::ErrorInternalServerError("cannot acquire config"))?
+        .server
+        .url
+        .clone()
+    {
+        Some(v) => v,
+        None => {
+            format!("{}://{}", connection.scheme(), connection.host(),)
+        }
+    };
     auth::check(
         host,
         request.headers(),
@@ -187,9 +199,8 @@ async fn upload(
                     .get_file(bytes_checksum)
                 {
                     urls.push(format!(
-                        "{}://{}/{}\n",
+                        "{}/{}\n",
-                        connection.scheme(),
+                        server_url,
-                        connection.host(),
                         file.path
                             .file_name()
                             .map(|v| v.to_string_lossy())
@@ -227,12 +238,7 @@ async fn upload(
                 Byte::from_bytes(paste.data.len() as u128).get_appropriate_unit(false),
                 host
             );
-            urls.push(format!(
+            urls.push(format!("{}/{}\n", server_url, file_name));
-                "{}://{}/{}\n",
-                connection.scheme(),
-                connection.host(),
-                file_name
-            ));
         } else {
             log::warn!("{} sent an invalid form field", host);
             return Err(error::ErrorBadRequest("invalid form field"));