edtris/vendor/lite-plugins/plugins/spellcheck.lua
2021-02-07 16:56:19 -05:00

180 lines
4.5 KiB
Lua

local core = require "core"
local style = require "core.style"
local config = require "core.config"
local command = require "core.command"
local common = require "core.common"
local DocView = require "core.docview"
local Doc = require "core.doc"
config.spellcheck_files = { "%.txt$", "%.md$", "%.markdown$" }
if PLATFORM == "Windows" then
config.dictionary_file = EXEDIR .. "/words.txt"
else
config.dictionary_file = "/usr/share/dict/words"
end
local last_input_time = 0
local word_pattern = "%a+"
local words
core.add_thread(function()
local t = {}
local i = 0
for line in io.lines(config.dictionary_file) do
for word in line:gmatch(word_pattern) do
t[word:lower()] = true
end
i = i + 1
if i % 1000 == 0 then coroutine.yield() end
end
words = t
core.redraw = true
core.log_quiet("Finished loading dictionary file: \"%s\"", config.dictionary_file)
end)
local function matches_any(filename, ptns)
for _, ptn in ipairs(ptns) do
if filename:find(ptn) then return true end
end
end
local function active_word(doc, line, tail)
local l, c = doc:get_selection()
return l == line and c == tail
and doc == core.active_view.doc
and system.get_time() - last_input_time < 0.5
end
local text_input = Doc.text_input
function Doc:text_input(...)
text_input(self, ...)
last_input_time = system.get_time()
end
local draw_line_text = DocView.draw_line_text
function DocView:draw_line_text(idx, x, y)
draw_line_text(self, idx, x, y)
if not words
or not matches_any(self.doc.filename or "", config.spellcheck_files) then
return
end
local s, e = 0, 0
local text = self.doc.lines[idx]
while true do
s, e = text:find(word_pattern, e + 1)
if not s then break end
local word = text:sub(s, e):lower()
if not words[word] and not active_word(self.doc, idx, e + 1) then
local color = style.spellcheck_error or style.syntax.keyword2
local x1 = x + self:get_col_x_offset(idx, s)
local x2 = x + self:get_col_x_offset(idx, e + 1)
local h = math.ceil(1 * SCALE)
renderer.draw_rect(x1, y + self:get_line_height() - h, x2 - x1, h, color)
end
end
end
local function get_word_at_caret()
local doc = core.active_view.doc
local l, c = doc:get_selection()
local s, e = 0, 0
local text = doc.lines[l]
while true do
s, e = text:find(word_pattern, e + 1)
if c >= s and c <= e + 1 then
return text:sub(s, e):lower(), s, e
end
end
end
local function compare_words(word1, word2)
local res = 0
for i = 1, math.max(#word1, #word2) do
if word1:byte(i) ~= word2:byte(i) then
res = res + 1
end
end
return res
end
command.add("core.docview", {
["spell-check:add-to-dictionary"] = function()
local word = get_word_at_caret()
if words[word] then
core.error("\"%s\" already exists in the dictionary", word)
return
end
if word then
local fp = assert(io.open(config.dictionary_file, "a"))
fp:write("\n" .. word .. "\n")
fp:close()
words[word] = true
core.log("Added \"%s\" to dictionary", word)
end
end,
["spell-check:replace"] = function()
local word, s, e = get_word_at_caret()
-- find suggestions
local suggestions = {}
local word_len = #word
for w in pairs(words) do
if math.abs(#w - word_len) <= 2 then
local diff = compare_words(word, w)
if diff < word_len * 0.5 then
table.insert(suggestions, { diff = diff, text = w })
end
end
end
if #suggestions == 0 then
core.error("Could not find any suggestions for \"%s\"", word)
return
end
-- sort suggestions table and convert to properly-capitalized text
table.sort(suggestions, function(a, b) return a.diff < b.diff end)
local doc = core.active_view.doc
local line = doc:get_selection()
local has_upper = doc.lines[line]:sub(s, s):match("[A-Z]")
for k, v in pairs(suggestions) do
if has_upper then
v.text = v.text:gsub("^.", string.upper)
end
suggestions[k] = v.text
end
-- select word and init replacement selector
local label = string.format("Replace \"%s\" With", word)
doc:set_selection(line, e + 1, line, s)
core.command_view:enter(label, function(text, item)
text = item and item.text or text
doc:replace(function() return text end)
end, function(text)
local t = {}
for _, w in ipairs(suggestions) do
if w:lower():find(text:lower(), 1, true) then
table.insert(t, w)
end
end
return t
end)
end,
})