add ssl support

1 files changed, 6 insertions, 3 deletions

diff --git a/irc.py b/irc.py
index 69cfe44..3fa579e 100755
--- a/irc.py
+++ b/irc.py
@@ -73,15 +73,18 @@ class Bot(asynchat.async_chat):
             self.__write(args, text)
         except Exception, e: pass
 
-    def run(self, host, port=6667):
-        self.initiate_connect(host, port)
+    def run(self, host, port=6667, ssl=False):
+        self.initiate_connect(host, port, ssl)
 
-    def initiate_connect(self, host, port):
+    def initiate_connect(self, host, port, ssl):
         if self.verbose:
             message = 'Connecting to %s:%s...' % (host, port)
             print >> sys.stderr, message,
         self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
         self.connect((host, port))
+        if ssl:
+            import ssl
+            self.socket = ssl.wrap_socket(self.socket)
         try: asyncore.loop()
         except KeyboardInterrupt:
             sys.exit()