pojokcodeid.nvim-lazy/lua/custom/plugins/nvimufo.lua

function TABLE_CONTAINS(tbl, x)
  local found = false
  for _, v in pairs(tbl) do
    if v == x then
      found = true
    end
  end
  return found
end

local icons = pcode.icons.folding

return {
  -- "kevinhwang91/nvim-ufo",
  -- dependencies = {
  --   "kevinhwang91/promise-async",
  --   {
  --     "luukvbaal/statuscol.nvim",
  --     config = function()
  --       local builtin = require "statuscol.builtin"
  --       require("statuscol").setup {
  --         -- relculright = true,
  --         -- segments = {
  --         -- 	{ text = { builtin.foldfunc }, click = "v:lua.ScFa" },
  --         -- 	{ text = { "%s" }, click = "v:lua.ScSa" },
  --         -- 	{ text = { builtin.lnumfunc, " " }, click = "v:lua.ScLa" },
  --         -- },
  --         setopt = true,
  --         relculright = true,
  --         segments = {
  --           { text = { "%s" }, click = "v:lua.ScSa" },
  --           {
  --             -- text = { builtin.foldfunc, " " },
  --             text = { builtin.foldfunc },
  --             condition = { builtin.not_empty, true, builtin.not_empty },
  --             click = "v:lua.ScFa",
  --           },
  --           { text = { builtin.lnumfunc, " " }, click = "v:lua.ScLa" },
  --           -- { text = { builtin.lnumfunc }, click = "v:lua.ScLa" },
  --         },
  --       }
  --     end,
  --   },
  -- },
  -- enabled = true,
  -- lazy = true,
  -- -- event = "BufReadPost",
  -- -- event = { "BufReadPost", "BufRead", "InsertEnter", "BufNewFile" },
  -- event = "VeryLazy",
  -- config = function()
  --   vim.o.foldcolumn = "1" -- '0' is not bad
  --   vim.o.foldlevel = 99 -- Using ufo provider need a large value, feel free to decrease the value
  --   vim.o.foldlevelstart = 99
  --   vim.o.foldenable = true
  --   -- vim.o.fillchars = [[eob: ,fold: ,foldopen:,foldsep: ,foldclose:]]
  --   -- vim.o.fillchars = [[eob: ,fold: ,foldopen:,foldsep: ,foldclose:]]
  --   -- vim.o.fillchars = [[eob: ,fold: ,foldopen:,foldsep:│,foldclose:]]
  --   -- vim.o.fillchars = [[eob: ,fold: ,foldopen:,foldsep:│,foldclose:]]
  --   -- vim.o.fillchars = [[eob: ,fold: ,foldopen:󰛲,foldsep:│,foldclose:󰜄]]
  --   -- vim.o.fillchars = [[eob: ,fold: ,foldopen:,foldsep:│,foldclose:]]
  --   -- vim.o.fillchars = [[eob: ,fold: ,foldopen:▾,foldsep:│,foldclose:▸]]
  --   vim.opt.fillchars = {
  --     vert = icons.vert,
  --     fold = icons.fold,
  --     eob = icons.eob,
  --     diff = icons.diff,
  --     msgsep = icons.msgsep,
  --     foldopen = icons.foldopen,
  --     foldsep = icons.foldsep,
  --     foldclose = icons.foldclose,
  --   }
  --   -- these are "extra", change them as you like
  --   vim.keymap.set("n", "zR", require("ufo").openAllFolds)
  --   vim.keymap.set("n", "zM", require("ufo").closeAllFolds)
  --   vim.cmd("highlight FoldColumn guifg=" .. vim.fn.synIDattr(vim.fn.synIDtrans(vim.fn.hlID "Comment"), "fg"))
  --   -- vim.cmd("highlight FoldColumn guifg=" .. vim.fn.synIDattr(vim.fn.synIDtrans(vim.fn.hlID("IblIndent")), "fg"))
  --   -- Option 3: treesitter as a main provider instead
  --   -- Only depend on `nvim-treesitter/queries/filetype/folds.scm`,
  --   -- performance and stability are better than `foldmethod=nvim_treesitter#foldexpr()`
  --
  --   --[[require("ufo").setup({
  -- 	-- provider_selector = function(bufnr, filetype, buftype)
  -- 	-- 	return { "treesitter", "indent" }
  -- 	-- end,
  -- 	open_fold_hl_timeout = 150,
  -- 	close_fold_kinds_for_ft = {
  -- 		default = { "imports", "comment" },
  -- 		json = { "array" },
  -- 		c = { "comment", "region" },
  -- 	},
  -- 	preview = {
  -- 		win_config = {
  -- 			border = { "", "─", "", "", "", "─", "", "" },
  -- 			-- winhighlight = 'Normal:Normal',
  -- 			-- winhighlight = 'IncSearch:Folded',
  -- 			winhighlight = "Normal:UfoPreviewNormal,FloatBorder:UfoPreviewBorder,CursorLine:UfoPreviewCursorLine",
  -- 			winblend = 0,
  -- 		},
  -- 		mappings = {
  -- 			scrollU = "<C-u>",
  -- 			scrollD = "<C-d>",
  -- 			jumpTop = "[",
  -- 			jumpBot = "]",
  -- 		},
  -- 	},
  -- 	provider_selector = function(_, filetype)
  -- 		return { "treesitter", "indent" }
  -- 	end,
  -- 	fold_virt_text_handler = function(virt_text, lnum, end_lnum, width, truncate)
  -- 		local result = {}
  -- 		local _end = end_lnum - 1
  -- 		local final_text = vim.trim(vim.api.nvim_buf_get_text(0, _end, 0, _end, -1, {})[1])
  -- 		local suffix = final_text:format(end_lnum - lnum)
  -- 		local suffix_width = vim.fn.strdisplaywidth(suffix)
  -- 		local target_width = width - suffix_width
  -- 		local cur_width = 0
  -- 		for _, chunk in ipairs(virt_text) do
  -- 			local chunk_text = chunk[1]
  -- 			local chunk_width = vim.fn.strdisplaywidth(chunk_text)
  -- 			if target_width > cur_width + chunk_width then
  -- 				table.insert(result, chunk)
  -- 			else
  -- 				chunk_text = truncate(chunk_text, target_width - cur_width)
  -- 				local hl_group = chunk[2]
  -- 				table.insert(result, { chunk_text, hl_group })
  -- 				chunk_width = vim.fn.strdisplaywidth(chunk_text)
  -- 				-- str width returned from truncate() may less than 2nd argument, need padding
  -- 				if cur_width + chunk_width < target_width then
  -- 					suffix = suffix .. (" "):rep(target_width - cur_width - chunk_width)
  -- 				end
  -- 				break
  -- 			end
  -- 			cur_width = cur_width + chunk_width
  -- 		end
  -- 		table.insert(result, { " ⋯ ", "NonText" })
  -- 		if vim.bo.filetype ~= "json" then
  -- 			table.insert(result, { suffix, "TSPunctBracket" })
  -- 		end
  -- 		return result
  -- 	end,
  -- })]]
  --   --
  --
  --   -- start ini bagian code support comment dan import
  --   local ftMap = {
  --     vim = "indent",
  --     python = { "indent" },
  --     git = "",
  --     -- javascriptreact = { "treesitter", "indent" },
  --     -- typescriptreact = { "treesitter", "indent" },
  --   }
  --
  --   local function customizeSelector(bufnr)
  --     local function handleFallbackException(err, providerName)
  --       if type(err) == "string" and err:match "UfoFallbackException" then
  --         return require("ufo").getFolds(bufnr, providerName)
  --       else
  --         return require("promise").reject(err)
  --       end
  --     end
  --
  --     return require("ufo")
  --       .getFolds(bufnr, "lsp")
  --       :catch(function(err)
  --         return handleFallbackException(err, "treesitter")
  --       end)
  --       :catch(function(err)
  --         return handleFallbackException(err, "indent")
  --       end)
  --   end
  --
  --   require("ufo").setup {
  --     open_fold_hl_timeout = 150,
  --     close_fold_kinds_for_ft = {
  --       -- default = { "imports", "comment" },
  --       -- json = { "array" },
  --       -- c = { "comment", "region" },
  --     },
  --     preview = {
  --       win_config = {
  --         border = { "", "─", "", "", "", "─", "", "" },
  --         winhighlight = "Normal:Folded",
  --         winblend = 0,
  --       },
  --       mappings = {
  --         scrollU = "<C-u>",
  --         scrollD = "<C-d>",
  --         jumpTop = "[",
  --         jumpBot = "]",
  --       },
  --     },
  --     provider_selector = function(bufnr, filetype, buftype)
  --       -- if you prefer treesitter provider rather than lsp,
  --       -- return ftMap[filetype]
  --       -- return ftMap[filetype] or {'treesitter', 'indent'}
  --       return ftMap[filetype] or customizeSelector
  --
  --       -- refer to ./doc/example.lua for detail
  --     end,
  --
  --     fold_virt_text_handler = function(virt_text, lnum, end_lnum, width, truncate)
  --       local result = {}
  --       local closed_fold_text = "comments ..." -- Teks yang ingin ditampilkan
  --       local import_fold_text = "import ..." -- Teks yang ingin ditampilkan
  --       local is_comment = false -- Variabel untuk mengecek apakah ini komentar
  --       local is_import = false
  --       local is_bracket = false
  --
  --       -- Memeriksa apakah baris awal dari fold adalah komentar
  --       local start_line = vim.api.nvim_buf_get_lines(0, lnum - 1, lnum, false)[1]
  --       -- cari comentar dengan awalan /* untuk generaal comment
  --       if start_line:find "^%s*%/%*" then
  --         is_comment = true
  --       -- cara commentar dengan awalan <!-- untuk html
  --       elseif start_line:find "^%s*<!--" then
  --         is_comment = true
  --       -- cari comentar dengan awalan -- untuk lua
  --       elseif start_line:find "^%s*%-%-" then
  --         is_comment = true
  --       end
  --       -- cek fold yang berawalan import
  --       if start_line:find "^%s*import" then
  --         is_import = true
  --       end
  --       -- cek fold dengan akhiran {
  --       if start_line:find "%s*{%s*$" then
  --         is_bracket = true
  --       end
  --       if is_comment then
  --         local suffix = string.format(" %s ", closed_fold_text)
  --         local target_width = width - vim.fn.strdisplaywidth(suffix)
  --         local cur_width = 0
  --         for _, chunk in ipairs(virt_text) do
  --           local chunk_text = chunk[1]
  --           local chunk_width = vim.fn.strdisplaywidth(chunk_text)
  --           if target_width > cur_width + chunk_width then
  --             table.insert(result, chunk)
  --           else
  --             chunk_text = truncate(chunk_text, target_width - cur_width)
  --             local hl_group = chunk[2]
  --             table.insert(result, { chunk_text, hl_group })
  --             break
  --           end
  --           cur_width = cur_width + chunk_width
  --         end
  --         -- Menambahkan teks 'Comments ...' ke akhir baris yang dilipat
  --         table.insert(result, { suffix, "NonText" })
  --       elseif is_import then
  --         --[[
  -- 			local suffix = string.format(" %s ", import_fold_text)
  -- 			local target_width = width - vim.fn.strdisplaywidth(suffix)
  -- 			local cur_width = 0
  -- 			for _, chunk in ipairs(virt_text) do
  -- 				local chunk_text = chunk[1]
  -- 				local chunk_width = vim.fn.strdisplaywidth(chunk_text)
  -- 				if target_width > cur_width + chunk_width then
  -- 					table.insert(result, chunk)
  -- 				else
  -- 					chunk_text = truncate(chunk_text, target_width - cur_width)
  -- 					local hl_group = chunk[2]
  -- 					table.insert(result, { chunk_text, hl_group })
  -- 					break
  -- 				end
  -- 				cur_width = cur_width + chunk_width
  -- 			end
  -- 			-- Menambahkan teks 'Comments ...' ke akhir baris yang dilipat
  -- 			table.insert(result, { suffix, "NonText" })
  -- 			]]
  --         --
  --
  --         local suffix = (" 󰁂 %d "):format(end_lnum - lnum)
  --         local sufWidth = vim.fn.strdisplaywidth(suffix)
  --         local targetWidth = width - sufWidth
  --         local curWidth = 0
  --         for _, chunk in ipairs(virt_text) do
  --           local chunkText = chunk[1]
  --           local chunkWidth = vim.fn.strdisplaywidth(chunkText)
  --           if targetWidth > curWidth + chunkWidth then
  --             table.insert(result, chunk)
  --           else
  --             chunkText = truncate(chunkText, targetWidth - curWidth)
  --             local hlGroup = chunk[2]
  --             table.insert(result, { chunkText, hlGroup })
  --             chunkWidth = vim.fn.strdisplaywidth(chunkText)
  --             -- str width returned from truncate() may less than 2nd argument, need padding
  --             if curWidth + chunkWidth < targetWidth then
  --               suffix = suffix .. (" "):rep(targetWidth - curWidth - chunkWidth)
  --             end
  --             break
  --           end
  --           curWidth = curWidth + chunkWidth
  --         end
  --         table.insert(result, { " import⋯ ", "NonText" })
  --         table.insert(result, { suffix, "MoreMsg" })
  --       else
  --         -- Jika bukan komentar, tampilkan teks asli
  --         -- for _, chunk in ipairs(virt_text) do
  --         -- 	table.insert(result, chunk)
  --         -- end
  --         local _end = end_lnum - 1
  --         local final_text = vim.trim(vim.api.nvim_buf_get_text(0, _end, 0, _end, -1, {})[1])
  --         local suffix = final_text:format(end_lnum - lnum)
  --         local suffix_width = vim.fn.strdisplaywidth(suffix)
  --         local target_width = width - suffix_width
  --         local cur_width = 0
  --         for _, chunk in ipairs(virt_text) do
  --           local chunk_text = chunk[1]
  --           local chunk_width = vim.fn.strdisplaywidth(chunk_text)
  --           if target_width > cur_width + chunk_width then
  --             table.insert(result, chunk)
  --           else
  --             chunk_text = truncate(chunk_text, target_width - cur_width)
  --             local hl_group = chunk[2]
  --             table.insert(result, { chunk_text, hl_group })
  --             chunk_width = vim.fn.strdisplaywidth(chunk_text)
  --             -- str width returned from truncate() may less than 2nd argument, need padding
  --             if cur_width + chunk_width < target_width then
  --               suffix = suffix .. (" "):rep(target_width - cur_width - chunk_width)
  --             end
  --             break
  --           end
  --           cur_width = cur_width + chunk_width
  --         end
  --         table.insert(result, { " ⋯ ", "NonText" })
  --         local l = { "javascriptreact", "typescriptreact" }
  --         if TABLE_CONTAINS(l, vim.bo.filetype) and not is_bracket then
  --           table.insert(result, { suffix, "TSPunctBracket" })
  --         end
  --       end
  --       return result
  --     end,
  --   }
  --   -- end bagian code support comment dan import
  -- end,
}