Fix expand

new/main.go

@@ -130,14 +130,13 @@ func (m *model) handleKey(msg tea.KeyMsg) (tea.Model, tea.Cmd) {
 		}
 
 	case key.Matches(msg, keyMap.GotoTop):
-		m.cursor = 0
 		m.head = m.top
+		m.cursor = 0
 
 	case key.Matches(msg, keyMap.GotoBottom):
-		m.cursor = 0
 		m.head = m.findBottom()
+		m.cursor = 0
 		m.scrollIntoView()
-		m.cursor = m.visibleLines() - 1
 
 	case key.Matches(msg, keyMap.NextSibling):
 		pointsTo := m.cursorPointsTo()

new/node.go

@@ -38,7 +38,7 @@ func (n *node) append(child *node) {
 func (n *node) collapseThisOrParent() *node {
 	if n.end == nil || n.isCollapsed() {
 		if n.parent() != nil {
-			return n.parent().collapse()
+			return n.parent().collapseThisOrParent()
 		}
 		return n
 	}
@@ -62,6 +62,7 @@ func (n *node) isCollapsed() bool {
 
 func (n *node) expand() {
 	if n.collapsed != nil {
+		n.next.prev = n.end
 		n.next = n.collapsed
 		n.collapsed = nil
 	}