From b36aaeaac878b6a3d8351d4a941af5cc93341dd3 Mon Sep 17 00:00:00 2001
From: Julian T <julian@jtle.dk>
Date: Thu, 10 Sep 2020 22:54:37 +0200
Subject: Added file upload

---
 main.go | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'main.go')

diff --git a/main.go b/main.go
index 74b38b5..e6c1def 100644
--- a/main.go
+++ b/main.go
@@ -92,6 +92,9 @@ func NewServer(conf *Config) *Server {
     s.mux = http.NewServeMux()
     s.mux.HandleFunc("/", s.httpRoot)
     s.mux.HandleFunc("/upload", s.httpUpload)
+    s.mux.Handle("/data/", http.StripPrefix("/data/",
+        http.FileServer(http.Dir(conf.DataPath))))
+
     s.conf = conf
 
     s.filestore = map[string][]string{}
@@ -202,8 +205,10 @@ func (s *Server) httpRoot(w http.ResponseWriter, r *http.Request) {
 
     var page struct {
         Notes []note
-        Msg string
+        Msg   string
+        Files *map[string][]string
     }
+    page.Files = &s.filestore
 
     err := s.db.SelectContext(r.Context(), &page.Notes, `
         SELECT hash, name, location FROM notes WHERE available = True`)
@@ -212,7 +217,9 @@ func (s *Server) httpRoot(w http.ResponseWriter, r *http.Request) {
         return
     }
 
+    s.fslock.RLock()
     s.renderTemplate(w, &page, "root.template")
+    s.fslock.RUnlock()
 }
 
 func (s *Server) httpUpload(w http.ResponseWriter, r *http.Request) {
@@ -262,6 +269,8 @@ func (s *Server) httpUpload(w http.ResponseWriter, r *http.Request) {
 
     s.saveFile(hash, fname)
     s.httpLog(r, "Uploaded file %s", fname)
+
+    http.Redirect(w, r, "/", http.StatusSeeOther)
 }
 
 func (s *Server) allocFile(hash string, t string) string {
-- 
cgit v1.2.3