small fix in enable hydra action

triplequote · Oct 23, 2017 · f6da3bd · f6da3bd
1 parent ebd31d9
commit f6da3bd
Showing 1 changed file with 15 additions and 4 deletions.
diff --git a/scala/scala-impl/src/org/jetbrains/plugins/scala/actions/EnableHydraSettingsAction.scala b/scala/scala-impl/src/org/jetbrains/plugins/scala/actions/EnableHydraSettingsAction.scala
@@ -1,20 +1,31 @@
 package org.jetbrains.plugins.scala.actions
 
-import com.intellij.openapi.actionSystem.{AnAction, AnActionEvent, CommonDataKeys}
+import com.intellij.openapi.actionSystem.{AnAction, AnActionEvent, CommonDataKeys, ToggleAction}
 import com.intellij.openapi.project.Project
 import org.jetbrains.plugins.scala.compiler.HydraCompilerSettings
 
 /**
   * @author Maris Alexandru
   */
-class EnableHydraSettingsAction extends AnAction{
-  def actionPerformed(e: AnActionEvent) {
+class EnableHydraSettingsAction extends ToggleAction{
+
+  override def setSelected(e: AnActionEvent, toSet: Boolean): Unit = {
     CommonDataKeys.PROJECT.getData(e.getDataContext) match {
       case project: Project => {
         val settings = HydraCompilerSettings.getInstance(project)
-        settings.isHydraSettingsEnabled = !settings.isHydraSettingsEnabled
+        settings.isHydraSettingsEnabled = toSet
       }
       case _ =>
     }
   }
+
+  override def isSelected(e: AnActionEvent): Boolean = {
+    CommonDataKeys.PROJECT.getData(e.getDataContext) match {
+      case project: Project => {
+        val settings = HydraCompilerSettings.getInstance(project)
+        settings.isHydraSettingsEnabled
+      }
+      case _ => false
+    }
+  }
 }