6bd14252e9281059785d12bae1a7d10647c18215
[idea/community.git] / 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  * Created by Yuli Fiterman on 9/13/2016.
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 }