micro: add textcount plugin, enable it for .md/.adoc

2 files changed, 86 insertions, 0 deletions

diff --git a/dot_config/micro/init.lua b/dot_config/micro/init.lua
new file mode 100644
index 0000000..03238df
--- /dev/null
+++ b/dot_config/micro/init.lua
@@ -0,0 +1,21 @@
+local config = import 'micro/config'
+
+local function wantwords(ft)
+	return ft == 'markdown'
+		or ft == 'asciidoc'
+end
+
+local function appendsep(s, a)
+	if s ~= '' then s = s .. ' | ' end
+	return s .. a
+end
+
+function onBufferOpen(b)
+	if wantwords(b:FileType()) then
+		local format = config.GetGlobalOption 'statusformatr'
+		local add    = '$(textcount.words) Words'
+		if not string.find(format, add) then
+			b:SetOption('statusformatr', appendsep(format, add))
+		end
+	end
+end
diff --git a/dot_config/micro/plug/textcount/textcount.lua b/dot_config/micro/plug/textcount/textcount.lua
new file mode 100644
index 0000000..2d123fa
--- /dev/null
+++ b/dot_config/micro/plug/textcount/textcount.lua
@@ -0,0 +1,65 @@
+VERSION = '0.1.0'
+
+local fmt    = import 'fmt'
+local micro  = import 'micro'
+local config = import 'micro/config'
+
+function init()
+	micro.SetStatusInfoFn 'textcount.lines'
+	micro.SetStatusInfoFn 'textcount.paragraphs'
+	micro.SetStatusInfoFn 'textcount.sentences'
+	micro.SetStatusInfoFn 'textcount.words'
+	config.MakeCommand('wc.l', linesi, 		config.NoComplete)
+	config.MakeCommand('wc.p', paragraphsi, config.NoComplete)
+	config.MakeCommand('wc.s', sentencesi, 	config.NoComplete)
+	config.MakeCommand('wc.w', wordsi, 		config.NoComplete)
+
+	config.MakeCommand('wc', wc, config.NoComplete)
+end
+
+-- utils
+local function pattern(b, pat)
+	local str  = fmt.Sprintf('%s', b:Bytes())
+	local _, n = str:gsub(pat, '')
+	return n
+end
+
+local function genb(...)
+	local args = {...}
+	return function(b)
+		local total = 0
+		for _, v in ipairs(args) do
+			total = total + pattern(b, v)
+		end
+		return tostring(total)
+	end
+end
+
+local function geni(f, label)
+	return function(bp)
+		local data = f(bp.Buf)
+		micro.InfoBar():Message(data .. ' ' .. label)
+	end
+end
+
+-- functions
+lines  = genb '\n'
+linesi = geni(lines, 'Lines')
+
+paragraphs  = genb('\n\n', '$') -- $ handles the final paragraph
+paragraphsi = geni(paragraphs, 'Paragraphs')
+
+sentences  = genb '%.[^%.]'
+sentencesi = geni(sentences, 'Sentences')
+
+words  = genb '%S+'
+wordsi = geni(words, 'Words')
+
+-- user-defined patterns
+function wc(bp, args)
+	local total = 0
+	for _, v in args() do
+		total = total + pattern(bp.Buf, v)
+	end
+	micro.InfoBar():Message(tostring(total))
+end