Initial commitHEADmaster

Signed-off-by: Agustin Zubiaga <aguz@sugarlabs.org>

1 files changed, 329 insertions, 0 deletions

diff --git a/pilas/console/console_widget.py b/pilas/console/console_widget.py
new file mode 100644
index 0000000..c96d702
--- /dev/null
+++ b/pilas/console/console_widget.py
@@ -0,0 +1,329 @@
+# -*- coding: utf-8 -*-
+from __future__ import absolute_import
+
+import re
+
+from PyQt4.QtGui import QPlainTextEdit
+from PyQt4.QtGui import QTextCursor
+from PyQt4.QtGui import QTextFormat
+from PyQt4.QtGui import QTextEdit
+from PyQt4.QtGui import QColor
+from PyQt4.QtCore import Qt
+from PyQt4.QtCore import SIGNAL
+
+from pilas.console import console
+from pilas.console import highlighter
+
+
+INDENT = 4
+
+BRACES = {"'": "'",
+    '"': '"',
+    '{': '}',
+    '[': ']',
+    '(': ')'}
+
+EDITOR_STYLE = """QPlainTextEdit {
+                    font-family: monospace;
+                    font-size: 10;
+                    color: black;
+                    background-color: white;
+                    selection-color: white;
+                    selection-background-color: #437DCD;
+                }"""
+
+
+class ConsoleWidget(QPlainTextEdit):
+
+    def __init__(self, locals):
+        QPlainTextEdit.__init__(self, u'>>> ')
+        self.setUndoRedoEnabled(False)
+        self.setStyleSheet(EDITOR_STYLE)
+        self.setToolTip(self.tr("Show/Hide (F4)"))
+
+        self._patIsWord = re.compile('\w+')
+        self.prompt = u'>>> '
+        self._console = console.Console(locals)
+        self._history = []
+        self._braces = None
+
+        self._highlighter = highlighter.Highlighter(self.document(), 'python',
+            highlighter.COLOR_SCHEME)
+
+        self.connect(self, SIGNAL("cursorPositionChanged()"),
+            self.highlight_current_line)
+        self.highlight_current_line()
+        self.setCursorPosition(0)
+
+    def setCursorPosition(self, position):
+        self.moveCursor(QTextCursor.StartOfLine)
+        for i in xrange(len(self.prompt) + position):
+            self.moveCursor(QTextCursor.Right)
+
+    def keyPressEvent(self, event):
+        if event.key() in (Qt.Key_Enter, Qt.Key_Return):
+            self._write_command()
+            return
+        if self._get_cursor_position() < 0:
+            self.setCursorPosition(0)
+        if event.key() == Qt.Key_Tab:
+            self.textCursor().insertText(' ' * INDENT)
+            return
+        if event.key() == Qt.Key_Home:
+            self.setCursorPosition(0)
+            return
+        if event.key() == Qt.Key_PageUp:
+            return
+        elif event.key() in (Qt.Key_Left, Qt.Key_Backspace):
+            if self._get_cursor_position() == 0:
+                return
+        elif event.key() == Qt.Key_Up:
+            self._set_command(self._get_prev_history_entry())
+            return
+        elif event.key() == Qt.Key_Down:
+            self._set_command(self._get_next_history_entry())
+            return
+
+        if event.key() == Qt.Key_Tab:
+            if self.textCursor().hasSelection():
+                self.indent_more()
+                return
+            else:
+                self.textCursor().insertText(' ' * INDENT)
+                return
+        elif event.key() == Qt.Key_Backspace:
+            if self.textCursor().hasSelection():
+                QPlainTextEdit.keyPressEvent(self, event)
+                return
+            for i in xrange(INDENT):
+                self.moveCursor(QTextCursor.Left, QTextCursor.KeepAnchor)
+            text = self.textCursor().selection()
+            if unicode(text.toPlainText()) == ' ' * INDENT:
+                self.textCursor().removeSelectedText()
+                return
+            else:
+                for i in xrange(text.toPlainText().size()):
+                    self.moveCursor(QTextCursor.Right)
+        elif event.key() == Qt.Key_Home:
+            if event.modifiers() == Qt.ShiftModifier:
+                move = QTextCursor.KeepAnchor
+            else:
+                move = QTextCursor.MoveAnchor
+            if self.textCursor().atBlockStart():
+                self.moveCursor(QTextCursor.WordRight, move)
+                return
+        elif event.key() in (Qt.Key_Enter, Qt.Key_Return) and \
+          event.modifiers() == Qt.ShiftModifier:
+            return
+        elif unicode(event.text()) in \
+        (set(BRACES.values()) - set(["'", '"'])):
+            self.moveCursor(QTextCursor.Left, QTextCursor.KeepAnchor)
+            brace = unicode(self.textCursor().selection().toPlainText())
+            self.moveCursor(QTextCursor.Right)
+            self.moveCursor(QTextCursor.Right, QTextCursor.KeepAnchor)
+            braceClose = unicode(self.textCursor().selection().toPlainText())
+            self.moveCursor(QTextCursor.Left)
+            if BRACES.get(brace, False) == unicode(event.text()) and \
+              braceClose == unicode(event.text()):
+                self.moveCursor(QTextCursor.Right)
+                return
+        selection = self.textCursor().selectedText()
+
+        QPlainTextEdit.keyPressEvent(self, event)
+
+        if unicode(event.text()) in BRACES:
+            cursor = self.textCursor()
+            cursor.movePosition(QTextCursor.StartOfLine,
+                QTextCursor.KeepAnchor)
+            self.textCursor().insertText(
+                BRACES[unicode(event.text())])
+            self.moveCursor(QTextCursor.Left)
+            self.textCursor().insertText(selection)
+
+    def highlight_current_line(self):
+        self.emit(SIGNAL("cursorPositionChange(int, int)"),
+            self.textCursor().blockNumber() + 1,
+            self.textCursor().columnNumber())
+        self.extraSelections = []
+
+        selection = QTextEdit.ExtraSelection()
+        lineColor = QColor(highlighter.COLOR_SCHEME.get('current-line',
+                    highlighter.COLOR_SCHEME['current-line']))
+        lineColor.setAlpha(20)
+        selection.format.setBackground(lineColor)
+        selection.format.setProperty(QTextFormat.FullWidthSelection,
+            True)
+        selection.cursor = self.textCursor()
+        selection.cursor.clearSelection()
+        self.extraSelections.append(selection)
+        self.setExtraSelections(self.extraSelections)
+
+        #Brace matching
+        if self._braces is not None:
+            self._braces = None
+        cursor = self.textCursor()
+        if cursor.position() == 0:
+            self.setExtraSelections(self.extraSelections)
+            return
+        cursor.movePosition(QTextCursor.PreviousCharacter,
+                             QTextCursor.KeepAnchor)
+        text = unicode(cursor.selectedText())
+        pos1 = cursor.position()
+        if text in (')', ']', '}'):
+            pos2 = self._match_braces(pos1, text, forward=False)
+        elif text in ('(', '[', '{'):
+            pos2 = self._match_braces(pos1, text, forward=True)
+        else:
+            self.setExtraSelections(self.extraSelections)
+            return
+        if pos2 is not None:
+            self._braces = (pos1, pos2)
+            selection = QTextEdit.ExtraSelection()
+            selection.format.setForeground(QColor(
+                highlighter.COLOR_SCHEME.get('brace-foreground',
+                highlighter.COLOR_SCHEME.get('brace-foreground'))))
+            selection.format.setBackground(QColor(
+                highlighter.COLOR_SCHEME.get('brace-background',
+                highlighter.COLOR_SCHEME.get('brace-background'))))
+            selection.cursor = cursor
+            self.extraSelections.append(selection)
+            selection = QTextEdit.ExtraSelection()
+            selection.format.setForeground(QColor(
+                highlighter.COLOR_SCHEME.get('brace-foreground',
+                highlighter.COLOR_SCHEME.get('brace-foreground'))))
+            selection.format.setBackground(QColor(
+                highlighter.COLOR_SCHEME.get('brace-background',
+                highlighter.COLOR_SCHEME.get('brace-background'))))
+            selection.cursor = self.textCursor()
+            selection.cursor.setPosition(pos2)
+            selection.cursor.movePosition(QTextCursor.NextCharacter,
+                             QTextCursor.KeepAnchor)
+            self.extraSelections.append(selection)
+        else:
+            self._braces = (pos1,)
+            selection = QTextEdit.ExtraSelection()
+            selection.format.setBackground(QColor(
+                highlighter.COLOR_SCHEME.get('brace-background',
+                highlighter.COLOR_SCHEME.get('brace-background'))))
+            selection.format.setForeground(QColor(
+                highlighter.COLOR_SCHEME.get('brace-foreground',
+                highlighter.COLOR_SCHEME.get('brace-foreground'))))
+            selection.cursor = cursor
+            self.extraSelections.append(selection)
+        self.setExtraSelections(self.extraSelections)
+
+    def _text_under_cursor(self):
+        tc = self.textCursor()
+        tc.select(QTextCursor.WordUnderCursor)
+        return tc.selectedText()
+
+    def get_selection(self, posStart, posEnd):
+        cursor = self.textCursor()
+        cursor.setPosition(posStart)
+        cursor2 = self.textCursor()
+        if posEnd == QTextCursor.End:
+            cursor2.movePosition(posEnd)
+            cursor.setPosition(cursor2.position(), QTextCursor.KeepAnchor)
+        else:
+            cursor.setPosition(posEnd, QTextCursor.KeepAnchor)
+        text = cursor.selectedText()
+        return unicode(text)
+
+    def _match_braces(self, position, brace, forward):
+        """based on: http://gitorious.org/khteditor"""
+        if forward:
+            braceMatch = {'(': ')', '[': ']', '{': '}'}
+            text = self.get_selection(position, QTextCursor.End)
+            braceOpen, braceClose = 1, 1
+        else:
+            braceMatch = {')': '(', ']': '[', '}': '{'}
+            text = self.get_selection(QTextCursor.Start, position)
+            braceOpen, braceClose = len(text) - 1, len(text) - 1
+        while True:
+            if forward:
+                posClose = text.find(braceMatch[brace], braceClose)
+            else:
+                posClose = text.rfind(braceMatch[brace], 0, braceClose + 1)
+            if posClose > -1:
+                if forward:
+                    braceClose = posClose + 1
+                    posOpen = text.find(brace, braceOpen, posClose)
+                else:
+                    braceClose = posClose - 1
+                    posOpen = text.rfind(brace, posClose, braceOpen + 1)
+                if posOpen > -1:
+                    if forward:
+                        braceOpen = posOpen + 1
+                    else:
+                        braceOpen = posOpen - 1
+                else:
+                    if forward:
+                        return position + posClose
+                    else:
+                        return position - (len(text) - posClose)
+            else:
+                return
+
+    def _add_prompt(self, incomplete):
+        if incomplete:
+            prompt = '.' * 3 + ' '
+        else:
+            prompt = self.prompt
+        self.appendPlainText(prompt)
+        self.moveCursor(QTextCursor.End)
+
+    def _get_cursor_position(self):
+        return self.textCursor().columnNumber() - len(self.prompt)
+
+    def _write_command(self):
+        command = self.document().findBlockByLineNumber(
+                    self.document().lineCount() - 1).text()
+        #remove the prompt from the QString
+        command = command.remove(0, len(self.prompt)).toUtf8().data()
+        self._add_history(command.decode('utf8'))
+        incomplete = self._write(command.decode('utf8'))
+        if not incomplete:
+            output = self._read()
+            if output is not None:
+                if output.__class__.__name__ == 'unicode':
+                    output = output.encode('utf8')
+                self.appendPlainText(output.decode('utf8'))
+        self._add_prompt(incomplete)
+
+    def _set_command(self, command):
+        self.moveCursor(QTextCursor.End)
+        self.moveCursor(QTextCursor.StartOfLine, QTextCursor.KeepAnchor)
+        for i in xrange(len(self.prompt)):
+            self.moveCursor(QTextCursor.Right, QTextCursor.KeepAnchor)
+        self.textCursor().removeSelectedText()
+        self.textCursor().insertText(command)
+        self.moveCursor(QTextCursor.End)
+
+    def mousePressEvent(self, event):
+        #to avoid selection
+        event.ignore()
+
+    def _write(self, line):
+        return self._console.push(line)
+
+    def _read(self):
+        return self._console.output
+
+    def _add_history(self, command):
+        if command and (not self._history or self._history[-1] != command):
+            self._history.append(command)
+        self.history_index = len(self._history)
+
+    def _get_prev_history_entry(self):
+        if self._history:
+            self.history_index = max(0, self.history_index - 1)
+            return self._history[self.history_index]
+        return ''
+
+    def _get_next_history_entry(self):
+        if self._history:
+            hist_len = len(self._history)
+            self.history_index = min(hist_len, self.history_index + 1)
+            if self.history_index < hist_len:
+                return self._history[self.history_index]
+        return ''