Add the distbuild libs

1 files changed, 45 insertions, 0 deletions

diff --git a/distbuild/sockserv.py b/distbuild/sockserv.py
new file mode 100644
index 00000000..6d8f216e
--- /dev/null
+++ b/distbuild/sockserv.py
@@ -0,0 +1,45 @@
+# mainloop/sockserv.py -- socket server state machines
+#
+# Copyright 2012 Codethink Limited
+# All rights reserved.
+
+
+import logging
+
+from sm import StateMachine
+from socketsrc import NewConnection, SocketError, ListeningSocketEventSource
+
+
+class ListenServer(StateMachine):
+
+    '''Listen for new connections on a port, send events for them.'''
+
+    def __init__(self, addr, port, machine, extra_args=None):
+        StateMachine.__init__(self, 'listening')
+        self._addr = addr
+        self._port = port
+        self._machine = machine
+        self._extra_args = extra_args or []
+        
+    def setup(self):
+        src = ListeningSocketEventSource(self._addr, self._port)
+        self.mainloop.add_event_source(src)
+
+        spec = [
+            ('listening', src, NewConnection, 'listening', self.new_conn),
+            ('listening', src, SocketError, None, self.report_error),
+        ]
+        self.add_transitions(spec)
+
+    def new_conn(self, event_source, event):
+        logging.debug(
+            'ListenServer: Creating new %s using %s and %s' %
+                (repr(self._machine),
+                 repr(event.connection),
+                 repr(self._extra_args)))
+        m = self._machine(event.connection, *self._extra_args)
+        self.mainloop.add_state_machine(m)
+
+    def report_error(self, event_source, event):
+        logging.error(str(event))
+