From 0a14d9d2a5f2b2d45a38970cd8ea05d98de79bbb Mon Sep 17 00:00:00 2001
From: pTalanov <Pavel.Talanov@jetbrains.com>
Date: Tue, 5 Jun 2012 21:56:04 +0400
Subject: [PATCH] Restart highlighting in the whole project after setting up
 K2JS module

---
 .../jet/plugin/quickfix/JsModuleSetUp.java       | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/idea/src/org/jetbrains/jet/plugin/quickfix/JsModuleSetUp.java b/idea/src/org/jetbrains/jet/plugin/quickfix/JsModuleSetUp.java
index 388c6dd3ab1..92fb1d9fd18 100644
--- a/idea/src/org/jetbrains/jet/plugin/quickfix/JsModuleSetUp.java
+++ b/idea/src/org/jetbrains/jet/plugin/quickfix/JsModuleSetUp.java
@@ -16,12 +16,15 @@
 
 package org.jetbrains.jet.plugin.quickfix;
 
+import com.intellij.codeInsight.daemon.DaemonCodeAnalyzer;
 import com.intellij.notification.Notification;
 import com.intellij.notification.NotificationType;
 import com.intellij.notification.Notifications;
 import com.intellij.openapi.module.Module;
 import com.intellij.openapi.roots.ModuleRootManager;
 import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.psi.PsiManager;
+import com.intellij.psi.impl.PsiModificationTrackerImpl;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 import org.jetbrains.jet.plugin.project.K2JSModuleComponent;
@@ -56,11 +59,22 @@ public final class JsModuleSetUp {
 
         if (!copyJsLibFiles(rootDir)) return;
 
+        setUpK2JSModuleComponent(module);
+
+        restartHighlightingInTheWholeProject(module);
+
+        refreshRootDir(module, continuation);
+    }
+
+    private static void setUpK2JSModuleComponent(@NotNull Module module) {
         K2JSModuleComponent jsModuleComponent = K2JSModuleComponent.getInstance(module);
         jsModuleComponent.setJavaScriptModule(true);
         jsModuleComponent.setPathToJavaScriptLibrary("/lib/" + PathUtil.JS_LIB_JAR_NAME);
+    }
 
-        refreshRootDir(module, continuation);
+    private static void restartHighlightingInTheWholeProject(@NotNull Module module) {
+        ((PsiModificationTrackerImpl) PsiManager.getInstance(module.getProject()).getModificationTracker()).incCounter();
+        DaemonCodeAnalyzer.getInstance(module.getProject()).restart();
     }
 
     private static boolean copyJsLibFiles(@NotNull File rootDir) {