wedb_embed 0.1.12

use sonic_rs::{JsonContainerTrait, JsonValueMutTrait, JsonValueTrait, Value};

/// JSON Merge Patch (RFC 7396).
/// JSON 合并补丁规范实现（RFC 7396）
pub fn json_merge_patch(target: &mut Value, patch: &Value) {
  if let Some(patch_obj) = patch.as_object() {
    if !target.is_object() {
      *target = sonic_rs::json!({});
    }
    let Some(target_obj) = target.as_object_mut() else {
      return;
    };
    for (k, v) in patch_obj.iter() {
      if v.is_null() {
        target_obj.remove(&k);
      } else {
        let entry = target_obj.entry(k).or_insert(sonic_rs::json!(null));
        json_merge_patch(entry, v);
      }
    }
  } else {
    *target = patch.clone();
  }
}

/// Formats JSON string output with INDENT / NEWLINE / SPACE options.
/// 格式化 JSON 字符串输出（对标 Kvrocks INDENT / NEWLINE / SPACE 选项）
pub fn format_json(
  val: &Value,
  indent: Option<&str>,
  newline: Option<&str>,
  space: Option<&str>,
) -> String {
  if indent.is_none() && newline.is_none() && space.is_none() {
    return sonic_rs::to_string(val).unwrap_or_default();
  }

  let mut out = String::with_capacity(128);
  format_value_recursive(val, 0, indent, newline, space, &mut out);
  out
}

fn format_value_recursive(
  val: &Value,
  depth: usize,
  indent: Option<&str>,
  newline: Option<&str>,
  space: Option<&str>,
  out: &mut String,
) {
  if val.is_null() {
    out.push_str("null");
  } else if let Some(b) = val.as_bool() {
    out.push_str(if b { "true" } else { "false" });
  } else if let Some(i) = val.as_i64() {
    let mut b = itoa::Buffer::new();
    out.push_str(b.format(i));
  } else if let Some(u) = val.as_u64() {
    let mut b = itoa::Buffer::new();
    out.push_str(b.format(u));
  } else if let Some(f) = val.as_f64() {
    let mut b = zmij::Buffer::new();
    out.push_str(b.format(f));
  } else if val.is_str() {
    out.push_str(&sonic_rs::to_string(val).unwrap_or_default());
  } else if let Some(arr) = val.as_array() {
    if arr.is_empty() {
      out.push_str("[]");
      return;
    }
    let nl = newline.unwrap_or("");
    let has_nl = !nl.is_empty();
    let ind = indent.unwrap_or("");

    out.push('[');
    for (i, elem) in arr.iter().enumerate() {
      if i > 0 {
        out.push(',');
      }
      if has_nl {
        out.push_str(nl);
        for _ in 0..=depth {
          out.push_str(ind);
        }
      }
      format_value_recursive(elem, depth + 1, indent, newline, space, out);
    }
    if has_nl {
      out.push_str(nl);
      for _ in 0..depth {
        out.push_str(ind);
      }
    }
    out.push(']');
  } else if let Some(obj) = val.as_object() {
    if obj.is_empty() {
      out.push_str("{}");
      return;
    }
    let nl = newline.unwrap_or("");
    let has_nl = !nl.is_empty();
    let ind = indent.unwrap_or("");
    let colon_sep = if space == Some(" ") { ": " } else { ":" };

    out.push('{');
    for (i, (k, v)) in obj.iter().enumerate() {
      if i > 0 {
        out.push(',');
      }
      if has_nl {
        out.push_str(nl);
        for _ in 0..=depth {
          out.push_str(ind);
        }
      }
      out.push('"');
      out.push_str(k);
      out.push('"');
      out.push_str(colon_sep);
      format_value_recursive(v, depth + 1, indent, newline, space, out);
    }
    if has_nl {
      out.push_str(nl);
      for _ in 0..depth {
        out.push_str(ind);
      }
    }
    out.push('}');
  }
}

/// Transforms a JSON value into Redis RESP format string aligned with Kvrocks Json::TransformResp / JSON.RESP.
/// 将 JSON 节点序列化为 RESP 格式字符串（对标 Kvrocks Json::TransformResp 与 Redis JSON.RESP）
pub fn json_transform_resp(origin: &Value, out: &mut String) {
  if let Some(obj) = origin.as_object() {
    let mut b = itoa::Buffer::new();
    out.push('*');
    out.push_str(b.format(obj.len() * 2 + 1));
    out.push_str("\r\n+{\r\n");
    for (k, v) in obj.iter() {
      let mut len_b = itoa::Buffer::new();
      out.push('$');
      out.push_str(len_b.format(k.len()));
      out.push_str("\r\n");
      out.push_str(k);
      out.push_str("\r\n");
      json_transform_resp(v, out);
    }
  } else if let Some(arr) = origin.as_array() {
    let mut b = itoa::Buffer::new();
    out.push('*');
    out.push_str(b.format(arr.len() + 1));
    out.push_str("\r\n+[\r\n");
    for item in arr.iter() {
      json_transform_resp(item, out);
    }
  } else if let Some(i) = origin.as_i64() {
    let mut b = itoa::Buffer::new();
    out.push(':');
    out.push_str(b.format(i));
    out.push_str("\r\n");
  } else if let Some(u) = origin.as_u64() {
    let mut b = itoa::Buffer::new();
    out.push(':');
    out.push_str(b.format(u));
    out.push_str("\r\n");
  } else if let Some(s) = origin.as_str() {
    let mut b = itoa::Buffer::new();
    out.push('$');
    out.push_str(b.format(s.len()));
    out.push_str("\r\n");
    out.push_str(s);
    out.push_str("\r\n");
  } else if let Some(f) = origin.as_f64() {
    let mut b = zmij::Buffer::new();
    let s = b.format(f);
    let mut len_b = itoa::Buffer::new();
    out.push('$');
    out.push_str(len_b.format(s.len()));
    out.push_str("\r\n");
    out.push_str(s);
    out.push_str("\r\n");
  } else if let Some(b) = origin.as_bool() {
    if b {
      out.push_str("+true\r\n");
    } else {
      out.push_str("+false\r\n");
    }
  } else if origin.is_null() {
    out.push_str("$-1\r\n");
  }
}