add Shrimp class as a nicer way to run code

src/index.ts

@@ -7,6 +7,34 @@ export { Compiler } from '#compiler/compiler'
 export { parser } from '#parser/shrimp'
 export { globals } from '#prelude'
 
+export class Shrimp {
+    vm: VM
+
+    constructor() {
+        this.vm = new VM({ instructions: [], constants: [], labels: new Map() }, shrimpGlobals)
+    }
+
+    async run(code: string | Bytecode): Promise<any> {
+        let bytecode
+
+        if (typeof code === 'string') {
+            const compiler = new Compiler(code, Object.keys(shrimpGlobals))
+            bytecode = compiler.bytecode
+        } else {
+            bytecode = code
+        }
+
+        this.vm.appendBytecode(bytecode)
+        await this.vm.continue()
+        return this.vm.stack.length ? fromValue(this.vm.stack.at(-1)!) : null
+    }
+
+    get(name: string): any {
+        const value = this.vm.scope.get(name)
+        return value ? fromValue(value) : null
+    }
+}
+
 export async function runFile(path: string, globals?: Record<string, any>): Promise<any> {
     const code = readFileSync(path, 'utf-8')
     return await runCode(code, globals)

src/tests/shrimp.test.ts

@@ -0,0 +1,21 @@
+import { describe } from 'bun:test'
+import { expect, test } from 'bun:test'
+import { Shrimp } from '..'
+
+describe('Shrimp', () => {
+  test('allows running Shrimp code', async () => {
+    const shrimp = new Shrimp()
+
+    expect(await shrimp.run(`1 + 5`)).toEqual(6)
+
+    await shrimp.run(`abc = true`)
+    expect(shrimp.get('abc')).toEqual(true)
+
+    await shrimp.run(`name = Bob`)
+    expect(shrimp.get('abc')).toEqual(true)
+    expect(shrimp.get('name')).toEqual('Bob')
+
+    await shrimp.run(`abc = false`)
+    expect(shrimp.get('abc')).toEqual(false)
+  })
+})