1 files changed, 27 insertions, 1 deletions
diff --git a/Swift/QtUI/QtTextEdit.h b/Swift/QtUI/QtTextEdit.h
index 075728b..a8df4d3 100644
--- a/Swift/QtUI/QtTextEdit.h
+++ b/Swift/QtUI/QtTextEdit.h
@@ -5,20 +5,46 @@
  */
 
 #pragma once
+
+#include <SwifTools/SpellParser.h>
+
+#include <Swift/Controllers/Settings/SettingsProvider.h>
+#include <Swift/Controllers/SettingConstants.h>
+
 #include <QTextEdit>
+#include <QPointer>
 
 namespace Swift {
+	class SpellChecker;
+	class QtSpellCheckerWindow;
 	class QtTextEdit : public QTextEdit {
 		Q_OBJECT
 	public:
-		QtTextEdit(QWidget* parent = 0);
+		QtTextEdit(SettingsProvider* settings, QWidget* parent = 0);
+		virtual ~QtTextEdit();
 		virtual QSize sizeHint() const;
 	signals:
+		void wordCorrected(QString& word);
 		void returnPressed();
 		void unhandledKeyPressEvent(QKeyEvent* event);
+	public slots:
+		void handleSettingChanged(const std::string& settings);
 	protected:
 		virtual void keyPressEvent(QKeyEvent* event);
+		virtual void contextMenuEvent(QContextMenuEvent* event);
 	private slots:
 		void handleTextChanged();
+	private:
+		SpellChecker *checker_;
+		std::vector<QAction*> replaceWordActions_;
+		PositionPairList misspelledPositions_;
+		SettingsProvider *settings_;
+		QPointer<QtSpellCheckerWindow> spellCheckerWindow_;
+		void addSuggestions(QMenu* menu, QContextMenuEvent* event);
+		void replaceMisspelledWord(const QString& word, int cursorPosition);
+		void setUpSpellChecker();
+		void underlineMisspells();
+		void spellCheckerSettingsWindow();
+		PositionPair getWordFromCursor(int cursorPosition);
 	};
 }