From cddbcfc6d1e71e6146ba36e9c91ff38e1afd6165 Mon Sep 17 00:00:00 2001
From: Mickael <mickael.kerjean@health.nsw.gov.au>
Date: Mon, 29 Mar 2021 23:44:04 +1100
Subject: [PATCH] feature (zip): configurable timeout on zip download

---
 server/ctrl/files.go | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/server/ctrl/files.go b/server/ctrl/files.go
index ac4a3e99..5ff1fe48 100644
--- a/server/ctrl/files.go
+++ b/server/ctrl/files.go
@@ -23,9 +23,10 @@ type FileInfo struct {
 	Time int64  `json:"time"`
 }
 
-const ZIP_TIMEOUT = 100
-
-var FileCache AppCache
+var (
+	FileCache AppCache
+	ZipTimeout int
+)
 
 func init() {
 	FileCache = NewAppCache()
@@ -33,6 +34,17 @@ func init() {
 	FileCache.OnEvict(func(key string, value interface{}) {
 		os.RemoveAll(filepath.Join(cachePath, key))
 	})
+	ZipTimeout = Config.Get("features.protection.zip_timeout").Schema(func(f *FormElement) *FormElement{
+		if f == nil {
+			f = &FormElement{}
+		}
+		f.Default = 60
+		f.Name = "zip_timeout"
+		f.Type = "number"
+		f.Description = "Timeout when user wants to download archive as a zip"
+		f.Placeholder = "Default: 60seconds"
+			return f
+	}).Int()
 }
 
 func FileLs(ctx App, res http.ResponseWriter, req *http.Request) {
@@ -426,7 +438,7 @@ func FileDownloader(ctx App, res http.ResponseWriter, req *http.Request) {
 	start := time.Now()
 	var addToZipRecursive func(App, *zip.Writer, string, string) error
 	addToZipRecursive = func(c App, zw *zip.Writer, backendPath string, zipRoot string) (err error) {
-		if time.Now().Sub(start) > ZIP_TIMEOUT * time.Second {
+		if time.Now().Sub(start) > time.Duration(ZipTimeout) * time.Second {
 			return ErrTimeout
 		}
 		if strings.HasSuffix(backendPath, "/") == false {