better architecture for file system

cmd/registry.go

@@ -20,14 +20,15 @@ var registryCmd = &cobra.Command{
 	Long: `Distributes tokens from a file or a direcotry
 	into a HashMap and prints the Map`,
 	Run: func(cmd *cobra.Command, args []string) {
-		tokens := splitter.SplitFile(args[0])
+		entrys := splitter.SplitFile(args[0])
 		tr := registry.TokenRegistry{ Map: make(map[string]int64)}
-		tr.DistributeTokens(tokens)
+
+		tr.DistributeTokens(entrys.CreateMatrix())
 
 		tw := new(tabwriter.Writer)
 		tw.Init(os.Stdout, 0, 8, 0, '\t', 0)
 		for v, k := range tr.Map {
-			fmt.Fprintf(tw, "value: \"%s\" \t key: %b \n", v, k)
+			fmt.Fprintf(tw, "value: \"%s\" \t key: %d \n", v, k)
 		}
 		tw.Flush()
 	},

internal/splitter/objects.go

@@ -0,0 +1,20 @@
+package splitter
+
+type RegistryEntry struct {
+	Name string
+	ChildEntrys []RegistryEntry
+	Tokens []string
+}
+
+func (re RegistryEntry) CreateMatrix() [][]string {
+	if re.ChildEntrys != nil {
+		mat := [][]string{}
+		for _, child := range re.ChildEntrys {
+			mat = addToMatrix(mat, child.CreateMatrix())
+		}
+		return mat
+	} else {
+		return [][]string{re.Tokens}
+	}
+}
+

internal/splitter/splitter.go

@@ -6,10 +6,7 @@ import (
 	"strings"
 )
 
-func SplitFile(path string) [][]string {
-	if path == "" {
-		return make([][]string, 0)
-	}
+func SplitFile(path string) RegistryEntry {
 	info, err := os.Stat(path)
 	if err != nil {
 		panic(err)
@@ -17,18 +14,26 @@ func SplitFile(path string) [][]string {
 
 	if info.IsDir() {
 		paths := getFilePathsFromDir(path)
-		matrix := make([][]string, len(paths))
-		for _, filePath := range paths {
-			matrix = addToMatrix(matrix, SplitFile(filePath))
-		}
-		return matrix
-	} else {
-		matrix := make([][]string, 1)
-		matrix[0] = tokenize(getFileContent(path))
-		return matrix
-	}
-}
+		arr := make([]RegistryEntry, len(paths))
 
+		for _, filePath := range paths {
+			if filePath != "" {
+				arr = append(arr, SplitFile(filePath))
+			}
+		}
+		return RegistryEntry{
+			Name: info.Name(), 
+			Tokens: nil,
+			ChildEntrys: arr,
+		}
+	} else {
+		return RegistryEntry{
+			Name: info.Name(),
+			Tokens: tokenize(getFileContent(path)),
+			ChildEntrys: nil,
+		}
+	}
+}
 
 func getFileContent(path string) string{
 	data, err := os.ReadFile(path)