From 61b14551c19fbfb40b25464df80be5ccb6b693b4 Mon Sep 17 00:00:00 2001
From: Ruben Fiszel <ruben@rubenfiszel.com>
Date: Sat, 23 Jul 2022 18:28:41 +0200
Subject: [PATCH] websocket auto-connect

---
 frontend/src/lib/components/Editor.svelte     | 20 ++++++++++++++++---
 .../src/lib/components/ScriptEditor.svelte    |  1 -
 2 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/frontend/src/lib/components/Editor.svelte b/frontend/src/lib/components/Editor.svelte
index c82c563bca..da40cfd332 100644
--- a/frontend/src/lib/components/Editor.svelte
+++ b/frontend/src/lib/components/Editor.svelte
@@ -1,7 +1,7 @@
 <script lang="ts">
 	import { browser, dev } from '$app/env'
 	import { page } from '$app/stores'
-	import { buildExtraLib } from '$lib/utils'
+	import { buildExtraLib, sendUserToast } from '$lib/utils'
 	import type monaco from 'monaco-editor'
 	import editorWorker from 'monaco-editor/esm/vs/editor/editor.worker?worker'
 	import jsonWorker from 'monaco-editor/esm/vs/language/json/json.worker?worker'
@@ -24,6 +24,7 @@
 	export let extraLibPath: string = 'file:///node_modules/@types/windmill/index.d.ts'
 
 	let websockets: WebSocket[] = []
+	let websocketInterval: NodeJS.Timer | undefined
 	let uri: string = ''
 	let disposeMethod: () => void | undefined
 	const dispatch = createEventDispatcher()
@@ -153,14 +154,16 @@
 						const writer = new WebSocketMessageWriter(socket)
 						const languageClient = createLanguageClient({ reader, writer }, name, options)
 						languageClient.start()
-						socket.onClose((_code, _reason) => {
-							websocketAlive[name] = false
+						reader.onClose(() => {
 							try {
 								languageClient.stop()
 							} catch (err) {
 								console.error(err)
 							}
 						})
+						socket.onClose((_code, _reason) => {
+							websocketAlive[name] = false
+						})
 						if (name == 'deno') {
 							vscode.commands.getCommands().then((v) => {
 								if (!v.includes('deno.cache')) {
@@ -236,6 +239,16 @@
 					}
 				})
 			}
+			websocketInterval = setInterval(() => {
+				if (document.visibilityState == 'visible') {
+					if (!websocketAlive.black && !websocketAlive.deno && !websocketAlive.pyright) {
+						sendUserToast(
+							'Smart assistant got disconnected. Reconnecting to windmill language server for smart assistance'
+						)
+						reloadWebsocket()
+					}
+				}
+			}, 5000)
 		}
 	}
 
@@ -248,6 +261,7 @@
 			}
 		}
 		websockets = []
+		websocketInterval && clearInterval(websocketInterval)
 	}
 
 	async function loadMonaco() {
diff --git a/frontend/src/lib/components/ScriptEditor.svelte b/frontend/src/lib/components/ScriptEditor.svelte
index b5ba1c74c9..0b4d28fae6 100644
--- a/frontend/src/lib/components/ScriptEditor.svelte
+++ b/frontend/src/lib/components/ScriptEditor.svelte
@@ -223,7 +223,6 @@
 >
 
 <VSplitPane
-	class="h-full"
 	topPanelSize={viewPreview ? '75%' : '90%'}
 	downPanelSize={viewPreview ? '25%' : '10%'}
 	updateCallback={() => {