Add filtering sandboxes by state in the SDKs

packages/js-sdk/src/sandbox/sandboxApi.ts

@@ -9,13 +9,18 @@ import { NotFoundError, TemplateError } from '../errors'
 export interface SandboxApiOpts
   extends Partial<
     Pick<ConnectionOpts, 'apiKey' | 'debug' | 'domain' | 'requestTimeoutMs'>
-  > { }
+  > {}
 
 export interface SandboxListOpts extends SandboxApiOpts {
   /**
    * Filter the list of sandboxes by metadata, e.g. `{"key": "value"}`, if there are multiple filters they are combined with AND.
    */
   filters?: Record<string, string>
+
+  /**
+   * Filter the list of sandboxes by state.
+   */
+  state?: 'running' | 'paused'
 }
 
 /**
@@ -46,10 +51,15 @@ export interface SandboxInfo {
    * Sandbox start time.
    */
   startedAt: Date
+
+  /**
+   * Sandbox state.
+   */
+  state: 'running' | 'paused'
 }
 
 export class SandboxApi {
-  protected constructor() { }
+  protected constructor() {}
 
   /**
    * Kill the sandbox specified by sandbox ID.
@@ -94,21 +104,26 @@ export class SandboxApi {
    *
    * @returns list of running sandboxes.
    */
-  static async list(
-      opts?: SandboxListOpts): Promise<SandboxInfo[]> {
+  static async list(opts?: SandboxListOpts): Promise<SandboxInfo[]> {
     const config = new ConnectionConfig(opts)
     const client = new ApiClient(config)
 
     let query = undefined
     if (opts?.filters) {
-      const encodedPairs: Record<string, string> = Object.fromEntries(Object.entries(opts.filters).map(([key, value]) => [encodeURIComponent(key),encodeURIComponent(value)]))
+      const encodedPairs: Record<string, string> = Object.fromEntries(
+        Object.entries(opts.filters).map(([key, value]) => [
+          encodeURIComponent(key),
+          encodeURIComponent(value),
+        ])
+      )
       query = new URLSearchParams(encodedPairs).toString()
     }
 
     const res = await client.api.GET('/sandboxes', {
-        params: {
-          query: {query},
-        },
+      params: {
+        query: { query },
+        state: opts?.state,
+      },
       signal: config.getSignal(opts?.requestTimeoutMs),
     })
 
@@ -127,6 +142,7 @@ export class SandboxApi {
         ...(sandbox.alias && { name: sandbox.alias }),
         metadata: sandbox.metadata ?? {},
         startedAt: new Date(sandbox.startedAt),
+        state: sandbox.state,
       })) ?? []
     )
   }
@@ -207,13 +223,13 @@ export class SandboxApi {
   }
 
   /**
- * Pause the sandbox specified by sandbox ID.
- *
- * @param sandboxId sandbox ID.
- * @param opts connection options.
- *
- * @returns `true` if the sandbox got paused, `false` if the sandbox was already paused.
- */
+   * Pause the sandbox specified by sandbox ID.
+   *
+   * @param sandboxId sandbox ID.
+   * @param opts connection options.
+   *
+   * @returns `true` if the sandbox got paused, `false` if the sandbox was already paused.
+   */
   protected static async pauseSandbox(
     sandboxId: string,
     opts?: SandboxApiOpts
@@ -247,7 +263,6 @@ export class SandboxApi {
     return true
   }
 
-
   protected static async resumeSandbox(
     sandboxId: string,
     timeoutMs: number,
@@ -324,7 +339,7 @@ export class SandboxApi {
       )
       throw new TemplateError(
         'You need to update the template to use the new SDK. ' +
-        'You can do this by running `e2b template build` in the directory with the template.'
+          'You can do this by running `e2b template build` in the directory with the template.'
       )
     }
     return {

packages/js-sdk/tests/api/list.test.ts

@@ -24,11 +24,11 @@ sandboxTest.skipIf(isDebug)('list sandboxes', async ({ sandbox }) => {
 sandboxTest.skipIf(isDebug)('list sandboxes with filter', async () => {
   const uniqueId = Date.now().toString()
   // Create an extra sandbox with a uniqueId
-  const extraSbx = await Sandbox.create({ })
+  const extraSbx = await Sandbox.create({})
   try {
-    const sbx = await Sandbox.create({metadata: {uniqueId: uniqueId}})
+    const sbx = await Sandbox.create({ metadata: { uniqueId: uniqueId } })
     try {
-      const sandboxes = await Sandbox.list({filters: {uniqueId}})
+      const sandboxes = await Sandbox.list({ filters: { uniqueId } })
       assert.equal(sandboxes.length, 1)
       assert.equal(sandboxes[0].sandboxId, sbx.sandboxId)
     } finally {
@@ -38,3 +38,15 @@ sandboxTest.skipIf(isDebug)('list sandboxes with filter', async () => {
     await extraSbx.kill()
   }
 })
+
+sandboxTest.skipIf(isDebug)('list paused sandboxes', async ({ sandbox }) => {
+  const pausedSandbox = await sandbox.pause()
+  const pausedSandboxId = pausedSandbox.split('-')[0] + '-' + '00000000'
+  const sandboxes = await Sandbox.list({ state: 'paused' })
+
+  assert.isAtLeast(sandboxes.length, 1)
+  assert.include(
+    sandboxes.map((s) => s.sandboxId),
+    pausedSandboxId
+  )
+})