ValueScript/valuescript_vm/src/virtual_machine.rs

use std::rc::Rc;

use crate::builtins::error_builtin::ToError;
use crate::bytecode::Bytecode;
use crate::bytecode::DecoderMaker;
use crate::first_stack_frame::FirstStackFrame;
use crate::stack_frame::FrameStepOk;
use crate::stack_frame::StackFrame;
use crate::vs_value::{LoadFunctionResult, Val, ValTrait};

pub struct VirtualMachine {
  pub frame: StackFrame,
  pub stack: Vec<StackFrame>,
}

impl VirtualMachine {
  pub fn run(
    &mut self,
    bytecode: Rc<Bytecode>,
    step_limit: Option<usize>,
    params: &[Val],
  ) -> Result<Val, Val> {
    let mut bd = bytecode.decoder(0);

    let main_fn = bd.decode_val(&Vec::new());

    let mut frame = match main_fn.load_function() {
      LoadFunctionResult::StackFrame(f) => f,
      _ => panic!("bytecode does start with function"),
    };

    for p in params {
      frame.write_param(p.clone());
    }

    self.push(frame);

    match step_limit {
      Some(step_limit) => {
        let mut step_count = 0;

        while step_count < step_limit {
          self.step()?;
          step_count += 1;

          if self.stack.len() == 0 {
            return Ok(self.frame.get_call_result().return_);
          }
        }

        Err("step limit reached".to_error())
      }
      None => {
        while self.stack.len() > 0 {
          self.step()?;
        }

        Ok(self.frame.get_call_result().return_)
      }
    }
  }

  pub fn new() -> VirtualMachine {
    let mut registers: Vec<Val> = Vec::with_capacity(2);
    registers.push(Val::Undefined);
    registers.push(Val::Undefined);

    return VirtualMachine {
      frame: Box::new(FirstStackFrame::new()),
      stack: Default::default(),
    };
  }

  pub fn step(&mut self) -> Result<(), Val> {
    let step_ok = match self.frame.step() {
      Ok(step_ok) => step_ok,
      Err(e) => return self.handle_exception(e),
    };

    match step_ok {
      FrameStepOk::Continue => {}
      FrameStepOk::Pop(call_result) => {
        self.pop();
        self.frame.apply_call_result(call_result);
      }
      FrameStepOk::Push(new_frame) => {
        self.push(new_frame);
      }
      // TODO: Internal errors
      FrameStepOk::Yield(_) => return self.handle_exception("Unexpected yield".to_error()),
      FrameStepOk::YieldStar(_) => return self.handle_exception("Unexpected yield*".to_error()),
    }

    Ok(())
  }

  pub fn push(&mut self, mut frame: StackFrame) {
    std::mem::swap(&mut self.frame, &mut frame);
    self.stack.push(frame);
  }

  pub fn pop(&mut self) {
    // This name is accurate after the swap
    let mut old_frame = self.stack.pop().unwrap();
    std::mem::swap(&mut self.frame, &mut old_frame);
  }

  pub fn handle_exception(&mut self, exception: Val) -> Result<(), Val> {
    while !self.stack.is_empty() {
      let handled = self.frame.catch_exception(exception.clone());

      if handled {
        return Ok(());
      }

      if self.stack.is_empty() {
        return Err(exception);
      }

      self.pop();
    }

    Err(exception)
  }

  pub fn read_default_export(bytecode: Rc<Bytecode>) -> Val {
    bytecode.decoder(0).decode_val(&Vec::new())
  }
}