Change statusline separator color on mode change

[thedeany]

I'm experimenting with the slanted gaps example and I've got the separators generated by the process_sections function to be colored like I want in normal mode, but I'd like the separators to change color when the mode changes. I have the following code which grabs the correct color from my theme per mode like so:

local mode_colors = require "lualine.themes.ayu"

-- Put proper separators and gaps between components in sections
local function process_sections(sections)
  for name, section in pairs(sections) do
    local left = name:sub(9, 10) < "x"
    for pos = 1, name ~= "lualine_z" and #section or #section - 2 do
      print "attempting mode change refresh"
      local mode = string.lower(require("lualine.utils.mode").get_mode())
      local bg = mode_colors[mode].a.bg
      table.insert(section, pos * 2, { empty, color = { fg = colors.white, bg = bg } })
    end
    for id, comp in ipairs(section) do
      if type(comp) ~= "table" then
        comp = { comp }
        section[id] = comp
      end
      comp.separator = left and { right = "" } or { left = "" }
    end
  end
  return sections
end

but this function seemingly never gets called again after the initial render. I also added an autocmd to try to force a refresh, but that doesn't seem to help either.

-- Add autocommand for ModeChanged to trigger redrawstatus
vim.api.nvim_create_autocmd("ModeChanged", {
  callback = function(c)
    require("lualine").refresh { scope = "all" }
  end,
})

Is there something obvious I'm missing? I'd appreciate any help! My full lualine.lua plugin file is like so:

return {
  "nvim-lualine/lualine.nvim",
  event = "VeryLazy",
  dependencies = {
    "nvim-tree/nvim-web-devicons",
  },
  config = function()
    local colors = require "ayu.colors"
    colors.generate(true) -- Pass `true` to enable mirage
    local mode_colors = require "lualine.themes.ayu"

    -- Add autocommand for ModeChanged to trigger redrawstatus
    vim.api.nvim_create_autocmd("ModeChanged", {
      callback = function(c)
        require("lualine").refresh { scope = "all" }
      end,
    })

    local empty = require("lualine.component"):extend()
    function empty:draw(default_highlight)
      self.status = ""
      self.applied_separator = ""
      self:apply_highlights(default_highlight)
      self:apply_section_separators()
      return self.status
    end

    -- Put proper separators and gaps between components in sections
    local function process_sections(sections)
      for name, section in pairs(sections) do
        local left = name:sub(9, 10) < "x"
        for pos = 1, name ~= "lualine_z" and #section or #section - 2 do
          print "attempting mode change refresh"
          local mode = string.lower(require("lualine.utils.mode").get_mode())
          local bg = mode_colors[mode].a.bg
          table.insert(section, pos * 2, { empty, color = { fg = colors.white, bg = bg } })
        end
        for id, comp in ipairs(section) do
          if type(comp) ~= "table" then
            comp = { comp }
            section[id] = comp
          end
          comp.separator = left and { right = "" } or { left = "" }
        end
      end
      return sections
    end

    local function search_result()
      if vim.v.hlsearch == 0 then return "" end
      local last_search = vim.fn.getreg "/"
      if not last_search or last_search == "" then return "" end
      local searchcount = vim.fn.searchcount { maxcount = 9999 }
      return last_search .. "(" .. searchcount.current .. "/" .. searchcount.total .. ")"
    end

    local function modified()
      if vim.bo.modified then
        return "+"
      elseif vim.bo.modifiable == false or vim.bo.readonly == true then
        return "-"
      end
      return ""
    end
    require("lualine").setup {

      options = {
        always_divide_middle = false,
        component_separators = "",
        icons_enabled = true,
        section_separators = { left = "", right = "" },
        theme = "ayu",
        globalstatus = true,
        use_mode_colors = true,
        refresh = {
          statusline = 100,
        },
      },
      sections = process_sections {
        lualine_a = { "mode" },
        lualine_b = {
          "branch",
          "diff",
          {
            "diagnostics",
            source = { "nvim" },
            sections = { "error" },
            diagnostics_color = { error = { bg = colors.red, fg = colors.white } },
          },
          {
            "diagnostics",
            source = { "nvim" },
            sections = { "warn" },
            diagnostics_color = { warn = { bg = colors.orange, fg = colors.white } },
          },
          { "filename", file_status = false, path = 1 },
          { modified, color = { bg = colors.red } },
          {
            "%w",
            cond = function() return vim.wo.previewwindow end,
          },
          {
            "%r",
            cond = function() return vim.bo.readonly end,
          },
          {
            "%q",
            cond = function() return vim.bo.buftype == "quickfix" end,
          },
        },
        lualine_c = {},
        lualine_x = {},
        lualine_y = { search_result, "filetype" },
        lualine_z = { "%l:%c", "%p%%/%L" },
      },
      inactive_sections = {
        lualine_c = { "%f %y %m" },
        lualine_x = {},
      },
      tabline = {
        lualine_a = {
          {
            "buffers",
            mode = 4,
            buffers_color = {
              active = { bg = colors.keyword, gui = "" },
              inactive = { fg = colors.fg, bg = colors.panel_border, gui = "" },
            },
          },
        },
      },
    }
  end,
}