chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,102 @@
|
||||
import { execSync, spawn, spawnSync } from 'node:child_process'
|
||||
import path from 'node:path'
|
||||
|
||||
export const isDebug = process.env.E2B_DEBUG !== undefined
|
||||
|
||||
type CliRunOptions = {
|
||||
timeoutMs: number
|
||||
env?: NodeJS.ProcessEnv
|
||||
}
|
||||
|
||||
type CliRunSyncOptions = CliRunOptions & {
|
||||
input?: string | Buffer
|
||||
}
|
||||
|
||||
export type CliRunResult = {
|
||||
status: number | null
|
||||
stdout: Buffer
|
||||
stderr: Buffer
|
||||
error?: Error
|
||||
}
|
||||
|
||||
const cliPath = path.join(process.cwd(), 'dist', 'index.js')
|
||||
|
||||
export async function setup() {
|
||||
execSync('pnpm build', { stdio: 'inherit' })
|
||||
}
|
||||
|
||||
export function runCli(
|
||||
args: string[],
|
||||
options: CliRunSyncOptions
|
||||
): ReturnType<typeof spawnSync> {
|
||||
return spawnSync('node', [cliPath, ...args], {
|
||||
env: options.env ?? process.env,
|
||||
input: options.input,
|
||||
encoding: 'utf8',
|
||||
timeout: options.timeoutMs,
|
||||
})
|
||||
}
|
||||
|
||||
export async function runCliWithPipedStdin(
|
||||
args: string[],
|
||||
input: Buffer,
|
||||
options: CliRunOptions
|
||||
): Promise<CliRunResult> {
|
||||
return await new Promise((resolve) => {
|
||||
const child = spawn('node', [cliPath, ...args], {
|
||||
env: options.env ?? process.env,
|
||||
stdio: ['pipe', 'pipe', 'pipe'],
|
||||
})
|
||||
|
||||
const stdoutChunks: Buffer[] = []
|
||||
const stderrChunks: Buffer[] = []
|
||||
let childError: Error | undefined
|
||||
let timedOut = false
|
||||
|
||||
const timer = setTimeout(() => {
|
||||
timedOut = true
|
||||
child.kill()
|
||||
}, options.timeoutMs)
|
||||
|
||||
child.stdout.on('data', (chunk) => stdoutChunks.push(Buffer.from(chunk)))
|
||||
child.stderr.on('data', (chunk) => stderrChunks.push(Buffer.from(chunk)))
|
||||
child.on('error', (err) => {
|
||||
childError = err
|
||||
})
|
||||
child.on('close', (code) => {
|
||||
clearTimeout(timer)
|
||||
const timeoutError = timedOut
|
||||
? Object.assign(new Error('CLI command timed out'), {
|
||||
code: 'ETIMEDOUT',
|
||||
} as NodeJS.ErrnoException)
|
||||
: undefined
|
||||
|
||||
resolve({
|
||||
status: code,
|
||||
stdout: Buffer.concat(stdoutChunks),
|
||||
stderr: Buffer.concat(stderrChunks),
|
||||
error: childError ?? timeoutError,
|
||||
})
|
||||
})
|
||||
|
||||
child.stdin.write(input)
|
||||
child.stdin.end()
|
||||
})
|
||||
}
|
||||
|
||||
export function bufferToText(value: Buffer | string | null | undefined): string {
|
||||
if (!value) {
|
||||
return ''
|
||||
}
|
||||
return typeof value === 'string' ? value : value.toString('utf8')
|
||||
}
|
||||
|
||||
export function parseEnvInt(name: string, fallback: number): number {
|
||||
const raw = process.env[name]
|
||||
if (!raw) {
|
||||
return fallback
|
||||
}
|
||||
|
||||
const parsed = Number.parseInt(raw, 10)
|
||||
return Number.isFinite(parsed) ? parsed : fallback
|
||||
}
|
||||
Reference in New Issue
Block a user