From 3b33e803e30528f1e9b68d5dc7581ece72edaa4e Mon Sep 17 00:00:00 2001
From: Dane Everitt <dane@daneeveritt.com>
Date: Sun, 23 Aug 2020 14:34:42 -0700
Subject: [PATCH] Handle deleting a directory

---
 server/filesystem.go | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/server/filesystem.go b/server/filesystem.go
index 6eddf96..0b36cd9 100644
--- a/server/filesystem.go
+++ b/server/filesystem.go
@@ -621,12 +621,15 @@ func (fs *Filesystem) Delete(p string) error {
 			fs.Server.Log().WithField("error", err).WithField("path", resolved).Warn("error while attempting to stat file before deletion")
 		}
 	} else {
-		// TODO: handle deleting a directory by finding its current size.
-		go func (st os.FileInfo) {
-			if !st.IsDir() {
-				atomic.SwapInt64(&fs.diskUsage, fs.diskUsage - st.Size())
-			}
-		}(st)
+		if !st.IsDir() {
+			atomic.SwapInt64(&fs.diskUsage, -st.Size())
+		} else {
+			go func(st os.FileInfo, resolved string) {
+				if s, err := fs.DirectorySize(resolved); err == nil {
+					atomic.AddInt64(&fs.diskUsage, -s)
+				}
+			}(st, resolved)
+		}
 	}
 
 	return os.RemoveAll(resolved)