From f1e66558e478e84b4a4a3d8bf39464093d9a4876 Mon Sep 17 00:00:00 2001
From: Mattias Andrée <maandree@operamail.com>
Date: Mon, 27 Aug 2012 01:26:03 +0200
Subject: message can coloured

---
 ponysay | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

(limited to 'ponysay')

diff --git a/ponysay b/ponysay
index 45e1d33..b0bc98b 100755
--- a/ponysay
+++ b/ponysay
@@ -1371,9 +1371,23 @@ class Backend():
         msg = self.message
         if wrap is not None:
             msg = self.__wrapMessage(msg, wrap)
-        lines = msg.split('\n')
         
-        return self.balloon.get(width, height, lines, self.__len)
+        if '\033' in msg:
+            AUTO_PUSH = '\033[01010~'
+            AUTO_POP  = '\033[10101~'
+            cstack = ColourStack(AUTO_PUSH, AUTO_POP)
+            buf = ''
+            for c in msg:
+                if c == '\n':
+                    for cc in ('%s\n%s' % (AUTO_PUSH, AUTO_POP)):
+                        buf += cc
+                        buf += cstack.feed(cc)
+                else:
+                    buf += c
+                    buf += cstack.feed(c)
+            msg = buf
+        
+        return self.balloon.get(width, height, msg.split('\n'), self.__len)
     
     
     '''
-- 
cgit