1 files changed, 120 insertions, 0 deletions
diff --git a/lua/blink/cmp/completion/windows/render/column.lua b/lua/blink/cmp/completion/windows/render/column.lua
new file mode 100644
index 0000000..b9a75d0
--- /dev/null
+++ b/lua/blink/cmp/completion/windows/render/column.lua
@@ -0,0 +1,120 @@
+--- @class blink.cmp.DrawColumn
+--- @field components blink.cmp.DrawComponent[]
+--- @field gap number
+--- @field lines string[][]
+--- @field width number
+--- @field ctxs blink.cmp.DrawItemContext[]
+---
+--- @field new fun(components: blink.cmp.DrawComponent[], gap: number): blink.cmp.DrawColumn
+--- @field render fun(self: blink.cmp.DrawColumn, ctxs: blink.cmp.DrawItemContext[])
+--- @field get_line_text fun(self: blink.cmp.DrawColumn, line_idx: number): string
+--- @field get_line_highlights fun(self: blink.cmp.DrawColumn, line_idx: number): blink.cmp.DrawHighlight[]
+
+local text_lib = require('blink.cmp.completion.windows.render.text')
+
+--- @type blink.cmp.DrawColumn
+--- @diagnostic disable-next-line: missing-fields
+local column = {}
+
+function column.new(components, gap)
+  local self = setmetatable({}, { __index = column })
+  self.components = components
+  self.gap = gap
+  self.lines = {}
+  self.width = 0
+  self.ctxs = {}
+  return self
+end
+
+function column:render(ctxs)
+  --- render text and get the max widths of each component
+  --- @type string[][]
+  local lines = {}
+  local max_component_widths = {}
+  for _, ctx in ipairs(ctxs) do
+    --- @type string[]
+    local line = {}
+    for component_idx, component in ipairs(self.components) do
+      local text = text_lib.apply_component_width(component.text(ctx) or '', component)
+      table.insert(line, text)
+      max_component_widths[component_idx] =
+        math.max(max_component_widths[component_idx] or 0, vim.api.nvim_strwidth(text))
+    end
+    table.insert(lines, line)
+  end
+
+  --- get the total width of the column
+  local column_width = 0
+  for _, max_component_width in ipairs(max_component_widths) do
+    if max_component_width > 0 then column_width = column_width + max_component_width + self.gap end
+  end
+  column_width = math.max(column_width - self.gap, 0)
+
+  --- find the component that will fill the empty space
+  local fill_idx = -1
+  for component_idx, component in ipairs(self.components) do
+    if component.width and component.width.fill then
+      fill_idx = component_idx
+      break
+    end
+  end
+  if fill_idx == -1 then fill_idx = #self.components end
+
+  --- and add extra spaces until we reach the column width
+  for _, line in ipairs(lines) do
+    local line_width = 0
+    for _, component_text in ipairs(line) do
+      if #component_text > 0 then line_width = line_width + vim.api.nvim_strwidth(component_text) + self.gap end
+    end
+    line_width = line_width - self.gap
+    local remaining_width = column_width - line_width
+    line[fill_idx] = text_lib.pad(line[fill_idx], vim.api.nvim_strwidth(line[fill_idx]) + remaining_width)
+  end
+
+  -- store results for later
+  self.width = column_width
+  self.lines = lines
+  self.ctxs = ctxs
+end
+
+function column:get_line_text(line_idx)
+  local text = ''
+  local line = self.lines[line_idx]
+  for _, component in ipairs(line) do
+    if #component > 0 then text = text .. component .. string.rep(' ', self.gap) end
+  end
+  return text:sub(1, -self.gap - 1)
+end
+
+function column:get_line_highlights(line_idx)
+  local ctx = self.ctxs[line_idx]
+  local offset = 0
+  local highlights = {}
+
+  for component_idx, component in ipairs(self.components) do
+    local text = self.lines[line_idx][component_idx]
+    if #text > 0 then
+      local column_highlights = type(component.highlight) == 'function' and component.highlight(ctx, text)
+        or component.highlight
+
+      if type(column_highlights) == 'string' then
+        table.insert(highlights, { offset, offset + #text, group = column_highlights })
+      elseif type(column_highlights) == 'table' then
+        for _, highlight in ipairs(column_highlights) do
+          table.insert(highlights, {
+            offset + highlight[1],
+            offset + highlight[2],
+            group = highlight.group,
+            params = highlight.params,
+          })
+        end
+      end
+
+      offset = offset + #text + self.gap
+    end
+  end
+
+  return highlights
+end
+
+return column