iframe-proxy | Sunbelt Computer Software

History

143 lines (128 loc) · 4.8 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

extern crate alloc;

pub mod browser_module;

pub mod convert;

pub mod js_module;

pub mod vm_class;

pub mod wasm_builtins;

#[macro_use]

extern crate rustpython_vm;

use js_sys::{Reflect, WebAssembly::RuntimeError};

use std::panic;

pub(crate) use vm_class::weak_vm;

use wasm_bindgen::prelude::*;

/// Sets error info on the window object, and prints the backtrace to console

pub fn panic_hook(info: &panic::PanicHookInfo<'_>) {

// If something errors, just ignore it; we don't want to panic in the panic hook

let try_set_info = || {

let msg = &info.to_string();

let window = match web_sys::window() {

Some(win) => win,

None => return,

};

let _ = Reflect::set(&window, &"__RUSTPYTHON_ERROR_MSG".into(), &msg.into());

let error = RuntimeError::new(msg);

let _ = Reflect::set(&window, &"__RUSTPYTHON_ERROR".into(), &error);

let stack = match Reflect::get(&error, &"stack".into()) {

Ok(stack) => stack,

Err(_) => return,

};

let _ = Reflect::set(&window, &"__RUSTPYTHON_ERROR_STACK".into(), &stack);

};

try_set_info();

console_error_panic_hook::hook(info);

}

#[doc(hidden)]

#[cfg(not(feature = "no-start-func"))]

#[wasm_bindgen(start)]

pub fn _setup_console_error() {

std::panic::set_hook(Box::new(panic_hook));

}

pub mod eval {

use crate::vm_class::VMStore;

use js_sys::{Object, Reflect, TypeError};

use rustpython_vm::compiler::Mode;

use wasm_bindgen::prelude::*;

const PY_EVAL_VM_ID: &str = "__py_eval_vm";

fn run_py(source: &str, options: Option<Object>, mode: Mode) -> Result<JsValue, JsValue> {

let vm = VMStore::init(PY_EVAL_VM_ID.into(), Some(true));

let options = options.unwrap_or_default();

let js_vars = {

let prop = Reflect::get(&options, &"vars".into())?;

if prop.is_undefined() {

None

} else if prop.is_object() {

Some(Object::from(prop))

} else {

return Err(TypeError::new("vars must be an object").into());

}

};

vm.set_stdout(Reflect::get(&options, &"stdout".into())?)?;

vm.set_stderr(Reflect::get(&options, &"stderr".into())?)?;

if let Some(js_vars) = js_vars {

vm.add_to_scope("js_vars".into(), js_vars.into())?;

}

if matches!(mode, Mode::Single) {

vm.exec_single(source, None)

} else {

vm.run(source, mode, None)

}

/// Evaluate Python code

///

/// ```js

/// var result = pyEval(code, options?);

/// ```

///

/// `code`: `string`: The Python code to run in eval mode

///

/// `options`:

///

/// - `vars?`: `{ [key: string]: any }`: Variables passed to the VM that can be

/// accessed in Python with the variable `js_vars`. Functions do work, and

/// receive the Python kwargs as the `this` argument.

/// - `stdout?`: `"console" | ((out: string) => void) | null`: A function to replace the

/// native print native print function, and it will be `console.log` when giving

/// `undefined` or "console", and it will be a dumb function when giving null.

/// - `stderr?`: `"console" | ((out: string) => void) | null`: A function to replace

/// `sys.stderr`, and it will be `console.error` when giving `undefined` or "console".

#[wasm_bindgen(js_name = pyEval)]

pub fn eval_py(source: &str, options: Option<Object>) -> Result<JsValue, JsValue> {

run_py(source, options, Mode::Eval)

}

/// Evaluate Python code

///

/// ```js

/// pyExec(code, options?);

/// ```

///

/// `code`: `string`: The Python code to run in exec mode

///

/// `options`: The options are the same as eval mode

#[wasm_bindgen(js_name = pyExec)]

pub fn exec_py(source: &str, options: Option<Object>) -> Result<(), JsValue> {

run_py(source, options, Mode::Exec).map(drop)

}

/// Evaluate Python code

///

/// ```js

/// var result = pyExecSingle(code, options?);

/// ```

///

/// `code`: `string`: The Python code to run in exec single mode

///

/// `options`: The options are the same as eval mode

#[wasm_bindgen(js_name = pyExecSingle)]

pub fn exec_single_py(source: &str, options: Option<Object>) -> Result<JsValue, JsValue> {

run_py(source, options, Mode::Single)

}

/// A module containing all the wasm-bindgen exports that rustpython_wasm has

/// Re-export as `pub use rustpython_wasm::exports::*;` in the root of your crate if you want your

/// wasm module to mimic rustpython_wasm's API

pub mod exports {

pub use crate::convert::PyError;

pub use crate::eval::{eval_py, exec_py, exec_single_py};

pub use crate::vm_class::{VMStore, WASMVirtualMachine};

}

#[doc(hidden)]

pub use exports::*;

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

lib.rs

Latest commit

History

lib.rs

File metadata and controls