SinTan1729/chhoto-url
SinTan1729/chhoto-url
1
0
Fork 0
mirror of https://github.com/SinTan1729/chhoto-url synced 2024-12-26 23:58:35 -06:00
Code Issues Releases Activity

chg: 404 response doesn't change the url

Browse source
This commit is contained in:
Sayantan Santra 2024-03-28 13:27:33 -05:00
parent 5d8dd6fb63
commit f526e7ec5b
Signed by: SinTan1729
GPG key ID: EB3E68BFBA25C85F
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
actix/src/main.rs
View file

@ -6,7 +6,7 @@ use actix_web::{
http::StatusCode, http::StatusCode,
middleware, post, middleware, post,
web::{self, Redirect}, web::{self, Redirect},
App, HttpResponse, HttpServer, Responder, App, Either, HttpResponse, HttpServer, Responder,
}; };
use rusqlite::Connection; use rusqlite::Connection;
use std::env; use std::env;
@ -82,13 +82,18 @@ async fn link_handler(shortlink: web::Path<String>, data: web::Data<AppState>) -
let redirect_method = env::var("redirect_method").unwrap_or(String::from("PERMANENT")); let redirect_method = env::var("redirect_method").unwrap_or(String::from("PERMANENT"));
database::add_hit(shortlink.as_str(), &data.db); database::add_hit(shortlink.as_str(), &data.db);
if redirect_method == "TEMPORARY" { if redirect_method == "TEMPORARY" {
Redirect::to(longlink) Either::Left(Redirect::to(longlink))
} else { } else {
// Defaults to permanent redirection // Defaults to permanent redirection
Redirect::to(longlink).permanent() Either::Left(Redirect::to(longlink).permanent())
} }
} else { } else {
Redirect::to("/err/404") Either::Right(
NamedFile::open_async("./resources/static/404.html")
.await
.customize()
.with_status(StatusCode::NOT_FOUND),
)
} }
} }