replace lazyvim with custom config

1 files changed, 13 insertions, 14 deletions

diff --git a/dot-config/nvim/lua/wordcount.lua b/dot-config/nvim/lua/wordcount.lua
index 92944e5..1ff5579 100644
--- a/dot-config/nvim/lua/wordcount.lua
+++ b/dot-config/nvim/lua/wordcount.lua
@@ -7,27 +7,26 @@
 local M = {}
 
 local config = {
-  filetypes = { "markdown", "text", "typst" },
+	filetypes = { "markdown", "text", "typst" },
 }
 
-function M.wordcount()
-  local wc = vim.fn.wordcount()
-
-  if wc.visual_words then
-    return tostring(wc.visual_words) .. " words"
-  end
+local function format_wc(n)
+	return tostring(n) .. " words"
+end
 
-  return tostring(wc.words) .. " words"
+function M.wordcount()
+	local wc = vim.fn.wordcount()
+	return (wc.visual_words and format_wc(wc.visual_words)) or format_wc(wc.words)
 end
 
 function M.is_available()
-  for _, ft in ipairs(config.filetypes) do
-    if ft == vim.bo.filetype then
-      return true
-    end
-  end
+	for _, ft in ipairs(config.filetypes) do
+		if ft == vim.bo.filetype then
+			return true
+		end
+	end
 
-  return false
+	return false
 end
 
 return M