local M = {}

M.filetypes = {
    'dap-repl',
    'dapui_console',
    '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