lualine.nvim/lua/lualine/utils/section.lua

-- Copyright (c) 2020-2021 hoob3rt
-- MIT license, see LICENSE for more details.
local M = {}
local utils = require('lualine.utils.utils')
local highlight = require('lualine.highlight')
-- Returns formated string for a section
function M.draw_section(section, section_name, is_focused)
  local highlight_name = highlight.format_highlight(is_focused,
                                                    'lualine_' .. section_name)

  local status = {}
  for _, component in pairs(section) do
    -- load components into status table
    if type(component) ~= 'table' or
        (type(component) == 'table' and not component.component_no) then
      return '' -- unknown element in section. section posibly not yet loaded
    end
    table.insert(status, component:draw(highlight_name, is_focused))
  end

  -- Flags required for knowing when to remove component separator
  local strip_next_component = false
  local last_component_found = false
  local first_component_no = #section

  -- Check through components to see when component separator need to be removed
  for component_no = #section, 1, -1 do
    if #status[component_no] > 0 then first_component_no = component_no end
    -- Remove component separator with highlight for last component
    if not last_component_found and #status[component_no] > 0 then
      last_component_found = true
      status[component_no] = section[component_no]:strip_separator()
      if section_name < 'c' then
        if type(section[first_component_no].options.separator) ~= 'table' and
            section[1].options.section_separators[1] ~= '' then
          status[component_no] = string.format('%s%%S{%s}',
                                               status[component_no], section[1]
                                                   .options.section_separators[1])
        end
      end
    end
    -- Remove component separator when color option is used in next component
    if strip_next_component then
      strip_next_component = false
      status[component_no] = section[component_no]:strip_separator()
    end
    -- Remove component separator when color option is used to color background
    if (type(section[component_no].options.color) == 'table' and
        section[component_no].options.color.bg) or
        type(section[component_no].options.color) == 'string' then
      strip_next_component = true
      status[component_no] = section[component_no]:strip_separator()
    end

    if (section[component_no].strip_previous_separator == true) then
      strip_next_component = true
    end
  end

  local left_sparator_string = ''
  if section_name > 'x' and section[first_component_no] and
      type(section[first_component_no].options.separator) ~= 'table' and
      section[1].options.section_separators[2] ~= '' then
    left_sparator_string = string.format('%%s{%s}',
                           section[first_component_no].options.ls_separator or
                               section[1].options.section_separators[2])
  end

  -- Remove empty strings from status
  status = utils.list_shrink(status)
  local status_str = table.concat(status)

  if #status_str == 0 then return '' end

  local needs_hl

  local find_start_trans_sep_start, find_start_trans_sep_end = status_str:find('^%%s{.-}')
  if find_start_trans_sep_start then
    -- the section doesn't need to be prepended with default hl when sections
    -- first component has trasitionals sep
    needs_hl = status_str:find('^%%#', find_start_trans_sep_end + 1)
  else needs_hl = status_str:find('^%%#') end

  if needs_hl then
    -- Don't prepend with old highlight when the component changes it imidiately
    return left_sparator_string .. status_str
  else
    return left_sparator_string .. highlight_name .. status_str
  end
end

return M