use crate::{
Span, Spanned,
expression::Expression,
visitor::{VisitWith, Visitor, VisitorMut},
};
use boa_interner::{Interner, ToInternedString};
use core::ops::ControlFlow;
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
#[derive(Clone, Debug, PartialEq)]
pub struct Conditional {
condition: Box<Expression>,
if_true: Box<Expression>,
if_false: Box<Expression>,
}
impl Conditional {
#[inline]
#[must_use]
pub const fn condition(&self) -> &Expression {
&self.condition
}
#[inline]
#[must_use]
pub const fn if_true(&self) -> &Expression {
&self.if_true
}
#[inline]
#[must_use]
pub const fn if_false(&self) -> &Expression {
&self.if_false
}
#[inline]
#[must_use]
pub fn new(condition: Expression, if_true: Expression, if_false: Expression) -> Self {
Self {
condition: Box::new(condition),
if_true: Box::new(if_true),
if_false: Box::new(if_false),
}
}
}
impl Spanned for Conditional {
#[inline]
fn span(&self) -> Span {
Span::new(self.condition.span().start(), self.if_false.span().end())
}
}
impl ToInternedString for Conditional {
#[inline]
fn to_interned_string(&self, interner: &Interner) -> String {
format!(
"{} ? {} : {}",
self.condition().to_interned_string(interner),
self.if_true().to_interned_string(interner),
self.if_false().to_interned_string(interner)
)
}
}
impl From<Conditional> for Expression {
#[inline]
fn from(cond_op: Conditional) -> Self {
Self::Conditional(cond_op)
}
}
impl VisitWith for Conditional {
fn visit_with<'a, V>(&'a self, visitor: &mut V) -> ControlFlow<V::BreakTy>
where
V: Visitor<'a>,
{
visitor.visit_expression(&self.condition)?;
visitor.visit_expression(&self.if_true)?;
visitor.visit_expression(&self.if_false)
}
fn visit_with_mut<'a, V>(&'a mut self, visitor: &mut V) -> ControlFlow<V::BreakTy>
where
V: VisitorMut<'a>,
{
visitor.visit_expression_mut(&mut self.condition)?;
visitor.visit_expression_mut(&mut self.if_true)?;
visitor.visit_expression_mut(&mut self.if_false)
}
}