#[cfg(feature = "dict")]
use crate::dict::{dict_new, dict_path_insert, dict_path_remove, list_to_dict};
use crate::interp::Interp;
use crate::types::*;
#[cfg(feature = "string-command")]
use crate::util;
use crate::*;
use core::fmt::Write as _;
use alloc::string::String;
use alloc::vec::Vec;
pub fn cmd_append(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 2, 0, "varName ?value value ...?")?;
let mut new_string: String = interp
.var(&argv[1])
.map(|val| val.to_string())
.unwrap_or_default();
for item in &argv[2..] {
new_string.push_str(item.as_str());
}
interp.set_var_return(&argv[1], new_string.into()).map(Some)
}
pub fn cmd_array(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
interp.call_subcommand(argv, 1, &ARRAY_SUBCOMMANDS)
}
static ARRAY_SUBCOMMANDS: [Subcommand; 6] = [
Subcommand("exists", cmd_array_exists),
Subcommand("get", cmd_array_get),
Subcommand("names", cmd_array_names),
Subcommand("set", cmd_array_set),
Subcommand("size", cmd_array_size),
Subcommand("unset", cmd_array_unset),
];
pub fn cmd_array_exists(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 3, "arrayName")?;
molt_opt_ok!(Value::from(interp.array_exists(argv[2].as_str())))
}
pub fn cmd_array_names(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 3, "arrayName")?;
molt_opt_ok!(Value::from(interp.array_names(argv[2].as_str())))
}
pub fn cmd_array_get(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 3, "arrayName")?;
molt_opt_ok!(Value::from(interp.array_get(argv[2].as_str())))
}
pub fn cmd_array_set(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 4, 4, "arrayName list")?;
let var_name = argv[2].as_var_name();
if var_name.index().is_none() {
interp.array_set(var_name.name(), &argv[3].as_list()?)?;
molt_opt_ok!()
} else {
interp.array_set(var_name.name(), &Value::empty().as_list()?)?;
molt_err!("can't set \"{}\": variable isn't array", &argv[2])
}
}
pub fn cmd_array_size(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 3, "arrayName")?;
molt_opt_ok!(Value::from(interp.array_size(argv[2].as_str()) as MoltInt))
}
pub fn cmd_array_unset(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 4, "arrayName ?index?")?;
if argv.len() == 3 {
interp.array_unset(argv[2].as_str());
} else {
interp.unset_element(argv[2].as_str(), argv[3].as_str());
}
molt_opt_ok!()
}
pub fn cmd_assert_eq(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 3, 3, "received expected")?;
if argv[1] == argv[2] {
molt_opt_ok!()
} else {
molt_err!(
"assertion failed: received \"{}\", expected \"{}\".",
argv[1],
argv[2]
)
}
}
pub fn cmd_break(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 1, 1, "")?;
Err(Exception::molt_break())
}
pub fn cmd_catch(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 2, 4, "script ?resultVarName? ?optionsVarName?")?;
let result = interp.eval_value(&argv[1]);
let (code, value) = match &result {
Ok(val) => (0, val.clone()),
Err(exception) => match exception.code() {
ResultCode::Okay => unreachable!(), ResultCode::Error => (1, exception.value()),
ResultCode::Return => (2, exception.value()),
ResultCode::Break => (3, exception.value()),
ResultCode::Continue => (4, exception.value()),
ResultCode::Other(_) => unimplemented!(), },
};
if argv.len() >= 3 {
interp.set_var(&argv[2], value)?;
}
if argv.len() == 4 {
#[cfg(feature = "dict")]
interp.set_var(&argv[3], interp.return_options(&result))?;
#[cfg(not(feature = "dict"))]
unimplemented!("return options");
}
Ok(Some(Value::from(code)))
}
pub fn cmd_continue(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 1, 1, "")?;
Err(Exception::molt_continue())
}
#[cfg(feature = "dict")]
pub fn cmd_dict(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
interp.call_subcommand(argv, 1, &DICT_SUBCOMMANDS)
}
#[cfg(feature = "dict")]
static DICT_SUBCOMMANDS: [Subcommand; 9] = [
Subcommand("create", cmd_dict_new),
Subcommand("exists", cmd_dict_exists),
Subcommand("get", cmd_dict_get),
Subcommand("keys", cmd_dict_keys),
Subcommand("remove", cmd_dict_remove),
Subcommand("set", cmd_dict_set),
Subcommand("size", cmd_dict_size),
Subcommand("unset", cmd_dict_unset),
Subcommand("values", cmd_dict_values),
];
#[cfg(feature = "dict")]
fn cmd_dict_new(_: &mut Interp, argv: &[Value]) -> MoltOptResult {
if argv.len() % 2 != 0 {
return molt_err!(
"wrong # args: should be \"{} {}\"",
Value::from(&argv[0..2]).to_string(),
"?key value?"
);
}
if argv.len() > 2 {
molt_opt_ok!(Value::from(list_to_dict(&argv[2..])))
} else {
molt_opt_ok!(Value::from(dict_new()))
}
}
#[cfg(feature = "dict")]
fn cmd_dict_exists(_: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 4, 0, "dictionary key ?key ...?")?;
let mut value: Value = argv[2].clone();
let indices = &argv[3..];
for index in indices {
if let Ok(dict) = value.as_dict() {
if let Some(val) = dict.get(index) {
value = val.clone();
} else {
return molt_opt_ok!(false);
}
} else {
return molt_opt_ok!(false);
}
}
molt_opt_ok!(true)
}
#[cfg(feature = "dict")]
fn cmd_dict_get(_: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 0, "dictionary ?key ...?")?;
let mut value: Value = argv[2].clone();
let indices = &argv[3..];
for index in indices {
let dict = value.as_dict()?;
if let Some(val) = dict.get(index) {
value = val.clone();
} else {
return molt_err!("key \"{}\" not known in dictionary", index);
}
}
molt_opt_ok!(value)
}
#[cfg(feature = "dict")]
fn cmd_dict_keys(_: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 3, "dictionary")?;
let dict = argv[2].as_dict()?;
let keys: MoltList = dict.keys().cloned().collect();
molt_opt_ok!(keys)
}
#[cfg(feature = "dict")]
fn cmd_dict_remove(_: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 0, "dictionary ?key ...?")?;
let mut dict = (*argv[2].as_dict()?).clone();
for key in &argv[3..] {
dict.shift_remove(key);
}
molt_opt_ok!(dict)
}
#[cfg(feature = "dict")]
fn cmd_dict_set(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 5, 0, "dictVarName key ?key ...? value")?;
let value = &argv[argv.len() - 1];
let keys = &argv[3..(argv.len() - 1)];
if let Ok(old_dict_val) = interp.var(&argv[2]) {
interp.set_var_return(&argv[2], dict_path_insert(&old_dict_val, keys, value)?).map(Some)
} else {
let new_val = Value::from(dict_new());
interp.set_var_return(&argv[2], dict_path_insert(&new_val, keys, value)?).map(Some)
}
}
#[cfg(feature = "dict")]
fn cmd_dict_size(_: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 3, "dictionary")?;
let dict = argv[2].as_dict()?;
molt_opt_ok!(dict.len() as MoltInt)
}
#[cfg(feature = "dict")]
fn cmd_dict_unset(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 4, 0, "dictVarName key ?key ...?")?;
let keys = &argv[3..];
if let Ok(old_dict_val) = interp.var(&argv[2]) {
interp.set_var_return(&argv[2], dict_path_remove(&old_dict_val, keys)?).map(Some)
} else {
let new_val = Value::from(dict_new());
interp.set_var_return(&argv[2], dict_path_remove(&new_val, keys)?).map(Some)
}
}
#[cfg(feature = "dict")]
fn cmd_dict_values(_: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 3, "dictionary")?;
let dict = argv[2].as_dict()?;
let values: MoltList = dict.values().cloned().collect();
molt_opt_ok!(values)
}
pub fn cmd_error(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 2, 2, "message")?;
molt_err!(argv[1].clone())
}
#[cfg(feature = "std")]
pub fn cmd_exit(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 1, 2, "?returnCode?")?;
let return_code: MoltInt = if argv.len() == 1 {
0
} else {
argv[1].as_int()?
};
std::process::exit(return_code as i32)
}
#[cfg(feature = "expr")]
pub fn cmd_expr(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 2, 2, "expr")?;
interp.expr(&argv[1]).map(Some)
}
pub fn cmd_for(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 5, 5, "start test next command")?;
let start = &argv[1];
let test = &argv[2];
let next = &argv[3];
let command = &argv[4];
interp.eval_value(start)?;
while interp.expr_bool(test)? {
let result = interp.eval_value(command);
if let Err(exception) = result {
match exception.code() {
ResultCode::Break => break,
ResultCode::Continue => (),
_ => return Err(exception),
}
}
let result = interp.eval_value(next);
if let Err(exception) = result {
match exception.code() {
ResultCode::Break => break,
ResultCode::Continue => {
return molt_err!("invoked \"continue\" outside of a loop");
}
_ => return Err(exception),
}
}
}
molt_opt_ok!()
}
pub fn cmd_foreach(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 4, 4, "varList list body")?;
let var_list = &*argv[1].as_list()?;
let list = &*argv[2].as_list()?;
let body = &argv[3];
let mut i = 0;
while i < list.len() {
for var in var_list {
if i < list.len() {
interp.set_var(var, list[i].clone())?;
i += 1;
} else {
interp.set_var(var, Value::empty())?;
}
}
let result = interp.eval_value(body);
if let Err(exception) = result {
match exception.code() {
ResultCode::Break => break,
ResultCode::Continue => (),
_ => return Err(exception),
}
}
}
molt_opt_ok!()
}
pub fn cmd_global(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
if interp.scope_level() > 0 {
for name in &argv[1..] {
interp.upvar(0, name.as_str());
}
}
molt_opt_ok!()
}
#[derive(Eq, PartialEq, Debug)]
enum IfWants {
Expr,
ThenBody,
SkipThenClause,
ElseClause,
ElseBody,
}
pub fn cmd_if(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
let mut argi = 1;
let mut wants = IfWants::Expr;
while argi < argv.len() {
match wants {
IfWants::Expr => {
wants = if interp.expr_bool(&argv[argi])? {
IfWants::ThenBody
} else {
IfWants::SkipThenClause
};
}
IfWants::ThenBody => {
if argv[argi].as_str() == "then" {
argi += 1;
}
if argi < argv.len() {
return interp.eval_value(&argv[argi]).map(Some);
} else {
break;
}
}
IfWants::SkipThenClause => {
if argv[argi].as_str() == "then" {
argi += 1;
}
if argi < argv.len() {
argi += 1;
wants = IfWants::ElseClause;
}
continue;
}
IfWants::ElseClause => {
if argv[argi].as_str() == "elseif" {
wants = IfWants::Expr;
} else {
wants = IfWants::ElseBody;
continue;
}
}
IfWants::ElseBody => {
if argv[argi].as_str() == "else" {
argi += 1;
if argi == argv.len() {
return molt_err!(
"wrong # args: no script following after \"{}\" argument",
argv[argi - 1]
);
}
}
if argi < argv.len() {
return interp.eval_value(&argv[argi]).map(Some);
} else {
break;
}
}
}
argi += 1;
}
if argi < argv.len() {
molt_err!("wrong # args: extra words after \"else\" clause in \"if\" command")
} else if wants == IfWants::Expr {
molt_err!(
"wrong # args: no expression after \"{}\" argument",
argv[argi - 1]
)
} else if wants == IfWants::ThenBody || wants == IfWants::SkipThenClause {
molt_err!(
"wrong # args: no script following after \"{}\" argument",
argv[argi - 1]
)
} else {
molt_opt_ok!() }
}
pub fn cmd_incr(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 2, 3, "varName ?increment?")?;
let increment: MoltInt = if argv.len() == 3 {
argv[2].as_int()?
} else {
1
};
let new_value = increment
+ interp
.var(&argv[1])
.and_then(|val| val.as_int())
.unwrap_or(0);
interp.set_var_return(&argv[1], new_value.into()).map(Some)
}
#[cfg(feature = "info")]
pub fn cmd_info(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
interp.call_subcommand(argv, 1, &INFO_SUBCOMMANDS)
}
#[cfg(feature = "info")]
static INFO_SUBCOMMANDS: [Subcommand; 11] = [
Subcommand("args", cmd_info_args),
Subcommand("body", cmd_info_body),
Subcommand("cmdtype", cmd_info_cmdtype),
Subcommand("commands", cmd_info_commands),
Subcommand("complete", cmd_info_complete),
Subcommand("default", cmd_info_default),
Subcommand("exists", cmd_info_exists),
Subcommand("globals", cmd_info_globals),
Subcommand("locals", cmd_info_locals),
Subcommand("procs", cmd_info_procs),
Subcommand("vars", cmd_info_vars),
];
#[cfg(feature = "info")]
pub fn cmd_info_args(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 3, "procname")?;
interp.proc_args(argv[2].as_str()).map(Some)
}
#[cfg(feature = "info")]
pub fn cmd_info_body(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 3, "procname")?;
interp.proc_body(argv[2].as_str()).map(Some)
}
#[cfg(feature = "info")]
pub fn cmd_info_cmdtype(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 3, "command")?;
interp.command_type(argv[2].as_str()).map(Some)
}
#[cfg(feature = "info")]
pub fn cmd_info_commands(interp: &mut Interp, _argv: &[Value]) -> MoltOptResult {
molt_opt_ok!(Value::from(interp.command_names()))
}
#[cfg(feature = "info")]
pub fn cmd_info_default(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 5, 5, "procname arg varname")?;
if let Some(val) = interp.proc_default(argv[2].as_str(), argv[3].as_str())? {
interp.set_var(&argv[4], val)?;
molt_opt_ok!(1)
} else {
interp.set_var(&argv[4], Value::empty())?;
molt_opt_ok!(0)
}
}
#[cfg(feature = "info")]
pub fn cmd_info_exists(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 3, "varname")?;
Ok(Some(interp.var_exists(&argv[2]).into()))
}
#[cfg(feature = "info")]
pub fn cmd_info_complete(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 3, "command")?;
if interp.complete(argv[2].as_str()) {
molt_opt_ok!(true)
} else {
molt_opt_ok!(false)
}
}
#[cfg(feature = "info")]
pub fn cmd_info_globals(interp: &mut Interp, _argv: &[Value]) -> MoltOptResult {
molt_opt_ok!(Value::from(interp.vars_in_global_scope()))
}
#[cfg(feature = "info")]
pub fn cmd_info_locals(interp: &mut Interp, _argv: &[Value]) -> MoltOptResult {
molt_opt_ok!(Value::from(interp.vars_in_local_scope()))
}
#[cfg(feature = "info")]
pub fn cmd_info_procs(interp: &mut Interp, _argv: &[Value]) -> MoltOptResult {
molt_opt_ok!(Value::from(interp.proc_names()))
}
#[cfg(feature = "info")]
pub fn cmd_info_vars(interp: &mut Interp, _argv: &[Value]) -> MoltOptResult {
molt_opt_ok!(Value::from(interp.vars_in_scope()))
}
pub fn cmd_join(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 2, 3, "list ?joinString?")?;
let list = argv[1].as_list()?;
let join_string = if argv.len() == 3 {
argv[2].as_str()
} else {
" "
};
let mut result = String::new();
let mut first = true;
for part in &*list {
if !first {
result.push_str(join_string);
}
first = false;
write!(result, "{}", part).unwrap(); }
molt_opt_ok!(result)
}
pub fn cmd_lappend(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 2, 0, "varName ?value ...?")?;
let var_result = interp.var(&argv[1]);
let mut list: MoltList = if let Ok(r) = var_result {
r.to_list()?
} else {
Vec::new()
};
let mut values = argv[2..].to_owned();
list.append(&mut values);
interp.set_var_return(&argv[1], Value::from(list)).map(Some)
}
pub fn cmd_lindex(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 2, 0, "list ?index ...?")?;
if argv.len() != 3 {
lindex_into(&argv[1], &argv[2..])
} else {
lindex_into(&argv[1], &argv[2].as_list()?)
}
}
pub fn lindex_into(list: &Value, indices: &[Value]) -> MoltOptResult {
let mut value: Value = list.clone();
for index_val in indices {
let next_value = {
let list = value.as_list()?;
let index = index_val.as_int()?;
let index = usize::try_from(index).unwrap_or(list.len());
list.get(index).cloned().unwrap_or_default()
};
value = next_value;
}
molt_opt_ok!(value)
}
pub fn cmd_list(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
molt_opt_ok!(&argv[1..])
}
pub fn cmd_llength(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 2, 2, "list")?;
molt_opt_ok!(argv[1].as_list()?.len() as MoltInt)
}
#[cfg(feature = "std")]
pub fn cmd_pdump(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 1, 1, "")?;
interp.profile_dump();
molt_opt_ok!()
}
#[cfg(feature = "std")]
pub fn cmd_pclear(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 1, 1, "")?;
interp.profile_clear();
molt_opt_ok!()
}
pub fn cmd_proc(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 4, 4, "name args body")?;
let name = argv[1].as_str();
let args = &*argv[2].as_list()?;
for arg in args {
let vec = arg.as_list()?;
if vec.is_empty() {
return molt_err!("argument with no name");
} else if vec.len() > 2 {
return molt_err!("too many fields in argument specifier \"{}\"", arg);
}
}
interp.add_proc(name, args, &argv[3]);
molt_opt_ok!()
}
#[cfg(feature = "std")]
pub fn cmd_puts(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 2, 2, "string")?;
println!("{}", argv[1]);
molt_opt_ok!()
}
pub fn cmd_rename(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 3, 3, "oldName newName")?;
let old_name = argv[1].as_str();
let new_name = argv[2].as_str();
if !interp.has_command(old_name) {
return molt_err!("can't rename \"{}\": command doesn't exist", old_name);
}
if new_name.is_empty() {
interp.remove_command(old_name);
} else {
interp.rename_command(old_name, new_name);
}
molt_opt_ok!()
}
pub fn cmd_return(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 1, 0, "?options...? ?value?")?;
let mut code = ResultCode::Okay;
let mut level: MoltInt = 1;
let mut error_code: Option<Value> = None;
let mut error_info: Option<Value> = None;
if argv.len() == 1 {
return Err(Exception::molt_return_ext(
None,
level as usize,
code,
));
}
let mut return_value = None;
let opt_args: &[Value] = if argv.len() % 2 == 0 {
return_value = Some(argv[argv.len() - 1].clone());
&argv[1..argv.len() - 1]
} else {
&argv[1..argv.len()]
};
for pair in opt_args.chunks(2) {
let val = &pair[1];
match pair[0].as_str() {
"-code" => {
code = ResultCode::from_value(val)?;
}
"-errorcode" => {
error_code = Some(val.clone());
}
"-errorinfo" => {
error_info = Some(val.clone());
}
"-level" => {
level = val.as_int()?;
}
_ => return molt_err!("invalid return option: \"{}\"", pair[0]),
}
}
if code == ResultCode::Error {
Err(Exception::molt_return_err(
return_value,
level as usize,
error_code,
error_info,
))
} else if level == 0 && code == ResultCode::Okay {
Ok(return_value)
} else {
Err(Exception::molt_return_ext(
return_value,
level as usize,
code,
))
}
}
pub fn cmd_set(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 2, 3, "varName ?newValue?")?;
if argv.len() == 3 {
interp.set_var_return(&argv[1], argv[2].clone()).map(Some)
} else {
molt_opt_ok!(interp.var(&argv[1])?)
}
}
#[cfg(feature = "std")]
pub fn cmd_source(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 2, 2, "filename")?;
let filename = argv[1].as_str();
match std::fs::read_to_string(filename) {
Ok(script) => interp.eval(&script).map(Some),
Err(e) => molt_err!("couldn't read file \"{}\": {}", filename, e),
}
}
#[cfg(feature = "string-command")]
pub fn cmd_string(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
interp.call_subcommand(argv, 1, STRING_SUBCOMMANDS)
}
#[cfg(feature = "string-command")]
static STRING_SUBCOMMANDS: &[Subcommand] = &[
Subcommand("cat", cmd_string_cat),
Subcommand("compare", cmd_string_compare),
Subcommand("equal", cmd_string_equal),
Subcommand("first", cmd_string_first),
Subcommand("last", cmd_string_last),
Subcommand("length", cmd_string_length),
#[cfg(feature = "dict")]
Subcommand("map", cmd_string_map),
Subcommand("range", cmd_string_range),
Subcommand("tolower", cmd_string_tolower),
Subcommand("toupper", cmd_string_toupper),
Subcommand("trim", cmd_string_trim),
Subcommand("trimleft", cmd_string_trim),
Subcommand("trimright", cmd_string_trim),
];
#[cfg(feature = "string-command")]
pub fn cmd_string_cat(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
let mut buff = String::new();
for arg in &argv[2..] {
buff.push_str(arg.as_str());
}
molt_opt_ok!(buff)
}
#[cfg(feature = "string-command")]
pub fn cmd_string_compare(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 4, 7, "?-nocase? ?-length length? string1 string2")?;
let arglen = argv.len();
let mut nocase = false;
let mut length: Option<MoltInt> = None;
let opt_args = &argv[2..arglen - 2];
let mut queue = opt_args.iter();
while let Some(opt) = queue.next() {
match opt.as_str() {
"-nocase" => nocase = true,
"-length" => {
if let Some(val) = queue.next() {
length = Some(val.as_int()?);
} else {
return molt_err!("wrong # args: should be \"string compare ?-nocase? ?-length length? string1 string2\"");
}
}
_ => return molt_err!("bad option \"{}\": must be -nocase or -length", opt),
}
}
let val1 = &argv[arglen - 2];
let val2 = &argv[arglen - 1];
if nocase {
let val1 = util::to_lowercase(val1.as_str());
let val2 = util::to_lowercase(val2.as_str());
molt_opt_ok!(util::compare_len(&val1, &val2, length))
} else {
molt_opt_ok!(util::compare_len(val1.as_str(), val2.as_str(), length))
}
}
#[cfg(feature = "string-command")]
pub fn cmd_string_equal(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 4, 7, "?-nocase? ?-length length? string1 string2")?;
let arglen = argv.len();
let mut nocase = false;
let mut length: Option<MoltInt> = None;
let opt_args = &argv[2..arglen - 2];
let mut queue = opt_args.iter();
while let Some(opt) = queue.next() {
match opt.as_str() {
"-nocase" => nocase = true,
"-length" => {
if let Some(val) = queue.next() {
length = Some(val.as_int()?);
} else {
return molt_err!("wrong # args: should be \"string equal ?-nocase? ?-length length? string1 string2\"");
}
}
_ => return molt_err!("bad option \"{}\": must be -nocase or -length", opt),
}
}
if nocase {
let val1 = &argv[arglen - 2];
let val2 = &argv[arglen - 1];
let val1 = Value::from(util::to_lowercase(val1.as_str()));
let val2 = Value::from(util::to_lowercase(val2.as_str()));
let flag = util::compare_len(val1.as_str(), val2.as_str(), length) == 0;
molt_opt_ok!(flag)
} else {
let flag = util::compare_len(argv[arglen - 2].as_str(), argv[arglen - 1].as_str(), length) == 0;
molt_opt_ok!(flag)
}
}
#[cfg(feature = "string-command")]
pub fn cmd_string_first(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 4, 5, "needleString haystackString ?startIndex?")?;
let needle = argv[2].as_str();
let haystack = argv[3].as_str();
let start_char: usize = if argv.len() == 5 {
let arg = argv[4].as_int()?;
if arg < 0 { 0 } else { arg as usize }
} else {
0
};
let pos_byte: Option<usize> = haystack
.char_indices()
.nth(start_char)
.and_then(|(start_byte, _)| haystack[start_byte..].find(needle));
let pos_char: MoltInt = match pos_byte {
None => -1,
Some(b) => haystack[b..]
.char_indices()
.take_while(|(i, _)| *i < b)
.count() as MoltInt
+ start_char as MoltInt
};
molt_opt_ok!(pos_char)
}
#[cfg(feature = "string-command")]
pub fn cmd_string_last(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 4, 5, "needleString haystackString ?lastIndex?")?;
let needle = argv[2].as_str();
let haystack = argv[3].as_str();
let count = haystack.chars().count();
let last: Option<usize> = if argv.len() == 5 {
let arg = argv[4].as_int()?;
if arg < 0 {
return molt_opt_ok!(-1);
}
if arg as usize >= count {
None
} else {
Some(arg as usize)
}
} else {
None
};
let slice = match last {
None => haystack,
Some(n) => match haystack.char_indices().nth(n + 1) {
None => haystack,
Some((byte, _)) => &haystack[..byte],
},
};
let pos_byte = slice.rfind(needle);
let pos_char: MoltInt = match pos_byte {
None => -1,
Some(b) => haystack
.char_indices()
.take_while(|(i, _)| *i < b)
.count() as MoltInt
};
molt_opt_ok!(pos_char)
}
#[cfg(feature = "string-command")]
pub fn cmd_string_length(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 3, "string")?;
let len: MoltInt = argv[2].as_str().chars().count() as MoltInt;
molt_opt_ok!(len)
}
#[cfg(all(feature = "dict", feature = "string-command"))]
pub fn cmd_string_map(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 4, 5, "?-nocase? charMap string")?;
let mut nocase = false;
if argv.len() == 5 {
let opt = argv[2].as_str();
if opt == "-nocase" {
nocase = true;
} else {
return molt_err!("bad option \"{}\": must be -nocase", opt);
}
}
let char_map = argv[argv.len() - 2].as_dict()?;
let string = argv[argv.len() - 1].as_str();
let filtered_keys = char_map
.iter()
.map(|(k, v)| {
let new_k = if nocase {
Value::from(util::to_lowercase(k.as_str()))
} else {
k.clone()
};
let count = new_k.as_str().chars().count();
(new_k, count, v.clone())
})
.filter(|(_, count, _)| *count > 0)
.collect::<Vec<_>>();
let string_lower: Option<String> = if nocase {
Some(util::to_lowercase(string))
} else {
None
};
let mut result = String::new();
let mut skip = 0;
for (i, c) in string.char_indices() {
if skip > 0 {
skip -= 1;
continue;
}
let mut matched = false;
for (from, from_char_count, to) in &filtered_keys {
let haystack: &str = match &string_lower {
Some(x) => &x[i..],
None => &string[i..],
};
if haystack.starts_with(from.as_str()) {
matched = true;
result.push_str(to.as_str());
skip = from_char_count - 1;
break;
}
}
if !matched {
result.push(c);
}
}
molt_opt_ok!(result)
}
#[cfg(feature = "string-command")]
pub fn cmd_string_range(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 5, 5, "string first last")?;
let string = argv[2].as_str();
let first = argv[3].as_int()?;
let last = argv[4].as_int()?;
if last < 0 {
return molt_opt_ok!("");
}
let clamp = { |i: MoltInt| if i < 0 {
0
} else {
i
}
};
let substr = string
.chars()
.skip(clamp(first) as usize)
.take((clamp(last) - clamp(first) + 1) as usize)
.collect::<String>();
molt_opt_ok!(substr)
}
#[cfg(feature = "string-command")]
pub fn cmd_string_tolower(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 3, "string")?;
let lower = util::to_lowercase(argv[2].as_str());
molt_opt_ok!(lower)
}
#[cfg(feature = "string-command")]
pub fn cmd_string_toupper(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 3, "string")?;
let upper = util::to_uppercase(argv[2].as_str());
molt_opt_ok!(upper)
}
#[cfg(feature = "string-command")]
pub fn cmd_string_trim(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(2, argv, 3, 3, "string")?;
let s = argv[2].as_str();
let trimmed = match argv[1].as_str() {
"trimleft" => s.trim_start_matches(util::is_whitespace),
"trimright" => s.trim_end_matches(util::is_whitespace),
_ => s.trim_matches(util::is_whitespace),
};
molt_opt_ok!(trimmed.to_string())
}
pub fn cmd_throw(_interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 3, 3, "type message")?;
Err(Exception::molt_err2(argv[1].clone(), argv[2].clone()))
}
#[cfg(feature = "std")]
pub fn cmd_time(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 2, 3, "command ?count?")?;
let command = &argv[1];
let count = if argv.len() == 3 {
argv[2].as_int()?
} else {
1
};
let start = std::time::Instant::now();
for _i in 0..count {
interp.eval_value(command)?;
}
let span = start.elapsed();
let avg = if count > 0 {
span.as_nanos() / (count as u128)
} else {
0
} as MoltInt;
molt_opt_ok!("{} nanoseconds per iteration", avg)
}
pub fn cmd_unset(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 1, 0, "?-nocomplain? ?--? ?name name name...?")?;
let mut options_ok = true;
for arg in argv {
let var = arg.as_str();
if options_ok {
if var == "--" {
options_ok = false;
continue;
} else if var == "-nocomplain" {
continue;
}
}
interp.unset_var(arg);
}
molt_opt_ok!()
}
pub fn cmd_while(interp: &mut Interp, argv: &[Value]) -> MoltOptResult {
check_args(1, argv, 3, 3, "test command")?;
while interp.expr_bool(&argv[1])? {
let result = interp.eval_value(&argv[2]);
if let Err(exception) = result {
match exception.code() {
ResultCode::Break => break,
ResultCode::Continue => (),
_ => return Err(exception),
}
}
}
molt_opt_ok!()
}