Exit Gitpod Task when terminal widget is closed, close terminal widget when Gitpod Task exits, and update terminal widget when Gitpod Task title changes

Author: felladrin

components/ide/jetbrains/backend-plugin/src/main/kotlin/io/gitpod/jetbrains/remote/GitpodTerminalService.kt

@@ -8,13 +8,14 @@ import com.intellij.openapi.client.ClientProjectSession
 import com.intellij.openapi.diagnostic.thisLogger
 import com.intellij.util.application
 import com.jetbrains.rdserver.terminal.BackendTerminalManager
-import io.gitpod.jetbrains.remote.GitpodManager
 import io.gitpod.supervisor.api.Status
 import io.gitpod.supervisor.api.StatusServiceGrpc
 import io.gitpod.supervisor.api.TerminalOuterClass
 import io.gitpod.supervisor.api.TerminalServiceGrpc
+import io.grpc.StatusRuntimeException
 import io.grpc.stub.ClientCallStreamObserver
 import io.grpc.stub.ClientResponseObserver
+import io.grpc.stub.StreamObserver
 import org.jetbrains.plugins.terminal.ShellTerminalWidget
 import org.jetbrains.plugins.terminal.TerminalView
 import java.util.*
@@ -30,6 +31,7 @@ class GitpodTerminalService(session: ClientProjectSession) {
     private val terminalView = TerminalView.getInstance(session.project)
     private val backendTerminalManager = BackendTerminalManager.getInstance(session.project)
     private val terminalServiceFutureStub = TerminalServiceGrpc.newFutureStub(GitpodManager.supervisorChannel)
+    private val terminalServiceStub = TerminalServiceGrpc.newStub(GitpodManager.supervisorChannel)
     private val statusServiceStub = StatusServiceGrpc.newStub(GitpodManager.supervisorChannel)
 
     init { start() }
@@ -49,7 +51,7 @@ class GitpodTerminalService(session: ClientProjectSession) {
         }
     }
 
-    private fun createSharedTerminalAndExecuteCommand(title: String, command: String) {
+    private fun createSharedTerminalAndExecuteCommand(title: String, command: String): ShellTerminalWidget? {
         val registeredTerminals = terminalView.widgets.toMutableList()
 
         backendTerminalManager.createNewSharedTerminal(UUID.randomUUID().toString(), title)
@@ -59,8 +61,14 @@ class GitpodTerminalService(session: ClientProjectSession) {
 
             widget.terminalTitle.change { applicationTitle = title }
 
-            (widget as ShellTerminalWidget).executeCommand(command)
+            val shellTerminalWidget = widget as ShellTerminalWidget
+
+            shellTerminalWidget.executeCommand(command)
+
+            return shellTerminalWidget
         }
+
+        return null
     }
 
     private fun createTerminalsAttachedToTasks(
@@ -120,8 +128,9 @@ class GitpodTerminalService(session: ClientProjectSession) {
             }
 
             thisLogger().error(
-                "gitpod: Got an error while trying to get tasks list from Supervisor. Trying again in on second.",
-                throwable
+                    "gitpod: Got an error while trying to get tasks list from Supervisor. " +
+                            "Trying again in one second.",
+                    throwable
             )
         }
 
@@ -150,8 +159,9 @@ class GitpodTerminalService(session: ClientProjectSession) {
             }
 
             thisLogger().error(
-                "gitpod: Got an error while trying to get terminals list from Supervisor. Trying again in on second.",
-                throwable
+                    "gitpod: Got an error while trying to get terminals list from Supervisor. " +
+                            "Trying again in one second.",
+                    throwable
             )
         }
 
@@ -164,9 +174,45 @@ class GitpodTerminalService(session: ClientProjectSession) {
     }
 
     private fun createAttachedSharedTerminal(supervisorTerminal: TerminalOuterClass.Terminal) {
-        createSharedTerminalAndExecuteCommand(
+        val shellTerminalWidget = createSharedTerminalAndExecuteCommand(
             supervisorTerminal.title,
             "gp tasks attach ${supervisorTerminal.alias}"
+        ) ?: return
+
+        shellTerminalWidget.addListener {
+            terminalServiceFutureStub.shutdown(
+                    TerminalOuterClass.ShutdownTerminalRequest.newBuilder()
+                            .setAlias(supervisorTerminal.alias)
+                            .build()
+            )
+        }
+
+        terminalServiceStub.listen(
+                TerminalOuterClass.ListenTerminalRequest.newBuilder()
+                        .setAlias(supervisorTerminal.alias)
+                        .build(),
+                object : StreamObserver<TerminalOuterClass.ListenTerminalResponse> {
+                    override fun onNext(response: TerminalOuterClass.ListenTerminalResponse) {
+                        when {
+                            response.hasTitle() -> shellTerminalWidget.terminalTitle.change {
+                                applicationTitle = response.title
+                            }
+                            response.hasExitCode() -> shellTerminalWidget.close()
+                        }
+                    }
+
+                    override fun onCompleted() = Unit
+
+                    override fun onError(throwable: Throwable) {
+                        if (throwable is StatusRuntimeException &&
+                                (throwable.status.description == "signal: terminated" ||
+                                        throwable.status.description == "signal: killed")) return
+
+                        thisLogger()
+                                .error("gitpod: Got an error while listening to " +
+                                        "'${supervisorTerminal.title}' terminal.", throwable)
+                    }
+                }
         )
     }
 }