From 5c53fd79dfb9cc32ad6009a496db49b0bc1da0e6 Mon Sep 17 00:00:00 2001
From: Jan Philipp Timme <jan.philipp@timme.it>
Date: Sat, 24 Oct 2015 18:12:02 +0200
Subject: [PATCH] [TASK] Initial import.

---
 .gitignore |   1 +
 bot.py     | 116 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 117 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 bot.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..0c4c8b5
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+.spyderproject
diff --git a/bot.py b/bot.py
new file mode 100644
index 0000000..363caf6
--- /dev/null
+++ b/bot.py
@@ -0,0 +1,116 @@
+# -*- coding: utf-8 -*-
+
+import asyncio
+import inspect
+import logging
+
+logging.basicConfig(format='[%(asctime)s] [%(levelname)s] %(message)s', level=logging.DEBUG, datefmt='%d.%m.%Y %H:%M:%S')
+logger = logging.getLogger(__name__)
+
+
+class ManagedProtocol(asyncio.Protocol):
+    """Basic managed protocol handler, registers itself to ConnectionManager.
+    """
+
+    def __init__(self, loop, connection_manager, endpoint):
+        self._loop = loop
+        self._connection_manager = connection_manager
+        self._endpoint = endpoint
+        self._transport = None
+
+    def _log(self, msg):
+        host, port = self._endpoint
+        logger.info("[{}:{}] ".format(host, port)+str(msg))
+
+    def connection_made(self, transport):
+        self._connection_manager.register_active_connection(self._endpoint, self)
+        self._transport = transport
+        self._log("Connection made!")
+        host, port = transport.get_extra_info('peername')
+        self._log('Connected to: {}:{}'.format(host, port))
+
+    def data_received(self, data):
+        self._log(data)
+        # TODO: parse and call more methodes / on itself for events or something
+
+    def eof_received(self):
+        self._log("Eof received!")
+
+    def connection_lost(self, exc):
+        self._log("Connection lost!")
+        self._connection_manager.unregister_active_connection(self._endpoint)
+
+    def destroy(self):
+        """ Triggered by ConnectionManager.remove_endpoint(). Closes transport. """
+        self._transport.close()
+
+
+class IrcProtocol(ManagedProtocol):
+    def __init__(self):
+        pass
+
+
+class ConnectionManager(object):
+    """Takes care of known endpoints that a connections shall be established to.
+    """
+
+    def __init__(self, loop):
+        self._loop = loop
+        self._endpoints = []
+        self._active_connections = {}
+        self._loop.set_exception_handler(self._handle_async_exception)
+
+    def add_endpoint(self, endpoint):
+        logger.debug("Endpoint added: {}:{}".format(endpoint[0],endpoint[1]))
+        self._endpoints.append(endpoint)
+        self._create_connection(endpoint)
+
+    def _create_connection(self, endpoint):
+        protocol = ManagedProtocol(self._loop, self, endpoint)
+        coroutine = self._loop.create_connection(lambda: protocol, endpoint[0], endpoint[1])
+        asyncio.async(coroutine)
+
+    def remove_endpoint(self, endpoint):
+        logger.debug("Endpoint removed: {}:{}".format(endpoint[0],endpoint[1]))
+        self._endpoints.remove(endpoint)
+        if endpoint in self._active_connections:
+            self._active_connections[endpoint].close()
+
+    def register_active_connection(self, endpoint, protocol):
+        self._active_connections[endpoint] = protocol
+
+    def unregister_active_connection(self, endpoint):
+        del self._active_connections[endpoint]
+        self._create_connection(endpoint)
+
+    def _handle_async_exception(self, loop, context):
+        """Trying to take care of connection related exceptions."""
+        logger.error("An async exception has been caught: "+str(context['exception']))
+        stack = context['future'].get_stack()
+        if len(stack) > 1 and stack[1].f_code.co_name == 'create_connection':
+            last_stackframe = stack[len(stack)-1]
+            call_args = inspect.getargvalues(last_stackframe)
+            host = call_args.locals['host']
+            port = call_args.locals['port']
+            logger.error("Bad endpoint: {}:{}".format(host, port))
+            self.remove_endpoint((host, port))
+            return True
+        return False
+
+
+if __name__ == '__main__':
+    freenode = ("irc.freenode.net", 6667)
+    euirc = ("irc.esduirc.net", 6267)
+
+    loop = asyncio.get_event_loop()
+
+    connection_manager = ConnectionManager(loop)
+    connection_manager.add_endpoint(euirc)
+    #connection_manager.add_endpoint(freenode)
+
+    try:
+        loop.run_forever()
+    except KeyboardInterrupt:
+        pass
+    finally:
+        loop.close()