probablycorey/shrimp
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

allow passing in extra globals

Browse Source
This commit is contained in:
Chris Wanstrath 2025-11-03 13:44:12 -08:00
parent 402748d1da
commit 90a1f63847
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/index.ts
View File

@ -1,25 +1,24 @@
import { readFileSync } from 'fs'
import { VM, fromValue, type Bytecode } from 'reefvm'
import { Compiler } from '#compiler/compiler'
import { globals, colors } from '#prelude'
import { globals as shrimpGlobals, colors } from '#prelude'
export { Compiler } from '#compiler/compiler'
export { parser } from '#parser/shrimp'
export { globals } from '#prelude'
export async function runFile(path: string): Promise<any> {
export async function runFile(path: string, globals?: Record<string, any>): Promise<any> {
const code = readFileSync(path, 'utf-8')
return await runCode(code)
return await runCode(code, globals)
}
export async function runCode(code: string): Promise<any> {
const compiler = new Compiler(code, Object.keys(globals))
return await runBytecode(compiler.bytecode)
export async function runCode(code: string, globals?: Record<string, any>): Promise<any> {
return await runBytecode(compileCode(code, globals), globals)
}
export async function runBytecode(bytecode: Bytecode): Promise<any> {
export async function runBytecode(bytecode: Bytecode, globals?: Record<string, any>): Promise<any> {
try {
const vm = new VM(bytecode, globals)
const vm = new VM(bytecode, Object.assign({}, shrimpGlobals, globals))
await vm.run()
return vm.stack.length ? fromValue(vm.stack[vm.stack.length - 1]!) : null
} catch (error: any) {
@ -28,12 +27,13 @@ export async function runBytecode(bytecode: Bytecode): Promise<any> {
}
}
export function compileFile(path: string): Bytecode {
export function compileFile(path: string, globals?: Record<string, any>): Bytecode {
const code = readFileSync(path, 'utf-8')
return compileCode(code)
return compileCode(code, globals)
}
export function compileCode(code: string): Bytecode {
const compiler = new Compiler(code, Object.keys(globals))
export function compileCode(code: string, globals?: Record<string, any>): Bytecode {
const globalNames = [...Object.keys(shrimpGlobals), ...(globals ? Object.keys(globals) : [])]
const compiler = new Compiler(code, globalNames)
return compiler.bytecode
}