Add code snippet highlight to CLI

.changeset/cold-yaks-dress.md

@@ -0,0 +1,5 @@
+---
+"@e2b/cli": patch
+---
+
+Add highlight to CLI code snippet examples

packages/cli/package.json

@@ -55,6 +55,7 @@
     "async-listen": "^3.0.1",
     "boxen": "^7.1.1",
     "chalk": "^5.3.0",
+    "cli-highlight": "^2.1.11",
     "commander": "^11.1.0",
     "common-tags": "^1.8.2",
     "console-table-printer": "^2.11.2",

packages/cli/src/commands/build.ts

@@ -14,6 +14,8 @@ import {
   asLocal,
   asLocalRelative,
   asPrimary,
+  asPython,
+  asTypescript,
 } from 'src/utils/format'
 import { pathOption } from 'src/options'
 import { createBlobFromFiles } from 'src/docker/archive'
@@ -122,7 +124,7 @@ export const buildCommand = new commander.Command('build')
         const name = newName || config?.name
 
         console.log(
-          `Preparing sandbox template building (${filePaths.length} files in Docker build context). ${!filePaths.length ? `If you are using ${asLocal('.dockerignore')} check if it is configured correctly.`:''}`,
+          `Preparing sandbox template building (${filePaths.length} files in Docker build context). ${!filePaths.length ? `If you are using ${asLocal('.dockerignore')} check if it is configured correctly.` : ''}`,
         )
 
         const { dockerfileContent, dockerfileRelativePath } = getDockerfile(
@@ -241,9 +243,10 @@ async function waitForBuildFinish(
           )} finished.\n
           Now you can start creating your sandboxes from this template. You can find more here:
           ${asPrimary('https://e2b.dev/docs/guide/custom-sandbox')}, section ${asBold('Spawn and control your sandbox.')}
+
           | ${asBold('Python SDK')} |
           ==============
-          from e2b import Sandbox
+          ${asPython(`from e2b import Sandbox
 
           # Start sandbox
           sandbox = Sandbox(id="${aliases?.length ? aliases[0] : template.data.envID}")
@@ -252,12 +255,12 @@ async function waitForBuildFinish(
           # https://e2b.dev/docs/sandbox/overview
 
           # Close sandbox once done
-          sandbox.close()
-
+          sandbox.close()`)
+          }
 
           | ${asBold('JS SDK')} |
           ==========
-          import { Sandbox } from '@e2b/sdk'
+          ${asTypescript(`import { Sandbox } from '@e2b/sdk'
 
           // Start sandbox
           const sandbox = await Sandbox.create({ id: '${aliases?.length ? aliases[0] : template.data.envID}' })
@@ -266,7 +269,8 @@ async function waitForBuildFinish(
           // https://e2b.dev/docs/sandbox/overview
 
           // Close sandbox once done
-          await sandbox.close()
+          await sandbox.close()`)
+          }
           `,
         )
         break

packages/cli/src/commands/shell.ts

@@ -23,7 +23,6 @@ export const shellCommand = new commander.Command('shell')
       const apiKey = ensureAPIKey()
       let envID = id
 
-
       const root = getRoot(opts.path)
       const configPath = getConfigPath(root)
 

packages/cli/src/utils/format.ts

@@ -1,5 +1,6 @@
 import * as chalk from 'chalk'
 import * as e2b from '@e2b/sdk'
+import * as highlight from 'cli-highlight'
 
 import { cwdRelative } from './filesystem'
 
@@ -53,7 +54,16 @@ export function asBuildLogs(content: string) {
 export function asHeadline(content: string) {
   return chalk.default.underline(asPrimary(asBold(content)))
 }
+
 export function listAliases(aliases: string[] | undefined) {
   if (!aliases) return undefined
   return aliases.join(' | ')
 }
+
+export function asTypescript(code: string) {
+  return highlight.default(code, { language: 'typescript', ignoreIllegals: true })
+}
+
+export function asPython(code: string) {
+  return highlight.default(code, { language: 'python', ignoreIllegals: true })
+}