From e655ccfd404561db81aef44725bc9a33c818b251 Mon Sep 17 00:00:00 2001
From: Arnas Udovic <zordsdavini@gmail.com>
Date: Sun, 16 Mar 2025 14:24:27 +0200
Subject: [PATCH] fix category count

---
 formatter.go | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/formatter.go b/formatter.go
index 2f42ce2..3916f85 100644
--- a/formatter.go
+++ b/formatter.go
@@ -1,8 +1,9 @@
 package main
 
 import (
-	zordTree "g.arns.lt/zordsdavini/zord-tree"
 	"strings"
+
+	zordTree "g.arns.lt/zordsdavini/zord-tree"
 )
 
 func FormatTreeSummery(tree zordTree.Tree) Tree {
@@ -24,11 +25,7 @@ func readTree(tree zordTree.Tree, ts *Tree, isRoot bool) {
 			if ts.Categories == nil {
 				ts.Categories = make(map[string]int32)
 			}
-			if _, ok := ts.Categories[category]; ok {
-				ts.Categories[category]++
-			} else {
-				ts.Categories[category] = 1
-			}
+			ts.Categories[category] = int32(len(subtree.Files))
 		}
 	}