Add option for temporary redirection

README.md

@@ -87,7 +87,7 @@ export site_url=<url>
 cd actix
 cargo run
 ```
-You can optionally set the port the server listens on by appending `--port=[port]`
+You can optionally set the port the server listens on by appending `--port=[port]`.
 ### 4. Navigate to `http://localhost:4567` in your browser, add links as you wish.
 
 ## Running with docker
@@ -122,6 +122,9 @@ docker run -p 4567:4567 \
     -d simply-shorten:latest
 ```
 
+You can also set the redirect method to Permanent 308 (default) or Temporary 307 by setting
+the `redirect_method` variable to `TEMPORARY` or `PERMANENT` (it's matched exactly).
+
 ## Disable authentication
 If you do not define a password environment variable when starting the docker image, authentication
 will be disabled.

actix/src/main.rs

@@ -71,8 +71,14 @@ async fn link_handler(shortlink: web::Path<String>, data: web::Data<AppState>) -
     if longlink == *"" {
         Redirect::to("/err/404").using_status_code(StatusCode::NOT_FOUND)
     } else {
+        let redirect_method = env::var("redirect_method").unwrap_or("PERMANENT".to_string());
         database::add_hit(shortlink.as_str(), &data.db);
-        Redirect::to(longlink).permanent()
+        if redirect_method == *"TEMPORARY" {
+            Redirect::to(longlink)
+        } else {
+            // Defaults to permanent redirection
+            Redirect::to(longlink).permanent()
+        }
     }
 }
 

compose.yaml

@@ -10,11 +10,17 @@ services:
             # In this case, you can get rid of the db volume below
             # and instead do a mount manually by specifying the location
             # - db_url=/urls.sqlite
+
             # Change it in case you want to set the website name
             # displayed in front of the shorturls, defaults to
             # the hostname you're accessing it from
             # - site_url=https://www.example.com
+
             - password=$3CuReP4S$W0rD
+            
+            # Pass the redirect method, if needed TEMPORARY and PERMANENT
+            # are accepted values, defaults to PERMANENT
+            # - redirect_method=TEMPORARY
         volumes:
             - db:/urls.sqlite
         networks: