From 7c6029aa996af182b343a289c6c82b4330c458f5 Mon Sep 17 00:00:00 2001
From: William Harrell <wharrell1@protonmail.com>
Date: Fri, 4 Nov 2022 22:43:18 -0400
Subject: [PATCH] watch file transfer dir for changes

---
 server/internal/websocket/websocket.go | 49 ++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)

diff --git a/server/internal/websocket/websocket.go b/server/internal/websocket/websocket.go
index 23c42bd6..f4617658 100644
--- a/server/internal/websocket/websocket.go
+++ b/server/internal/websocket/websocket.go
@@ -8,6 +8,7 @@ import (
 	"sync/atomic"
 	"time"
 
+	"github.com/fsnotify/fsnotify"
 	"github.com/gorilla/websocket"
 	"github.com/rs/zerolog"
 	"github.com/rs/zerolog/log"
@@ -211,6 +212,30 @@ func (ws *WebSocketHandler) Start() {
 
 		ws.logger.Err(err).Msg("sync clipboard")
 	})
+
+	// watch for file changes
+	if ws.conf.FileTransfer {
+		watcher, err := fsnotify.NewWatcher()
+		if err != nil {
+			ws.logger.Err(err).Msg("unable to start file transfer dir watcher")
+			return
+		}
+
+		go func() {
+			for {
+				select {
+				case <-watcher.Events:
+					ws.sendFileTransferUpdate()
+				case err := <-watcher.Errors:
+					ws.logger.Err(err).Msg("error in file transfer dir watcher")
+				}
+			}
+		}()
+
+		if err := watcher.Add(ws.conf.FileTransferPath); err != nil {
+			ws.logger.Err(err).Msg("unable to add file transfer path to watcher")
+		}
+	}
 }
 
 func (ws *WebSocketHandler) Shutdown() error {
@@ -354,6 +379,30 @@ func (ws *WebSocketHandler) MakeFilePath(filename string) string {
 	return fmt.Sprintf("%s%s", ws.conf.FileTransferPath, filename)
 }
 
+func (ws *WebSocketHandler) sendFileTransferUpdate() {
+	files, err := utils.ListFiles(ws.conf.FileTransferPath)
+	if err != nil {
+		ws.logger.Err(err).Msg("unable to ls file transfer path")
+		return
+	}
+
+	message := message.FileList{
+		Event: event.FILETRANSFER_LIST,
+		Cwd:   ws.conf.FileTransferPath,
+		Files: *files,
+	}
+
+	var broadcastErr error
+	if ws.conf.UnprivFileTransfer {
+		broadcastErr = ws.sessions.Broadcast(message, nil)
+	} else {
+		broadcastErr = ws.sessions.AdminBroadcast(message, nil)
+	}
+	if broadcastErr != nil {
+		ws.logger.Err(broadcastErr).Msg("unable to broadcast file list")
+	}
+}
+
 func (ws *WebSocketHandler) authenticate(r *http.Request) (bool, error) {
 	passwords, ok := r.URL.Query()["password"]
 	if !ok || len(passwords[0]) < 1 {