From 9220c377656fa88096d4052bc691f783bfd5a6f1 Mon Sep 17 00:00:00 2001
From: Arnas Udovic <zordsdavini@gmail.com>
Date: Sun, 16 Mar 2025 14:51:43 +0200
Subject: [PATCH] fix category count #2

---
 formatter.go | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/formatter.go b/formatter.go
index 3916f85..166de28 100644
--- a/formatter.go
+++ b/formatter.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"fmt"
 	"strings"
 
 	zordTree "g.arns.lt/zordsdavini/zord-tree"
@@ -17,8 +18,8 @@ func FormatTreeSummery(tree zordTree.Tree) Tree {
 }
 
 func readTree(tree zordTree.Tree, ts *Tree, isRoot bool) {
+	fmt.Println(ts.Categories, isRoot)
 	for _, subtree := range tree.Dirs {
-		readTree(subtree, ts, false)
 		category := strings.Replace(subtree.Path, tree.Path, "", 1)
 		category = strings.Replace(category, "/", "", 1)
 		if isRoot {
@@ -26,7 +27,18 @@ func readTree(tree zordTree.Tree, ts *Tree, isRoot bool) {
 				ts.Categories = make(map[string]int32)
 			}
 			ts.Categories[category] = int32(len(subtree.Files))
+		} else {
+			if len(tree.Files) > 0 {
+				for _, subcategory := range tree.Files[0].Category {
+					if val, ok := ts.Categories[subcategory]; ok {
+						fmt.Println(val, ok, subcategory)
+						ts.Categories[subcategory] += int32(len(subtree.Files))
+					}
+				}
+			}
 		}
+
+		readTree(subtree, ts, false)
 	}
 
 	for _, file := range tree.Files {