init commitHEADmaster

1 files changed, 77 insertions, 0 deletions

diff --git a/autoload/tools/python/jedi.kak b/autoload/tools/python/jedi.kak
new file mode 100644
index 0000000..82d799a
--- /dev/null
+++ b/autoload/tools/python/jedi.kak
@@ -0,0 +1,77 @@
+hook -once global BufSetOption filetype=python %{
+    require-module jedi
+}
+
+provide-module jedi %{
+
+declare-option -hidden str jedi_tmp_dir
+declare-option -hidden completions jedi_completions
+declare-option -docstring "colon separated list of path added to `python`'s $PYTHONPATH environment variable" \
+    str jedi_python_path
+
+define-command jedi-complete -docstring "Complete the current selection" %{
+    evaluate-commands %sh{
+        dir=$(mktemp -d "${TMPDIR:-/tmp}"/kak-jedi.XXXXXXXX)
+        mkfifo ${dir}/fifo
+        printf %s\\n "set-option buffer jedi_tmp_dir ${dir}"
+        printf %s\\n "evaluate-commands -no-hooks write -sync ${dir}/buf"
+    }
+    evaluate-commands %sh{
+        dir=${kak_opt_jedi_tmp_dir}
+        printf %s\\n "evaluate-commands -draft %{ edit! -fifo ${dir}/fifo *jedi-output* }"
+        ((
+            trap - INT QUIT
+            cd $(dirname ${kak_buffile})
+
+            export PYTHONPATH="$kak_opt_jedi_python_path:$PYTHONPATH"
+            python 2> "${dir}/fifo" -c 'if 1:
+                import os
+                dir = os.environ["kak_opt_jedi_tmp_dir"]
+                buffile = os.environ["kak_buffile"]
+                line = int(os.environ["kak_cursor_line"])
+                column = int(os.environ["kak_cursor_column"])
+                timestamp = os.environ["kak_timestamp"]
+                client = os.environ["kak_client"]
+                pipe_escape = lambda s: s.replace("|", "\\|")
+                def quote(s):
+                    c = chr(39) # single quote
+                    return c + s.replace(c, c+c) + c
+                import jedi
+                script = jedi.Script(code=open(dir + "/buf", "r").read(), path=buffile)
+                completions = (
+                    quote(
+                        pipe_escape(str(c.name)) + "|" +
+                        pipe_escape("info -style menu -- " + quote(c.docstring())) + "|" +
+                        pipe_escape(str(c.name))
+                    )
+                    for c in script.complete(line=line, column=column-1)
+                )
+                header = str(line) + "." + str(column) + "@" + timestamp
+                cmds = [
+                    "echo completed",
+                    " ".join(("set-option", quote("buffer=" + buffile), "jedi_completions", header, *completions)),
+                ]
+                print("evaluate-commands -client", quote(client), quote("\n".join(cmds)))
+            ' | kak -p "${kak_session}"
+            rm -r ${dir}
+        ) & ) > /dev/null 2>&1 < /dev/null
+    }
+}
+
+define-command jedi-enable-autocomplete -docstring "Add jedi completion candidates to the completer" %{
+    set-option window completers option=jedi_completions %opt{completers}
+    hook window -group jedi-autocomplete InsertIdle .* %{ try %{
+        execute-keys -draft <a-h><a-k>\..\z<ret>
+        echo 'completing...'
+        jedi-complete
+    } }
+    alias window complete jedi-complete
+}
+
+define-command jedi-disable-autocomplete -docstring "Disable jedi completion" %{
+    set-option window completers %sh{ printf %s\\n "'${kak_opt_completers}'" | sed -e 's/option=jedi_completions://g' }
+    remove-hooks window jedi-autocomplete
+    unalias window complete jedi-complete
+}
+
+}