rust-store-core 4.6.1

//! 跨模块共享的小工具

use serde_json::{json, Map, Value};

use crate::command::T2Q_MAX_ROWS;
use crate::schema::{Profile, Registry};

/// 从 params 中按 `@ref` 取值；ref 为空/缺席返回 None
pub(crate) fn param<'a>(params: &'a Map<String, Value>, r: Option<&String>) -> Option<&'a Value> {
    let r = r?;
    if r.is_empty() {
        return None;
    }
    // 对应 JS `ref.slice(1)`：无条件去掉首字符 '@'
    let key = r.get(1..).unwrap_or("");
    params.get(key)
}

/// 收集 `$having` 引用的聚合别名（含 `$and` / `$or` / `$nor` 内部）；
/// `is_agg` 判定某键是否为本块聚合别名（根级 `$group` 与关系聚合谓词共用同一遍历）。
pub(crate) fn collect_having_agg_refs(
    having: &Value,
    is_agg: &dyn Fn(&str) -> bool,
    out: &mut std::collections::HashSet<String>,
) {
    let Some(m) = having.as_object() else {
        return;
    };
    for (k, v) in m {
        if matches!(k.as_str(), "$and" | "$or" | "$nor") {
            if let Some(arr) = v.as_array() {
                for it in arr {
                    collect_having_agg_refs(it, is_agg, out);
                }
            }
        } else if !k.starts_with('$') && is_agg(k) {
            out.insert(k.clone());
        }
    }
}

/// JS `x != null`：undefined / null 均视为缺席
pub(crate) fn is_nullish(v: Option<&Value>) -> bool {
    match v {
        None => true,
        Some(Value::Null) => true,
        Some(_) => false,
    }
}

/// [`is_nullish`] 的 checked 版：nullish（None / `null`）→ None，否则原值返回。
///
/// 用于替换 `if !is_nullish(x) { … x.unwrap() … }` 这类**条件与取值重复求值**的
/// 脆弱模式 —— 条件若被重构而 guard 漂移，`unwrap()` 立即 panic；改为一次判定后
/// 由 `Option` 携带值，前提不再靠人工维护。
pub(crate) fn non_nullish(v: Option<&Value>) -> Option<&Value> {
    v.filter(|x| !x.is_null())
}

/// text2query 档行数封顶（**省略即视为上限**；`standard` 档原样返回）。
///
/// 用于**根级取数**（最终结果规模）：AI 问数不写 `$limit` 时不得全表拉取。
pub(crate) fn force_t2q_limit(registry: &Registry, limit: Option<&Value>) -> Option<Value> {
    if registry.profile() != Profile::Text2Query {
        return limit.cloned();
    }
    let l = limit.and_then(|v| v.as_f64()).unwrap_or(T2Q_MAX_ROWS);
    Some(Value::from(l.min(T2Q_MAX_ROWS) as i64))
}

/// text2query 档行数封顶（**仅夹上限，不强加**；`standard` 档原样返回）。
///
/// 用于**关系级取数**：用户未显式给 `$limit` 时保持「该父行全部子行」语义，
/// 避免改变分页深度计数（`has_paginated`）判定。
pub(crate) fn clamp_t2q_limit(registry: &Registry, limit: Option<&Value>) -> Option<Value> {
    let v = limit?;
    if registry.profile() != Profile::Text2Query {
        return Some(v.clone());
    }
    match v.as_f64() {
        Some(l) => Some(Value::from(l.min(T2Q_MAX_ROWS) as i64)),
        None => Some(v.clone()),
    }
}

/// 定位首个「含指定阶段键」的阶段下标（自定义 `$pipeline` 的覆盖/追加定位用）。
pub(super) fn find_stage_idx(stages: &[Value], key: &str) -> Option<usize> {
    stages
        .iter()
        .position(|s| s.as_object().map(|o| o.contains_key(key)).unwrap_or(false))
}

/// 按序追加 $sort/$skip/$limit
pub(super) fn append_order(
    stages: &mut Vec<Value>,
    sort: Option<&Value>,
    skip_val: Option<&Value>,
    limit_val: Option<&Value>,
) {
    if let Some(v) = non_nullish(sort) {
        stages.push(json!({ "$sort": v.clone() }));
    }
    if let Some(v) = non_nullish(skip_val) {
        stages.push(json!({ "$skip": v.clone() }));
    }
    if let Some(v) = non_nullish(limit_val) {
        stages.push(json!({ "$limit": v.clone() }));
    }
}