Added TypeScript support to the code interpreter (#91)

* added typescript support via SWC

* added ts tests

* changed target node version

* minor nits by @0div

* added test for typescript kernel failure

* changed swc path in prod Dockerfile

* added changeset

* skip test here so that the template can be built

Author: mishushakov

.changeset/tiny-adults-breathe.md

@@ -0,0 +1,5 @@
+---
+'@e2b/code-interpreter-template': patch
+---
+
+added typescript support

js/tests/defaultKernels.test.ts

@@ -3,6 +3,8 @@ import { expect } from 'vitest'
 import { sandboxTest } from './setup'
 
 sandboxTest('test js kernel', async ({ sandbox }) => {
-  const output = await sandbox.runCode('console.log("Hello World!")', { language: 'js' })
+  const output = await sandbox.runCode('console.log("Hello World!")', {
+    language: 'js',
+  })
   expect(output.logs.stdout).toEqual(['Hello World!\n'])
 })

template/.ts.swcrc

@@ -12,4 +12,4 @@
     "targets": "node 20"
   },
   "isModule": false
-}
+}

template/server/contexts.py

@@ -11,6 +11,11 @@
 
 logger = logging.Logger(__name__)
 
+def get_kernel_for_language(language: str) -> str:
+    if language == "typescript":
+        return "javascript"
+
+    return language
 
 def normalize_language(language: Optional[str]) -> str:
     if not language:
@@ -21,13 +26,16 @@ def normalize_language(language: Optional[str]) -> str:
     if language == "js":
         return "javascript"
 
+    if language == "ts":
+        return "typescript"
+
     return language
 
 
 async def create_context(client, websockets: dict, language: str, cwd: str) -> Context:
     data = {
         "path": str(uuid.uuid4()),
-        "kernel": {"name": language},
+        "kernel": {"name": get_kernel_for_language(language)},
         "type": "notebook",
         "name": str(uuid.uuid4()),
     }

template/server/messaging.py

@@ -3,6 +3,7 @@
 import logging
 import uuid
 import asyncio
+import subprocess
 
 from asyncio import Queue
 from envs import get_envs
@@ -27,7 +28,6 @@
 
 logger = logging.getLogger(__name__)
 
-
 class Execution:
     def __init__(self, in_background: bool = False):
         self.queue = Queue[
@@ -199,6 +199,32 @@ async def execute(
                         + code
                     )
 
+            if self.language == "typescript":
+                logger.info("Compiling TypeScript: %s", code)
+
+                # call SWC to compile the typescript code
+                try:
+                    compile_result = subprocess.run("swc --config-file .ts.swcrc --filename index.ts".split(), input=code.encode(), capture_output=True)
+
+                    if compile_result.returncode != 0:
+                        logger.error("Error during TypeScript compilation: %s", compile_result.stderr.decode())
+                        yield Error(
+                            name="TypeScriptCompilerError",
+                            value=compile_result.stderr.decode(),
+                            traceback="",
+                        )
+                        return
+
+                    code = compile_result.stdout.decode()
+                except Exception as e:
+                    logger.error("Error starting SWC process: %s", e)
+                    yield Error(
+                        name="TypeScriptCompilerError",
+                        value=str(e),
+                        traceback="",
+                    )
+                    return
+
             logger.info(code)
             request = self._get_execute_request(message_id, code, False)