import base64 import vim import sys import socket import json class CommError(Exception): pass 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' % 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_request'])) set_buffer_content(b2, base64.b64decode(rspdata['full_response'])) def submit_current_buffer(): curbuf = vim.current.buffer b2_id = vim.eval("s:b2") b2 = vim.buffers[int(b2_id)] vim.command("let s:b1=bufnr('$')") vim.command("only") vim.command("rightbelow vertical new") vim.command("b %s" % b2_id) vim.command("wincmd h") full_request = '\n'.join(curbuf) commdata = {'action': 'submit', 'full_request': base64.b64encode(full_request)} result = communicate(commdata) set_buffer_content(b2, base64.b64decode(result['response']['full_response'])) # (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"))