Close mongo and casssandra terminal tabs once the shells are exited (#1183)

* initial commit for closing terminal

* added extra case

* lint changes and hostee explorer fixes

* fixed lint errors

* fixed compile error

* fixed review comments

src/Terminal/JupyterLabAppFactory.ts

@@ -2,15 +2,48 @@
  * JupyterLab applications based on jupyterLab components
  */
 import { ServerConnection, TerminalManager } from "@jupyterlab/services";
+import { IMessage } from "@jupyterlab/services/lib/terminal/terminal";
 import { Terminal } from "@jupyterlab/terminal";
 import { Panel, Widget } from "@phosphor/widgets";
+import { userContext } from "UserContext";
 
 export class JupyterLabAppFactory {
-  public static async createTerminalApp(serverSettings: ServerConnection.ISettings) {
+  private isShellClosed: boolean;
+  private onShellExited: () => void;
+  private checkShellClosed: ((content: string | undefined) => boolean | undefined) | undefined;
+
+  constructor(closeTab: () => void) {
+    this.onShellExited = closeTab;
+    this.isShellClosed = false;
+    this.checkShellClosed = undefined;
+
+    switch (userContext.apiType) {
+      case "Mongo":
+        this.checkShellClosed = JupyterLabAppFactory.isMongoShellClosed;
+        break;
+      case "Cassandra":
+        this.checkShellClosed = JupyterLabAppFactory.isCassandraShellClosed;
+        break;
+    }
+  }
+
+  public async createTerminalApp(serverSettings: ServerConnection.ISettings) {
     const manager = new TerminalManager({
       serverSettings: serverSettings,
     });
     const session = await manager.startNew();
+    session.messageReceived.connect(async (_, message: IMessage) => {
+      const content = message.content && message.content[0]?.toString();
+      if (this.checkShellClosed && message.type == "stdout") {
+        //Close the terminal tab once the shell closed messages are received
+        if (this.checkShellClosed(content)) {
+          this.isShellClosed = true;
+        } else if (content?.includes("cosmosuser@") && this.isShellClosed) {
+          this.onShellExited();
+        }
+      }
+    }, this);
+
     const term = new Terminal(session, { theme: "dark", shutdownOnClose: true });
 
     if (!term) {
@@ -38,4 +71,12 @@ export class JupyterLabAppFactory {
       panel.dispose();
     });
   }
+
+  private static isMongoShellClosed(content: string | undefined) {
+    return content?.endsWith("bye\r\n") || (content?.includes("Stopped") && content?.includes("mongo --host"));
+  }
+
+  private static isCassandraShellClosed(content: string | undefined) {
+    return content == "\r\n" || (content?.includes("Stopped") && content?.includes("cqlsh"));
+  }
 }

src/Terminal/TerminalProps.ts

@@ -10,4 +10,5 @@ export interface TerminalProps {
   authType: AuthType;
   apiType: ApiType;
   subscriptionId: string;
+  tabId: string;
 }

src/Terminal/index.ts

@@ -1,5 +1,6 @@
 import { ServerConnection } from "@jupyterlab/services";
 import "@jupyterlab/terminal/style/index.css";
+import { MessageTypes } from "Contracts/ExplorerContracts";
 import postRobot from "post-robot";
 import { HttpHeaders } from "../Common/Constants";
 import { Action } from "../Shared/Telemetry/TelemetryConstants";
@@ -54,13 +55,20 @@ const initTerminal = async (props: TerminalProps) => {
   const startTime = TelemetryProcessor.traceStart(Action.OpenTerminal, data);
 
   try {
-    await JupyterLabAppFactory.createTerminalApp(serverSettings);
+    await new JupyterLabAppFactory(() => closeTab(props.tabId)).createTerminalApp(serverSettings);
     TelemetryProcessor.traceSuccess(Action.OpenTerminal, data, startTime);
   } catch (error) {
     TelemetryProcessor.traceFailure(Action.OpenTerminal, data, startTime);
   }
 };
 
+const closeTab = (tabId: string): void => {
+  window.parent.postMessage(
+    { type: MessageTypes.CloseTab, data: { tabId: tabId }, signature: "pcIframe" },
+    window.document.referrer
+  );
+};
+
 const main = async (): Promise<void> => {
   postRobot.on(
     "props",