Change default behavior of CommentDialog.

Now accepts on Enter/Return.

1 files changed, 21 insertions, 3 deletions

diff --git a/ui/commentdialog.h b/ui/commentdialog.h
index f1c40ca5..c25e1857 100644
--- a/ui/commentdialog.h
+++ b/ui/commentdialog.h
@@ -2,8 +2,8 @@
 
 #include <QtWidgets/QDialog>
 #include <QtWidgets/QComboBox>
+#include <QtWidgets/QTextEdit>
 #include "binaryninjaapi.h"
-#include "dialogtextedit.h"
 #include "uicontext.h"
 #include "uicomment.h"
 
@@ -17,11 +17,29 @@
 
     \ingroup commentdialog
 */
+class BINARYNINJAUIAPI CommentDialogTextEdit : public QTextEdit
+{
+	Q_OBJECT
+
+public:
+	CommentDialogTextEdit(QWidget* parent);
+
+protected:
+	virtual void keyPressEvent(QKeyEvent* event) override;
+
+Q_SIGNALS:
+	void contentAccepted();
+};
+
+/*!
+
+    \ingroup commentdialog
+*/
 class BINARYNINJAUIAPI CommentDialog : public QDialog
 {
 	Q_OBJECT
 
-	DialogTextEdit* m_comment;
+	CommentDialogTextEdit* m_comment;
 	UIComment m_uicomment;
 
   public:
@@ -32,4 +50,4 @@ class BINARYNINJAUIAPI CommentDialog : public QDialog
 	const BinaryViewRef& getCommentBackingData();
 	UICommentType getCommentType();
 	uint64_t getCommentAddress();
-};
+};
\ No newline at end of file