initial testing of qcodeeditor version

Author: psifertex

QCodeEditor.py

@@ -0,0 +1 @@
+/Users/jwiens/dev/QCodeEditor/QCodeEditor.py

__init__.py

@@ -17,7 +17,7 @@
 from binaryninjaui import (getMonospaceFont, UIAction, UIActionHandler, Menu, DockHandler,
      getThemeColor, ThemeColor)
 import numbers
-from .lntextedit import LNTextEdit
+from .QCodeEditor import QCodeEditor, PythonHighlighter
 
 snippetPath = os.path.realpath(os.path.join(user_plugin_path(), "..", "snippets"))
 try:
@@ -140,9 +140,8 @@ def __init__(self, context, parent=None):
         self.browseButton.setIcon(QIcon.fromTheme("edit-undo"))
         self.deleteSnippetButton = QPushButton("Delete")
         self.newSnippetButton = QPushButton("New Snippet")
-        self.edit = LNTextEdit()
-        self.edit.edit.setPlaceholderText("python code")
-        self.edit.edit.highlight_color = getThemeColor(ThemeColor.SelectionColor)
+        self.edit = QCodeEditor(HIGHLIGHT_CURRENT_LINE=False, SyntaxHighlighter=PythonHighlighter)
+        self.edit.setPlaceholderText("python code")
         self.resetting = False
         self.columns = 3
         self.context = context
@@ -157,9 +156,9 @@ def __init__(self, context, parent=None):
 
         #Set Editbox Size
         font = getMonospaceFont(self)
-        self.edit.edit.setFont(font)
+        self.edit.setFont(font)
         font = QFontMetrics(font)
-        self.edit.edit.setTabStopWidth(4 * font.width(' ')); #TODO, replace with settings API
+        self.edit.setTabStopWidth(4 * font.width(' ')); #TODO, replace with settings API
 
         #Files
         self.files = QFileSystemModel()
@@ -246,9 +245,9 @@ def __init__(self, context, parent=None):
             if self.tree.selectionModel().hasSelection():
                 self.selectFile(self.tree.selectionModel().selection(), None)
                 self.edit.setFocus()
-                cursor = self.edit.edit.textCursor()
-                cursor.setPosition(self.edit.edit.document().characterCount()-1)
-                self.edit.edit.setTextCursor(cursor)
+                cursor = self.edit.textCursor()
+                cursor.setPosition(self.edit.document().characterCount()-1)
+                self.edit.setTextCursor(cursor)
             else:
                 self.readOnly(True)
         else:
@@ -282,7 +281,7 @@ def clearSelection(self):
         self.currentHotkeyLabel.setText("")
         self.currentFileLabel.setText("")
         self.snippetDescription.setText("")
-        self.edit.edit.setPlainText("")
+        self.edit.setPlainText("")
         self.currentFile = ""
 
     def reject(self):
@@ -337,7 +336,7 @@ def loadSnippet(self):
         (snippetDescription, snippetKeys, snippetCode) = loadSnippetFromFile(self.currentFile)
         self.snippetDescription.setText(snippetDescription) if snippetDescription else self.snippetDescription.setText("")
         self.keySequenceEdit.setKeySequence(snippetKeys) if snippetKeys else self.keySequenceEdit.setKeySequence(QKeySequence(""))
-        self.edit.edit.setPlainText(snippetCode) if snippetCode else self.edit.edit.setPlainText("")
+        self.edit.setPlainText(snippetCode) if snippetCode else self.edit.setPlainText("")
         self.readOnly(False)
 
     def newFileDialog(self):
@@ -359,7 +358,7 @@ def newFileDialog(self):
     def readOnly(self, flag):
         self.keySequenceEdit.setEnabled(not flag)
         self.snippetDescription.setReadOnly(flag)
-        self.edit.edit.setReadOnly(flag)
+        self.edit.setReadOnly(flag)
         if flag:
             self.snippetDescription.setDisabled(True)
             self.edit.setDisabled(True)
@@ -385,15 +384,15 @@ def snippetChanged(self):
             return True
         if snippetKeys != None and snippetKeys != self.keySequenceEdit.keySequence().toString():
             return True
-        return self.edit.edit.toPlainText() != snippetCode or \
+        return self.edit.toPlainText() != snippetCode or \
                self.snippetDescription.text() != snippetDescription
 
     def save(self):
         log_debug("Saving snippet %s" % self.currentFile)
         outputSnippet = codecs.open(self.currentFile, "w", "utf-8")
         outputSnippet.write("#" + self.snippetDescription.text() + "\n")
         outputSnippet.write("#" + self.keySequenceEdit.keySequence().toString() + "\n")
-        outputSnippet.write(self.edit.edit.toPlainText())
+        outputSnippet.write(self.edit.toPlainText())
         outputSnippet.close()
         self.registerAllSnippets()
 
@@ -414,7 +413,7 @@ def run(self):
         outputSnippet = codecs.open(self.currentFile, "w", "utf-8")
         outputSnippet.write("#" + self.snippetDescription.text() + "\n")
         outputSnippet.write("#" + self.keySequenceEdit.keySequence().toString() + "\n")
-        outputSnippet.write(self.edit.edit.toPlainText())
+        outputSnippet.write(self.edit.toPlainText())
         outputSnippet.close()
         self.registerAllSnippets()