dirtree: add custom virtual directory role

Add a new directory role to indicate virtual nodes in the directory
tree. This allows to style the virtual nodes differently and apply
different behaviors in some commands (i.e. rmdir).

Signed-off-by: Koni Marti <koni.marti@gmail.com>
Acked-by: Robin Jarry <robin@jarry.cc>

1 files changed, 24 insertions, 0 deletions

diff --git a/app/dirtree.go b/app/dirtree.go
index 8f8730b4..1e0e76fb 100644
--- a/app/dirtree.go
+++ b/app/dirtree.go
@@ -43,6 +43,30 @@ func (dt *DirectoryTree) OnVirtualNode(cb func()) {
 	dt.virtualCb = cb
 }
 
+func (dt *DirectoryTree) Selected() string {
+	if dt.listIdx < 0 || dt.listIdx >= len(dt.list) {
+		return dt.DirectoryList.Selected()
+	}
+	node := dt.list[dt.listIdx]
+	sep := dt.DirectoryList.worker.PathSeparator()
+	elems := strings.Split(dt.treeDirs[getAnyUid(node)], sep)
+	n := countLevels(node)
+	if n < 0 || n >= len(elems) {
+		return ""
+	}
+	return strings.Join(elems[:(n+1)], sep)
+}
+
+func (dt *DirectoryTree) SelectedDirectory() *models.Directory {
+	if dt.virtual {
+		return &models.Directory{
+			Name: dt.Selected(),
+			Role: models.VirtualRole,
+		}
+	}
+	return dt.DirectoryList.SelectedDirectory()
+}
+
 func (dt *DirectoryTree) ClearList() {
 	dt.list = make([]*types.Thread, 0)
 }