123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178 |
- use crate::ast::*;
- use crate::interp::*;
- use anyhow::{bail, Error};
- /// The list of builtins provided at startup.
- pub fn builtins() -> Vec<BuiltinFunc> {
- vec![
- BuiltinFunc {
- name: "rep",
- callback: Box::new(|state: &State, exprs: &[ExprRef], env: &Env| -> Result<Value, Error> {
- if let [rep, expr] = exprs {
- let mut buf = String::new();
- let num = state.eval(*rep, env)?.as_num(&state.ast.borrow())?;
- for _ in 0..num {
- buf.push_str(
- &state
- .eval(*expr, env)?
- .as_str(&state.ast.borrow())?
- .to_string(),
- );
- }
- Ok(Value::Lit(Literal::Str(buf)))
- } else {
- bail!("`rep`: expected two arguments, got {}", exprs.len())
- }
- }),
- },
- BuiltinFunc {
- name: "str/upper",
- callback: Box::new(|state: &State, exprs: &[ExprRef], env: &Env| -> Result<Value, Error> {
- if let [expr] = exprs {
- let s = state.eval(*expr, env)?;
- Ok(Value::Lit(Literal::Str(
- s.as_str(&state.ast.borrow())?.to_uppercase(),
- )))
- } else {
- bail!("`str/capitalize`: expected 1 argument1, got {}", exprs.len());
- }
- }),
- },
- BuiltinFunc {
- name: "str/capitalize",
- callback: Box::new(|state: &State, exprs: &[ExprRef], env: &Env| -> Result<Value, Error> {
- if let [expr] = exprs {
- let s = state.eval(*expr, env)?;
- Ok(Value::Lit(Literal::Str(titlecase::titlecase(
- s.as_str(&state.ast.borrow())?,
- ))))
- } else {
- bail!("`str/capitalize`: expected 1 argument1, got {}", exprs.len());
- }
- }),
- },
- BuiltinFunc {
- name: "str/lower",
- callback: Box::new(|state: &State, exprs: &[ExprRef], env: &Env| -> Result<Value, Error> {
- if let [expr] = exprs {
- let s = state.eval(*expr, env)?;
- Ok(Value::Lit(Literal::Str(
- s.as_str(&state.ast.borrow())?.to_lowercase(),
- )))
- } else {
- bail!("`str/lower`: expected 1 argument1, got {}", exprs.len());
- }
- }),
- },
- BuiltinFunc {
- name: "sub",
- callback: Box::new(|state: &State, exprs: &[ExprRef], env: &Env| -> Result<Value, Error> {
- if let [x, y] = exprs {
- let x = state.eval(*x, env)?.as_num(&state.ast.borrow())?;
- let y = state.eval(*y, env)?.as_num(&state.ast.borrow())?;
- Ok(Value::Lit(Literal::Num(x - y)))
- } else {
- bail!("`sub`: expected 2 arguments, got {}", exprs.len());
- }
- }),
- },
- BuiltinFunc {
- name: "tuple/len",
- callback: Box::new(|state: &State, exprs: &[ExprRef], env: &Env| -> Result<Value, Error> {
- if let [expr] = exprs {
- let tup = state.eval(*expr, env)?;
- Ok(Value::Lit(Literal::Num(tup.as_tup(&state.ast.borrow())?.len() as i64)))
- } else {
- bail!("`tuple/len`: expected 1 argument, got {}", exprs.len())
- }
- }),
- },
- BuiltinFunc {
- name: "tuple/concat",
- callback: Box::new(|state: &State, exprs: &[ExprRef], env: &Env| -> Result<Value, Error> {
- if let [expr] = exprs {
- let val = state.eval(*expr, env)?;
- let tup = val.as_tup(&state.ast.borrow())?;
- let mut contents = Vec::new();
- for elem in tup {
- for th in state.hnf(elem)?.as_tup(&state.ast.borrow())? {
- contents.push(th.clone());
- }
- }
- Ok(Value::Tup(contents))
- } else {
- bail!("tuple/concat: expected 1 argument, got {}", exprs.len());
- }
- }),
- },
- BuiltinFunc {
- name: "tuple/index",
- callback: Box::new(|state: &State, exprs: &[ExprRef], env: &Env| -> Result<Value, Error> {
- if let [tup, idx] = exprs {
- let tup = state.eval(*tup, env)?;
- let idx = state.eval(*idx, env)?;
- state.hnf(
- &tup.as_tup(&state.ast.borrow())?
- [idx.as_num(&state.ast.borrow())? as usize],
- )
- } else {
- bail!("`tuple/index`: expected 2 arguments, got {}", exprs.len());
- }
- }),
- },
- BuiltinFunc {
- name: "tuple/replace",
- callback: Box::new(|state: &State, exprs: &[ExprRef], env: &Env| -> Result<Value, Error> {
- if let [tup, idx, new] = exprs {
- let tup_val = state.eval(*tup, env)?;
- let tup = tup_val.as_tup(&state.ast.borrow())?;
- let idx = state.eval(*idx, env)?.as_num(&state.ast.borrow())?;
- let mut modified = Vec::with_capacity(tup.len());
- for i in 0..idx {
- modified.push(tup[i as usize].clone());
- }
- modified.push(Thunk::Expr(*new, env.clone()));
- for i in (idx + 1)..(tup.len() as i64) {
- modified.push(tup[i as usize].clone());
- }
- Ok(Value::Tup(modified))
- } else {
- bail!("`tuple/replace`: expected 3 arguments, got {}", exprs.len());
- }
- }),
- },
- BuiltinFunc {
- name: "tuple/fold",
- callback: Box::new(|state: &State, exprs: &[ExprRef], env: &Env| -> Result<Value, Error> {
- if let [func, init, tup] = exprs {
- let func = state.eval(*func, env)?;
- let tup = state.eval(*tup, env)?;
- let mut result = Thunk::Expr(*init, env.clone());
- for t in tup.as_tup(&state.ast.borrow())? {
- result = Thunk::Value(
- state.eval_closure(
- func.as_closure(&state.ast.borrow())?,
- vec![result, t.clone()],
- )?);
- }
- state.hnf(&result)
- } else {
- bail!("`tuple/fold`: expected 3 arguments, got {}", exprs.len());
- }
- }),
- },
- ]
- }
|