From 2f8371588c289cf89b1269012e2454d186e8e240 Mon Sep 17 00:00:00 2001
From: Iordanis Petkakis <12776461+dpetka2001@users.noreply.github.com>
Date: Tue, 1 Oct 2024 23:07:03 +0300
Subject: [PATCH] feat(toggle): add more `M.wk` fields for better user
 customization

---
 lua/lazyvim/util/toggle.lua | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/lua/lazyvim/util/toggle.lua b/lua/lazyvim/util/toggle.lua
index 46e92c78..d693dd9d 100644
--- a/lua/lazyvim/util/toggle.lua
+++ b/lua/lazyvim/util/toggle.lua
@@ -3,6 +3,9 @@ local M = {}
 
 ---@class lazyvim.Toggle
 ---@field name string
+---@field icon? string
+---@field color_enabled? string
+---@field color_disabled? string
 ---@field get fun():boolean
 ---@field set fun(state:boolean)
 
@@ -50,7 +53,8 @@ function M.wk(lhs, toggle)
     {
       lhs,
       icon = function()
-        return safe_get() and { icon = " ", color = "green" } or { icon = " ", color = "yellow" }
+        return safe_get() and { icon = toggle.icon or " ", color = toggle.color_enabled or "green" }
+          or { icon = toggle.icon or " ", color = toggle.color_disabled or "yellow" }
       end,
       desc = function()
         return (safe_get() and "Disable " or "Enable ") .. toggle.name