platform/lang-impl/src/com/intellij/execution/runners/ConsoleTitleGen.kt

   1 /*
   2  * Copyright 2000-2016 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package com.intellij.execution.runners
  17
  18 import com.intellij.execution.ExecutionHelper
  19 import com.intellij.execution.process.ProcessHandler
  20 import com.intellij.execution.ui.RunContentDescriptor
  21 import com.intellij.openapi.project.Project
  22 import java.util.stream.Collectors
  23
  24 /**
  25  * @author traff
  26  */
  27 open class ConsoleTitleGen @JvmOverloads constructor(private val myProject: Project, private val consoleTitle: String, private val shouldAddNumberToTitle: Boolean = true) {
  28
  29   fun makeTitle(): String {
  30
  31     if (shouldAddNumberToTitle) {
  32       val activeConsoleNames = getActiveConsoles(consoleTitle)
  33       var max = 0
  34       for (name in activeConsoleNames) {
  35         if (max == 0) {
  36           max = 1
  37         }
  38         try {
  39           val num = Integer.parseInt(name.substring(consoleTitle.length + 1, name.length - 1))
  40           if (num > max) {
  41             max = num
  42           }
  43         }
  44         catch (ignored: Exception) {
  45           //skip
  46         }
  47
  48       }
  49       if (max >= 1) {
  50         return consoleTitle + "(" + (max + 1) + ")"
  51       }
  52     }
  53
  54     return consoleTitle
  55   }
  56
  57
  58   open protected fun getActiveConsoles(consoleTitle: String): List<String> {
  59     val consoles = ExecutionHelper.collectConsolesByDisplayName(myProject) { dom -> dom.contains(consoleTitle) }
  60
  61     return consoles.filter({ input ->
  62       val handler = input.processHandler
  63       handler != null && !handler.isProcessTerminated
  64     }).map({ it.displayName });
  65   }
  66 }