update size on deletion of children

frogfreg · Oct 16, 2024 · d10e37a · d10e37a
1 parent 5691d6e
commit d10e37a
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/fileinfo/fileInfo.go b/fileinfo/fileInfo.go
@@ -218,9 +218,13 @@ func GetSortedDirs(m map[string]FileInfo, root string) []FileInfo {
 }
 
 func CleanChildren(m map[string]FileInfo, dir string) {
+	size := m[dir].Size
 	delete(m, dir)
 
 	for path, fi := range m {
+		if slices.Contains(fi.Children, dir) {
+			fi.Size -= size
+		}
 		fi.Children = slices.DeleteFunc(fi.Children, func(item string) bool {
 			return item == dir
 		})

diff --git a/fileinfo/fileInfo_test.go b/fileinfo/fileInfo_test.go
@@ -70,11 +70,10 @@ func TestCleanChildren(t *testing.T) {
 	}
 
 	expected := map[string]FileInfo{
-		"testfiles": {Name: "testfiles", FileType: "dir", Size: 13, Children: []string{"testfiles/file1.txt", "testfiles/file2.txt"}, Checked: true},
+		"testfiles": {Name: "testfiles", FileType: "dir", Size: 3, Children: []string{"testfiles/file1.txt", "testfiles/file2.txt"}, Checked: true},
 	}
 
 	CleanChildren(m, "testfiles/dir1")
-	t.Log(m)
 
 	if !reflect.DeepEqual(expected, m) {
 		t.Errorf("expected %v, got %v", expected, m)

diff --git a/main.go b/main.go
@@ -127,16 +127,16 @@ func (m model) updateCurrentDir(dir string, replace bool) model {
 	return m
 }
 
-func deleteCmd(m map[string]fileinfo.FileInfo, dir string) tea.Cmd {
+func deleteCmd(m map[string]fileinfo.FileInfo, path string) tea.Cmd {
 	f := func() tea.Msg {
 		var res deleteResponse
 
-		err := os.RemoveAll(dir)
+		err := os.RemoveAll(path)
 		if err != nil {
 			res.err = err
 		}
 
-		fileinfo.CleanChildren(m, dir)
+		fileinfo.CleanChildren(m, path)
 
 		return res
 	}