From 389276ee3461414cbb5768bc15ebca5ea8472455 Mon Sep 17 00:00:00 2001
From: "Luke I. Wilson" <thelukaswils@gmail.com>
Date: Tue, 6 Apr 2021 11:14:17 -0500
Subject: [PATCH] PanelContainer: Bug fixes and integration

---
 main.go              | 15 ++++++++++++++-
 ui/panelcontainer.go | 46 +++++++++++++++++++++++++++++++++++---------
 2 files changed, 51 insertions(+), 10 deletions(-)

diff --git a/main.go b/main.go
index 62be962..f5f855c 100644
--- a/main.go
+++ b/main.go
@@ -199,6 +199,10 @@ func main() {
 	fileMenu.AddItems([]ui.Item{&ui.ItemEntry{Name: "New File", Shortcut: "Ctrl+N", Callback: func() {
 		textEdit := ui.NewTextEdit(screen, "", []byte{}, &theme) // No file path, no contents
 		tabContainer := getActiveTabContainer()
+		if tabContainer == nil {
+			tabContainer = ui.NewTabContainer(&theme)
+			panelContainer.SetSelected(tabContainer)
+		}
 		tabContainer.AddTab("noname", textEdit)
 		tabContainer.FocusTab(tabContainer.GetTabCount() - 1)
 		changeFocus(panelContainer)
@@ -224,6 +228,10 @@ func main() {
 				}
 
 				textEdit := ui.NewTextEdit(screen, path, bytes, &theme)
+				if tabContainer == nil {
+					tabContainer = ui.NewTabContainer(&theme)
+					panelContainer.SetSelected(tabContainer)
+				}
 				tabContainer.AddTab(path, textEdit)
 			}
 
@@ -276,7 +284,12 @@ func main() {
 			if tabContainer != nil && tabContainer.GetTabCount() > 0 {
 				tabContainer.RemoveTab(tabContainer.GetSelectedTabIdx())
 			} else { // No tabs open; close the editor
-				closing = true
+				// if the selected is root: close editor. otherwise close panel
+				if panelContainer.IsRootSelected() {
+					closing = true
+				} else {
+					panelContainer.DeleteSelected()
+				}
 			}
 		}}})
 
diff --git a/ui/panelcontainer.go b/ui/panelcontainer.go
index 077361d..a22c0a0 100755
--- a/ui/panelcontainer.go
+++ b/ui/panelcontainer.go
@@ -35,7 +35,9 @@ func (c *PanelContainer) ClearSelected() Component {
 	item := (**c.selected).Left
 	(**c.selected).Left = nil
 	(**c.selected).Kind = PanelKindEmpty
-	(*c.selected).UpdateSplits()
+	if p := (**c.selected).Parent; p != nil {
+		p.UpdateSplits()
+	}
 	return item
 }
 
@@ -52,6 +54,13 @@ func (c *PanelContainer) DeleteSelected() Component {
 	} else {
 		item := (**c.selected).Left
 		p := (**c.selected).Parent
+
+		if c.focused {
+			(*c.selected).SetFocused(false) // Unfocus item
+		}
+
+		// we're shifting panel right to left,
+		// need to focus left
 		if p != nil {
 			if *c.selected == (*p).Left { // If we're deleting the parent's Left
 				(*p).Left = (*p).Right
@@ -59,16 +68,28 @@ func (c *PanelContainer) DeleteSelected() Component {
 			} else { // Deleting parent's Right
 				(*p).Right = nil
 			}
-			(*p).Kind = PanelKindSingle
 
-			if c.focused {
-				(*c.selected).SetFocused(false) // Unfocus item
+			if (*p).Left != nil { // Left == panel  ; SHOULD NOT BE PANEL
+				// asserting left is panel:
+				// if left is not a Leaf !.IsLeaf():
+				//   make the parent match the left:
+				//   p.Left = panel's Left
+				//   p.Right = panel's Right
+				//   p.Kind = panel's kind
+				// else:
+				//   parent left = panel's left
+				//   parent's kind = panel's kind
+				panel := (*p).Left.(*Panel)
+
+				(*p).Left = (*panel).Left
+				(*p).Right = (*panel).Right
+				(*p).Kind = (*panel).Kind
+			} else {
+				(*p).Kind = PanelKindEmpty
 			}
 			c.selected = &p
+			(*p).UpdateSplits()
 		} else if c.floatingMode { // Deleting a floating Panel without a parent
-			if c.focused {
-				c.floating[0].SetFocused(false) // Unfocus Panel and item
-			}
 			c.floating[0] = nil
 			copy(c.floating, c.floating[1:]) // Shift items to front
 			c.floating = c.floating[:len(c.floating)-1] // Shrink slice's len by one
@@ -81,7 +102,7 @@ func (c *PanelContainer) DeleteSelected() Component {
 		} else {
 			panic("Panel does not have parent and is not floating")
 		}
-		(*c.selected).UpdateSplits()
+		
 		if c.focused {
 			(*c.selected).SetFocused(c.focused)
 		}
@@ -106,7 +127,7 @@ func (c *PanelContainer) SwapNeighborsSelected() {
 // Turns the selected Panel into a split panel, moving its contents to its Left field,
 // and putting the given Panel at the Right field. `panel` cannot be nil.
 func (c *PanelContainer) splitSelectedWithPanel(kind SplitKind, panel *Panel) {
-	(**c.selected).Left = &Panel{Parent: *c.selected, Left: (**c.selected).Left, Kind: PanelKindSingle}
+	(**c.selected).Left = &Panel{Parent: *c.selected, Left: (**c.selected).Left, Kind: (**c.selected).Kind}
 	(**c.selected).Right = panel
 	(**c.selected).Right.(*Panel).Parent = *c.selected
 
@@ -145,7 +166,14 @@ func (c *PanelContainer) SplitSelected(kind SplitKind, item Component) {
 	}
 }
 
+func (c *PanelContainer) IsRootSelected() bool {
+	return *c.selected == c.root
+}
+
 func (c *PanelContainer) GetSelected() Component {
+	if !(*c.selected).IsLeaf() {
+		panic("selected is not leaf")
+	}
 	return (**c.selected).Left
 }