90692edc35
Now instead of wrapping the code up in a try/catch, we just use a series of evals to grab *e and then restore *1/*2/*3.
286 lines
8.8 KiB
VimL
286 lines
8.8 KiB
VimL
" autoload/nrepl/fireplace_connection.vim
|
|
" Maintainer: Tim Pope <http://tpo.pe/>
|
|
|
|
if exists("g:autoloaded_nrepl_fireplace_connection") || &cp
|
|
finish
|
|
endif
|
|
let g:autoloaded_nrepl_fireplace_connection = 1
|
|
|
|
function! s:function(name) abort
|
|
return function(substitute(a:name,'^s:',matchstr(expand('<sfile>'), '<SNR>\d\+_'),''))
|
|
endfunction
|
|
|
|
" Bencode {{{1
|
|
|
|
function! nrepl#fireplace_connection#bencode(value) abort
|
|
if type(a:value) == type(0)
|
|
return 'i'.a:value.'e'
|
|
elseif type(a:value) == type('')
|
|
return strlen(a:value).':'.a:value
|
|
elseif type(a:value) == type([])
|
|
return 'l'.join(map(copy(a:value),'nrepl#fireplace_connection#bencode(v:val)'),'').'e'
|
|
elseif type(a:value) == type({})
|
|
return 'd'.join(values(map(copy(a:value),'nrepl#fireplace_connection#bencode(v:key).nrepl#fireplace_connection#bencode(v:val)')),'').'e'
|
|
else
|
|
throw "Can't bencode ".string(a:value)
|
|
endif
|
|
endfunction
|
|
|
|
function! nrepl#fireplace_connection#bdecode(value) abort
|
|
return s:bdecode({'pos': 0, 'value': a:value})
|
|
endfunction
|
|
|
|
function! s:bdecode(state) abort
|
|
let value = a:state.value
|
|
if value[a:state.pos] =~# '\d'
|
|
let pos = a:state.pos
|
|
let length = matchstr(value[pos : -1], '^\d\+')
|
|
let a:state.pos += strlen(length) + length + 1
|
|
return value[pos+strlen(length)+1 : pos+strlen(length)+length]
|
|
elseif value[a:state.pos] ==# 'i'
|
|
let int = matchstr(value[a:state.pos+1:-1], '[^e]*')
|
|
let a:state.pos += 2 + strlen(int)
|
|
return str2nr(int)
|
|
elseif value[a:state.pos] ==# 'l'
|
|
let values = []
|
|
let a:state.pos += 1
|
|
while value[a:state.pos] !=# 'e' && value[a:state.pos] !=# ''
|
|
call add(values, s:bdecode(a:state))
|
|
endwhile
|
|
let a:state.pos += 1
|
|
return values
|
|
elseif value[a:state.pos] ==# 'd'
|
|
let values = {}
|
|
let a:state.pos += 1
|
|
while value[a:state.pos] !=# 'e' && value[a:state.pos] !=# ''
|
|
let key = s:bdecode(a:state)
|
|
let values[key] = s:bdecode(a:state)
|
|
endwhile
|
|
let a:state.pos += 1
|
|
return values
|
|
else
|
|
throw 'bencode parse error: '.string(a:state)
|
|
endif
|
|
endfunction
|
|
|
|
" }}}1
|
|
|
|
function! s:shellesc(arg) abort
|
|
if a:arg =~ '^[A-Za-z0-9_/.-]\+$'
|
|
return a:arg
|
|
elseif &shell =~# 'cmd'
|
|
return '"'.substitute(substitute(a:arg, '"', '""""', 'g'), '%', '"%"', 'g').'"'
|
|
else
|
|
let escaped = shellescape(a:arg)
|
|
if &shell =~# 'sh' && &shell !~# 'csh'
|
|
return substitute(escaped, '\\\n', '\n', 'g')
|
|
else
|
|
return escaped
|
|
endif
|
|
endif
|
|
endfunction
|
|
|
|
function! nrepl#fireplace_connection#prompt() abort
|
|
return fireplace#input_host_port()
|
|
endfunction
|
|
|
|
function! nrepl#fireplace_connection#open(arg) abort
|
|
if a:arg =~# '^\d\+$'
|
|
let host = 'localhost'
|
|
let port = a:arg
|
|
elseif a:arg =~# ':\d\+$'
|
|
let host = matchstr(a:arg, '.*\ze:')
|
|
let port = matchstr(a:arg, ':\zs.*')
|
|
else
|
|
throw "nREPL: Couldn't find [host:]port in " . a:arg
|
|
endif
|
|
let client = deepcopy(s:nrepl)
|
|
let client.host = host
|
|
let client.port = port
|
|
let session = client.process({'op': 'clone'})['new-session']
|
|
let response = client.process({'op': 'eval', 'session': session, 'code':
|
|
\ '(do (println "success") (symbol (str (System/getProperty "path.separator") (System/getProperty "java.class.path"))))'})
|
|
let client._path = response.value[-1]
|
|
if has_key(response, 'out')
|
|
let client.session = session
|
|
endif
|
|
return client
|
|
endfunction
|
|
|
|
function! s:nrepl_path() dict abort
|
|
return split(self._path[1:-1], self._path[0])
|
|
endfunction
|
|
|
|
function! s:nrepl_process(payload) dict abort
|
|
let combined = {'status': [], 'session': []}
|
|
for response in self.call(a:payload)
|
|
for key in keys(response)
|
|
if key ==# 'id' || key ==# 'ns'
|
|
let combined[key] = response[key]
|
|
elseif key ==# 'value'
|
|
let combined.value = extend(get(combined, 'value', []), [response.value])
|
|
elseif key ==# 'status'
|
|
for entry in response[key]
|
|
if index(combined[key], entry) < 0
|
|
call extend(combined[key], [entry])
|
|
endif
|
|
endfor
|
|
elseif key ==# 'session'
|
|
if index(combined[key], response[key]) < 0
|
|
call extend(combined[key], [response[key]])
|
|
endif
|
|
elseif type(response[key]) == type('')
|
|
let combined[key] = get(combined, key, '') . response[key]
|
|
else
|
|
let combined[key] = response[key]
|
|
endif
|
|
endfor
|
|
endfor
|
|
if index(combined.status, 'error') >= 0
|
|
throw 'nREPL: ' . tr(combined.status[0], '-', ' ')
|
|
endif
|
|
return combined
|
|
endfunction
|
|
|
|
function! s:nrepl_eval(expr, ...) dict abort
|
|
let payload = {"op": "eval"}
|
|
let payload.code = a:expr
|
|
let options = a:0 ? a:1 : {}
|
|
if has_key(options, 'ns')
|
|
let payload.ns = options.ns
|
|
elseif has_key(self, 'ns')
|
|
let payload.ns = self.ns
|
|
endif
|
|
if get(options, 'session', 1)
|
|
if has_key(self, 'session')
|
|
let payload.session = self.session
|
|
elseif &verbose
|
|
echohl WarningMSG
|
|
echo "nREPL: server has bug preventing session support"
|
|
echohl None
|
|
endif
|
|
endif
|
|
if has_key(options, 'file_path')
|
|
let payload.op = 'load-file'
|
|
let payload['file-path'] = options.file_path
|
|
let payload['file-name'] = fnamemodify(options.file_path, ':t')
|
|
if has_key(payload, 'ns')
|
|
let payload.file = "(in-ns '".payload.ns.") ".payload.code
|
|
call remove(payload, 'ns')
|
|
else
|
|
let payload.file = payload.code
|
|
endif
|
|
call remove(payload, 'code')
|
|
endif
|
|
let response = self.process(payload)
|
|
if has_key(response, 'ns') && !a:0
|
|
let self.ns = response.ns
|
|
endif
|
|
|
|
if has_key(response, 'ex') && has_key(payload, 'session')
|
|
let response.stacktrace = s:extract_last_stacktrace(self)
|
|
endif
|
|
|
|
if has_key(response, 'value')
|
|
let response.value = response.value[-1]
|
|
endif
|
|
return response
|
|
endfunction
|
|
|
|
function! s:extract_last_stacktrace(nrepl)
|
|
let format_st = '(clojure.core/symbol (clojure.core/str "\n\b" (clojure.core/apply clojure.core/str (clojure.core/interleave (clojure.core/repeat "\n") (clojure.core/map clojure.core/str (.getStackTrace *e)))) "\n\b\n"))'
|
|
let stacktrace = split(get(split(a:nrepl.process({'op': 'eval', 'code': '['.format_st.' *3 *2 *1]', 'session': a:nrepl.session}).value[0], "\n\b\n"), 1, ""), "\n")
|
|
call a:nrepl.call({'op': 'eval', 'code': '(nth *1 1)', 'session': a:nrepl.session})
|
|
call a:nrepl.call({'op': 'eval', 'code': '(nth *2 2)', 'session': a:nrepl.session})
|
|
call a:nrepl.call({'op': 'eval', 'code': '(nth *3 3)', 'session': a:nrepl.session})
|
|
return stacktrace
|
|
endfunction
|
|
|
|
function! s:nrepl_call(payload) dict abort
|
|
let in = 'ruby -rsocket -e '.s:shellesc(
|
|
\ 'begin;' .
|
|
\ 'TCPSocket.open(%(' . self.host . '), ' . self.port . ') {|s|' .
|
|
\ 's.write(ARGV.first); loop {' .
|
|
\ 'body = s.readpartial(8192);' .
|
|
\ 'raise %(not an nREPL server: upgrade to Leiningen 2) if body =~ /=> $/;' .
|
|
\ 'print body;' .
|
|
\ 'break if body =~ /6:statusl(5:error|14:session-closed)?4:done/ }};' .
|
|
\ 'rescue; abort $!.to_s;' .
|
|
\ 'end') . ' ' .
|
|
\ s:shellesc(nrepl#fireplace_connection#bencode(a:payload))
|
|
let out = system(in)
|
|
if !v:shell_error
|
|
return nrepl#fireplace_connection#bdecode('l'.out.'e')
|
|
endif
|
|
throw 'nREPL: '.split(out, "\n")[0]
|
|
endfunction
|
|
|
|
let s:nrepl = {
|
|
\ 'call': s:function('s:nrepl_call'),
|
|
\ 'eval': s:function('s:nrepl_eval'),
|
|
\ 'path': s:function('s:nrepl_path'),
|
|
\ 'process': s:function('s:nrepl_process')}
|
|
|
|
if !has('python')
|
|
finish
|
|
endif
|
|
|
|
python << EOF
|
|
import vim
|
|
import select
|
|
import socket
|
|
import string
|
|
import re
|
|
|
|
def fireplace_string_encode(input):
|
|
str_list = []
|
|
for c in input:
|
|
if (000 <= ord(c) and ord(c) <= 037) or c == '"' or c == "\\":
|
|
str_list.append("\\{0:03o}".format(ord(c)))
|
|
else:
|
|
str_list.append(c)
|
|
return '"' + ''.join(str_list) + '"'
|
|
|
|
def fireplace_let(var, value):
|
|
return vim.command('let ' + var + " = " + fireplace_string_encode(value))
|
|
|
|
def fireplace_repl_interact():
|
|
buffer = ''
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
host = vim.eval('self.host')
|
|
port = int(vim.eval('self.port'))
|
|
s.settimeout(8)
|
|
try:
|
|
try:
|
|
s.connect((host, port))
|
|
s.setblocking(1)
|
|
s.sendall(vim.eval('payload'))
|
|
while True:
|
|
while len(select.select([s], [], [], 0.1)[0]) == 0:
|
|
vim.eval('getchar(1)')
|
|
body = s.recv(8192)
|
|
if re.search("=> $", body) != None:
|
|
raise Exception("not an nREPL server: upgrade to Leiningen 2")
|
|
buffer += body
|
|
if re.search('6:statusl(5:error|14:session-closed)?4:done', body):
|
|
break
|
|
fireplace_let('out', buffer)
|
|
except Exception, e:
|
|
fireplace_let('err', str(e))
|
|
finally:
|
|
s.close()
|
|
EOF
|
|
|
|
function! s:nrepl_call(payload) dict abort
|
|
let payload = nrepl#fireplace_connection#bencode(a:payload)
|
|
python << EOF
|
|
fireplace_repl_interact()
|
|
EOF
|
|
if !exists('err')
|
|
return nrepl#fireplace_connection#bdecode('l'.out.'e')
|
|
endif
|
|
throw 'nREPL Connection Error: '.err
|
|
endfunction
|
|
|
|
" vim:set et sw=2:
|