diff --git a/src/Terminal/JupyterLabAppFactory.ts b/src/Terminal/JupyterLabAppFactory.ts index fa2ce4772..a9a64740a 100644 --- a/src/Terminal/JupyterLabAppFactory.ts +++ b/src/Terminal/JupyterLabAppFactory.ts @@ -17,13 +17,17 @@ export class JupyterLabAppFactory { } private isMongoShellStarted(content: string | undefined) { - this.isShellStarted = content?.includes("MongoDB shell version") || content?.includes("citus=>"); + this.isShellStarted = content?.includes("MongoDB shell version"); } private isCassandraShellStarted(content: string | undefined) { this.isShellStarted = content?.includes("Connected to") && content?.includes("cqlsh"); } + private isPostgresShellStarted(content: string | undefined) { + this.isShellStarted = content?.includes("cqlsh"); + } + constructor(closeTab: () => void) { this.onShellExited = closeTab; this.isShellStarted = false; @@ -36,6 +40,9 @@ export class JupyterLabAppFactory { case "Cassandra": this.checkShellStarted = this.isCassandraShellStarted; break; + case "Postgres": + this.checkShellStarted = this.isPostgresShellStarted; + break; } }