thymeleaf 0.1.0-beta.1

use std::sync::Arc;

use crate::util::{Utf16String, ValidateError};

use super::{IStandardExpression, StandardExpressionResult};

/// `th:each` 解析后的迭代声明。
///
/// 对应 Java: `org.thymeleaf.standard.expression.Each`。
pub struct Each {
    iter_var: Arc<dyn IStandardExpression>,
    status_var: Option<Arc<dyn IStandardExpression>>,
    iterable: Arc<dyn IStandardExpression>,
}

impl Each {
    /// 创建迭代声明，并按 Java 顺序校验迭代变量和 iterable。
    /// 对应 Java 语义：`Each` 的 `new` 行为（Rust 侧辅助/私有路径）。
    pub fn new(
        iter_var: Option<Arc<dyn IStandardExpression>>,
        status_var: Option<Arc<dyn IStandardExpression>>,
        iterable: Option<Arc<dyn IStandardExpression>>,
    ) -> Result<Self, ValidateError> {
        let iter_var = iter_var.ok_or_else(|| ValidateError::IllegalArgument {
            message: Some("Iteration variable cannot be null".to_owned()),
        })?;
        let iterable = iterable.ok_or_else(|| ValidateError::IllegalArgument {
            message: Some("Iterable cannot be null".to_owned()),
        })?;
        Ok(Self {
            iter_var,
            status_var,
            iterable,
        })
    }
    /// 返回迭代变量表达式。
    /// 对应 Java: `Each#getIterVar()`。
    pub fn get_iter_var(&self) -> &dyn IStandardExpression {
        self.iter_var.as_ref()
    }
    /// 判断是否声明状态变量。
    /// 对应 Java: `Each#hasStatusVar()`。
    pub fn has_status_var(&self) -> bool {
        self.status_var.is_some()
    }
    /// 返回可空状态变量表达式。
    /// 对应 Java: `Each#getStatusVar()`。
    pub fn get_status_var(&self) -> Option<&dyn IStandardExpression> {
        self.status_var.as_deref()
    }
    /// 返回 iterable 表达式。
    /// 对应 Java: `Each#getIterable()`。
    pub fn get_iterable(&self) -> &dyn IStandardExpression {
        self.iterable.as_ref()
    }
    /// 返回 `iter[,status] : iterable` 规范文本。
    /// 对应 Java: `Each#getStringRepresentation()`。
    pub fn get_string_representation(&self) -> StandardExpressionResult<Utf16String> {
        let mut units = self
            .iter_var
            .get_string_representation()?
            .as_utf16()
            .to_vec();
        if let Some(status_var) = &self.status_var {
            units.push(b',' as u16);
            units.extend_from_slice(status_var.get_string_representation()?.as_utf16());
        }
        units.extend_from_slice(&[b' ' as u16, b':' as u16, b' ' as u16]);
        units.extend_from_slice(self.iterable.get_string_representation()?.as_utf16());
        Ok(Utf16String::from_utf16(units))
    }
}