feat: deselect quick highlighters on same shortcut (#270)

variar · Jul 25, 2021 · cac0afe · cac0afe
1 parent fb509d2
commit cac0afe
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/src/ui/src/crawlerwidget.cpp b/src/ui/src/crawlerwidget.cpp
@@ -69,8 +69,8 @@
 
 #include "configuration.h"
 #include "dispatch_to.h"
-#include "infoline.h"
 #include "fontutils.h"
+#include "infoline.h"
 #include "overview.h"
 #include "quickfindpattern.h"
 #include "quickfindwidget.h"
@@ -1219,7 +1219,20 @@ void CrawlerWidget::registerShortcuts()
                                       } );
 
     const auto addColorLabelToSelection = [ this ]( size_t label ) {
-        std::get<0>( wordsHighlighters_[ label ] ).append( getSelectedText() );
+        auto& wordsHighlighters = std::get<0>( wordsHighlighters_[ label ] );
+        auto selectedPattern = getSelectedText();
+
+        const auto existingPattern = std::find_if(
+            wordsHighlighters.begin(), wordsHighlighters.end(),
+            [ &selectedPattern ]( const auto& pattern ) { return selectedPattern == pattern; } );
+
+        if ( existingPattern == wordsHighlighters.end() ) {
+            wordsHighlighters.append( std::move( selectedPattern ) );
+        }
+        else {
+            wordsHighlighters.erase( existingPattern );
+        }
+
         logMainView_->setWordsHighlighters( wordsHighlighters_ );
         filteredView_->setWordsHighlighters( wordsHighlighters_ );
     };