import base64 import vim import sys import socket import json class CommError(Exception): pass def escape(s): return s.replace("'", "''") def communicate(data): global PAPPY_PORT # Submits data to the comm port of the proxy s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) s.connect(('127.0.0.1', int(vim.eval('s:commport')))) datastr = json.dumps(data) # Send our data total_sent = 0 while total_sent < len(data): sent = s.send(datastr[total_sent:]) assert sent != 0 total_sent += sent s.send('\n') # Get our response retstr = '' c = '' while c != '\n': retstr = retstr + c c = s.recv(1) assert c != '' result = json.loads(retstr) if not result['success']: vim.command("echoerr '%s'" % escape(result['message'])) raise CommError(result['message']) return result def read_line(conn): data = '' c = '' while c != '\n': data = data + c c = conn.read(1) return data def run_command(command): funcs = { "setup": set_up_windows, "submit": submit_current_buffer, } if command in funcs: funcs[command]() def set_buffer_content(buf, text): buf[:] = None first = True for l in text.split('\n'): if first: buf[0] = l first = False else: buf.append(l) def set_up_windows(): reqid = vim.eval("a:2") comm_port = vim.eval("a:3") vim.command("let s:commport=%d"%int(comm_port)) # Get the left buffer vim.command("new") vim.command("only") b2 = vim.current.buffer vim.command("let s:b2=bufnr('$')") # Vsplit new file vim.command("vnew") b1 = vim.current.buffer vim.command("let s:b1=bufnr('$')") # Get the request comm_data = {"action": "get_request", "reqid": reqid} try: reqdata = communicate(comm_data) except CommError: return comm_data = {"action": "get_response", "reqid": reqid} try: rspdata = communicate(comm_data) except CommError: return # Set up the buffers set_buffer_content(b1, base64.b64decode(reqdata['full_message'])) if 'full_message' in rspdata: set_buffer_content(b2, base64.b64decode(rspdata['full_message'])) # Save the port, ssl, host setting vim.command("let s:repport=%d" % int(reqdata['port'])) vim.command("let s:rephost='%s'" % escape(reqdata['host'])) if reqdata['is_ssl']: vim.command("let s:repisssl=1") else: vim.command("let s:repisssl=0") def submit_current_buffer(): curbuf = vim.current.buffer b2_id = int(vim.eval("s:b2")) b2 = vim.buffers[b2_id] vim.command("let s:b1=bufnr('$')") vim.command("only") vim.command("rightbelow vertical new") vim.command("b %d" % b2_id) vim.command("wincmd h") full_request = '\n'.join(curbuf) commdata = {'action': 'submit', 'full_message': base64.b64encode(full_request), 'tags': ['repeater'], 'port': int(vim.eval("s:repport")), 'host': vim.eval("s:rephost")} if vim.eval("s:repisssl") == '1': commdata["is_ssl"] = True else: commdata["is_ssl"] = False result = communicate(commdata) set_buffer_content(b2, base64.b64decode(result['response']['full_message'])) # (left, right) = set_up_windows() # set_buffer_content(left, 'Hello\nWorld') # set_buffer_content(right, 'Hello\nOther\nWorld') #print "Arg is %s" % vim.eval("a:arg") run_command(vim.eval("a:1"))