Wonkiness on layout

wdgt_prompt.go

@@ -46,7 +46,7 @@ type Prompt struct {
 	btnOk, btnCancel *Button
 	onOk             func(string) bool
 
-	keyMap KeyMap
+	keyMap, customKeyMap KeyMap
 }
 
 var _ Widget = (*Prompt)(nil)
@@ -68,6 +68,7 @@ func (w *Prompt) Init(id string, style tcell.Style) {
 	w.btnCancel.SetLabel("Cancel")
 	w.focusable = true
 	w.keyMap = BlankKeyMap()
+	w.customKeyMap = BlankKeyMap()
 }
 func (w *Prompt) Id() string { return w.id }
 func (w *Prompt) HandleResize(ev *tcell.EventResize) {
@@ -83,14 +84,20 @@ func (w *Prompt) HandleResize(ev *tcell.EventResize) {
 	w.btnCancel.SetPos(Coord{X: w.x + 1, Y: w.y + w.h - 1})
 }
 
-func (w *Prompt) SetKeyMap(km KeyMap)   { w.keyMap = km }
-func (w *Prompt) AddToKeyMap(km KeyMap) { w.keyMap.Merge(km) }
+func (w *Prompt) SetKeyMap(km KeyMap, def bool) {
+	if def {
+		w.keyMap = km
+	} else {
+		w.customKeyMap = km
+	}
+}
+func (w *Prompt) AddToKeyMap(km KeyMap) { w.customKeyMap.Merge(km) }
 func (w *Prompt) RemoveFromKeyMap(km KeyMap) {
 	for k := range km.Keys {
-		w.keyMap.Remove(k)
+		w.customKeyMap.Remove(k)
 	}
 	for r := range km.Runes {
-		w.keyMap.RemoveRune(r)
+		w.customKeyMap.RemoveRune(r)
 	}
 }
 
@@ -98,7 +105,8 @@ func (w *Prompt) HandleKey(ev *tcell.EventKey) bool {
 	if !w.active {
 		return false
 	}
-	return w.keyMap.Handle(ev)
+	b1, b2 := w.keyMap.Handle(ev), w.customKeyMap.Handle(ev)
+	return b1 || b2
 }
 
 func (w *Prompt) HandleTime(ev *tcell.EventTime) {