feat: Update file handling API endpoints

Author: Gerome El-assaad

app/api/files/content/route.ts

@@ -1,5 +1,11 @@
 import { NextResponse } from 'next/server'
 import { Sandbox } from '@e2b/code-interpreter'
+import { LRUCache } from 'lru-cache'
+
+const cache = new LRUCache<string, string>({
+  max: 500,
+  ttl: 1000 * 60 * 5, // 5 minutes
+})
 
 export const runtime = 'nodejs'
 export const dynamic = 'force-dynamic'
@@ -41,8 +47,18 @@ export async function GET(req: Request) {
       )
     }
 
+    const cacheKey = `${sessionID}:${path}`
+    const cachedContent = cache.get(cacheKey)
+
+    if (cachedContent) {
+      return NextResponse.json({ content: cachedContent })
+    }
+
     const sandbox = await getSandbox(sessionID, template || undefined)
     const content = await sandbox.files.read(path)
+
+    cache.set(cacheKey, content)
+
     return NextResponse.json({ content })
   } catch (error: any) {
     return NextResponse.json(
@@ -65,6 +81,10 @@ export async function POST(req: Request) {
 
     const sandbox = await getSandbox(sessionID, template || undefined)
     await sandbox.files.write(path, content)
+
+    const cacheKey = `${sessionID}:${path}`
+    cache.delete(cacheKey)
+
     return NextResponse.json({ success: true })
   } catch (error: any) {
     return NextResponse.json(

app/api/files/route.ts

@@ -1,6 +1,7 @@
 import { NextRequest, NextResponse } from 'next/server'
 import { Sandbox, FileType } from '@e2b/code-interpreter'
 import { FileSystemNode } from '@/components/file-tree'
+import { getSandbox } from '@/lib/sandbox'
 
 export const runtime = 'nodejs'
 export const dynamic = 'force-dynamic'
@@ -31,26 +32,6 @@ async function listFilesRecursively(
   return nodes
 }
 
-const E2B_API_KEY = process.env.E2B_API_KEY
-
-const sandboxTimeout = 10 * 60 * 1000
-
-async function getSandbox(sessionID: string, template?: string) {
-  if (!E2B_API_KEY) {
-    throw new Error('E2B_API_KEY environment variable not found')
-  }
-
-  const sandbox = await Sandbox.create(template || 'code-interpreter-v1', {
-    apiKey: E2B_API_KEY,
-    metadata: {
-      sessionID,
-      template: template || 'code-interpreter-v1',
-    },
-    timeoutMs: sandboxTimeout,
-  })
-  return sandbox
-}
-
 export async function GET(request: NextRequest) {
   const searchParams = request.nextUrl.searchParams
   const sessionID = searchParams.get('sessionID')