Update AstNodeUtils.kt

saveourtool · Jun 22, 2022 · 0fb80d7 · 0fb80d7
1 parent f24f4ee
commit 0fb80d7
Showing 1 changed file with 9 additions and 13 deletions.
diff --git a/diktat-rules/src/main/kotlin/org/cqfn/diktat/ruleset/utils/AstNodeUtils.kt b/diktat-rules/src/main/kotlin/org/cqfn/diktat/ruleset/utils/AstNodeUtils.kt
@@ -723,19 +723,6 @@ fun ASTNode.extractLineOfText(): String {
     return text.joinToString(separator = "").trim()
 }
 
-/**
- * @param pred a predicate
- * @return filtered sequence
- */
-private fun <T> Sequence<T>.takeWhileInclusive(pred: (T) -> Boolean): Sequence<T> {
-    var shouldContinue = true
-    return takeWhile {
-        val result = shouldContinue
-        shouldContinue = pred(it)
-        result
-    }
-}
-
 /**
  * Checks node has `@Test` annotation
  */
@@ -846,6 +833,15 @@ fun ASTNode.takeByChainOfTypes(vararg types: IElementType): ASTNode? {
     return node
 }
 
+private fun <T> Sequence<T>.takeWhileInclusive(pred: (T) -> Boolean): Sequence<T> {
+    var shouldContinue = true
+    return takeWhile {
+        val result = shouldContinue
+        shouldContinue = pred(it)
+        result
+    }
+}
+
 private fun Collection<KtAnnotationEntry>.containSuppressWithName(name: String) =
     this.any {
         it.shortName.toString() == (Suppress::class.simpleName) &&