treeprinter: support indented lines under nodes

This change adds the vertical connector when a multi-line node leaves
spaces for it.

Author: RaduBerinde

internal/treeprinter/tree_printer.go

@@ -257,7 +257,10 @@ func (n Node) AddLine(text string) {
 		text = "  " + text
 	}
 	rowIdx := t.addRow(n.level-1, text)
-	if t.style != BulletStyle {
+	// We have to start the line for the first child later, unless we are using
+	// the bullet style or the text starts with spaces.
+	if (t.style == CompactStyle && !strings.HasPrefix(text, " ")) ||
+		(t.style == DefaultStyle && !strings.HasPrefix(text, "   ")) {
 		t.stack[n.level-1].firstChildConnectRow = rowIdx
 	}
 }

internal/treeprinter/tree_printer_test.go

@@ -25,9 +25,12 @@ func TestTreePrinter(t *testing.T) {
 		n13.AddEmptyLine()
 		n131 := n13.Child("1.3.1")
 		n131.AddLine("1.3.1a")
-		n13.Child("1.3.2\n1.3.2a")
+		n132 := n13.Child("1.3.2\n1.3.2a")
 		n13.AddEmptyLine()
-		n131.Child("1.3.1.1\n1.3.1.1a")
+		n132.Child("1.3.2.1\n1.3.2.1a")
+		n133 := n13.Child("1.3.3\n   some info\n   more info")
+		n133.AddEmptyLine()
+		n133.Child("1.3.3.1")
 		n1.Child("1.4")
 		r.Child("2")
 	}
@@ -51,11 +54,16 @@ root
  │    │    │
  │    │    ├── 1.3.1
  │    │    │   1.3.1a
- │    │    └── 1.3.2
- │    │        1.3.2a
+ │    │    ├── 1.3.2
+ │    │    │   1.3.2a
+ │    │    │    │
+ │    │    │    └── 1.3.2.1
+ │    │    │        1.3.2.1a
+ │    │    └── 1.3.3
+ │    │         │ some info
+ │    │         │ more info
  │    │         │
- │    │         └── 1.3.1.1
- │    │             1.3.1.1a
+ │    │         └── 1.3.3.1
  │    └── 1.4
  └── 2
 `
@@ -82,11 +90,16 @@ root
 │ │ │
 │ │ ├ 1.3.1
 │ │ │ 1.3.1a
-│ │ └ 1.3.2
-│ │   1.3.2a
+│ │ ├ 1.3.2
+│ │ │ 1.3.2a
+│ │ │ │
+│ │ │ └ 1.3.2.1
+│ │ │   1.3.2.1a
+│ │ └ 1.3.3
+│ │   │  some info
+│ │   │  more info
 │ │   │
-│ │   └ 1.3.1.1
-│ │     1.3.1.1a
+│ │   └ 1.3.3.1
 │ └ 1.4
 └ 2
 `
@@ -121,12 +134,19 @@ root
 │   │   ├── • 1.3.1
 │   │   │     1.3.1a
 │   │   │
-│   │   └── • 1.3.2
-│   │       │ 1.3.2a
+│   │   ├── • 1.3.2
+│   │   │   │ 1.3.2a
+│   │   │   │
+│   │   │   │
+│   │   │   └── • 1.3.2.1
+│   │   │         1.3.2.1a
+│   │   │
+│   │   └── • 1.3.3
+│   │       │    some info
+│   │       │    more info
 │   │       │
 │   │       │
-│   │       └── • 1.3.1.1
-│   │             1.3.1.1a
+│   │       └── • 1.3.3.1
 │   │
 │   └── • 1.4
 │