X-Git-Url: https://plomlompom.com/repos/?p=plomrogue2-experiments;a=blobdiff_plain;f=new2%2Frogue_chat_curses.py;fp=new2%2Frogue_chat_curses.py;h=c8a17c888388dcfb9317896a7d830ac2bd00b9cf;hp=15d0053a8bca770f385aa9e8a70a2f707f415c55;hb=478d293913c37ed1bc98ab65db9658c58d6f7081;hpb=621e33482140c399f883b6e858f45e9c04471729

diff --git a/new2/rogue_chat_curses.py b/new2/rogue_chat_curses.py
index 15d0053..c8a17c8 100755
--- a/new2/rogue_chat_curses.py
+++ b/new2/rogue_chat_curses.py
@@ -39,8 +39,12 @@ class PlomSocketClient(PlomSocket):
         self.socket.close()
 
     def run(self):
+        import ssl
         try:
             for msg in self.recv():
+                if msg == 'NEED_SSL':
+                    self.socket = ssl.wrap_socket(self.socket)
+                    continue
                 self.recv_handler(msg)
         except BrokenSocketConnection:
             pass  # we assume socket will be known as dead by now
@@ -606,4 +610,4 @@ class TUI:
                 self.send('TASK:WRITE ' + key)
                 self.switch_mode('play')
 
-TUI('127.0.0.1:5000')
+TUI('localhost:5000')