elements/checkbox.go

@@ -126,6 +126,10 @@ func (element *Checkbox) SetConfig (new tomo.Config) {
 	element.entity.Invalidate()
 }
 
+func (element *Checkbox) HandleFocusChange () {
+	element.entity.Invalidate()
+}
+
 func (element *Checkbox) HandleMouseDown (x, y int, button input.Button) {
 	if !element.Enabled() { return }
 	element.Focus()

elements/slider.go

@@ -94,6 +94,10 @@ func (element *Slider) SetEnabled (enabled bool) {
 	element.entity.Invalidate()
 }
 
+func (element *Slider) HandleFocusChange () {
+	element.entity.Invalidate()
+}
+
 func (element *Slider) HandleMouseDown (x, y int, button input.Button) {
 	if !element.Enabled() { return }
 	element.Focus()

elements/switch.go

@@ -97,6 +97,10 @@ func (element *Switch) Draw (destination canvas.Canvas) {
 	element.drawer.Draw(destination, foreground, offset)
 }
 
+func (element *Switch) HandleFocusChange () {
+	element.entity.Invalidate()
+}
+
 func (element *Switch) HandleMouseDown (x, y int, button input.Button) {
 	if !element.Enabled() { return }
 	element.Focus()

elements/textbox.go

@@ -134,6 +134,10 @@ func (element *TextBox) Draw (destination canvas.Canvas) {
 	}
 }
 
+func (element *TextBox) HandleFocusChange () {
+	element.entity.Invalidate()
+}
+
 func (element *TextBox) HandleMouseDown (x, y int, button input.Button) {
 	if !element.Enabled() { return }
 	element.Focus()