ValueScript/valuescript_vm/src/array_higher_functions/array_reduce_right.rs

use std::rc::Rc;

use super::super::vs_value::{Val, ValTrait, LoadFunctionResult};
use super::super::vs_array::VsArray;
use super::super::native_frame_function::NativeFrameFunction;
use super::super::stack_frame::{StackFrameTrait, FrameStepResult, CallResult};

pub static REDUCE_RIGHT: NativeFrameFunction = NativeFrameFunction {
  make_frame: || Box::new(ReduceRightFrame {
    this: None,
    array_i: 0,
    reducer: Val::Void,
    param_i: 0,
    value: None,
  }),
};

struct ReduceRightFrame {
  this: Option<Rc<VsArray>>,
  array_i: usize,

  reducer: Val,
  param_i: usize,
  value: Option<Val>,
}

impl StackFrameTrait for ReduceRightFrame {
  fn write_this(&mut self, this: Val) {
    self.this = this.as_array_data();

    match &self.this {
      None => {},
      Some(ad) => {
        self.array_i = ad.elements.len()
      },
    };
  }

  fn write_param(&mut self, param: Val) {
    match self.param_i {
      0 => { self.reducer = param; }
      1 => { self.value = Some(param); }
      _ => {},
    };

    self.param_i += 1;
  }

  fn step(&mut self) -> FrameStepResult {
    let array_data = match &self.this {
      None => std::panic!("Not implemented: exception: reduceRight called on non-array"),
      Some(ad) => ad,
    };

    if self.array_i == 0 {
      match &self.value {
        None => {
          std::panic!("Not implemented: exception: reduceRight of empty array with no initial value");
        },
        Some(value) => {
          return FrameStepResult::Pop(CallResult {
            return_: value.clone(),
            this: Val::Array(array_data.clone()),
          });
        },
      }
    }

    self.array_i -= 1;
    let array_i = self.array_i;

    let el = &array_data.elements[array_i];

    match el {
      Val::Void => {
        return FrameStepResult::Continue;
      },
      _ => match &self.value {
        None => {
          self.value = Some(el.clone());
          return FrameStepResult::Continue;
        },
        Some(value) => match self.reducer.load_function() {
          LoadFunctionResult::NotAFunction =>
            std::panic!("Not implemented: exception: reduceRight fn is not a function")
          ,
          LoadFunctionResult::NativeFunction(native_fn) => {
            self.value = Some(native_fn(
              &mut Val::Undefined,
              vec![
                value.clone(),
                el.clone(),
                Val::Number(array_i as f64),
                Val::Array(array_data.clone()),
              ],
            ));
  
            return FrameStepResult::Continue;
          },
          LoadFunctionResult::StackFrame(mut new_frame) => {
            new_frame.write_param(value.clone());
            new_frame.write_param(el.clone());
            new_frame.write_param(Val::Number(array_i as f64));
            new_frame.write_param(Val::Array(array_data.clone()));
            return FrameStepResult::Push(new_frame);
          },
        },
      },
    };
  }

  fn apply_call_result(&mut self, call_result: CallResult) {
    self.value = Some(call_result.return_);
  }

  fn get_call_result(&mut self) -> CallResult {
    std::panic!("Not appropriate for ReduceRightFrame")
  }
}