chore: import upstream snapshot with attribution

2026-07-13 12:47:58 +08:00
commit b16403ea71
789 changed files with 115226 additions and 0 deletions
@@ -0,0 +1,148 @@
+import * as chalk from 'chalk'
+import * as commander from 'commander'
+
+import { asBold, asTimestamp, withUnderline } from 'src/utils/format'
+import { wait } from 'src/utils/wait'
+import { formatEnum, Format, isRunning } from './utils'
+import { Sandbox } from 'e2b'
+import { ensureAPIKey } from '../../api'
+
+export const metricsCommand = new commander.Command('metrics')
+  .description('show metrics for sandbox')
+  .argument(
+    '<sandboxID>',
+    `show metrics for sandbox specified by ${asBold('<sandboxID>')}`
+  )
+  .alias('mt')
+  .option('-f, --follow', 'keep streaming metrics until the sandbox is closed')
+  .option(
+    '--format <format>',
+    `specify format for printing metrics (${formatEnum(Format)})`,
+    Format.PRETTY
+  )
+  .action(
+    async (
+      sandboxID: string,
+      opts?: {
+        follow: boolean
+        format: Format
+      }
+    ) => {
+      try {
+        const format = opts?.format.toLowerCase() as Format | undefined
+        if (format && !Object.values(Format).includes(format)) {
+          throw new Error(`Invalid log format: ${format}`)
+        }
+
+        let start: Date | undefined
+        let isFirstRun = true
+        let firstMetricsPrinted = false
+
+        if (format === Format.PRETTY) {
+          console.log(`\nMetrics for sandbox ${asBold(sandboxID)}:`)
+        }
+
+        const apiKey = ensureAPIKey()
+        const isRunningPromise = isRunning(sandboxID)
+
+        do {
+          const metrics = await Sandbox.getMetrics(sandboxID, { start, apiKey })
+
+          if (metrics.length !== 0 && !firstMetricsPrinted) {
+            firstMetricsPrinted = true
+            process.stdout.write('\n')
+          }
+
+          for (const metric of metrics) {
+            if (start && metric.timestamp <= start) {
+              // Skip the metric if it has the same timestamp as the last one
+              continue
+            }
+            start = metric.timestamp
+
+            printMetric(metric.timestamp, JSON.stringify(metric), format)
+          }
+
+          const isRunning = await isRunningPromise
+
+          if (!isRunning && metrics.length === 0 && isFirstRun) {
+            if (format === Format.PRETTY) {
+              console.log(
+                `\nStopped printing metrics — sandbox ${withUnderline(
+                  'not found'
+                )}`
+              )
+            }
+            break
+          }
+
+          if (!isRunning) {
+            if (format === Format.PRETTY) {
+              console.log(
+                `\nStopped printing metrics — sandbox is ${withUnderline(
+                  'closed'
+                )}`
+              )
+            }
+            break
+          }
+
+          await wait(400)
+          isFirstRun = false
+        } while (opts?.follow)
+      } catch (err: any) {
+        console.error(err)
+        process.exit(1)
+      }
+    }
+  )
+
+function printMetric(
+  timestamp: Date,
+  line: string,
+  format: Format | undefined
+) {
+  const metric = JSON.parse(line)
+  const level = chalk.default.green()
+
+  if (format === Format.JSON) {
+    console.log(
+      JSON.stringify({
+        timestamp: timestamp.toISOString(),
+        ...metric,
+      })
+    )
+  } else {
+    const time = `[${timestamp
+      .toISOString()
+      .replace(/\.\d{3}Z/, 'Z')
+      .replace(/T/, ' ')}]`
+    delete metric['timestamp']
+    const multipleCores = metric.cpuCount > 1
+    metric.cpuCount += 0
+    console.log(
+      `${asTimestamp(time)} ${level} ` +
+        asBold('CPU') +
+        `: ${metric.cpuUsedPct.toString().padStart(5)}% / ${metric.cpuCount
+          .toString()
+          .padStart(2)} Core${multipleCores && 's'} | ` +
+        asBold('Memory') +
+        `: ${toMB(metric.memUsed).toFixed(0).padStart(5)} / ${toMB(
+          metric.memTotal
+        )
+          .toFixed(0)
+          .padEnd(5)} MiB | ` +
+        asBold('Disk') +
+        `: ${toMB(metric.diskUsed).toFixed(0).padStart(5)} / ${toMB(
+          metric.diskTotal
+        )
+          .toFixed(0)
+          .padEnd(5)} MiB`
+    )
+  }
+}
+
+// we can't use bite shift here because shift is 32 bit operation and disk sizes can be greater than 2^32
+function toMB(bytes: number): number {
+  return bytes / 1024 / 1024
+}