Persist theme - Reload everything on a disconnect (#3250)

* Reload everything on a disconnect

* fix unit-integration tests

* Further improvements to connection manager; persist theme across reconnects

* Fix up artifactGraph.test

* Actually pass the callback

* Kurt hmmm (#3308)

* kurts attempts

* we're almost sane

* get tests working, praise be

---------

Co-authored-by: 49lf <ircsurfer33@gmail.com>

* typo

---------

Co-authored-by: Kurt Hutten Irev-Dev <k.hutten@protonmail.ch>

src/lang/std/artifactGraph.test.ts

@@ -14,6 +14,10 @@ import {
 } from './artifactGraph'
 import { err } from 'lib/trap'
 import { engineCommandManager, kclManager } from 'lib/singletons'
+import {
+  EngineCommandManagerEvents,
+  EngineConnectionEvents,
+} from 'lang/std/engineConnection'
 import { CI, VITE_KC_DEV_TOKEN } from 'env'
 import fsp from 'fs/promises'
 import fs from 'fs'
@@ -113,42 +117,44 @@ beforeAll(async () => {
   }
 
   // THESE TEST WILL FAIL without VITE_KC_DEV_TOKEN set in .env.development.local
-  engineCommandManager.start({
-    disableWebRTC: true,
-    token: VITE_KC_DEV_TOKEN,
-    // there does seem to be a minimum resolution, not sure what it is but 256 works ok.
-    width: 256,
-    height: 256,
-    executeCode: () => {},
-    makeDefaultPlanes: () => makeDefaultPlanes(engineCommandManager),
-    setMediaStream: () => {},
-    setIsStreamReady: () => {},
-    modifyGrid: async () => {},
+  await new Promise((resolve) => {
+    engineCommandManager.start({
+      // disableWebRTC: true,
+      token: VITE_KC_DEV_TOKEN,
+      // there does seem to be a minimum resolution, not sure what it is but 256 works ok.
+      width: 256,
+      height: 256,
+      makeDefaultPlanes: () => makeDefaultPlanes(engineCommandManager),
+      setMediaStream: () => {},
+      setIsStreamReady: () => {},
+      modifyGrid: async () => {},
+      callbackOnEngineLiteConnect: async () => {
+        const cacheEntries = Object.entries(codeToWriteCacheFor) as [
+          CodeKey,
+          string
+        ][]
+        const cacheToWriteToFileTemp: Partial<CacheShape> = {}
+        for (const [codeKey, code] of cacheEntries) {
+          const ast = parse(code)
+          if (err(ast)) {
+            console.error(ast)
+            return Promise.reject(ast)
+          }
+          const result = await kclManager.executeAst(ast)
+
+          cacheToWriteToFileTemp[codeKey] = {
+            orderedCommands: engineCommandManager.orderedCommands,
+            responseMap: engineCommandManager.responseMap,
+          }
+        }
+        const cache = JSON.stringify(cacheToWriteToFileTemp)
+
+        await fsp.mkdir(pathStart, { recursive: true })
+        await fsp.writeFile(fullPath, cache)
+        resolve(true)
+      },
+    })
   })
-  await engineCommandManager.waitForReady
-
-  const cacheEntries = Object.entries(codeToWriteCacheFor) as [
-    CodeKey,
-    string
-  ][]
-  const cacheToWriteToFileTemp: Partial<CacheShape> = {}
-  for (const [codeKey, code] of cacheEntries) {
-    const ast = parse(code)
-    if (err(ast)) {
-      console.error(ast)
-      throw ast
-    }
-    await kclManager.executeAst(ast)
-
-    cacheToWriteToFileTemp[codeKey] = {
-      orderedCommands: engineCommandManager.orderedCommands,
-      responseMap: engineCommandManager.responseMap,
-    }
-  }
-  const cache = JSON.stringify(cacheToWriteToFileTemp)
-
-  await fsp.mkdir(pathStart, { recursive: true })
-  await fsp.writeFile(fullPath, cache)
 }, 20_000)
 
 afterAll(() => {