From 5201da5b1b4c3ee043e88d9f9b87364215d1ddf2 Mon Sep 17 00:00:00 2001
From: Aravinth Manivannan <realaravinth@batsense.net>
Date: Wed, 20 Mar 2024 17:56:41 +0530
Subject: [PATCH] feat: redirect to store hostname after reset

---
 src/main.rs | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 9c2089f..1ea05d4 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -10,8 +10,8 @@ use actix_web::web::ServiceConfig;
 use actix_web::HttpResponse;
 use actix_web::{get, http::header, post, web, App, HttpServer, Responder};
 use clap::Parser;
-use url::Url;
 use log::info;
+use url::Url;
 
 fn thirappu(ribbon: &str, store: &str) {
     let ribbon_link = format!("/etc/nginx/sites-enabled/{ribbon}");
@@ -64,11 +64,17 @@ struct Args {
 
 impl Display for Args {
     fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
-        write!(f, "port: {} ribbon: {} store: {} store_weburl: {}", self.port, self.ribbon, self.store, self.store_weburl.to_string())
+        write!(
+            f,
+            "port: {} ribbon: {} store: {} store_weburl: {}",
+            self.port,
+            self.ribbon,
+            self.store,
+            self.store_weburl.to_string()
+        )
     }
 }
 
-
 #[get("/reset")]
 async fn reset_get() -> impl Responder {
     let reset = include_str!("./reset.html");
@@ -83,7 +89,7 @@ async fn reset_post(ctx: web::Data<Args>) -> impl Responder {
     reset(&ctx.ribbon, &ctx.store);
 
     HttpResponse::Found()
-        .append_header((header::LOCATION, "/"))
+        .append_header((header::LOCATION, ctx.store_weburl.to_string()))
         .finish()
 }