use provider.onDidRegisterContainerConnection to know when providers …

…are up
containers · Jan 31, 2024 · 0357897 · 0357897
1 parent 44c74bb
commit 0357897
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/packages/backend/src/managers/playground.ts b/packages/backend/src/managers/playground.ts
@@ -64,6 +64,14 @@ export class PlayGroundManager {
   }
 
   async adoptRunningPlaygrounds() {
+    provider.onDidRegisterContainerConnection(async () => {
+      await this.doAdoptRunningPlaygrounds();
+    });
+    // Do it now in case providers are already registered
+    await this.doAdoptRunningPlaygrounds();
+  }
+
+  private async doAdoptRunningPlaygrounds() {
     const containers = await containerEngine.listContainers();
     const playgroundContainers = containers.filter(
       c => LABEL_MODEL_ID in c.Labels && LABEL_MODEL_PORT in c.Labels && c.State === 'running',

diff --git a/packages/backend/src/studio.spec.ts b/packages/backend/src/studio.spec.ts
@@ -54,6 +54,9 @@ vi.mock('@podman-desktop/api', async () => {
       onEvent: vi.fn(),
       listContainers: mocks.listContainers,
     },
+    provider: {
+      onDidRegisterContainerConnection: vi.fn(),
+    },
   };
 });