1 files changed, 185 insertions, 0 deletions

diff --git a/presentty/ansiparser.py b/presentty/ansiparser.py
new file mode 100644
index 0000000..a780b42
--- /dev/null
+++ b/presentty/ansiparser.py
@@ -0,0 +1,185 @@
+# Copyright (C) 2015 James E. Blair <corvus@gnu.org>
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import re
+
+import urwid
+
+class ANSIParser(object):
+    colors = [
+        urwid.BLACK,
+        urwid.DARK_RED,
+        urwid.DARK_GREEN,
+        urwid.BROWN,
+        urwid.DARK_BLUE,
+        urwid.DARK_MAGENTA,
+        urwid.DARK_CYAN,
+        urwid.LIGHT_GRAY,
+        urwid.DARK_GRAY,
+        urwid.LIGHT_RED,
+        urwid.LIGHT_GREEN,
+        urwid.YELLOW,
+        urwid.LIGHT_BLUE,
+        urwid.LIGHT_MAGENTA,
+        urwid.LIGHT_CYAN,
+        urwid.WHITE,
+        ]
+
+    colors256 = ['0', '6', '8', 'a', 'd', 'f']
+    colorsgray = ['3', '7', '11', '13', '15', '19', '23', '27', '31',
+                  '35', '38', '42', '46', '50', '52', '58', '62', '66',
+                  '70', '74', '78', '82', '85', '89', '93']
+
+    def __init__(self):
+        self.x = 0
+        self.y = 0
+        self.text_lines = []
+        self.attr_lines = []
+        self.background = urwid.AttrSpec('light gray', 'black')
+        self.attr = self.background
+        self.resetColor()
+        self.moveTo(0,0)
+
+    def resetColor(self):
+        self.bold = False
+        self.blink = False
+        self.fg = 7
+        self.bg = 0
+
+    def moveTo(self, x, y):
+        while x>80:
+            x-=80
+            y+=1
+        while y+1 > len(self.text_lines):
+            self.text_lines.append([u' ' for i in range(80)])
+            self.attr_lines.append([self.attr for i in range(80)])
+        self.x = x
+        self.y = y
+
+    def parseSequence(self, seq):
+        values = []
+        buf = ''
+        for c in seq:
+            if c in ['\x1b', '[']:
+                continue
+            if c == ';':
+                values.append(int(buf))
+                buf = ''
+                continue
+            if ord(c) < 64:
+                buf += c
+        if buf:
+            values.append(int(buf))
+        if c == 'm':
+            if not values:
+                values = [0]
+            fg256 = None
+            for v in values:
+                if fg256 is True:
+                    if v <= 0x08:
+                        self.fg = v
+                    elif v <= 0x0f:
+                        self.fg = v - 0x08
+                        self.bold = True
+                    elif v <= 0xe7:
+                        r, x = divmod(v-16, 36)
+                        g, x = divmod(x, 6)
+                        b = x % 6
+                        fg256 = ('#' +
+                                 self.colors256[r] +
+                                 self.colors256[g] +
+                                 self.colors256[b])
+                    else:
+                        fg256 = 'g' + str(self.colorsgray[v-232])
+                elif v == 0:
+                    self.resetColor()
+                elif v == 1:
+                    self.bold = True
+                elif v == 5:
+                    self.blink = True
+                elif v>29 and v<38:
+                    self.fg = v-30
+                elif v>39 and v<48:
+                    self.bg = v-40
+                elif v==38:
+                    fg256=True
+            fg = self.fg
+            if self.bold:
+                fg += 8
+            fgattrs = []
+            if self.blink:
+                fgattrs.append('blink')
+            if fg256:
+                fgattrs.append(fg256)
+            else:
+                fgattrs.append(self.colors[fg])
+            self.attr = urwid.AttrSpec(', '.join(fgattrs), self.colors[self.bg])
+        if c == 'A':
+            if not values:
+                values = [1]
+            y = max(self.y-values[0], 0)
+            self.moveTo(self.x, y)
+        if c == 'C':
+            if not values:
+                values = [1]
+            x = self.x + values[0]
+            self.moveTo(x, self.y)
+        if c == 'H':
+            self.moveTo(values[1]-1, values[0]-1)
+
+    def parse(self, data):
+        seq = ''
+        for char in data:
+            if seq:
+                seq += char
+                if ord(char) >= 64 and char != '[':
+                    self.parseSequence(seq)
+                    seq = ''
+                continue
+            if char == '\x1a':
+                continue
+            if char == '\x1b':
+                seq = char
+                continue
+            if char == '\r':
+                self.moveTo(0, self.y)
+                continue
+            if char == '\n':
+                self.moveTo(self.x, self.y+1)
+                continue
+            if not seq:
+                self.text_lines[self.y][self.x] = char
+                self.attr_lines[self.y][self.x] = self.attr
+                x = self.x + 1
+                self.moveTo(x, self.y)
+        text = []
+        current_attr = self.attr_lines[0][0]
+        current_text = u''
+        for y in range(len(self.text_lines)):
+            for x in range(80):
+                char = self.text_lines[y][x]
+                attr = self.attr_lines[y][x]
+                if (attr.foreground_number != current_attr.foreground_number or
+                    attr.background_number != current_attr.background_number):
+                    text.append((current_attr, current_text))
+                    current_attr = attr
+                    current_text = u''
+                current_text += char
+            if (current_attr.background_number==0):
+                current_text = current_text.rstrip(' ')
+            current_text += u'\n'
+        current_text = re.sub('\n+$', '\n', current_text)
+        text.append((current_attr, current_text))
+        return text