local config = require("java-deps.config")

local icons = require("java-deps.views.icons")

local M = {}

local function str_to_table(str)

local t = {}

for i = 1, #str do

t[i] = str:sub(i, i)

end

return t

end

local function table_to_str(t)

local ret = ""

for _, value in ipairs(t) do

ret = ret .. tostring(value)

end

return ret

end

local guides = {

markers = {

bottom = "└",

middle = "├",

vertical = "│",

horizontal = "─",

},

}

---@param flattened_outline_items TreeItem

---@return table

---@return table

function M.get_lines(flattened_outline_items)

local lines = {}

local hl_info = {}

for node_line, node in ipairs(flattened_outline_items) do

local depth = node.depth

local marker_space = config.options.fold_markers and 1 or 0

local line = str_to_table(string.rep(" ", depth + marker_space))

local folded = node:is_foldable()

for index, _ in ipairs(line) do

-- all items start with a space (or two)

if config.options.show_guides then

if index == #line then

-- add fold markers

if config.options.fold_markers and folded then

if node:is_expanded() then

line[index] = config.options.fold_markers[2]

else

line[index] = config.options.fold_markers[1]

end

elseif depth > 1 then

if node.isLast then

line[index] = guides.markers.bottom

else

line[index] = guides.markers.middle

end

end

elseif not node.hierarchy[index] and depth > 1 then

line[index + marker_space] = guides.markers.vertical

end

end

line[index] = line[index] .. " "

end

local string_prefix = ""

for _, value in ipairs(line) do

string_prefix = string_prefix .. tostring(value)

end

local hl_icon = icons.get_icon(node.data)

local icon = hl_icon.icon

table.insert(lines, string_prefix .. icon .. " " .. node.label)

local hl_start = #string_prefix

local hl_end = #string_prefix + #icon

local hl_type = hl_icon.hl or "Type"

table.insert(hl_info, { node_line, hl_start, hl_end, hl_type })

node.prefix_length = #string_prefix + #icon + 1

end

return lines, hl_info

end

return M