From a1525ca022c51ef5db881879b878d244685edce8 Mon Sep 17 00:00:00 2001
From: "Vitaliy.Bibaev" <vitaliy.bibaev@jetbrains.com>
Date: Wed, 20 Dec 2017 20:07:16 +0300
Subject: [PATCH] Add support for some intermediate operations if possible

---
 .../lib/sequence/KotlinSequencesSupport.kt    | 36 +++++++++++++++++--
 1 file changed, 34 insertions(+), 2 deletions(-)

diff --git a/idea/idea-jvm/src/com/intellij/debugger/streams/kotlin/lib/sequence/KotlinSequencesSupport.kt b/idea/idea-jvm/src/com/intellij/debugger/streams/kotlin/lib/sequence/KotlinSequencesSupport.kt
index 1b6c8acaf61..52aa6f329cb 100644
--- a/idea/idea-jvm/src/com/intellij/debugger/streams/kotlin/lib/sequence/KotlinSequencesSupport.kt
+++ b/idea/idea-jvm/src/com/intellij/debugger/streams/kotlin/lib/sequence/KotlinSequencesSupport.kt
@@ -1,13 +1,45 @@
 // Copyright 2000-2017 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
 package com.intellij.debugger.streams.kotlin.lib.sequence
 
-import com.intellij.debugger.streams.lib.impl.LibrarySupportBase
+import com.intellij.debugger.streams.lib.IntermediateOperation
+import com.intellij.debugger.streams.lib.impl.*
+import com.intellij.debugger.streams.resolve.AppendResolver
+import com.intellij.debugger.streams.trace.impl.handler.unified.DistinctByKeyHandler
+import com.intellij.debugger.streams.trace.impl.handler.unified.DistinctTraceHandler
 
 /**
  * @author Vitaliy.Bibaev
  */
 class KotlinSequencesSupport : LibrarySupportBase() {
   init {
-    // todo[bibaev]: add supported operations
+    addIntermediateOperationsSupport(*filterOperations("filter", "filterNot", "filterIndexed",
+        "filterIsInstance", "drop", "dropWhile", "minus", "minusElement", "take", "takeWhile", "onEach"))
+
+    addIntermediateOperationsSupport(*mapOperations("map", "mapNotNull", "mapIndexed",
+        "requireNoNulls", "withIndex", "zip", "constrainOnce"))
+
+    addIntermediateOperationsSupport(*flatMapOperations("flatMap", "flatten"))
+
+    addIntermediateOperationsSupport(*sortedOperations("sorted", "sortedBy", "sortedDescending", "sortedWith"))
+
+    addIntermediateOperationsSupport(DistinctOperation("distinct", ::DistinctTraceHandler))
+    addIntermediateOperationsSupport(DistinctOperation("distinctBy", ::DistinctByKeyHandler))
+
+    addIntermediateOperationsSupport(ConcatOperation("plus", AppendResolver()))
+    addIntermediateOperationsSupport(ConcatOperation("plusElement", AppendResolver()))
+
+    addIntermediateOperationsSupport()
   }
+
+  private fun filterOperations(vararg names: String): Array<IntermediateOperation> =
+      names.map { FilterOperation(it) }.toTypedArray()
+
+  private fun mapOperations(vararg names: String): Array<IntermediateOperation> =
+      names.map { MappingOperation(it) }.toTypedArray()
+
+  private fun flatMapOperations(vararg names: String): Array<IntermediateOperation> =
+      names.map { FlatMappingOperation(it) }.toTypedArray()
+
+  private fun sortedOperations(vararg names: String): Array<IntermediateOperation> =
+      names.map { SortedOperation(it) }.toTypedArray()
 }
\ No newline at end of file