From a4a0fb2b48afa1a37bfd0e5f4fb8d9770b8cdfe3 Mon Sep 17 00:00:00 2001
From: aceberg <1502200+aceberg@users.noreply.github.com>
Date: Fri, 6 Oct 2023 22:37:01 +0700
Subject: [PATCH] Config for history

---
 internal/web/config.go             | 1 +
 internal/web/line.go               | 2 --
 internal/web/sort.go               | 6 +-----
 internal/web/templates/config.html | 4 ++++
 4 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/internal/web/config.go b/internal/web/config.go
index ecf99e6..e14b3bc 100644
--- a/internal/web/config.go
+++ b/internal/web/config.go
@@ -41,6 +41,7 @@ func saveConfigHandler(w http.ResponseWriter, r *http.Request) {
 	AppConfig.Color = r.FormValue("color")
 	AppConfig.IgnoreIP = r.FormValue("ignoreip")
 	AppConfig.LogLevel = r.FormValue("loglevel")
+	AppConfig.HistDays = r.FormValue("history")
 
 	timeout := r.FormValue("timeout")
 	AppConfig.Timeout, err = strconv.Atoi(timeout)
diff --git a/internal/web/line.go b/internal/web/line.go
index 411cc32..2cce657 100644
--- a/internal/web/line.go
+++ b/internal/web/line.go
@@ -9,8 +9,6 @@ import (
 func lineHandler(w http.ResponseWriter, r *http.Request) {
 	var guiData models.GuiData
 
-	updateAllHosts()
-
 	guiData.Config = AppConfig
 	guiData.Hosts = []models.Host{}
 
diff --git a/internal/web/sort.go b/internal/web/sort.go
index 86a5b74..a26df0a 100644
--- a/internal/web/sort.go
+++ b/internal/web/sort.go
@@ -43,7 +43,6 @@ func sortByIPs(method string) {
 }
 
 func sortHandler(w http.ResponseWriter, r *http.Request) {
-	var guiData models.GuiData
 
 	sortMethod := r.FormValue("sort_method")
 
@@ -76,10 +75,7 @@ func sortHandler(w http.ResponseWriter, r *http.Request) {
 		updateAllHosts()
 	}
 
-	guiData.Config = AppConfig
-	guiData.Hosts = AllHosts
-
-	execTemplate(w, "line", guiData)
+	http.Redirect(w, r, r.Header.Get("Referer"), 302)
 }
 
 func sortByField(method string, field string) {
diff --git a/internal/web/templates/config.html b/internal/web/templates/config.html
index 07c4f75..e39a2d2 100644
--- a/internal/web/templates/config.html
+++ b/internal/web/templates/config.html
@@ -69,6 +69,10 @@
                         </select>
                     </td>
                 </tr>
+                <tr>
+                    <td>Keep history (days)</td>
+                    <td><input name="history" type="text" class="form-control" value="{{ .Config.HistDays }}"></td>
+                </tr>
                 <tr>
                     <td><button type="submit" class="btn btn-primary">Save</button></td>
             </form>