nvim/lua/plugins/fold.lua

local M = {
  "bbjornstad/pretty-fold.nvim",
}

M.config = function()
  local global_setup = {
    sections = {
      left = { "content" },
      right = {
        " ",
        function()
          return ("[%dL]"):format(vim.v.foldend - vim.v.foldstart)
        end,
        "[",
        "percentage",
        "]",
      },
    },
    matchup_patterns = {
      { "{", "}" },
      { "%(", ")" },
      { "%[", "]" },
    },
    process_comment_signs = ({ "delete", "spaces", false })[2],
  }

  local function ft_setup(lang, options) -- {{{
    local opts = vim.tbl_deep_extend("force", global_setup, options)
    if opts and opts.matchup_patterns and global_setup.matchup_patterns then
      opts.matchup_patterns = vim.list_extend(opts.matchup_patterns, global_setup.matchup_patterns)
    end
    require("pretty-fold").ft_setup(lang, opts)
  end -- }}}

  require("pretty-fold").setup(global_setup)

  ft_setup("lua", {
    matchup_patterns = {
      { "^%s*do$", "end" },
      { "^%s*if", "end" },
      { "^%s*for", "end" },
      { "function[^%(]*%(", "end" },
    },
  })

  ft_setup("vim", {
    matchup_patterns = {
      { "^%s*function!?[^%(]*%(", "endfunction" },
    },
  })
end

return M