feat(text): center text

folke · Nov 29, 2022 · 88869e6 · 88869e6
1 parent 7303017
commit 88869e6
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/lua/lazy/view/text.lua b/lua/lazy/view/text.lua
@@ -120,6 +120,22 @@ function Text:highlight(patterns)
   end
 end
 
+function Text:center()
+  local last = self._lines[#self._lines]
+  if not last then
+    return
+  end
+  local width = 0
+  for _, segment in ipairs(last) do
+    width = width + vim.fn.strwidth(segment.str)
+  end
+  width = vim.api.nvim_win_get_width(self.win) - 2 * self.padding - width
+  table.insert(last, 1, {
+    str = string.rep(" ", math.floor(width / 2 + 0.5)),
+  })
+  return self
+end
+
 function Text:trim()
   while #self._lines > 0 and #self._lines[1] == 0 do
     table.remove(self._lines, 1)