thymeleaf 0.1.0-beta.1

use std::sync::{Arc, RwLock, RwLockReadGuard, RwLockWriteGuard, Weak};

use indexmap::IndexMap;
use thiserror::Error;

use crate::context::IExpressionContext;
use crate::util::Utf16String;

use super::{
    ExpressionObjectNames, IExpressionObjectFactory, IExpressionObjects, StandardExpressionResult,
    TemplateValue,
};

/// 表达式工具对象容器。
///
/// 对应 Java: `org.thymeleaf.expression.ExpressionObjects`。
///
/// 容器保存工厂在构造时返回的同一共享名称集合。对象仅在第一次读取时创建；可缓存
/// 对象连同 Java `null` 结果一起缓存，非缓存对象每次读取都重新调用工厂。容器按
/// 模板执行创建，默认预留三个缓存项。
pub struct ExpressionObjects {
    context: Weak<dyn IExpressionContext>,
    expression_object_factory: Arc<dyn IExpressionObjectFactory>,
    expression_object_names: ExpressionObjectNames,
    objects: RwLock<IndexMap<Option<Utf16String>, Option<Arc<TemplateValue>>>>,
}

/// 创建或使用表达式对象容器时可能出现的结构错误。
#[derive(Debug, Error, PartialEq, Eq)]
/// 对应 Java 语义：`ExpressionObjects` 的 Rust 侧类型 `ExpressionObjectsError`。
pub enum ExpressionObjectsError {
    /// Java `Validate.notNull` 对应错误。
    #[error("{message}")]
    IllegalArgument {
        /// Java 异常消息。
        message: &'static str,
    },
}

impl ExpressionObjects {
    /// 使用表达式上下文和工厂创建请求级对象容器。
    ///
    /// 对应 Java: `ExpressionObjects#ExpressionObjects(IExpressionContext,
    /// IExpressionObjectFactory)`。
    ///
    /// # 参数
    /// - `context`：当前表达式上下文的弱引用；Rust 借用关系保证容器不会脱离其所属
    ///   Context 使用，同时避免 Java 垃圾收集可处理而 `Arc` 无法自动回收的引用环。
    /// - `expression_object_factory`：声明并创建表达式对象的工厂。
    ///
    /// # 错误
    /// 任一参数为 `None` 时按 Java 校验顺序返回对应参数错误。
    pub fn new(
        context: Option<Weak<dyn IExpressionContext>>,
        expression_object_factory: Option<Arc<dyn IExpressionObjectFactory>>,
    ) -> Result<Self, ExpressionObjectsError> {
        let context = context.ok_or(ExpressionObjectsError::IllegalArgument {
            message: "Context cannot be null",
        })?;
        let expression_object_factory =
            expression_object_factory.ok_or(ExpressionObjectsError::IllegalArgument {
                message: "Expression Object Factory cannot be null",
            })?;
        let expression_object_names: ExpressionObjectNames = expression_object_factory
            .get_all_expression_object_names()
            .unwrap_or_else(|| Arc::from([]));

        Ok(Self {
            context,
            expression_object_factory,
            expression_object_names,
            objects: RwLock::new(IndexMap::with_capacity(3)),
        })
    }
}

impl IExpressionObjects for ExpressionObjects {
    fn size(&self) -> i32 {
        i32::try_from(self.expression_object_names.len()).unwrap_or(i32::MAX)
    }

    fn contains_object(&self, name: Option<&Utf16String>) -> bool {
        self.expression_object_names
            .iter()
            .any(|candidate| candidate.as_ref() == name)
    }

    fn get_object_names(&self) -> ExpressionObjectNames {
        Arc::clone(&self.expression_object_names)
    }

    fn get_object(
        &self,
        name: Option<&Utf16String>,
    ) -> StandardExpressionResult<Option<Arc<TemplateValue>>> {
        let key = name.cloned();

        // HashMap#containsKey 与 get 分开，确保缓存的 Java null 不会被误判成未缓存。
        if let Some(object) = read_recovering_poison(&self.objects).get(&key) {
            return Ok(object.clone());
        }
        if !self
            .expression_object_names
            .iter()
            .any(|candidate| candidate == &key)
        {
            return Ok(None);
        }

        let Some(context) = self.context.upgrade() else {
            return Ok(None);
        };
        let object = self.expression_object_factory.build_object(context, name)?;
        if !self.expression_object_factory.is_cacheable(name) {
            return Ok(object);
        }

        // Java 在并发竞争时允许多个线程同时构建；最后写入缓存的结果供后续调用复用。
        write_recovering_poison(&self.objects).insert(key, object.clone());
        Ok(object)
    }
}

fn read_recovering_poison<T>(lock: &RwLock<T>) -> RwLockReadGuard<'_, T> {
    lock.read()
        .unwrap_or_else(std::sync::PoisonError::into_inner)
}

fn write_recovering_poison<T>(lock: &RwLock<T>) -> RwLockWriteGuard<'_, T> {
    lock.write()
        .unwrap_or_else(std::sync::PoisonError::into_inner)
}