use std::collections::{BTreeMap, HashSet};
use std::fmt;
use serde::Serialize;
use crate::compiler::ast;
use crate::compiler::instructions::Instructions;
use crate::compiler::meta::find_undeclared;
use crate::compiler::parser::parse_expr;
use crate::environment::Environment;
use crate::error::Error;
use crate::output::Output;
use crate::value::Value;
use crate::vm::Vm;
pub struct Expression<'env, 'source> {
env: &'env Environment<'source>,
instr: ExpressionBacking<'source>,
}
enum ExpressionBacking<'source> {
Borrowed(Instructions<'source>),
Owned(crate::loader::OwnedInstructions),
}
impl fmt::Debug for Expression<'_, '_> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("Expression")
.field("env", &self.env)
.finish()
}
}
impl<'env, 'source> Expression<'env, 'source> {
pub(crate) fn new(
env: &'env Environment<'source>,
instructions: Instructions<'source>,
) -> Expression<'env, 'source> {
Expression {
env,
instr: ExpressionBacking::Borrowed(instructions),
}
}
pub(crate) fn new_owned(
env: &'env Environment<'source>,
instructions: crate::loader::OwnedInstructions,
) -> Expression<'env, 'source> {
Expression {
env,
instr: ExpressionBacking::Owned(instructions),
}
}
fn instructions(&self) -> &Instructions<'_> {
match self.instr {
ExpressionBacking::Borrowed(ref x) => x,
ExpressionBacking::Owned(ref x) => x.borrow_dependent(),
}
}
pub fn eval<S: Serialize>(&self, ctx: S) -> Result<Value, Error> {
self._eval(Value::from_serialize(&ctx))
}
pub fn undeclared_variables(&self, nested: bool) -> HashSet<String> {
match parse_expr(self.instructions().source()) {
Ok(expr) => find_undeclared(
&ast::Stmt::EmitExpr(ast::Spanned::new(
ast::EmitExpr { expr },
Default::default(),
)),
nested,
),
Err(_) => HashSet::new(),
}
}
fn _eval(&self, root: Value) -> Result<Value, Error> {
Ok(ok!(Vm::new(self.env).eval(
self.instructions(),
root,
&BTreeMap::new(),
&mut Output::null(),
crate::AutoEscape::None,
))
.0
.expect("expression evaluation did not leave value on stack"))
}
}