Make DapUI extension work with the control bar

lua/lualine/extensions/nvim-dap-ui.lua

@@ -1,18 +1,58 @@
--- MIT license, see LICENSE for more details.
--- Extension for nvim-dap-ui
 local M = {}
 
-M.sections = {
-  lualine_a = { { 'filename', file_status = false } },
+M.filetypes = {
+    "dap-repl",
+    "dapui_console",
+    "dapui_console",
+    "dapui_watches",
+    "dapui_stacks",
+    "dapui_breakpoints",
+    "dapui_scopes",
 }
 
-M.filetypes = {
-  'dap-repl',
-  'dapui_console',
-  'dapui_watches',
-  'dapui_stacks',
-  'dapui_breakpoints',
-  'dapui_scopes',
+local function merge_colors(foreground, background)
+    local new_name = foreground .. background
+
+    local hl_fg = vim.api.nvim_get_hl(0, { name = foreground })
+    local hl_bg = vim.api.nvim_get_hl(0, { name = background })
+
+    local fg = string.format("#%06x", hl_fg.fg and hl_fg.fg or 0)
+    local bg = string.format("#%06x", hl_bg.bg and hl_bg.bg or 0)
+
+    vim.api.nvim_set_hl(0, new_name, { fg = fg, bg = bg })
+    return new_name
+end
+
+local function get_dap_repl_winbar(active)
+    local get_mode = require("lualine.highlight").get_mode_suffix
+
+    return function()
+        local filetype = vim.bo.filetype
+        local disabled_filetypes = { "dap-repl" }
+
+        if not vim.tbl_contains(disabled_filetypes, filetype) then
+            return ""
+        end
+
+        local background_color = string.format("lualine_b" .. "%s", active and get_mode() or "_inactive")
+
+        local controls_string = "%#" .. background_color .. "#"
+        for element in require("dapui.controls").controls():gmatch("%S+") do
+            local color, action = string.match(element, "%%#(.*)#(%%.*)%%#0#")
+            controls_string = controls_string .. " %#" .. merge_colors(color, background_color) .. "#" .. action
+        end
+        return controls_string
+    end
+end
+
+M.winbar = {
+    lualine_a = { { "filename", file_status = false } },
+    lualine_b = { get_dap_repl_winbar(true) },
+}
+
+M.inactive_winbar = {
+    lualine_a = { { "filename", file_status = false } },
+    lualine_b = { get_dap_repl_winbar(false) },
 }
 
 return M