WIP

start moving to new interface

WIP

WIP

WIP

WIP

WIP

1 files changed, 47 insertions, 0 deletions

diff --git a/pkg/gui/status_tree.go b/pkg/gui/status_tree.go
new file mode 100644
index 000000000..758f308a7
--- /dev/null
+++ b/pkg/gui/status_tree.go
@@ -0,0 +1,47 @@
+package gui
+
+import (
+	"os"
+	"sort"
+	"strings"
+
+	"github.com/jesseduffield/lazygit/pkg/commands/models"
+)
+
+func GetTreeFromStatusFiles(files []*models.File) *models.StatusLineNode {
+	root := &models.StatusLineNode{}
+
+	sort.SliceStable(files, func(i, j int) bool {
+		return files[i].Name < files[j].Name
+	})
+
+	var curr *models.StatusLineNode
+	for _, file := range files {
+		split := strings.Split(file.Name, string(os.PathSeparator))
+		curr = root
+	outer:
+		for i, dir := range split {
+			var setFile *models.File
+			if i == len(split)-1 {
+				setFile = file
+			}
+			for _, existingChild := range curr.Children {
+				if existingChild.Name == dir {
+					curr = existingChild
+					continue outer
+				}
+			}
+			newChild := &models.StatusLineNode{
+				Name: dir,
+				File: setFile,
+			}
+			curr.Children = append(curr.Children, newChild)
+
+			curr = newChild
+		}
+	}
+
+	root.SortTree()
+
+	return root
+}