CodeContext

interface CodeContext {
  readonly id: string;
  readonly language: string;
  readonly cwd: string;
  readonly createdAt: Date;
  readonly lastUsed: Date;
}

import { getSandbox } from '@cloudflare/sandbox';

const sandbox = getSandbox(env.SANDBOX, 'my-sandbox');

// Create a Python context
const context = await sandbox.createCodeContext({
  language: 'python',
  cwd: '/workspace/analysis'
});

console.log('Context ID:', context.id);
console.log('Language:', context.language);
console.log('Working dir:', context.cwd);
console.log('Created:', context.createdAt.toISOString());

// Use the context
await sandbox.runCode('import pandas as pd', { context });
await sandbox.runCode('df = pd.read_csv("data.csv")', { context });

// Context maintains state between executions
await sandbox.runCode('print(df.head())', { context });

// Create separate contexts for different tasks
const analysisContext = await sandbox.createCodeContext({
  language: 'python',
  cwd: '/workspace/analysis'
});

const visualizationContext = await sandbox.createCodeContext({
  language: 'python',
  cwd: '/workspace/viz'
});

// Set up analysis context
await sandbox.runCode(
  'import pandas as pd\ndf = pd.read_csv("data.csv")',
  { context: analysisContext }
);

// Set up visualization context
await sandbox.runCode(
  'import matplotlib.pyplot as plt',
  { context: visualizationContext }
);

// Contexts are isolated - variables don't leak
await sandbox.runCode('print(df)', { context: analysisContext }); // Works
await sandbox.runCode('print(df)', { context: visualizationContext }); // Error: df not defined

// Create isolated contexts for different users
const contexts = new Map<string, CodeContext>();

async function getUserContext(userId: string) {
  if (!contexts.has(userId)) {
    const context = await sandbox.createCodeContext({
      language: 'python',
      cwd: `/workspace/users/${userId}`
    });
    contexts.set(userId, context);
  }
  return contexts.get(userId)!;
}

// Each user gets their own isolated context
const user1Context = await getUserContext('user-1');
const user2Context = await getUserContext('user-2');

await sandbox.runCode('x = 42', { context: user1Context });
await sandbox.runCode('x = 99', { context: user2Context });

const result1 = await sandbox.runCode('print(x)', { context: user1Context });
console.log(result1.logs.stdout); // ["42"]

const result2 = await sandbox.runCode('print(x)', { context: user2Context });
console.log(result2.logs.stdout); // ["99"]

// List and clean up old contexts
const contexts = await sandbox.listCodeContexts();
const oneHourAgo = new Date(Date.now() - 60 * 60 * 1000);

for (const context of contexts) {
  if (context.lastUsed < oneHourAgo) {
    console.log(`Deleting unused context ${context.id}`);
    await sandbox.deleteCodeContext(context.id);
  }
}