dynamodb-expression 0.2.0-beta.9

use core::fmt::{self, Write};

use crate::operand::Operand;

/// Represents a [DynamoDB `IN` condition][1]. True if the value from the
/// [`Operand`] (the `op` parameter) is equal to any value in the list (the
/// `items` parameter).
///
/// The DynamoDB allows the list to contain up to 100 values. It must have at least 1.
///
/// See also: [`Path::in_`]
///
/// ```
/// # fn main() -> Result<(), Box<dyn std::error::Error>> {
/// use dynamodb_expression::{condition::In, operand::Operand, Path};
/// # use pretty_assertions::assert_eq;
///
/// let condition = "name".parse::<Path>()?.in_(["Jack", "Jill"]);
/// assert_eq!(r#"name IN ("Jack","Jill")"#, condition.to_string());
/// #
/// # Ok(())
/// # }
/// ```
///
/// [1]: https://docs.aws.amazon.com/amazondynamodb/latest/developerguide/Expressions.OperatorsAndFunctions.html#Expressions.OperatorsAndFunctions.Comparators
/// [`Path::in_`]: crate::path::Path::in_
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct In {
    pub(crate) op: Operand,
    pub(crate) items: Vec<Operand>,
}

impl In {
    /// Allows for manually creating an `In` instance.
    ///
    /// See also: [`Path::in_`]
    ///
    /// ```
    /// # fn main() -> Result<(), Box<dyn std::error::Error>> {
    /// use dynamodb_expression::{condition::In, operand::Operand, Path};
    /// # use pretty_assertions::assert_eq;
    ///
    /// let condition = In::new("name".parse::<Path>()?, ["Jack", "Jill"]);
    /// assert_eq!(r#"name IN ("Jack","Jill")"#, condition.to_string());
    /// #
    /// # Ok(())
    /// # }
    /// ```
    ///
    /// [1]: https://docs.aws.amazon.com/amazondynamodb/latest/developerguide/Expressions.OperatorsAndFunctions.html#Expressions.OperatorsAndFunctions.Comparators
    /// [`Path::in_`]: crate::path::Path::in_
    pub fn new<O, I, T>(op: O, items: I) -> Self
    where
        O: Into<Operand>,
        I: IntoIterator<Item = T>,
        T: Into<Operand>,
    {
        Self {
            op: op.into(),
            items: items.into_iter().map(Into::into).collect(),
        }
    }
}

impl fmt::Display for In {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        self.op.fmt(f)?;
        f.write_str(" IN (")?;

        let mut first = true;
        self.items.iter().try_for_each(|item| {
            if first {
                first = false;
            } else {
                f.write_char(',')?;
            }

            item.fmt(f)
        })?;

        f.write_char(')')
    }
}