def on_select(self, item, cursor):
focus_window(self.last_window)
mark()
item = item[0]
if 'offset' in item:
line = vfunc.byte2line(item['offset'] + 1)
else:
line = item['line']
vim.command('normal! {}Gzz'.format(line))
def on_select(self, item, cursor):
focus_window(self.last_window)
mark()
item = item[0]
if 'offset' in item:
line = vfunc.byte2line(item['offset'] + 1)
else:
line = item['line']
vim.command('normal! {}Gzz'.format(line))
def escape():
if len(vim.windows) < 2:
return
cur = vfunc.winnr()
for n, w in reversed(list(enumerate(vim.windows, 1))):
if not buffer_with_file(w.buffer):
if not '[Command Line]'in w.buffer.name:
focus_window(n)
vim.command('q')
if n != cur:
if cur > n:
cur -= 1
focus_window(cur)
return
def escape():
if len(vim.windows) < 2:
return
cur = vfunc.winnr()
for n, w in reversed(list(enumerate(vim.windows, 1))):
if not buffer_with_file(w.buffer):
if not '[Command Line]' in w.buffer.name:
focus_window(n)
vim.command('q')
if n != cur:
if cur > n:
cur -= 1
focus_window(cur)
return
def execute(mode):
cbuf = vim.current.buffer
cwin = vim.current.window
if mode == 0: # paragraph
start = cbuf.mark('{')[0]
stop = cbuf.mark('}')[0]
elif mode == 1: # visual
start = max(0, cbuf.mark('<')[0] - 1)
stop = cbuf.mark('>')[0]
else: # whole buffer
start = 0
stop = len(cbuf)
input = '\n'.join(cbuf[start:stop])
executable = cbuf[0].lstrip('#! ')
with open('/tmp/vial-pipe-result.txt', 'wb') as f:
Popen(executable, shell=True, stderr=f,
stdout=f, stdin=PIPE).communicate(input)
make_scratch('vial-pipe', title='Result')
vim.command('norm! ggdG')
vim.command('0read /tmp/vial-pipe-result.txt')
focus_window(cwin)
def on_cancel(self):
focus_window(self.last_window)
def on_select(self, item, cursor):
focus_window(self.last_window)
mark()
vim.command('e {}'.format(item[2]))
def http():
lines = vim.current.buffer[:]
line, _ = vim.current.window.cursor
line -= 1
headers, templates = get_headers_and_templates(lines, line)
pwd_func = lambda p: vfunc.inputsecret('{}: '.format(p))
input_func = lambda p: vfunc.input('{}: '.format(p))
try:
method, url, query, body, tlist, rend = prepare_request(
lines, line, headers, input_func, pwd_func)
except PrepareException as e:
echoerr(str(e))
return
rctx = RequestContext()
rctx.request(method, url, query, body, headers)
cwin = vim.current.window
win, buf = make_scratch('__vial_http_req__', title='Request')
rlines = rctx.raw_request.splitlines()
hlines = []
for r in rctx.history[:-1]:
hlines.append(
bstr('Redirect {} from {}'.format(r.status, r.request[1]),
'utf-8'))
if hlines:
hlines.append(b'----------------')
buf[:] = hlines + rlines
win.cursor = 1, 0
win, buf = make_scratch('__vial_http_hdr__', title='Response headers')
if PY2:
buf[:] = [r.rstrip('\r\n') for r in rctx.response.msg.headers]
else:
buf[:] = [
'{}: {}'.format(*r).encode('utf-8')
for r in rctx.response.msg._headers
]
win.cursor = 1, 0
size = len(rctx.content)
win, buf = make_scratch('__vial_http_raw__', title='Raw Response')
buf[:] = rctx.content.splitlines()
win.cursor = 1, 0
if PY2:
rcontent_type = rctx.response.msg.gettype()
else:
rcontent_type = rctx.response.msg.get_content_type()
content, ctype, jdata = format_content(rcontent_type, rctx.content)
win, buf = make_scratch('__vial_http__')
win.options['statusline'] = 'Response: {} {} {}ms {}ms {}'.format(
rctx.response.status, rctx.response.reason, rctx.ctime, rctx.rtime,
sizeof_fmt(size))
vim.command('set filetype={}'.format(ctype))
buf[:] = content.splitlines(False)
win.cursor = 1, 0
focus_window(cwin)
def set_cookies(*args):
cookies = rctx.cookies
args = args or sorted(cookies.keys())
return 'Cookie: ' + ';'.join('{}={}'.format(k, cookies[k])
for k in args)
ctx = {
'body': content,
'json': jdata,
'headers': Headers(rctx.response.getheaders()),
'cookies': rctx.cookies,
'rcookies': rctx.rcookies,
'set_cookies': set_cookies
}
for t in tlist:
if t in templates:
lines = render_template(templates[t], **ctx).splitlines()
else:
lines = ['ERROR: template {} not found'.format(t)]
vfunc.append(rend + 1, [''] + lines)
rend += 1 + len(lines)
def http():
lines = vim.current.buffer[:]
line, _ = vim.current.window.cursor
line -= 1
headers, templates = get_headers_and_templates(lines, line)
pwd_func = lambda p: vfunc.inputsecret('{}: '.format(p))
input_func = lambda p: vfunc.input('{}: '.format(p))
try:
method, url, query, body, tlist, rend = prepare_request(lines, line, headers, input_func, pwd_func)
except PrepareException as e:
echoerr(str(e))
return
rctx = RequestContext()
rctx.request(method, url, query, body, headers)
cwin = vim.current.window
win, buf = make_scratch('__vial_http_req__', title='Request')
rlines = rctx.raw_request.splitlines()
hlines = []
for r in rctx.history[:-1]:
hlines.append(bstr('Redirect {} from {}'.format(
r.status, r.request[1]), 'utf-8'))
if hlines:
hlines.append(b'----------------')
buf[:] = hlines + rlines
win.cursor = 1, 0
win, buf = make_scratch('__vial_http_hdr__', title='Response headers')
if PY2:
buf[:] = [r.rstrip('\r\n') for r in rctx.response.msg.headers]
else:
buf[:] = ['{}: {}'.format(*r).encode('utf-8') for r in rctx.response.msg._headers]
win.cursor = 1, 0
size = len(rctx.content)
win, buf = make_scratch('__vial_http_raw__', title='Raw Response')
buf[:] = rctx.content.splitlines()
win.cursor = 1, 0
if PY2:
rcontent_type = rctx.response.msg.gettype()
else:
rcontent_type = rctx.response.msg.get_content_type()
content, ctype, jdata = format_content(rcontent_type, rctx.content)
win, buf = make_scratch('__vial_http__')
win.options['statusline'] = 'Response: {} {} {}ms {}ms {}'.format(
rctx.response.status, rctx.response.reason,
rctx.ctime, rctx.rtime, sizeof_fmt(size))
vim.command('set filetype={}'.format(ctype))
buf[:] = content.splitlines(False)
win.cursor = 1, 0
focus_window(cwin)
def set_cookies(*args):
cookies = rctx.cookies
args = args or sorted(cookies.keys())
return 'Cookie: ' + ';'.join('{}={}'.format(k, cookies[k]) for k in args)
ctx = {'body': content, 'json': jdata,
'headers': Headers(rctx.response.getheaders()),
'cookies': rctx.cookies,
'rcookies': rctx.rcookies,
'set_cookies': set_cookies}
for t in tlist:
if t in templates:
lines = render_template(templates[t], **ctx).splitlines()
else:
lines = ['ERROR: template {} not found'.format(t)]
vfunc.append(rend + 1, [''] + lines)
rend += 1 + len(lines)
def on_cancel(self):
focus_window(self.last_window)
def on_select(self, item, cursor):
focus_window(self.last_window)
mark()
vim.command('e {}'.format(item[2]))
def on_select(self, item, cursor):
focus_window(self.last_window)
mark()
vim.command('normal! {}Gzz^'.format(item[1]))
def on_select(self, item, cursor):
focus_window(self.last_window)
mark()
vim.command('normal! {}Gzz^'.format(item[1]))
def reset(self):
cwin = vim.current.window
_, self.buf = make_scratch('__vial_pytest__', self.init, 'pytest')
vim.command('normal! ggdG')
focus_window(cwin)
redraw()
def goto_file():
filename, line = vfunc.expand('<cWORD>').split(':')[:2]
for win in vim.windows:
if vfunc.buflisted(win.buffer.number):
focus_window(win)
vim.command('e +{} {}'.format(line, filename))
def http():
lines = vim.current.buffer[:]
line, _ = vim.current.window.cursor
line -= 1
headers, templates = get_headers_and_templates(lines, line)
pwd_func = lambda p: vfunc.inputsecret('{}: '.format(p))
input_func = lambda p: vfunc.input('{}: '.format(p))
try:
method, url, query, body, tlist, rend = prepare_request(
lines, line, headers, input_func, pwd_func)
except PrepareException as e:
echoerr(str(e))
return
u = urlparse.urlsplit(url)
if not u.hostname:
host = headers.pop('host', '')
if not host.startswith('http://') and not host.startswith('https://'):
host = 'http://' + host
u = urlparse.urlsplit(host + url)
path = u.path
if u.query:
path += '?' + u.query
if query:
path += ('&' if u.query else '?') + urllib.urlencode(query)
if u.scheme == 'https':
import ssl
cn = httplib.HTTPSConnection(u.hostname,
u.port or 443,
timeout=CONNECT_TIMEOUT,
context=ssl._create_unverified_context())
else:
cn = httplib.HTTPConnection(u.hostname,
u.port or 80,
timeout=CONNECT_TIMEOUT)
cn = send_collector(cn)
start = time.time()
cn.connect()
ctime = int((time.time() - start) * 1000)
cn.sock.settimeout(READ_TIMEOUT)
cn.request(method, path, body, headers)
response = cn.getresponse()
rtime = int((time.time() - start) * 1000)
cwin = vim.current.window
win, buf = make_scratch('__vial_http_req__', title='Request')
buf[:] = cn._sdata.splitlines()
win.cursor = 1, 0
win, buf = make_scratch('__vial_http_hdr__', title='Response headers')
if PY2:
buf[:] = [r.rstrip('\r\n') for r in response.msg.headers]
else:
buf[:] = [
'{}: {}'.format(*r).encode('utf-8') for r in response.msg._headers
]
win.cursor = 1, 0
content = response.read()
size = len(content)
win, buf = make_scratch('__vial_http_raw__', title='Raw Response')
buf[:] = content.splitlines()
win.cursor = 1, 0
if PY2:
rcontent_type = response.msg.gettype()
else:
rcontent_type = response.msg.get_content_type()
content, ctype, jdata = format_content(rcontent_type, content)
win, buf = make_scratch('__vial_http__')
win.options['statusline'] = 'Response: {} {} {}ms {}ms {}'.format(
response.status, response.reason, ctime, rtime, sizeof_fmt(size))
vim.command('set filetype={}'.format(ctype))
buf[:] = content.splitlines(False)
win.cursor = 1, 0
focus_window(cwin)
cj = Cookie.SimpleCookie()
if PY2:
cheaders = response.msg.getheaders('set-cookie')
else:
cheaders = response.msg.get_all('set-cookie')
for h in cheaders or []:
cj.load(h)
rcookies = {k: v.value for k, v in iteritems(cj)}
cookies = {k: v.coded_value for k, v in iteritems(cj)}
def set_cookies(*args):
args = args or sorted(cookies.keys())
return 'Cookie: ' + ';'.join('{}={}'.format(k, cookies[k])
for k in args)
ctx = {
'body': content,
'json': jdata,
'headers': Headers(response.getheaders()),
'cookies': cookies,
'rcookies': rcookies,
'set_cookies': set_cookies
}
for t in tlist:
if t in templates:
lines = render_template(templates[t], **ctx).splitlines()
else:
lines = ['ERROR: template {} not found'.format(t)]
vfunc.append(rend + 1, [''] + lines)
rend += 1 + len(lines)
