From 60b17c473a0edbd806c9f0c8e4b3b238cb60227e Mon Sep 17 00:00:00 2001
From: Drew DeVault <sir@cmpwn.com>
Date: Sun, 17 Mar 2019 17:39:22 -0400
Subject: [PATCH] Wrap Terminal in TermHost

---
 commands/term-close.go | 12 +++-------
 commands/term.go       | 13 +++--------
 widgets/aerc.go        |  5 ++++
 widgets/termhost.go    | 52 ++++++++++++++++++++++++++++++++++++++++++
 widgets/terminal.go    |  2 +-
 5 files changed, 64 insertions(+), 20 deletions(-)
 create mode 100644 widgets/termhost.go

diff --git a/commands/term-close.go b/commands/term-close.go
index 38fcc27..30f3392 100644
--- a/commands/term-close.go
+++ b/commands/term-close.go
@@ -3,7 +3,6 @@ package commands
 import (
 	"errors"
 
-	"git.sr.ht/~sircmpwn/aerc2/lib/ui"
 	"git.sr.ht/~sircmpwn/aerc2/widgets"
 )
 
@@ -15,15 +14,10 @@ func TermClose(aerc *widgets.Aerc, args []string) error {
 	if len(args) != 1 {
 		return errors.New("Usage: term-close")
 	}
-	grid, ok := aerc.SelectedTab().(*ui.Grid)
+	thost, ok := aerc.SelectedTab().(*widgets.TermHost)
 	if !ok {
 		return errors.New("Error: not a terminal")
 	}
-	for _, child := range grid.Children() {
-		if term, ok := child.(*widgets.Terminal); ok {
-			term.Close(nil)
-			return nil
-		}
-	}
-	return errors.New("Error: not a terminal")
+	thost.Terminal().Close(nil)
+	return nil
 }
diff --git a/commands/term.go b/commands/term.go
index 976ce29..1981e62 100644
--- a/commands/term.go
+++ b/commands/term.go
@@ -4,7 +4,6 @@ import (
 	"os/exec"
 	"time"
 
-	"git.sr.ht/~sircmpwn/aerc2/lib/ui"
 	"git.sr.ht/~sircmpwn/aerc2/widgets"
 
 	"github.com/gdamore/tcell"
@@ -27,14 +26,8 @@ func Term(aerc *widgets.Aerc, args []string) error {
 	if err != nil {
 		return err
 	}
-	grid := ui.NewGrid().Rows([]ui.GridSpec{
-		{ui.SIZE_WEIGHT, 1},
-	}).Columns([]ui.GridSpec{
-		{ui.SIZE_EXACT, aerc.Config().Ui.SidebarWidth},
-		{ui.SIZE_WEIGHT, 1},
-	})
-	grid.AddChild(term).At(0, 1)
-	tab := aerc.NewTab(grid, args[1])
+	host := widgets.NewTermHost(term, aerc.Config())
+	tab := aerc.NewTab(host, args[1])
 	term.OnTitle = func(title string) {
 		if title == "" {
 			title = args[1]
@@ -43,7 +36,7 @@ func Term(aerc *widgets.Aerc, args []string) error {
 		tab.Content.Invalidate()
 	}
 	term.OnClose = func(err error) {
-		aerc.RemoveTab(grid)
+		aerc.RemoveTab(host)
 		if err != nil {
 			aerc.PushStatus(" "+err.Error(), 10*time.Second).
 				Color(tcell.ColorRed, tcell.ColorWhite)
diff --git a/widgets/aerc.go b/widgets/aerc.go
index 49a61bd..af2d0df 100644
--- a/widgets/aerc.go
+++ b/widgets/aerc.go
@@ -122,6 +122,11 @@ func (aerc *Aerc) Event(event tcell.Event) bool {
 				aerc.BeginExCommand()
 				return true
 			}
+			interactive, ok := aerc.tabs.Tabs[aerc.tabs.Selected].Content.(ui.Interactive)
+			if ok {
+				return interactive.Event(event)
+			}
+			return false
 		}
 	}
 	return false
diff --git a/widgets/termhost.go b/widgets/termhost.go
new file mode 100644
index 0000000..7898b44
--- /dev/null
+++ b/widgets/termhost.go
@@ -0,0 +1,52 @@
+package widgets
+
+import (
+	"github.com/gdamore/tcell"
+
+	"git.sr.ht/~sircmpwn/aerc2/config"
+	"git.sr.ht/~sircmpwn/aerc2/lib/ui"
+)
+
+type TermHost struct {
+	grid *ui.Grid
+	term *Terminal
+}
+
+// Thin wrapper around terminal which puts it in a grid and passes through
+// input events. A bit of a hack tbh
+func NewTermHost(term *Terminal, conf *config.AercConfig) *TermHost {
+	grid := ui.NewGrid().Rows([]ui.GridSpec{
+		{ui.SIZE_WEIGHT, 1},
+	}).Columns([]ui.GridSpec{
+		{ui.SIZE_EXACT, conf.Ui.SidebarWidth},
+		{ui.SIZE_WEIGHT, 1},
+	})
+	grid.AddChild(term).At(0, 1)
+	return &TermHost{grid, term}
+}
+
+func (th *TermHost) Draw(ctx *ui.Context) {
+	th.grid.Draw(ctx)
+}
+
+func (th TermHost) Invalidate() {
+	th.grid.Invalidate()
+}
+
+func (th *TermHost) OnInvalidate(fn func(d ui.Drawable)) {
+	th.grid.OnInvalidate(func(_ ui.Drawable) {
+		fn(th)
+	})
+}
+
+func (th *TermHost) Event(event tcell.Event) bool {
+	return th.term.Event(event)
+}
+
+func (th *TermHost) Focus(focus bool) {
+	th.term.Focus(focus)
+}
+
+func (th *TermHost) Terminal() *Terminal {
+	return th.term
+}
diff --git a/widgets/terminal.go b/widgets/terminal.go
index 7726273..098543f 100644
--- a/widgets/terminal.go
+++ b/widgets/terminal.go
@@ -1,6 +1,7 @@
 package widgets
 
 import (
+	"fmt"
 	gocolor "image/color"
 	"os"
 	"os/exec"
@@ -188,7 +189,6 @@ func (term *Terminal) Focus(focus bool) {
 }
 
 func (term *Terminal) Event(event tcell.Event) bool {
-	// TODO
 	return false
 }