50 lines
2.2 KiB
Rust
50 lines
2.2 KiB
Rust
//! 内置函数模块：按功能分组注册，便于扩展和维护。
|
|
|
|
mod core;
|
|
mod io;
|
|
mod os;
|
|
mod string;
|
|
|
|
use crate::interpreter::{Env, Value};
|
|
use std::cell::RefCell;
|
|
use std::collections::HashMap;
|
|
use std::rc::Rc;
|
|
|
|
/// 向全局环境注册所有标准库（io、os 等）
|
|
pub fn register_all(env: &Rc<RefCell<Env>>) {
|
|
let mut e = env.borrow_mut();
|
|
|
|
// io
|
|
let mut io = HashMap::new();
|
|
io.insert("print".into(), Value::NativeFunction(io::print));
|
|
io.insert("input".into(), Value::NativeFunction(io::input));
|
|
e.define("io".into(), Value::Object(Rc::new(RefCell::new(io))), true);
|
|
// input and print can be used as global functions for convenience
|
|
e.define("print".into(), Value::NativeFunction(io::print), true);
|
|
e.define("input".into(), Value::NativeFunction(io::input), true);
|
|
|
|
// os
|
|
let mut os = HashMap::new();
|
|
os.insert("clock".into(), Value::NativeFunction(os::clock));
|
|
e.define("os".into(), Value::Object(Rc::new(RefCell::new(os))), true);
|
|
// clock can also be used as a global function for convenience
|
|
e.define("clock".into(), Value::NativeFunction(os::clock), true);
|
|
|
|
// core — len, typeof, push, pop
|
|
e.define("len".into(), Value::NativeFunction(core::len), true);
|
|
e.define("typeof".into(), Value::NativeFunction(core::typeof_fn), true);
|
|
e.define("push".into(), Value::NativeFunction(core::push), true);
|
|
e.define("pop".into(), Value::NativeFunction(core::pop), true);
|
|
|
|
// string — split, trim, substring, replace, contains, upper, lower, starts_with, ends_with
|
|
e.define("split".into(), Value::NativeFunction(string::split), true);
|
|
e.define("trim".into(), Value::NativeFunction(string::trim), true);
|
|
e.define("substring".into(), Value::NativeFunction(string::substring), true);
|
|
e.define("replace".into(), Value::NativeFunction(string::replace), true);
|
|
e.define("contains".into(), Value::NativeFunction(string::contains), true);
|
|
e.define("upper".into(), Value::NativeFunction(string::upper), true);
|
|
e.define("lower".into(), Value::NativeFunction(string::lower), true);
|
|
e.define("starts_with".into(), Value::NativeFunction(string::starts_with), true);
|
|
e.define("ends_with".into(), Value::NativeFunction(string::ends_with), true);
|
|
}
|