diff options
Diffstat (limited to 'autoload/tools/spell.kak')
| -rw-r--r-- | autoload/tools/spell.kak | 184 |
1 files changed, 184 insertions, 0 deletions
diff --git a/autoload/tools/spell.kak b/autoload/tools/spell.kak new file mode 100644 index 0000000..4bd3305 --- /dev/null +++ b/autoload/tools/spell.kak @@ -0,0 +1,184 @@ +declare-option -hidden range-specs spell_regions +declare-option -hidden str spell_last_lang + +declare-option -docstring "default language to use when none is passed to the spell-check command" str spell_lang + +define-command -params ..1 -docstring %{ + spell [<language>]: spell check the current buffer + + The first optional argument is the language against which the check will be performed (overrides `spell_lang`) + Formats of language supported: + - ISO language code, e.g. 'en' + - language code above followed by a dash or underscore with an ISO country code, e.g. 'en-US' + } spell %{ + try %{ add-highlighter window/ ranges 'spell_regions' } + evaluate-commands %sh{ + use_lang() { + if ! printf %s "$1" | grep -qE '^[a-z]{2,3}([_-][A-Z]{2})?$'; then + echo "fail 'Invalid language code (examples of expected format: en, en_US, en-US)'" + exit 1 + else + options="-l '$1'" + printf 'set-option buffer spell_last_lang %s\n' "$1" + fi + } + + if [ $# -ge 1 ]; then + use_lang "$1" + elif [ -n "${kak_opt_spell_lang}" ]; then + use_lang "${kak_opt_spell_lang}" + fi + + printf 'eval -no-hooks write %s\n' "${kak_response_fifo}" > $kak_command_fifo + + { + trap - INT QUIT + sed 's/^/^/' | eval "aspell --byte-offsets -a $options" 2>&1 | awk ' + BEGIN { + line_num = 1 + regions = ENVIRON["kak_timestamp"] + server_command = sprintf("kak -p \"%s\"", ENVIRON["kak_session"]) + } + + { + if (/^@\(#\)/) { + # drop the identification message + } + + else if (/^\*/) { + # nothing + } + + else if (/^[+-]/) { + # required to ignore undocumented aspell functionality + } + + else if (/^$/) { + line_num++ + } + + else if (/^[#&]/) { + word_len = length($2) + word_pos = substr($0, 1, 1) == "&" ? substr($4, 1, length($4) - 1) : $3; + regions = regions " " line_num "." word_pos "+" word_len "|DiagnosticError" + } + + else { + line = $0 + gsub(/"/, "&&", line) + command = "fail \"" line "\"" + exit + } + } + + END { + if (!length(command)) + command = "set-option \"buffer=" ENVIRON["kak_bufname"] "\" spell_regions " regions + + print command | server_command + close(server_command) + } + ' + } <$kak_response_fifo >/dev/null 2>&1 & + } +} + +define-command spell-clear %{ + unset-option buffer spell_regions +} + +define-command spell-next %{ evaluate-commands %sh{ + anchor_line="${kak_selection_desc%%.*}" + anchor_col="${kak_selection_desc%%,*}" + anchor_col="${anchor_col##*.}" + + start_first="${kak_opt_spell_regions%%|*}" + start_first="${start_first#* }" + + # Make sure properly formatted selection descriptions are in `%opt{spell_regions}` + if ! printf %s "${start_first}" | grep -qE '^[0-9]+\.[0-9]+,[0-9]+\.[0-9]+$'; then + exit + fi + + printf %s "${kak_opt_spell_regions#* }" | awk -v start_first="${start_first}" \ + -v anchor_line="${anchor_line}" \ + -v anchor_col="${anchor_col}" ' + BEGIN { + anchor_line = int(anchor_line) + anchor_col = int(anchor_col) + } + + { + for (i = 1; i <= NF; i++) { + sel = $i + sub(/\|.+$/, "", sel) + + start_line = sel + sub(/\..+$/, "", start_line) + start_line = int(start_line) + + start_col = sel + sub(/,.+$/, "", start_col) + sub(/^.+\./, "", start_col) + start_col = int(start_col) + + if (start_line < anchor_line \ + || (start_line == anchor_line && start_col <= anchor_col)) + continue + + target_sel = sel + break + } + } + + END { + if (!target_sel) + target_sel = start_first + + printf "select %s\n", target_sel + }' +} } + +define-command \ + -docstring "Suggest replacement words for the current selection, against the last language used by the spell-check command" \ + spell-replace %{ + prompt \ + -shell-script-candidates %{ + options="" + if [ -n "$kak_opt_spell_last_lang" ]; then + options="-l '$kak_opt_spell_last_lang'" + fi + printf %s "$kak_selection" | + eval "aspell -a $options" | + sed -n -e '/^&/ { s/^[^:]*: //; s/, /\n/g; p;}' + } \ + "Replace with: " \ + %{ + evaluate-commands -save-regs a %{ + set-register a %val{text} + execute-keys c <c-r>a <esc> + } + } +} + + +define-command -params 0.. \ + -docstring "Add the current selection to the dictionary" \ + spell-add %{ evaluate-commands %sh{ + options="" + if [ -n "$kak_opt_spell_last_lang" ]; then + options="-l '$kak_opt_spell_last_lang'" + fi + if [ $# -eq 0 ]; then + # use selections + eval set -- "$kak_quoted_selections" + fi + while [ $# -gt 0 ]; do + word="$1" + if ! printf '*%s\n#\n' "${word}" | eval "aspell -a $options" >/dev/null; then + printf 'fail "Unable to add word: %s"' "$(printf %s "${word}" | sed 's/"/&&/g')" + exit 1 + fi + shift + done +}} |