Add a <root> item if files at top level have changed

pkg/gui/filetree/build_tree.go

@@ -14,7 +14,7 @@ func BuildTreeFromFiles(files []*models.File) *Node[models.File] {
 
 	var curr *Node[models.File]
 	for _, file := range files {
-		splitPath := split(file.Path)
+		splitPath := split("./" + file.Path)
 		curr = root
 	outer:
 		for i := range splitPath {
@@ -70,7 +70,7 @@ func BuildTreeFromCommitFiles(files []*models.CommitFile) *Node[models.CommitFil
 
 	var curr *Node[models.CommitFile]
 	for _, file := range files {
-		splitPath := split(file.Path)
+		splitPath := split("./" + file.Path)
 		curr = root
 	outer:
 		for i := range splitPath {

pkg/gui/filetree/node.go

@@ -2,6 +2,7 @@ package filetree
 
 import (
 	"path"
+	"strings"
 
 	"github.com/jesseduffield/lazygit/pkg/commands/models"
 	"github.com/jesseduffield/lazygit/pkg/gui/types"
@@ -47,6 +48,10 @@ func (self *Node[T]) GetFile() *T {
 }
 
 func (self *Node[T]) GetPath() string {
+	return strings.TrimPrefix(self.path, "./")
+}
+
+func (self *Node[T]) GetFullPath() string {
 	return self.path
 }
 

pkg/gui/presentation/files.go

@@ -293,7 +293,13 @@ func getColorForChangeStatus(changeStatus string) style.TextStyle {
 }
 
 func fileNameAtDepth(node *filetree.Node[models.File], depth int) string {
-	splitName := split(node.GetPath())
+	splitName := split(node.GetFullPath())
+	if depth == 0 {
+		if len(splitName) == 1 {
+			return "<root>"
+		}
+		depth = 1
+	}
 	name := join(splitName[depth:])
 
 	if node.File != nil && node.File.IsRename() {
@@ -314,7 +320,13 @@ func fileNameAtDepth(node *filetree.Node[models.File], depth int) string {
 }
 
 func commitFileNameAtDepth(node *filetree.Node[models.CommitFile], depth int) string {
-	splitName := split(node.GetPath())
+	splitName := split(node.GetFullPath())
+	if depth == 0 {
+		if len(splitName) == 1 {
+			return "<root>"
+		}
+		depth = 1
+	}
 	name := join(splitName[depth:])
 
 	return name