From f64aaed205c396ae000632bce4de4e2a58d2a94d Mon Sep 17 00:00:00 2001
From: Dmitry Jemerov <yole@jetbrains.com>
Date: Tue, 9 May 2017 09:15:36 +0200
Subject: [PATCH] Allow updating of fold region for imports when it's collapsed

 #KT-12856 Fixed
---
 idea/src/org/jetbrains/kotlin/idea/KotlinFoldingBuilder.kt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/idea/src/org/jetbrains/kotlin/idea/KotlinFoldingBuilder.kt b/idea/src/org/jetbrains/kotlin/idea/KotlinFoldingBuilder.kt
index 6922bb403e6..12ecc8b0522 100644
--- a/idea/src/org/jetbrains/kotlin/idea/KotlinFoldingBuilder.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/KotlinFoldingBuilder.kt
@@ -44,7 +44,7 @@ class KotlinFoldingBuilder : CustomFoldingBuilder(), DumbAware {
         }
         val imports = root.importDirectives
         if (imports.size > 1) {
-            val importKeyword = imports.get(0).firstChild
+            val importKeyword = imports[0].firstChild
             val startOffset = importKeyword.endOffset + 1
 
             val importList = root.importList
@@ -52,7 +52,7 @@ class KotlinFoldingBuilder : CustomFoldingBuilder(), DumbAware {
                 val endOffset = importList.endOffset
 
                 val range = TextRange(startOffset, endOffset)
-                descriptors.add(FoldingDescriptor(importList, range))
+                descriptors.add(FoldingDescriptor(importList, range).apply { setCanBeRemovedWhenCollapsed(true) })
             }
         }