From 0b02918dfbc177ab4fc1d247c4d1ed5032f9c72a Mon Sep 17 00:00:00 2001 From: Diff Date: Wed, 24 Mar 2021 05:44:09 +0000 Subject: [PATCH] Add success message for uploads. --- main.go | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/main.go b/main.go index ad0af23..bf01e7b 100644 --- a/main.go +++ b/main.go @@ -155,7 +155,7 @@ var baseHeader = ` var baseFooter = ` ` -var uploadTemplate = `{{template "BaseHeader" "RUFF Upload Form"}} +var uploadTemplate = `{{template "BaseHeader" "RUFF - Upload Form"}}
@@ -163,16 +163,21 @@ var uploadTemplate = `{{template "BaseHeader" "RUFF Upload Form"}}
{{template "BaseFooter"}}` -var errorTemplate = `{{template "BaseHeader" "Upload Error"}} +var errorTemplate = `{{template "BaseHeader" "RUFF - Upload Error"}}

{{.}}

Go back

{{template "BaseFooter"}}` +var messageTemplate = `{{template "BaseHeader" (print "RUFF - " .)}} +

{{.}}

+{{template "BaseFooter"}}` + func setupUpload(server *http.Server, conf Config) { tpl := template.Must(template.New("BaseHeader").Parse(baseHeader)) template.Must(tpl.New("BaseFooter").Parse(baseFooter)) template.Must(tpl.New("UploadForm").Parse(uploadTemplate)) template.Must(tpl.New("UploadError").Parse(errorTemplate)) + template.Must(tpl.New("UploadMessage").Parse(messageTemplate)) http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { // Upload form @@ -206,5 +211,8 @@ func setupUpload(server *http.Server, conf Config) { tpl.ExecuteTemplate(w, "UploadError", err) return } + + tpl.ExecuteTemplate(w, "UploadMessage", "Upload successful!") + server.Shutdown(context.Background()) }) }