package kk_file

import (
	"os"
	"path/filepath"
)

type FileNode struct {
	FileName string
	FilePath string
	Children []FileNode
}

// DirTree collects all files and directories in a directory.
func DirTree(dir string, result *[]FileNode) error {
	files, err := os.ReadDir(dir)
	if err != nil {
		return err
	}

	for _, file := range files {
		abs, err := filepath.Abs(filepath.Join(dir, file.Name()))
		if err != nil {
			return err
		}
		fileInfo := FileNode{
			FileName: file.Name(),
			FilePath: abs,
		}

		if file.IsDir() {
			subDir := filepath.Join(dir, file.Name())
			subFiles := make([]FileNode, 0)
			if err := DirTree(subDir, &subFiles); err != nil {
				return err
			}
			fileInfo.Children = subFiles
		}

		*result = append(*result, fileInfo)
	}

	return nil
}