From a34be9eb36d26afdd723146673694db70662429c Mon Sep 17 00:00:00 2001
From: Koni Marti <koni.marti@gmail.com>
Date: Sun, 17 Apr 2022 01:03:50 +0200
Subject: [PATCH] status: use contextual ui styleset for statusline

Use contextual ui styleset for statusline widget.

Signed-off-by: Koni Marti <koni.marti@gmail.com>
Acked-by: Robin Jarry <robin@jarry.cc>
---
 widgets/status.go | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/widgets/status.go b/widgets/status.go
index 6c501b2..9944e0d 100644
--- a/widgets/status.go
+++ b/widgets/status.go
@@ -16,7 +16,6 @@ type StatusLine struct {
 	stack    []*StatusMessage
 	fallback StatusMessage
 	aerc     *Aerc
-	uiConfig config.UIConfig
 }
 
 type StatusMessage struct {
@@ -30,7 +29,6 @@ func NewStatusLine(uiConfig config.UIConfig) *StatusLine {
 			style:   uiConfig.GetStyle(config.STYLE_STATUSLINE_DEFAULT),
 			message: "Idle",
 		},
-		uiConfig: uiConfig,
 	}
 }
 
@@ -56,7 +54,7 @@ func (status *StatusLine) Draw(ctx *ui.Context) {
 
 func (status *StatusLine) Set(text string) *StatusMessage {
 	status.fallback = StatusMessage{
-		style:   status.uiConfig.GetStyle(config.STYLE_STATUSLINE_DEFAULT),
+		style:   status.uiConfig().GetStyle(config.STYLE_STATUSLINE_DEFAULT),
 		message: text,
 	}
 	status.Invalidate()
@@ -65,7 +63,7 @@ func (status *StatusLine) Set(text string) *StatusMessage {
 
 func (status *StatusLine) SetError(text string) *StatusMessage {
 	status.fallback = StatusMessage{
-		style:   status.uiConfig.GetStyle(config.STYLE_STATUSLINE_ERROR),
+		style:   status.uiConfig().GetStyle(config.STYLE_STATUSLINE_ERROR),
 		message: text,
 	}
 	status.Invalidate()
@@ -74,7 +72,7 @@ func (status *StatusLine) SetError(text string) *StatusMessage {
 
 func (status *StatusLine) Push(text string, expiry time.Duration) *StatusMessage {
 	msg := &StatusMessage{
-		style:   status.uiConfig.GetStyle(config.STYLE_STATUSLINE_DEFAULT),
+		style:   status.uiConfig().GetStyle(config.STYLE_STATUSLINE_DEFAULT),
 		message: text,
 	}
 	status.stack = append(status.stack, msg)
@@ -96,13 +94,13 @@ func (status *StatusLine) Push(text string, expiry time.Duration) *StatusMessage
 
 func (status *StatusLine) PushError(text string) *StatusMessage {
 	msg := status.Push(text, 10*time.Second)
-	msg.Color(status.uiConfig.GetStyle(config.STYLE_STATUSLINE_ERROR))
+	msg.Color(status.uiConfig().GetStyle(config.STYLE_STATUSLINE_ERROR))
 	return msg
 }
 
 func (status *StatusLine) PushSuccess(text string) *StatusMessage {
 	msg := status.Push(text, 10*time.Second)
-	msg.Color(status.uiConfig.GetStyle(config.STYLE_STATUSLINE_SUCCESS))
+	msg.Color(status.uiConfig().GetStyle(config.STYLE_STATUSLINE_SUCCESS))
 	return msg
 }
 
@@ -110,6 +108,10 @@ func (status *StatusLine) Expire() {
 	status.stack = nil
 }
 
+func (status *StatusLine) uiConfig() config.UIConfig {
+	return status.aerc.SelectedAccountUiConfig()
+}
+
 func (status *StatusLine) SetAerc(aerc *Aerc) {
 	status.aerc = aerc
 }