diff options
Diffstat (limited to 'dot-config/nvim/lua/wordcount.lua')
| -rw-r--r-- | dot-config/nvim/lua/wordcount.lua | 27 |
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 |