astyle: Added client/server

tools/astyle_client.py

@@ -0,0 +1,62 @@
+import glob
+import os
+import socket
+import time
+
+def receive_data(conn):
+    data = ''
+    for t in range(1000):
+        d = conn.recv(8196)
+        if d:
+            data += d
+            if data.endswith('\nDONE'):
+                return data[:-5]
+        time.sleep(0.01)
+    return ''
+
+
+def astyle(server_address, code):
+    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    try:
+        sock.connect(server_address)
+        sock.sendall(code + '\nDONE')
+        return receive_data(sock)
+    except socket.error as err:
+        print('Network error: ' + str(err))
+    sock.close()
+    return None
+
+
+def get_source_files(path):
+    files = []
+    for g in glob.glob(path + '*'):
+        if g.startswith('.'):
+            continue
+        if os.path.isdir(g):
+            files += get_source_files(g + '/')
+        if os.path.isfile(g) and (g.endswith('.cpp') or g.endswith('.h')):
+            files.append(g)
+    return files
+
+if __name__ == "__main__":
+    server_address = ('localhost', 18000)
+
+    source_files = []
+    for d in ['cli', 'gui', 'lib', 'test', 'tools']:
+        source_files += get_source_files(d + '/')
+
+    for filename in source_files:
+        f = open(filename, 'rt')
+        code = f.read()
+        f.close()
+        formatted_code = astyle(server_address, code)
+        if formatted_code is None:
+            break
+        if code != formatted_code:
+            print('Changed: ' + filename)
+            f = open(filename, 'wt')
+            f.write(formatted_code)
+            f.close()
+        else:
+            print('Unchanged: ' + filename)
+

tools/astyle_server.py

@@ -0,0 +1,44 @@
+
+import logging
+import socket
+import subprocess
+
+from astyle_client import receive_data
+
+logger = logging.getLogger()
+
+
+def server(port):
+    socket.setdefaulttimeout(30)
+    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+    server_address = ('', port)
+    sock.bind(server_address)
+
+    sock.listen(1)
+
+    while True:
+        # Wait for a connection
+        connection, client_address = sock.accept()
+
+        # Read data from client
+        try:
+            data = receive_data(connection)
+        except socket.error:
+            connection.close()
+            continue
+
+        # Format
+        process = subprocess.Popen(['astyle', '--options=../.astylerc'],
+                                   stdin=subprocess.PIPE,
+                                   stdout=subprocess.PIPE,
+                                   stderr=subprocess.PIPE)
+        comm = process.communicate(input=data)
+
+        connection.sendall(comm[0] + '\nDONE')
+        connection.close()
+
+
+if __name__ == "__main__":
+    server(port=18000)
+