1#!/usr/bin/python 2# 3# Server that will communicate with Vim through the netbeans interface. 4# Used by test_netbeans.vim. 5# 6# This requires Python 2.6 or later. 7 8from __future__ import print_function 9import socket 10import sys 11import time 12import threading 13import re 14 15try: 16 # Python 3 17 import socketserver 18except ImportError: 19 # Python 2 20 import SocketServer as socketserver 21 22class ThreadedTCPRequestHandler(socketserver.BaseRequestHandler): 23 24 def process_msgs(self, msgbuf): 25 # Process all the received netbeans commands/responses/events from Vim. 26 # Each one is separated by a newline character. If a partial command 27 # is received, process it later after the rest of it is received. 28 while True: 29 (line, sep, rest) = msgbuf.partition('\n') 30 if sep == '': 31 # received partial line 32 return line 33 msgbuf = rest 34 35 # Process a command only after receiving a newline. 36 response = '' 37 if line.find('Xcmdbuf') > 0: 38 name = line.split('"')[1] 39 response = '1:putBufferNumber!15 "' + name + '"\n' 40 response += '1:startDocumentListen!16\n' 41 elif re.match('1:insert=.* "\\\\n"', line): 42 # extract the command from the previous line 43 cmd = re.search('.*"(.*)"', self.prev_line).group(1) 44 testmap = { 45 'getCursor_Test' : '0:getCursor/30\n', 46 'E627_Test' : '0 setReadOnly!31\n', 47 'E628_Test' : '0:setReadOnly 32\n', 48 'E632_Test' : '0:getLength/33\n', 49 'E633_Test' : '0:getText/34\n', 50 'E634_Test' : '0:remove/35 1 1\n', 51 'E635_Test' : '0:insert/36 0 "line1\\n"\n', 52 'E636_Test' : '0:create!37\n', 53 'E637_Test' : '0:startDocumentListen!38\n', 54 'E638_Test' : '0:stopDocumentListen!39\n', 55 'E639_Test' : '0:setTitle!40 "Title"\n', 56 'E640_Test' : '0:initDone!41\n', 57 'E641_Test' : '0:putBufferNumber!42 "XSomeBuf"\n', 58 'E642_Test' : '9:putBufferNumber!43 "XInvalidBuf"\n', 59 'E643_Test' : '0:setFullName!44 "XSomeBuf"\n', 60 'E644_Test' : '0:editFile!45 "Xfile3"\n', 61 'E645_Test' : '0:setVisible!46 T\n', 62 'E646_Test' : '0:setModified!47 T\n', 63 'E647_Test' : '0:setDot!48 1/1\n', 64 'E648_Test' : '0:close!49\n', 65 'E650_Test' : '0:defineAnnoType!50 1 "abc" "a" "a" 1 1\n', 66 'E651_Test' : '0:addAnno!51 1 1 1 1\n', 67 'E652_Test' : '0:getAnno/52 8\n', 68 'editFile_Test' : '2:editFile!53 "Xfile3"\n', 69 'getLength_Test' : '2:getLength/54\n', 70 'getModified_Test' : '2:getModified/55\n', 71 'getText_Test' : '2:getText/56\n', 72 'setDot_Test' : '2:setDot!57 3/6\n', 73 'startDocumentListen_Test' : '2:startDocumentListen!58\n', 74 'stopDocumentListen_Test' : '2:stopDocumentListen!59\n', 75 'define_anno_Test' : '2:defineAnnoType!60 1 "s1" "x" "=>" blue none\n', 76 'E532_Test' : '2:defineAnnoType!61 1 "s1" "x" "=>" aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa none\n', 77 'add_anno_Test' : '2:addAnno!62 1 1 2/1 0\n', 78 'get_anno_Test' : '2:getAnno/63 1\n', 79 'remove_anno_Test' : '2:removeAnno!64 1\n', 80 'getModifiedAll_Test' : '0:getModified/65\n', 81 'create_Test' : '3:create!66\n', 82 'setTitle_Test' : '3:setTitle!67 "Xfile4"\n', 83 'setFullName_Test' : '3:setFullName!68 "Xfile4"\n', 84 'initDone_Test' : '3:initDone!69\n', 85 'setVisible_Test' : '3:setVisible!70 T\n', 86 'setModtime_Test' : '3:setModtime!71 6\n', 87 'insert_Test' : '3:insert/72 0 "line1\\nline2\\n"\n', 88 'remove_Test' : '3:remove/73 3 4\n', 89 'remove_invalid_offset_Test' : '3:remove/74 900 4\n', 90 'remove_invalid_count_Test' : '3:remove/75 1 800\n', 91 'guard_Test' : '3:guard!76 8 7\n', 92 'setModified_Test' : '3:setModified!77 T\n', 93 'insertDone_Test' : '3:insertDone!78 T F\n', 94 'saveDone_Test' : '3:saveDone!79\n', 95 'invalidcmd_Test' : '3:invalidcmd!80\n', 96 'invalidfunc_Test' : '3:invalidfunc/81\n', 97 'removeAnno_fail_Test' : '0:removeAnno/82 1\n', 98 'guard_fail_Test' : '0:guard/83 1 1\n', 99 'save_fail_Test' : '0:save/84\n', 100 'netbeansBuffer_fail_Test' : '0:netbeansBuffer/85 T\n', 101 'setExitDelay_Test' : '0:setExitDelay!86 2\n', 102 'setReadOnly_Test' : '3:setReadOnly!87\n', 103 'close_Test' : '3:close!88\n', 104 'specialKeys_Test' : '0:specialKeys!89 "F12 F13"\n', 105 'nbbufwrite_Test' : '4:editFile!90 "XnbBuffer"\n4:netbeansBuffer!91 T\n', 106 'detach_Test' : '2:close!92\n1:close!93\nDETACH\n' 107 } 108 # execute the specified test 109 if cmd not in testmap: 110 print("=== invalid command %s ===" % (cmd)) 111 else: 112 response = testmap[cmd] 113 elif line.find('disconnect') > 0: 114 # we're done 115 self.server.shutdown() 116 return 117 118 # save the current line, this is used as the test to run after 119 # receiving a newline only line. 120 self.prev_line = line 121 122 if len(response) > 0: 123 self.request.sendall(response.encode('utf-8')) 124 # Write the response into the file, so that the test can knows 125 # the command was sent. 126 with open("Xnetbeans", "a") as myfile: 127 myfile.write('send: ' + response) 128 if self.debug: 129 with open("save_Xnetbeans", "a") as myfile: 130 myfile.write('send: ' + response) 131 132 def handle(self): 133 print("=== socket opened ===") 134 # To preserve the Xnetbeans file as save_Xnetbeans, set debug to 1 135 self.debug = 0 136 self.prev_line = '' 137 msgbuf = '' 138 while True: 139 try: 140 received = self.request.recv(4096).decode('utf-8') 141 except socket.error: 142 print("=== socket error ===") 143 break 144 except IOError: 145 print("=== socket closed ===") 146 break 147 if received == '': 148 print("=== socket closed ===") 149 break 150 print("received: {0}".format(received)) 151 152 # Write the received line into the file, so that the test can check 153 # what happened. 154 with open("Xnetbeans", "a") as myfile: 155 myfile.write(received) 156 if self.debug: 157 with open("save_Xnetbeans", "a") as myfile: 158 myfile.write(received) 159 160 # Can receive more than one line in a response or a partial line. 161 # Accumulate all the received characters and process one line at 162 # a time. 163 msgbuf += received 164 msgbuf = self.process_msgs(msgbuf) 165 166class ThreadedTCPServer(socketserver.ThreadingMixIn, socketserver.TCPServer): 167 pass 168 169def writePortInFile(port): 170 # Write the port number in Xportnr, so that the test knows it. 171 f = open("Xportnr", "w") 172 f.write("{0}".format(port)) 173 f.close() 174 175if __name__ == "__main__": 176 HOST, PORT = "localhost", 0 177 178 server = ThreadedTCPServer((HOST, PORT), ThreadedTCPRequestHandler) 179 ip, port = server.server_address 180 181 # Start a thread with the server. That thread will then start a new thread 182 # for each connection. 183 server_thread = threading.Thread(target=server.serve_forever) 184 server_thread.start() 185 186 writePortInFile(port) 187 188 print("Listening on port {0}".format(port)) 189 190 # Main thread terminates, but the server continues running 191 # until server.shutdown() is called. 192 try: 193 while server_thread.isAlive(): 194 server_thread.join(1) 195 except (KeyboardInterrupt, SystemExit): 196 server.shutdown() 197