use endbasic_core::*;
use std::borrow::Cow;
use std::rc::Rc;
pub(super) struct SumIntegersFunction {
metadata: Rc<CallableMetadata>,
}
impl SumIntegersFunction {
pub(super) fn new() -> Rc<Self> {
Rc::from(Self {
metadata: CallableMetadataBuilder::new("SUM_INTEGERS")
.with_return_type(ExprType::Integer)
.with_syntax(&[(
&[],
Some(&RepeatedSyntax {
name: Cow::Borrowed("arg"),
type_syn: RepeatedTypeSyntax::AnyValue,
sep: ArgSepSyntax::Exactly(ArgSep::Long),
require_one: false,
allow_missing: true,
}),
)])
.test_build(),
})
}
}
impl Callable for SumIntegersFunction {
fn metadata(&self) -> Rc<CallableMetadata> {
self.metadata.clone()
}
fn exec(&self, scope: Scope<'_>) -> CallResult<()> {
let mut total: i32 = 0;
let mut reg = 0;
loop {
let sep = match scope.get_type(reg) {
VarArgTag::Immediate(sep, etype) => {
reg += 1;
match etype {
ExprType::Integer => total += scope.get_integer(reg),
_ => {
return Err(CallError::Argument(
"Only accepts integer values".to_owned(),
));
}
}
sep
}
_ => return Err(CallError::Argument("Only accepts integer values".to_owned())),
};
reg += 1;
if sep == ArgSep::End {
break;
}
}
scope.return_integer(total)
}
}