From 8de899c3f736ea3adae7c02e676409ede1311003 Mon Sep 17 00:00:00 2001
From: Damien Elmes <git@ichi2.net>
Date: Wed, 11 Apr 2012 18:37:07 +0900
Subject: [PATCH] save and restore selection when setting colour

---
 aqt/editor.py | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/aqt/editor.py b/aqt/editor.py
index f8dbdf66d..f980c17b9 100644
--- a/aqt/editor.py
+++ b/aqt/editor.py
@@ -43,7 +43,9 @@ body { margin: 5px; }
 </style><script>
 %s
 
-insertHTMLOK = %s;
+var insertHTMLOK = %s;
+var savedSel = null;
+
 
 String.prototype.format = function() {
     var args = arguments;
@@ -116,6 +118,17 @@ function onFocus(elem) {
     }
 }
 
+function saveSel() {
+    sel = document.getSelection();
+    savedSel = sel.getRangeAt(0);
+}
+
+function restoreSel() {
+    sel = document.getSelection();
+    sel.removeAllRanges();
+    sel.addRange(savedSel);
+}
+
 function onDragOver(elem) {
     elem.focus();
 }
@@ -675,6 +688,7 @@ class Editor(object):
         self._updateForegroundButton(recent[-1])
 
     def onForeground(self):
+        self.web.eval("saveSel();")
         class ColourPopup(QDialog):
             def __init__(self, parent):
                 QDialog.__init__(self, parent, Qt.FramelessWindowHint)
@@ -752,7 +766,7 @@ class Editor(object):
         recent = self.mw.pm.profile['recentColours']
         recent.remove(colour)
         recent.append(colour)
-        self._eval("setFormat('forecolor', '%s')" % colour)
+        self._eval("restoreSel(); setFormat('forecolor', '%s')" % colour)
         self.colourDiag.close()
         self.colourChanged()