From 5fa8938b957e38a930934d615e0f803e8c5a5336 Mon Sep 17 00:00:00 2001
From: Mattias Andrée <maandree@operamail.com>
Date: Tue, 21 Aug 2012 02:53:01 +0200
Subject: work on colour stack + make fix

---
 ponysay | 23 +++++++++++++++++++++--
 1 file changed, 21 insertions(+), 2 deletions(-)

(limited to 'ponysay')

diff --git a/ponysay b/ponysay
index db5c4a5..3a4efa3 100755
--- a/ponysay
+++ b/ponysay
@@ -1085,16 +1085,35 @@ class ColourStack():
     Constructor
     '''
     def __init__(self, autopush, autopop):
-        pass
+        self.autopush = autopush
+        self.autopop = autopop
+        self.lenpush = len(autopush)
+        self.lenpop = len(autopop)
+        self.bufproto = ' ' * (self.lenpush if self.lenpush > self.lenpop else self.lenpop)
+        self.stack = []
+        self.push()
+    
     
     def push(self):
+        self.stack = [[self.bufproto]] + self.stack
+        if len(self.stack) == 1:
+            return ''
         return ''
     
+    
     def pop(self):
         return ''
     
+    
     def feed(self, char):
-        return ''
+        buf = self.stack[0][0]
+        rc = ''
+        if buf[:-self.lenpush] == self.autopush:
+            rc = self.push()
+        elif buf[:-self.lenpop] == self.autopop:
+            rc = self.pop()
+        self.stack[0][0] = buf[1:] + char
+        return rc
 
 
 
-- 
cgit