ディレクトリ数が合わないパターンがあったため修正

owlinux1000 · Sep 30, 2023 · c427104 · c427104
1 parent ca2b12e
commit c427104
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/internal/counter.go b/internal/counter.go
@@ -3,6 +3,7 @@ package internal
 import (
 	"fmt"
 	"path/filepath"
+	"strings"
 )
 
 type counter struct {
@@ -19,8 +20,11 @@ func newCounter() *counter {
 
 func (c *counter) count(path string) {
 	dir, file := filepath.Split(path)
+	splited := strings.Split(dir, "/")
 	if len(dir) > 0 {
-		c.dirs[dir] = struct{}{}
+		for i := range splited {
+			c.dirs[filepath.Join(splited[:i+1]...)] = struct{}{}
+		}
 	}
 	if len(file) > 0 {
 		c.files[path] = struct{}{}