#!/usr/bin/python # # Server that will communicate with Vim through the netbeans interface. # Used by test_netbeans.vim. # # This requires Python 2.6 or later. from __future__ import print_function import socket import sys import time import threading import re try: # Python 3 import socketserver except ImportError: # Python 2 import SocketServer as socketserver class ThreadedTCPRequestHandler(socketserver.BaseRequestHandler): def process_msgs(self, msgbuf): # Process all the received netbeans commands/responses/events from Vim. # Each one is separated by a newline character. If a partial command # is received, process it later after the rest of it is received. while True: (line, sep, rest) = msgbuf.partition('\n') if sep == '': # received partial line return line msgbuf = rest # Process a command only after receiving a newline. response = '' if line.find('Xcmdbuf') > 0: name = line.split('"')[1] response = '1:putBufferNumber!15 "' + name + '"\n' response += '1:startDocumentListen!16\n' elif re.match('1:insert=.* "\\\\n"', line): # extract the command from the previous line cmd = re.search('.*"(.*)"', self.prev_line).group(1) testmap = { 'getCursor_Test' : '0:getCursor/30\n', 'E627_Test' : '0 setReadOnly!31\n', 'E628_Test' : '0:setReadOnly 32\n', 'E632_Test' : '0:getLength/33\n', 'E633_Test' : '0:getText/34\n', 'E634_Test' : '0:remove/35 1 1\n', 'E635_Test' : '0:insert/36 0 "line1\\n"\n', 'E636_Test' : '0:create!37\n', 'E637_Test' : '0:startDocumentListen!38\n', 'E638_Test' : '0:stopDocumentListen!39\n', 'E639_Test' : '0:setTitle!40 "Title"\n', 'E640_Test' : '0:initDone!41\n', 'E641_Test' : '0:putBufferNumber!42 "XSomeBuf"\n', 'E642_Test' : '9:putBufferNumber!43 "XInvalidBuf"\n', 'E643_Test' : '0:setFullName!44 "XSomeBuf"\n', 'E644_Test' : '0:editFile!45 "Xfile3"\n', 'E645_Test' : '0:setVisible!46 T\n', 'E646_Test' : '0:setModified!47 T\n', 'E647_Test' : '0:setDot!48 1/1\n', 'E648_Test' : '0:close!49\n', 'E650_Test' : '0:defineAnnoType!50 1 "abc" "a" "a" 1 1\n', 'E651_Test' : '0:addAnno!51 1 1 1 1\n', 'E652_Test' : '0:getAnno/52 8\n', 'editFile_Test' : '2:editFile!53 "Xfile3"\n', 'getLength_Test' : '2:getLength/54\n', 'getModified_Test' : '2:getModified/55\n', 'getText_Test' : '2:getText/56\n', 'setDot_Test' : '2:setDot!57 3/6\n', 'startDocumentListen_Test' : '2:startDocumentListen!58\n', 'stopDocumentListen_Test' : '2:stopDocumentListen!59\n', 'define_anno_Test' : '2:defineAnnoType!60 1 "s1" "x" "=>" blue none\n', 'E532_Test' : '2:defineAnnoType!61 1 "s1" "x" "=>" aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa none\n', 'add_anno_Test' : '2:addAnno!62 1 1 2/1 0\n', 'get_anno_Test' : '2:getAnno/63 1\n', 'remove_anno_Test' : '2:removeAnno!64 1\n', 'getModifiedAll_Test' : '0:getModified/65\n', 'create_Test' : '3:create!66\n', 'setTitle_Test' : '3:setTitle!67 "Xfile4"\n', 'setFullName_Test' : '3:setFullName!68 "Xfile4"\n', 'initDone_Test' : '3:initDone!69\n', 'setVisible_Test' : '3:setVisible!70 T\n', 'setModtime_Test' : '3:setModtime!71 6\n', 'insert_Test' : '3:insert/72 0 "line1\\nline2\\n"\n', 'remove_Test' : '3:remove/73 3 4\n', 'remove_invalid_offset_Test' : '3:remove/74 900 4\n', 'remove_invalid_count_Test' : '3:remove/75 1 800\n', 'guard_Test' : '3:guard!76 8 7\n', 'setModified_Test' : '3:setModified!77 T\n', 'insertDone_Test' : '3:insertDone!78 T F\n', 'saveDone_Test' : '3:saveDone!79\n', 'invalidcmd_Test' : '3:invalidcmd!80\n', 'invalidfunc_Test' : '3:invalidfunc/81\n', 'removeAnno_fail_Test' : '0:removeAnno/82 1\n', 'guard_fail_Test' : '0:guard/83 1 1\n', 'save_fail_Test' : '0:save/84\n', 'netbeansBuffer_fail_Test' : '0:netbeansBuffer/85 T\n', 'setExitDelay_Test' : '0:setExitDelay!86 2\n', 'setReadOnly_Test' : '3:setReadOnly!87\n', 'close_Test' : '3:close!88\n', 'specialKeys_Test' : '0:specialKeys!89 "F12 F13"\n', 'nbbufwrite_Test' : '4:editFile!90 "XnbBuffer"\n4:netbeansBuffer!91 T\n', 'detach_Test' : '2:close!92\n1:close!93\nDETACH\n' } # execute the specified test if cmd not in testmap: print("=== invalid command %s ===" % (cmd)) else: response = testmap[cmd] elif line.find('disconnect') > 0: # we're done self.server.shutdown() return # save the current line, this is used as the test to run after # receiving a newline only line. self.prev_line = line if len(response) > 0: self.request.sendall(response.encode('utf-8')) # Write the response into the file, so that the test can knows # the command was sent. with open("Xnetbeans", "a") as myfile: myfile.write('send: ' + response) if self.debug: with open("save_Xnetbeans", "a") as myfile: myfile.write('send: ' + response) def handle(self): print("=== socket opened ===") # To preserve the Xnetbeans file as save_Xnetbeans, set debug to 1 self.debug = 0 self.prev_line = '' msgbuf = '' while True: try: received = self.request.recv(4096).decode('utf-8') except socket.error: print("=== socket error ===") break except IOError: print("=== socket closed ===") break if received == '': print("=== socket closed ===") break print("received: {0}".format(received)) # Write the received line into the file, so that the test can check # what happened. with open("Xnetbeans", "a") as myfile: myfile.write(received) if self.debug: with open("save_Xnetbeans", "a") as myfile: myfile.write(received) # Can receive more than one line in a response or a partial line. # Accumulate all the received characters and process one line at # a time. msgbuf += received msgbuf = self.process_msgs(msgbuf) class ThreadedTCPServer(socketserver.ThreadingMixIn, socketserver.TCPServer): pass def writePortInFile(port): # Write the port number in Xportnr, so that the test knows it. f = open("Xportnr", "w") f.write("{0}".format(port)) f.close() if __name__ == "__main__": HOST, PORT = "localhost", 0 server = ThreadedTCPServer((HOST, PORT), ThreadedTCPRequestHandler) ip, port = server.server_address # Start a thread with the server. That thread will then start a new thread # for each connection. server_thread = threading.Thread(target=server.serve_forever) server_thread.start() writePortInFile(port) print("Listening on port {0}".format(port)) # Main thread terminates, but the server continues running # until server.shutdown() is called. try: while server_thread.isAlive(): server_thread.join(1) except (KeyboardInterrupt, SystemExit): server.shutdown()