From 200da820b1a4fe170be8aef7262059ca27fd3e33 Mon Sep 17 00:00:00 2001
From: Peter Gromov <peter@jetbrains.com>
Date: Tue, 18 May 2021 11:26:19 +0200
Subject: [PATCH] grazie: restart highlighting on setting change in a more
 centralized place

GitOrigin-RevId: d0499588bfd246531467f1e4a2006f3769c52986
---
 .../grazie/ide/inspection/grammar/GrazieInspection.kt      | 7 -------
 .../com/intellij/grazie/ide/msg/GrazieStateLifecycle.kt    | 6 ++++++
 2 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/plugins/grazie/src/main/kotlin/com/intellij/grazie/ide/inspection/grammar/GrazieInspection.kt b/plugins/grazie/src/main/kotlin/com/intellij/grazie/ide/inspection/grammar/GrazieInspection.kt
index b91c1f0f2ce4..12a22306309c 100644
--- a/plugins/grazie/src/main/kotlin/com/intellij/grazie/ide/inspection/grammar/GrazieInspection.kt
+++ b/plugins/grazie/src/main/kotlin/com/intellij/grazie/ide/inspection/grammar/GrazieInspection.kt
@@ -1,7 +1,6 @@
 // Copyright 2000-2019 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
 package com.intellij.grazie.ide.inspection.grammar
 
-import com.intellij.codeInsight.daemon.DaemonCodeAnalyzer
 import com.intellij.codeInspection.LocalInspectionTool
 import com.intellij.codeInspection.LocalInspectionToolSession
 import com.intellij.codeInspection.ProblemsHolder
@@ -17,14 +16,12 @@ import com.intellij.grazie.ide.language.LanguageGrammarChecking
 import com.intellij.grazie.ide.msg.GrazieStateLifecycle
 import com.intellij.grazie.utils.lazyConfig
 import com.intellij.lang.injection.InjectedLanguageManager
-import com.intellij.openapi.project.ProjectManager
 import com.intellij.openapi.util.Key
 import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiElementVisitor
 import com.intellij.psi.PsiLanguageInjectionHost
 import com.intellij.psi.PsiWhiteSpace
 import com.intellij.psi.util.elementType
-import java.util.*
 
 class GrazieInspection : LocalInspectionTool() {
   companion object : GrazieStateLifecycle {
@@ -46,10 +43,6 @@ class GrazieInspection : LocalInspectionTool() {
       disabledStrategiesIDs = newState.disabledGrammarStrategies
       suppression = newState.suppressingContext
       checking = newState.checkingContext
-
-      ProjectManager.getInstance().openProjects.forEach {
-        DaemonCodeAnalyzer.getInstance(it).restart()
-      }
     }
 
     private fun isCheckInElementTextDomainEnabled(domain: GrammarCheckingStrategy.TextDomain): Boolean {
diff --git a/plugins/grazie/src/main/kotlin/com/intellij/grazie/ide/msg/GrazieStateLifecycle.kt b/plugins/grazie/src/main/kotlin/com/intellij/grazie/ide/msg/GrazieStateLifecycle.kt
index 4db367d2010a..f810d6c64d71 100644
--- a/plugins/grazie/src/main/kotlin/com/intellij/grazie/ide/msg/GrazieStateLifecycle.kt
+++ b/plugins/grazie/src/main/kotlin/com/intellij/grazie/ide/msg/GrazieStateLifecycle.kt
@@ -1,6 +1,7 @@
 // Copyright 2000-2019 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
 package com.intellij.grazie.ide.msg
 
+import com.intellij.codeInsight.daemon.DaemonCodeAnalyzer
 import com.intellij.grazie.GrazieConfig
 import com.intellij.grazie.detection.LangDetector
 import com.intellij.grazie.ide.inspection.detection.LanguageDetectionInspection
@@ -10,6 +11,7 @@ import com.intellij.grazie.jlanguage.LangTool
 import com.intellij.grazie.spellcheck.GrazieSpellchecker
 import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.components.Service
+import com.intellij.openapi.project.ProjectManager
 import com.intellij.util.messages.MessageBusConnection
 import com.intellij.util.messages.Topic
 
@@ -36,6 +38,10 @@ internal class GrazieInitializerManager {
     connection.subscribe(topic, GrazieCommitInspection)
     connection.subscribe(topic, GrazieInspection)
     connection.subscribe(topic, LanguageDetectionInspection)
+
+    ProjectManager.getInstance().openProjects.forEach {
+      DaemonCodeAnalyzer.getInstance(it).restart()
+    }
   }
 
   fun register(subscriber: GrazieStateLifecycle): MessageBusConnection {