Initial support for indexed transformations

idea/testData/intentions/loopToCallChain/any_indexNeeded.kt

@@ -0,0 +1,11 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with 'filterIndexed{}.any()'"
+fun foo(list: List<String>) {
+    var found = false
+    <caret>for ((index, s) in list.withIndex()) {
+        if (s.length > index) {
+            found = true
+            break
+        }
+    }
+}

idea/testData/intentions/loopToCallChain/any_indexNeeded.kt.after

@@ -0,0 +1,7 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with 'filterIndexed{}.any()'"
+fun foo(list: List<String>) {
+    val <caret>found = list
+            .filterIndexed { index, s -> s.length > index }
+            .any()
+}

idea/testData/intentions/loopToCallChain/filterIndexed.kt

@@ -0,0 +1,10 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with 'filterIndexed{}.firstOrNull()'"
+fun foo(list: List<String>): String? {
+    <caret>for ((index, s) in list.withIndex()) {
+        if (s.length > index) {
+            return s
+        }
+    }
+    return null
+}

idea/testData/intentions/loopToCallChain/filterIndexed.kt.after

@@ -0,0 +1,7 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with 'filterIndexed{}.firstOrNull()'"
+fun foo(list: List<String>): String? {
+    <caret>return list
+            .filterIndexed { index, s -> s.length > index }
+            .firstOrNull()
+}

idea/testData/intentions/loopToCallChain/flatMap_indexUsed.kt

@@ -0,0 +1,10 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with 'mapIndexed{}.flatMap{}.firstOrNull{}'"
+fun foo(list: List<String>): String? {
+    <caret>for ((index, s) in list.withIndex()) {
+        for (line in s.lines().take(index)) {
+            if (line.isNotBlank()) return line
+        }
+    }
+    return null
+}

idea/testData/intentions/loopToCallChain/flatMap_indexUsed.kt.after

@@ -0,0 +1,8 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with 'mapIndexed{}.flatMap{}.firstOrNull{}'"
+fun foo(list: List<String>): String? {
+    <caret>return list
+            .mapIndexed { index, s -> s.lines().take(index) }
+            .flatMap { it }
+            .firstOrNull { it.isNotBlank() }
+}

idea/testData/intentions/loopToCallChain/mapAndFilterIndexed.kt

@@ -0,0 +1,11 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with 'map{}.filterIndexed{}.firstOrNull()'"
+fun foo(list: List<String>): Int? {
+    <caret>for ((index, s) in list.withIndex()) {
+        val l = s.length
+        if (l > index) {
+            return l
+        }
+    }
+    return null
+}

idea/testData/intentions/loopToCallChain/mapAndFilterIndexed.kt.after

@@ -0,0 +1,8 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with 'map{}.filterIndexed{}.firstOrNull()'"
+fun foo(list: List<String>): Int? {
+    <caret>return list
+            .map { it.length }
+            .filterIndexed { index, l -> l > index }
+            .firstOrNull()
+}

idea/testData/intentions/loopToCallChain/mapIndexed.kt

@@ -0,0 +1,9 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with 'mapIndexed{}.firstOrNull{}'"
+fun foo(list: List<String>): Int? {
+    <caret>for ((index, s) in list.withIndex()) {
+        val x = s.length * index
+        if (x > 0) return x
+    }
+    return null
+}

idea/testData/intentions/loopToCallChain/mapIndexed.kt.after

@@ -0,0 +1,7 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with 'mapIndexed{}.firstOrNull{}'"
+fun foo(list: List<String>): Int? {
+    <caret>return list
+            .mapIndexed { index, s -> s.length * index }
+            .firstOrNull { it > 0 }
+}

idea/testData/intentions/loopToCallChain/mapIndexed_twice.kt

@@ -0,0 +1,10 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with 'mapIndexed{}.mapIndexed{}.firstOrNull{}'"
+fun foo(list: List<String>): Int? {
+    <caret>for ((index, s) in list.withIndex()) {
+        val x = s.length * index
+        val y = x + index
+        if (y > 0) return y
+    }
+    return null
+}

idea/testData/intentions/loopToCallChain/mapIndexed_twice.kt.after

@@ -0,0 +1,8 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with 'mapIndexed{}.mapIndexed{}.firstOrNull{}'"
+fun foo(list: List<String>): Int? {
+    <caret>return list
+            .mapIndexed { index, s -> s.length * index }
+            .mapIndexed { index, x -> x + index }
+            .firstOrNull { it > 0 }
+}

idea/testData/intentions/loopToCallChain/mapTo_indexUsed.kt

@@ -0,0 +1,7 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with '+= mapIndexed{}'"
+fun foo(list: List<String>, target: MutableList<Int>) {
+    <caret>for ((index, s) in list.withIndex()) {
+        target.add(s.hashCode() * index)
+    }
+}

idea/testData/intentions/loopToCallChain/mapTo_indexUsed.kt.after

@@ -0,0 +1,5 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with '+= mapIndexed{}'"
+fun foo(list: List<String>, target: MutableList<Int>) {
+    <caret>target += list.mapIndexed { index, s -> s.hashCode() * index }
+}

idea/testData/intentions/loopToCallChain/mapUsesOldIndexAfterFilter.kt

@@ -0,0 +1,10 @@
+// WITH_RUNTIME
+// IS_APPLICABLE: false
+fun foo(list: List<String>): Int? {
+    <caret>for ((index, s) in list.withIndex()) {
+        if (s.isBlank()) continue
+        val x = s.length * index
+        if (x > 1000) return x
+    }
+    return null
+}