stak_wasm/
lib.rs

1//! A Stak Scheme interpreter in WASM.
2
3mod repl;
4
5use core::str;
6pub use repl::repl;
7use stak_compiler::compile_r7rs;
8use stak_device::ReadWriteDevice;
9use stak_file::{MemoryFileSystem, VoidFileSystem};
10use stak_macro::include_module;
11use stak_module::Module;
12use stak_process_context::{MemoryProcessContext, VoidProcessContext};
13use stak_r7rs::SmallPrimitiveSet;
14use stak_time::VoidClock;
15use stak_vm::Vm;
16use wasm_bindgen::prelude::*;
17
18/// Compiles source codes in Scheme.
19#[wasm_bindgen]
20pub fn compile(source: &str) -> Result<Vec<u8>, JsError> {
21    let mut target = vec![];
22    compile_r7rs(source.as_bytes(), &mut target)?;
23    Ok(target)
24}
25
26/// Interprets bytecode with standard input and returns its standard output.
27#[wasm_bindgen]
28pub fn interpret(bytecode: &[u8], input: &[u8], heap_size: usize) -> Result<Vec<u8>, JsError> {
29    let mut output = vec![];
30    let mut error = vec![];
31
32    Vm::new(
33        vec![Default::default(); heap_size],
34        SmallPrimitiveSet::new(
35            ReadWriteDevice::new(input, &mut output, &mut error),
36            VoidFileSystem::new(),
37            VoidProcessContext::new(),
38            VoidClock::new(),
39        ),
40    )?
41    .run(bytecode.iter().copied())?;
42
43    Ok(output)
44}
45
46/// Runs a Scheme script with standard input and returns its standard output.
47#[wasm_bindgen]
48pub fn run(source: &str, input: &[u8], heap_size: usize) -> Result<Vec<u8>, JsError> {
49    const MAIN_FILE: &str = "main.scm";
50
51    let mut output = vec![];
52    let mut error = vec![];
53    let files = [(MAIN_FILE.as_bytes(), source.as_bytes())];
54    let mut file_entries = [Default::default(); 1];
55
56    Vm::new(
57        vec![Default::default(); heap_size],
58        SmallPrimitiveSet::new(
59            ReadWriteDevice::new(input, &mut output, &mut error),
60            MemoryFileSystem::new(&files, &mut file_entries),
61            MemoryProcessContext::new(&["scheme", MAIN_FILE], &[]),
62            VoidClock::new(),
63        ),
64    )?
65    .run(
66        include_module!("run.scm", stak_module)
67            .bytecode()
68            .iter()
69            .copied(),
70    )
71    .map_err(|vm_error| match str::from_utf8(&error) {
72        Ok(error) if !error.is_empty() => JsError::new(error),
73        Ok(_) => JsError::from(vm_error),
74        Err(error) => JsError::from(error),
75    })?;
76
77    Ok(output)
78}
stak_wasm/lib.rs

stak_wasm/
lib.rs
