From 4adfd79f7fa41b49944836b6e373dffba300bd60 Mon Sep 17 00:00:00 2001
From: mikrise2 <mikrise2@mail.ru>
Date: Fri, 1 Dec 2023 13:43:54 +0100
Subject: [PATCH 1/6] [ML4SE-239] Repeat scenario.

---
 .../config/scenario/models/Scenario.kt        | 33 +++++++++++--------
 .../main/panel/panelStates/DefaultStates.kt   |  5 +--
 2 files changed, 22 insertions(+), 16 deletions(-)

diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/scenario/models/Scenario.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/scenario/models/Scenario.kt
index 1291f520..e0496dce 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/scenario/models/Scenario.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/scenario/models/Scenario.kt
@@ -19,6 +19,9 @@ data class Scenario(
     @Transient
     private val logger: Logger = Logger.getInstance(javaClass)
 
+    @Transient
+    private var currentSteps = LinkedList(steps)
+
     @Transient
     private var currentStepIterator: Iterator<ScenarioUnit>? = null
 
@@ -26,12 +29,12 @@ data class Scenario(
         var isValid: Boolean
         var step: ScenarioStep
         do {
-            if (steps.isEmpty()) {
+            if (currentSteps.isEmpty()) {
                 logger.warn("No steps found!")
                 return null
             }
 
-            step = steps.poll()
+            step = currentSteps.poll()
             isValid = step.isValid()
             if (!isValid) {
                 logger.warn("Found useless step, all configs are null. Skip it")
@@ -42,21 +45,23 @@ data class Scenario(
 
     @Suppress("ReturnCount")
     fun getNextUnit(project: Project): ScenarioUnit? {
-        if (!currentStepIterator.notNullAndHasNext()) {
-            return null
-        }
-        cleanStepSettings()
-        val currentStep = getNextStep() ?: return null
-        currentStep.prepareSettings(project)
-        currentStepIterator = currentStep.getUnits().iterator()
-        return if (currentStepIterator.notNullAndHasNext()) {
-            currentStepIterator?.next()
-        } else {
-            null
+        if (currentStepIterator.notNullAndHasNext()) {
+            cleanStepSettings()
+            val currentStep = getNextStep() ?: return null
+            currentStep.prepareSettings(project)
+            currentStepIterator = currentStep.getUnits().iterator()
+            if (currentStepIterator.notNullAndHasNext()) {
+                return null
+            }
         }
+        return currentStepIterator?.next()
+    }
+
+    fun reset() {
+        currentSteps = LinkedList(steps)
     }
 
-    private fun Iterator<ScenarioUnit>?.notNullAndHasNext() = this?.hasNext() ?: false
+    private fun Iterator<ScenarioUnit>?.notNullAndHasNext() = this?.hasNext() != true
 
     private fun cleanStepSettings() =
         MainPanelStorage.activeIdeHandlers.forEach {
diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt
index 3ee93412..6703e410 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt
@@ -43,7 +43,7 @@ fun Panel.agreementAcceptance() {
  * Switches the panel to the plugin description window.
  */
 fun Panel.welcomePage() {
-    loadBasePage(MainPageTemplate.loadCurrentTemplate(), "ui.button.next", true)
+    loadBasePage(MainPageTemplate.loadCurrentTemplate(), "ui.button.next", false)
     setNextAction {
         processScenario()
     }
@@ -122,7 +122,7 @@ fun Panel.serverErrorPage() {
 }
 
 fun Panel.finalPage() {
-    loadBasePage(FinalPageTemplate.loadCurrentTemplate(), "ui.button.welcome")
+    loadBasePage(FinalPageTemplate.loadCurrentTemplate(), "ui.button.welcome", false)
     setNextAction {
         welcomePage()
     }
@@ -162,6 +162,7 @@ fun Panel.processScenario() {
         }
 
         null -> {
+            scenario.reset()
             finalPage()
         }
     }

From 52b55a290cf9406404420cf0c5201773394c0454 Mon Sep 17 00:00:00 2001
From: mikrise2 <mikrise2@mail.ru>
Date: Fri, 1 Dec 2023 17:27:54 +0100
Subject: [PATCH 2/6] [ML4SE-239] Fixed survey front.

---
 .../tasktracker/config/survey/HtmlQuestion.kt | 35 ++++++++----
 .../tasktracker/config/survey/Survey.kt       |  2 +-
 .../ui/main/panel/MainPluginPanelFactory.kt   |  6 ++-
 .../main/panel/panelStates/DefaultStates.kt   |  2 +-
 .../tasktracker/config/survey_default.yaml    |  1 +
 .../ui/main/panel/template/css/dark.css       |  6 +--
 .../ui/main/panel/template/css/light.css      |  6 +--
 .../ui/main/panel/template/css/survey.css     | 17 +++++-
 .../ui/main/panel/template/survey.html        | 53 +++++++++++++++++--
 9 files changed, 101 insertions(+), 27 deletions(-)

diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/survey/HtmlQuestion.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/survey/HtmlQuestion.kt
index c108efab..b2fcbaaf 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/survey/HtmlQuestion.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/survey/HtmlQuestion.kt
@@ -2,6 +2,7 @@ package org.jetbrains.research.tasktracker.config.survey
 
 import kotlinx.serialization.SerialName
 import kotlinx.serialization.Serializable
+import kotlinx.serialization.Transient
 
 /**
  * Radio button id and value
@@ -23,24 +24,38 @@ sealed class HtmlQuestion {
     /**
      * `h` tag size. For example, h1, h2, and so on.
      */
-    private val textSize: Int = defaultTextSize
-        get() = if (field < minTextSize || field > maxTextSize) defaultTextSize else field
+    private val textSize: Int = DEFAULT_TEXT_SIZE
+        get() = if (field < MIN_TEXT_SIZE || field > MAX_TEXT_SIZE) DEFAULT_TEXT_SIZE else field
 
     /**
      * Sets 'required' on the input if this field is mandatory to fill out.
      */
     @SerialName("required")
-    private val isRequired: Boolean = false
-    abstract fun toHtml(): String
+    protected val isRequired: Boolean = false
+    internal abstract fun toHtml(): String
+
+    val html: String
+        get() = buildString {
+            append("<div class=\"box\" id=\"${elementId}___container\">")
+            append(toHtml())
+            append("</div>")
+        }
+
+    @Transient
+    private var alreadyRequired = false
 
     protected fun Any?.asParameter(name: String): String = this?.let { "$name=\"$it\"" } ?: ""
-    protected fun isRequiredString(): String = if (isRequired) "required" else ""
-    protected fun htmlText() = "<h$textSize>$text</h$textSize>"
+    protected open fun isRequiredString(): String =
+        if (isRequired && !alreadyRequired) "required".also { alreadyRequired = true } else ""
+
+    protected fun htmlText() = "<h$textSize ${isRequiredClass()}>$text</h$textSize>"
+
+    private fun isRequiredClass(): String = if (isRequired) "class=\"required\"" else ""
 
     companion object {
-        private const val minTextSize = 1
-        private const val maxTextSize = 5
-        private const val defaultTextSize = 3
+        private const val MIN_TEXT_SIZE = 1
+        private const val MAX_TEXT_SIZE = 5
+        private const val DEFAULT_TEXT_SIZE = 3
     }
 }
 
@@ -118,6 +133,8 @@ data class TextAreaHtmlQuestion(
         append("<textarea name=\"$elementId\" id=\"$elementId\" ${rows.asParameter("rows")} ")
         append("${cols.asParameter("cols")} ${isRequiredString()}></textarea>")
     }
+
+    override fun isRequiredString(): String = if (isRequired) "data-req=\"required\"" else ""
 }
 
 /**
diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/survey/Survey.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/survey/Survey.kt
index 78c15bb7..f3d17bdb 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/survey/Survey.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/survey/Survey.kt
@@ -4,5 +4,5 @@ import kotlinx.serialization.Serializable
 
 @Serializable
 data class Survey(val id: String, val htmlQuestions: List<HtmlQuestion>) {
-    fun toHtml() = htmlQuestions.joinToString(System.lineSeparator()) { it.toHtml() }
+    fun toHtml() = htmlQuestions.joinToString(System.lineSeparator()) { it.html }
 }
diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/MainPluginPanelFactory.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/MainPluginPanelFactory.kt
index 4cd29e61..74dce546 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/MainPluginPanelFactory.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/MainPluginPanelFactory.kt
@@ -185,7 +185,7 @@ class MainPluginPanelFactory : ToolWindowFactory {
      *
      * @return **true** if any required field is not filled. **false** otherwise.
      */
-    fun checkInputs(): Promise<Boolean> =
+    fun checkAgreementInputs(): Promise<Boolean> =
         mainWindow.executeJavaScriptAsync("checkAllInputs()").then {
             val agreementChecker = Json.decodeFromString(AgreementChecker.serializer(), it.toString())
             if (agreementChecker.allRequiredChecked()) {
@@ -195,6 +195,10 @@ class MainPluginPanelFactory : ToolWindowFactory {
             true
         }
 
+    fun checkSurveyInputs(): Promise<Boolean> = mainWindow.executeJavaScriptAsync("checkAllInputs()").then {
+        it.toBoolean()
+    }
+
     fun openExternalUrl(url: String) = mainWindow.openExternalUrl(url)
 
     fun setNextAction(listener: ActionListener) = nextButton.setListener(listener)
diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt
index 6703e410..319e5099 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt
@@ -25,7 +25,7 @@ typealias Panel = MainPluginPanelFactory
 fun Panel.agreementAcceptance() {
     loadBasePage(AgreementTemplate.loadCurrentTemplate(), "ui.button.next", false)
     setNextAction {
-        checkInputs()
+        checkAgreementInputs()
             .onSuccess {
                 if (!it) {
                     welcomePage()
diff --git a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/config/survey_default.yaml b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/config/survey_default.yaml
index 9bfa72ea..1c76321d 100644
--- a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/config/survey_default.yaml
+++ b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/config/survey_default.yaml
@@ -11,6 +11,7 @@ surveys:
       - !<Radio>
         text: 2. On a scale of 1 (very unlikely) to 5 (very likely), how likely are you to continue using the Emotion Tracker in the future?
         elementId: likely
+        required: true
         info:
           - id: likely1
             value: 1
diff --git a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/css/dark.css b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/css/dark.css
index 854f5b46..7a2212cd 100644
--- a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/css/dark.css
+++ b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/css/dark.css
@@ -2,11 +2,7 @@ body {
     background: black;
 }
 
-p {
-    color: #fefefe;
-}
-
-label {
+p, label, h1, h2, h3, h4, h5 {
     color: #fefefe;
 }
 
diff --git a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/css/light.css b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/css/light.css
index a3f18b2c..937c44fa 100644
--- a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/css/light.css
+++ b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/css/light.css
@@ -2,11 +2,7 @@ body {
     background: #fefefe;
 }
 
-p {
-    color: black;
-}
-
-label {
+p, label, h1, h2, h3, h4, h5 {
     color: black;
 }
 
diff --git a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/css/survey.css b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/css/survey.css
index c4ff86e8..65773046 100644
--- a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/css/survey.css
+++ b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/css/survey.css
@@ -1,3 +1,16 @@
-h1, h2, h3, h4, h5, label {
-    color: #fefefe;
+.form-container{
+    align-items: flex-start;
+    text-align: left;
+    justify-content: left;
+}
+
+.required:after {
+    content: " *";
+    font-weight: bold;
+    color: red;
+}
+
+.box{
+    padding: 3px;
+    margin: 6px;
 }
diff --git a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/survey.html b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/survey.html
index 267a2c47..af10fc1b 100644
--- a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/survey.html
+++ b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/survey.html
@@ -1,3 +1,50 @@
-<form id="theForm" action="%s" method="post">
-%s
-</form>
+<div class="container">
+    <form class="form-container" id="theForm" action="%s" method="post">
+        %s
+    </form>
+</div>
+<script>
+
+    let requiredFields = document.querySelectorAll('input[required], textarea[data-req="required"]')
+    requiredFields.forEach(function (fieldTemp) {
+        let fields = document.querySelectorAll(`input[name=${fieldTemp.name}]`)
+        fields.forEach(function (field){
+            field.onclick = function () {
+                if (field.type === "radio") {
+                    defaultBorder(field.name)
+                } else {
+                    defaultBorder(field.id)
+                }
+            };
+        })
+    });
+
+    function checkAllInputs() {
+        let checker = true
+        requiredFields.forEach(function (field) {
+            if (field.type === "checkbox" && !field.checked) {
+                redBorder(field.id)
+                checker = false
+            } else if (field.type === "radio") {
+                let checked = document.querySelector(`input[name = ${field.name}]:checked`);
+                if (checked == null) {
+                    redBorder(field.name)
+                    checker = false
+                }
+            } else if (field.value.trim() === "") {
+                redBorder(field.id)
+                checker = false
+            }
+        });
+        return checker
+    }
+
+    function redBorder(id) {
+        document.getElementById(`${id}___container`).style.outline = '1px solid red'
+    }
+
+    function defaultBorder(id) {
+        document.getElementById(`${id}___container`).style.outline = '1px solid black'
+    }
+
+</script>
\ No newline at end of file

From 4bf857c964c1ccc3392aaa7555741d4b7fd26def Mon Sep 17 00:00:00 2001
From: mikrise2 <mikrise2@mail.ru>
Date: Sat, 2 Dec 2023 08:47:31 +0100
Subject: [PATCH 3/6] [ML4SE-239] Survey is ready.

---
 .../ui/main/panel/panelStates/DefaultStates.kt     | 14 ++++++++++----
 .../tasktracker/config/survey_default.yaml         |  1 +
 .../ui/main/panel/template/css/survey.css          |  1 +
 .../tasktracker/ui/main/panel/template/survey.html |  9 ++++-----
 4 files changed, 16 insertions(+), 9 deletions(-)

diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt
index 211d0ac5..04f3dafd 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt
@@ -103,11 +103,17 @@ fun Panel.survey(id: String) {
         ?: error("Survey with id `$id` hasn't been found.")
     loadBasePage(SurveyTemplate(survey))
     setNextAction {
-        val surveyParser = SurveyParser(mainWindow, project)
-        GlobalScope.launch {
-            surveyParser.parseAndLog(survey)
+        checkSurveyInputs().runOnSuccess {
+            if (it) {
+                val surveyParser = SurveyParser(mainWindow, project)
+                GlobalScope.launch {
+                    surveyParser.parseAndLog(survey)
+                }
+                processScenario()
+            } else {
+                notifyError(project, UIBundle.message("ui.please.fill"))
+            }
         }
-        processScenario()
     }
 }
 
diff --git a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/config/survey_default.yaml b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/config/survey_default.yaml
index 1c76321d..af8106fa 100644
--- a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/config/survey_default.yaml
+++ b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/config/survey_default.yaml
@@ -205,5 +205,6 @@ surveys:
       - !<Textarea>
         text: 8. Please share any additional thoughts or comments you have about your experience using the Emotion Tracker plugin.
         elementId: additionalThoughts
+        required: true
         rows: 5
         cols: 50
diff --git a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/css/survey.css b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/css/survey.css
index 65773046..9ffcc4c5 100644
--- a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/css/survey.css
+++ b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/css/survey.css
@@ -13,4 +13,5 @@
 .box{
     padding: 3px;
     margin: 6px;
+    border-radius: 3px;
 }
diff --git a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/survey.html b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/survey.html
index af10fc1b..d32bfe9b 100644
--- a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/survey.html
+++ b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/ui/main/panel/template/survey.html
@@ -7,8 +7,8 @@
 
     let requiredFields = document.querySelectorAll('input[required], textarea[data-req="required"]')
     requiredFields.forEach(function (fieldTemp) {
-        let fields = document.querySelectorAll(`input[name=${fieldTemp.name}]`)
-        fields.forEach(function (field){
+        let fields = document.getElementsByName(fieldTemp.name);
+        fields.forEach(function (field) {
             field.onclick = function () {
                 if (field.type === "radio") {
                     defaultBorder(field.name)
@@ -44,7 +44,6 @@
     }
 
     function defaultBorder(id) {
-        document.getElementById(`${id}___container`).style.outline = '1px solid black'
+        document.getElementById(`${id}___container`).style.outline = '0px'
     }
-
-</script>
\ No newline at end of file
+</script>

From b88ef49022349266505c9ab1ca5de594c2c1c671 Mon Sep 17 00:00:00 2001
From: mikrise2 <mikrise2@mail.ru>
Date: Mon, 4 Dec 2023 11:11:54 +0100
Subject: [PATCH 4/6] [ML4SE-239] destroy settings on the end of the step.

---
 .../research/tasktracker/activities/InitActivity.kt    |  5 +----
 .../research/tasktracker/handler/ide/IdeHandler.kt     | 10 ++++++----
 .../tasktracker/handler/ide/InspectionHandler.kt       |  2 +-
 .../ui/main/panel/panelStates/DefaultStates.kt         |  2 +-
 .../research/tasktracker/config/scenario_default.yaml  |  9 ++++++++-
 .../research/tasktracker/config/settings_default.yaml  |  4 ++--
 6 files changed, 19 insertions(+), 13 deletions(-)

diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/activities/InitActivity.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/activities/InitActivity.kt
index 97dc2c3d..2eb7cd60 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/activities/InitActivity.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/activities/InitActivity.kt
@@ -15,8 +15,5 @@ class InitActivity : StartupActivity {
         TaskTrackerPlugin.initPlugin()
     }
 
-    // TODO: show an error message to the user if an error occurs
-    override fun runActivity(project: Project) {
-        TaskTrackerPlugin.initializationHandler.setupEnvironment(project) // TODO: move to the start point of the task
-    }
+    override fun runActivity(project: Project) = Unit
 }
diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/handler/ide/IdeHandler.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/handler/ide/IdeHandler.kt
index 828c7be1..a716c5f1 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/handler/ide/IdeHandler.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/handler/ide/IdeHandler.kt
@@ -5,21 +5,23 @@ import org.jetbrains.research.tasktracker.config.ide.MainIdeConfig
 import org.jetbrains.research.tasktracker.handler.BaseProjectHandler
 
 class IdeHandler(override val config: MainIdeConfig, override val project: Project) : BaseProjectHandler {
-    private val childHandlers: List<BaseProjectHandler?>
+    private val childHandlers: List<BaseProjectHandler>
 
     init {
         with(config) {
-            childHandlers = listOfNotNull(inspectionConfig, settingsConfig).map { it.buildHandler(project) }
+            childHandlers = listOfNotNull(inspectionConfig, settingsConfig).mapNotNull { it.buildHandler(project) }
         }
     }
 
     override fun setup() {
         childHandlers.forEach {
-            it?.setup()
+            it.setup()
         }
     }
 
     override fun destroy() {
-        // TODO destroy
+        childHandlers.forEach {
+            it.destroy()
+        }
     }
 }
diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/handler/ide/InspectionHandler.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/handler/ide/InspectionHandler.kt
index 6f96dc84..aa159795 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/handler/ide/InspectionHandler.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/handler/ide/InspectionHandler.kt
@@ -63,7 +63,7 @@ class InspectionHandler(override val config: InspectionConfig, override val proj
         initialProfile?.let {
             ProjectInspectionProfileManager.getInstance(project).setCurrentProfile(it)
         }
-        inspectionDisposable?.let { Disposer.dispose(it) }
+//        inspectionDisposable?.let { Disposer.dispose(it) }
     }
 
     private fun Collection<String>.enableInspections(profile: InspectionProfileImpl, project: Project) =
diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt
index 04f3dafd..c28c84ec 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt
@@ -42,11 +42,11 @@ fun Panel.agreementAcceptance() {
 fun Panel.welcomePage() {
     loadBasePage(MainPageTemplate.loadCurrentTemplate(), "ui.button.next", false)
     setNextAction {
+        TaskTrackerPlugin.initializationHandler.setupEnvironment(project)
         processScenario()
     }
 }
 
-// TODO refactor it for many configs
 /**
  * Switches the panel to the task selection window.
  */
diff --git a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/config/scenario_default.yaml b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/config/scenario_default.yaml
index d2a4e93e..f7fc01a7 100644
--- a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/config/scenario_default.yaml
+++ b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/config/scenario_default.yaml
@@ -6,11 +6,18 @@ scenario:
         - !<Survey>
           id: "default"
       mode: SHUFFLED
+      ideConfig:
+        settingsConfig:
+          theme: LIGHT
+        inspectionConfig:
+          mode: DISABLE_SELECTED
+          inspectionNames:
+            - "EmptyFinallyBlock"
     - units:
         - !<Setting>
           mainIdeConfig:
             settingsConfig:
-              theme: LIGHT
+              enableCodeCompletion: OFF
         - !<List>
           taskIds:
             - "hello"
diff --git a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/config/settings_default.yaml b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/config/settings_default.yaml
index de3e58cd..7d71b9f4 100644
--- a/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/config/settings_default.yaml
+++ b/ij-plugin/src/main/resources/org/jetbrains/research/tasktracker/config/settings_default.yaml
@@ -1,3 +1,3 @@
-enableCodeCompletion: OFF
-enableZenMode: OFF
+enableCodeCompletion: DEFAULT
+enableZenMode: DEFAULT
 theme: DEFAULT
\ No newline at end of file

From 0cfaf9a7a95a9c1404f27c32520c665be3553655 Mon Sep 17 00:00:00 2001
From: mikrise2 <mikrise2@mail.ru>
Date: Mon, 4 Dec 2023 12:13:04 +0100
Subject: [PATCH 5/6] [ML4SE-239] Kdoc for Scenario, ScenarioStep,
 ScenarioUnit.

---
 .../config/scenario/models/Scenario.kt        |  4 +++
 .../config/scenario/models/ScenarioStep.kt    |  7 +++++
 .../config/scenario/models/ScenarioUnit.kt    | 26 +++++++++++++++++++
 3 files changed, 37 insertions(+)

diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/scenario/models/Scenario.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/scenario/models/Scenario.kt
index e0496dce..5cc5a47a 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/scenario/models/Scenario.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/scenario/models/Scenario.kt
@@ -12,6 +12,10 @@ import kotlinx.serialization.encoding.Encoder
 import org.jetbrains.research.tasktracker.ui.main.panel.storage.MainPanelStorage
 import java.util.*
 
+/**
+ * Structure of the research scenario. It consists of steps.
+ * @see [ScenarioStep]
+ */
 @Serializable
 data class Scenario(
     @Serializable(with = QueueSerializer::class) val steps: Queue<ScenarioStep>
diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/scenario/models/ScenarioStep.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/scenario/models/ScenarioStep.kt
index e61580be..0779736f 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/scenario/models/ScenarioStep.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/scenario/models/ScenarioStep.kt
@@ -7,6 +7,13 @@ import org.jetbrains.research.tasktracker.config.ide.MainIdeConfig
 import org.jetbrains.research.tasktracker.handler.BaseProjectHandler
 import org.jetbrains.research.tasktracker.ui.main.panel.storage.MainPanelStorage
 
+/**
+ * This is the part of the research scenario. Each step contains Units.
+ * Units can be shuffled or ordered, depends on the [mode] field.
+ * Also, Scenario step can contain [ideConfig] settings which will
+ * be setup on the step start and destroyed on the step end.
+ * @see [ScenarioUnit]
+ */
 @Serializable
 data class ScenarioStep(
     private val units: List<ScenarioUnit>,
diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/scenario/models/ScenarioUnit.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/scenario/models/ScenarioUnit.kt
index 85066a14..4c9bab1d 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/scenario/models/ScenarioUnit.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/config/scenario/models/ScenarioUnit.kt
@@ -7,26 +7,52 @@ import org.jetbrains.research.tasktracker.config.ide.MainIdeConfig
 @Serializable
 sealed interface ScenarioUnit
 
+/**
+ * List of tasks from which user selects a task to run.
+ * The chosen task is removed from the list after completion.
+ * The process runs until there are no tasks left.
+ */
 @Serializable
 @SerialName("List")
 class TaskListUnit(val taskIds: List<String>) : ScenarioUnit
 
+/**
+ * List of tasks in which user can select one task.
+ * After completion of this task next unit will be processed.
+ * Unselected tasks won't be present.
+ */
 @Serializable
 @SerialName("SingleList")
 class TaskListWithSingleChoiceUnit(val taskIds: List<String>) : ScenarioUnit
 
+/**
+ * Task unit loaded by the id.
+ * @see [org.jetbrains.research.tasktracker.config.content.TaskContentConfig]
+ * And [org.jetbrains.research.tasktracker.config.content.task.ProgrammingTask]
+ */
 @Serializable
 @SerialName("Task")
 class TaskUnit(val id: String) : ScenarioUnit
 
+/**
+ * Setting Unit represents by MainIdeConfig format.
+ * @see [org.jetbrains.research.tasktracker.config.ide.MainIdeConfig].
+ */
 @Serializable
 @SerialName("Setting")
 class IdeSettingUnit(val mainIdeConfig: MainIdeConfig) : ScenarioUnit
 
+/**
+ * Survey unit. Uses *id* of the survey defined in the SurveyConfig.
+ * @see [org.jetbrains.research.tasktracker.config.survey.SurveyConfig].
+ */
 @Serializable
 @SerialName("Survey")
 class SurveyUnit(val id: String) : ScenarioUnit
 
+/**
+ * Opens external url in the JCEF.
+ */
 @Serializable
 @SerialName("External")
 class ExternalSourceUnit(val url: String) : ScenarioUnit

From e4aab25d77daa21c85d7e65696d3d66c13ced404 Mon Sep 17 00:00:00 2001
From: mikrise2 <mikrise2@mail.ru>
Date: Mon, 4 Dec 2023 15:01:25 +0100
Subject: [PATCH 6/6] [ML4SE-239] added tracking, inspection handler marked to
 be rewritten.

---
 .../tasktracker/handler/ide/InspectionHandler.kt |  7 ++++---
 .../ui/main/panel/MainPluginPanelFactory.kt      | 16 ++++++----------
 .../ui/main/panel/panelStates/DefaultStates.kt   |  6 ++----
 3 files changed, 12 insertions(+), 17 deletions(-)

diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/handler/ide/InspectionHandler.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/handler/ide/InspectionHandler.kt
index aa159795..09c001a3 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/handler/ide/InspectionHandler.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/handler/ide/InspectionHandler.kt
@@ -60,9 +60,10 @@ class InspectionHandler(override val config: InspectionConfig, override val proj
     }
 
     override fun destroy() {
-        initialProfile?.let {
-            ProjectInspectionProfileManager.getInstance(project).setCurrentProfile(it)
-        }
+        // TODO rewrite inspection handler, now it raises an error.
+//        initialProfile?.let {
+//            ProjectInspectionProfileManager.getInstance(project).setCurrentProfile(it)
+//        }
 //        inspectionDisposable?.let { Disposer.dispose(it) }
     }
 
diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/MainPluginPanelFactory.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/MainPluginPanelFactory.kt
index 74dce546..8789286b 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/MainPluginPanelFactory.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/MainPluginPanelFactory.kt
@@ -15,15 +15,12 @@ import com.intellij.ui.jcef.JBCefApp
 import com.intellij.util.ui.JBUI
 import kotlinx.serialization.json.Json
 import org.jetbrains.concurrency.Promise
-import org.jetbrains.research.tasktracker.TaskTrackerPlugin
 import org.jetbrains.research.tasktracker.config.content.task.base.Task
-import org.jetbrains.research.tasktracker.modelInference.model.EmoModel
 import org.jetbrains.research.tasktracker.tracking.BaseTracker
 import org.jetbrains.research.tasktracker.tracking.TaskFileHandler
 import org.jetbrains.research.tasktracker.tracking.activity.ActivityTracker
 import org.jetbrains.research.tasktracker.tracking.fileEditor.FileEditorTracker
 import org.jetbrains.research.tasktracker.tracking.toolWindow.ToolWindowTracker
-import org.jetbrains.research.tasktracker.tracking.webcam.WebCamTracker
 import org.jetbrains.research.tasktracker.tracking.webcam.collectAllDevices
 import org.jetbrains.research.tasktracker.ui.main.panel.models.AgreementChecker
 import org.jetbrains.research.tasktracker.ui.main.panel.models.ButtonState
@@ -79,10 +76,6 @@ class MainPluginPanelFactory : ToolWindowFactory {
         if (trackers.isNotEmpty()) { // Otherwise we can lose data
             return
         }
-        TaskTrackerPlugin.mainConfig.emotionConfig?.let {
-            GlobalPluginStorage.emoPredictor = EmoModel(it)
-        } ?: error("emotion config must exist by this moment")
-
         trackers.addAll(
             listOf(
                 ActivityTracker(project),
@@ -90,12 +83,15 @@ class MainPluginPanelFactory : ToolWindowFactory {
                 FileEditorTracker(project)
             )
         )
-        GlobalPluginStorage.emoPredictor?.let {
-            trackers.add(WebCamTracker(project, it))
-        }
         trackers.forEach { it.startTracking() }
     }
 
+    fun stopTracking() {
+        trackers.forEach {
+            it.stopTracking()
+        }
+    }
+
     fun loadBasePage(
         template: HtmlTemplate,
         buttonTextKey: String? = null,
diff --git a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt
index c28c84ec..d504f4b5 100644
--- a/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt
+++ b/ij-plugin/src/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt
@@ -9,7 +9,6 @@ import org.jetbrains.research.tasktracker.config.content.task.base.Task
 import org.jetbrains.research.tasktracker.config.content.task.base.TaskWithFiles
 import org.jetbrains.research.tasktracker.config.scenario.models.*
 import org.jetbrains.research.tasktracker.tracking.TaskFileHandler
-import org.jetbrains.research.tasktracker.tracking.activity.ActivityTracker
 import org.jetbrains.research.tasktracker.ui.main.panel.MainPluginPanelFactory
 import org.jetbrains.research.tasktracker.ui.main.panel.runOnSuccess
 import org.jetbrains.research.tasktracker.ui.main.panel.storage.MainPanelStorage
@@ -43,6 +42,7 @@ fun Panel.welcomePage() {
     loadBasePage(MainPageTemplate.loadCurrentTemplate(), "ui.button.next", false)
     setNextAction {
         TaskTrackerPlugin.initializationHandler.setupEnvironment(project)
+        startTracking()
         processScenario()
     }
 }
@@ -65,7 +65,6 @@ private fun Panel.selectTask(taskIds: List<String>, allRequired: Boolean = true)
  * Loads configs by selected task and language
  */
 fun Panel.processTask(id: String): Task {
-    startTracking() // TODO
     val task =
         MainPanelStorage.taskIdTask.values.find { it.id == id } ?: error("Can't find task with id '$id'")
     ApplicationManager.getApplication().invokeAndWait {
@@ -83,8 +82,6 @@ fun Panel.processTask(id: String): Task {
  */
 private fun Panel.solveTask(id: String, nextTasks: List<String> = emptyList()) {
     val task = processTask(id)
-    val activityTracker = ActivityTracker(project)
-    activityTracker.startTracking() // TODO start tracking for all trackers instead of this one
     loadBasePage(SolvePageTemplate(task))
     setNextAction {
         TaskFileHandler.disposeTask(project, task)
@@ -164,6 +161,7 @@ fun Panel.processScenario() {
 
         null -> {
             scenario.reset()
+            stopTracking()
             finalPage()
         }
     }