965 lines
29 KiB
Rust
Raw Normal View History

2019-06-09 22:55:35 +09:00
use std::collections::HashSet;
2019-06-19 15:09:42 +09:00
use std::fmt;
2019-06-09 22:55:35 +09:00
2019-06-02 22:03:35 +09:00
use serde_json::{Number, Value};
use serde_json::map::Entry;
2019-06-02 22:03:35 +09:00
2019-06-25 22:07:59 +09:00
use parser::*;
2020-03-22 23:48:24 +09:00
use self::expr_term::*;
use self::value_walker::ValueWalker;
2019-06-02 22:03:35 +09:00
2020-03-22 23:48:24 +09:00
mod cmp;
mod expr_term;
mod value_walker;
2019-06-02 22:03:35 +09:00
fn to_f64(n: &Number) -> f64 {
if n.is_i64() {
n.as_i64().unwrap() as f64
} else if n.is_f64() {
n.as_f64().unwrap()
} else {
n.as_u64().unwrap() as f64
}
}
fn abs_index(n: isize, len: usize) -> usize {
if n < 0_isize {
(n + len as isize).max(0) as usize
} else {
n.min(len as isize) as usize
}
}
2019-06-26 15:07:49 +09:00
#[derive(Debug, PartialEq)]
2019-06-02 22:03:35 +09:00
enum FilterKey {
String(String),
All,
}
pub enum JsonPathError {
EmptyPath,
EmptyValue,
Path(String),
Serde(String),
}
2019-06-19 15:09:42 +09:00
impl fmt::Debug for JsonPathError {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "{}", self)
}
}
impl fmt::Display for JsonPathError {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self {
JsonPathError::EmptyPath => f.write_str("path not set"),
JsonPathError::EmptyValue => f.write_str("json value not set"),
JsonPathError::Path(msg) => f.write_str(&format!("path error: \n{}\n", msg)),
JsonPathError::Serde(msg) => f.write_str(&format!("serde error: \n{}\n", msg)),
}
}
}
2019-06-25 18:58:25 +09:00
#[derive(Debug, Default)]
pub struct Selector<'a, 'b> {
2019-06-02 22:03:35 +09:00
node: Option<Node>,
node_ref: Option<&'b Node>,
2019-06-02 22:03:35 +09:00
value: Option<&'a Value>,
tokens: Vec<ParseToken>,
terms: Vec<Option<ExprTerm<'a>>>,
current: Option<Vec<&'a Value>>,
selectors: Vec<Selector<'a, 'b>>,
2019-06-02 22:03:35 +09:00
}
impl<'a, 'b> Selector<'a, 'b> {
2019-06-02 22:03:35 +09:00
pub fn new() -> Self {
2019-06-25 18:58:25 +09:00
Selector::default()
2019-06-02 22:03:35 +09:00
}
2019-06-09 22:55:35 +09:00
pub fn str_path(&mut self, path: &str) -> Result<&mut Self, JsonPathError> {
2019-06-02 22:03:35 +09:00
debug!("path : {}", path);
2019-06-26 15:07:49 +09:00
self.node_ref.take();
2019-06-25 18:58:25 +09:00
self.node = Some(Parser::compile(path).map_err(JsonPathError::Path)?);
2019-06-02 22:03:35 +09:00
Ok(self)
}
pub fn node_ref(&self) -> Option<&Node> {
if let Some(node) = &self.node {
2019-06-26 15:07:49 +09:00
return Some(node);
}
2019-06-26 15:07:49 +09:00
if let Some(node) = &self.node_ref {
return Some(*node);
}
2019-06-26 15:07:49 +09:00
None
}
pub fn compiled_path(&mut self, node: &'b Node) -> &mut Self {
self.node.take();
self.node_ref = Some(node);
2019-06-02 22:03:35 +09:00
self
}
2019-06-09 22:55:35 +09:00
pub fn reset_value(&mut self) -> &mut Self {
self.current = None;
2019-06-02 22:03:35 +09:00
self
}
pub fn value(&mut self, v: &'a Value) -> &mut Self {
self.value = Some(v);
self
}
fn _select(&mut self) -> Result<(), JsonPathError> {
if self.node_ref.is_some() {
let node_ref = self.node_ref.take().unwrap();
self.visit(node_ref);
return Ok(());
2019-06-02 22:03:35 +09:00
}
if self.node.is_none() {
return Err(JsonPathError::EmptyPath);
}
let node = self.node.take().unwrap();
self.visit(&node);
self.node = Some(node);
Ok(())
2019-06-02 22:03:35 +09:00
}
pub fn select_as<T: serde::de::DeserializeOwned>(&mut self) -> Result<Vec<T>, JsonPathError> {
self._select()?;
match &self.current {
Some(vec) => {
let mut ret = Vec::new();
for v in vec {
match T::deserialize(*v) {
Ok(v) => ret.push(v),
2019-06-19 11:34:47 +09:00
Err(e) => return Err(JsonPathError::Serde(e.to_string())),
2019-06-02 22:03:35 +09:00
}
}
Ok(ret)
}
2019-06-19 11:34:47 +09:00
_ => Err(JsonPathError::EmptyValue),
2019-06-02 22:03:35 +09:00
}
}
pub fn select_as_str(&mut self) -> Result<String, JsonPathError> {
self._select()?;
match &self.current {
Some(r) => {
2019-06-19 11:34:47 +09:00
Ok(serde_json::to_string(r).map_err(|e| JsonPathError::Serde(e.to_string()))?)
2019-06-02 22:03:35 +09:00
}
2019-06-19 11:34:47 +09:00
_ => Err(JsonPathError::EmptyValue),
2019-06-02 22:03:35 +09:00
}
}
pub fn select(&mut self) -> Result<Vec<&'a Value>, JsonPathError> {
self._select()?;
match &self.current {
Some(r) => Ok(r.to_vec()),
2019-06-19 11:34:47 +09:00
_ => Err(JsonPathError::EmptyValue),
2019-06-02 22:03:35 +09:00
}
}
fn new_filter_context(&mut self) {
self.terms.push(None);
debug!("new_filter_context: {:?}", self.terms);
}
fn in_filter<F: Fn(&Vec<&'a Value>, &mut Vec<&'a Value>, &mut HashSet<usize>) -> FilterKey>(&mut self, fun: F) {
2020-02-10 23:14:30 +09:00
fn get_parent<'a>(prev: Option<Vec<&'a Value>>, current_value: &[&'a Value], not_matched: HashSet<usize>) -> Option<Vec<&'a Value>> {
if prev.is_some() {
return prev;
}
let filtered: Vec<&Value> = current_value.iter().enumerate().filter(|(idx, _)| !not_matched.contains(idx))
.map(|(_, v)| *v)
.collect();
Some(filtered)
}
2019-06-25 18:58:25 +09:00
if let Some(peek) = self.terms.pop() {
match peek {
2019-06-03 21:41:08 +09:00
Some(v) => {
debug!("in_filter 1.: {:?}", v);
2019-06-02 22:03:35 +09:00
2019-06-03 21:41:08 +09:00
match v {
ExprTerm::Json(rel, fk, vec) => {
2019-06-02 22:03:35 +09:00
let mut tmp = Vec::new();
let mut not_matched = HashSet::new();
let filter_key = if let Some(FilterKey::String(key)) = fk {
let key_contained = &vec.iter().map(|v| match v {
Value::Object(map) if map.contains_key(&key) => map.get(&key).unwrap(),
_ => v,
}).collect();
fun(key_contained, &mut tmp, &mut not_matched)
} else {
fun(&vec, &mut tmp, &mut not_matched)
};
let parent = get_parent(rel, &vec, not_matched);
self.terms.push(Some(ExprTerm::Json(parent, Some(filter_key), tmp)));
2019-06-02 22:03:35 +09:00
}
2019-06-19 11:34:47 +09:00
_ => unreachable!(),
2019-06-03 21:41:08 +09:00
};
}
_ => {
debug!("in_filter 2.: {:?}", &self.current);
if let Some(current) = &self.current {
let mut tmp = Vec::new();
let mut not_matched = HashSet::new();
let filter_key = fun(current, &mut tmp, &mut not_matched);
self.terms.push(Some(ExprTerm::Json(None, Some(filter_key), tmp)));
2019-06-02 22:03:35 +09:00
}
2019-06-03 21:41:08 +09:00
}
2019-06-25 18:58:25 +09:00
}
2019-06-02 22:03:35 +09:00
}
}
fn all_in_filter_with_str(&mut self, key: &str) {
self.in_filter(|vec, tmp, _| {
ValueWalker::all_with_str(&vec, tmp, key, true);
2019-06-02 22:03:35 +09:00
FilterKey::All
});
debug!("all_in_filter_with_str : {}, {:?}", key, self.terms);
}
fn next_in_filter_with_str(&mut self, key: &str) {
2019-06-19 11:34:47 +09:00
fn _collect<'a>(
v: &'a Value,
tmp: &mut Vec<&'a Value>,
key: &str,
visited: &mut HashSet<*const Value>,
not_matched: &mut HashSet<usize>,
2019-06-19 11:34:47 +09:00
) {
2019-06-02 22:03:35 +09:00
match v {
2019-06-19 11:34:47 +09:00
Value::Object(map) => {
if map.contains_key(key) {
let ptr = v as *const Value;
if !visited.contains(&ptr) {
visited.insert(ptr);
tmp.push(v)
}
}
}
Value::Array(vec) => {
for v in vec {
_collect(v, tmp, key, visited, not_matched);
2019-06-02 22:03:35 +09:00
}
}
_ => {}
}
}
self.in_filter(|vec, tmp, not_matched| {
2019-06-02 22:03:35 +09:00
let mut visited = HashSet::new();
for (idx, v) in vec.iter().enumerate() {
match v {
Value::Object(map) => {
if map.contains_key(key) {
let ptr = *v as *const Value;
if !visited.contains(&ptr) {
visited.insert(ptr);
tmp.push(v)
}
} else {
not_matched.insert(idx);
}
}
Value::Array(vec) => {
not_matched.insert(idx);
for v in vec {
_collect(v, tmp, key, &mut visited, not_matched);
}
}
_ => {
not_matched.insert(idx);
}
}
2019-06-02 22:03:35 +09:00
}
2019-06-02 22:03:35 +09:00
FilterKey::String(key.to_owned())
});
debug!("next_in_filter_with_str : {}, {:?}", key, self.terms);
}
fn next_from_current_with_num(&mut self, index: f64) {
fn _collect<'a>(tmp: &mut Vec<&'a Value>, vec: &'a [Value], index: f64) {
let index = abs_index(index as isize, vec.len());
if let Some(v) = vec.get(index) {
tmp.push(v);
2019-06-02 22:03:35 +09:00
}
}
2019-06-23 18:21:59 +09:00
if let Some(current) = self.current.take() {
let mut tmp = Vec::new();
for c in current {
match c {
Value::Object(map) => {
for k in map.keys() {
if let Some(Value::Array(vec)) = map.get(k) {
_collect(&mut tmp, vec, index);
}
2019-06-02 22:03:35 +09:00
}
}
Value::Array(vec) => {
_collect(&mut tmp, vec, index);
}
_ => {}
2019-06-02 22:03:35 +09:00
}
}
if tmp.is_empty() {
self.terms.pop();
self.current = Some(vec![&Value::Null]);
} else {
self.current = Some(tmp);
}
2019-06-02 22:03:35 +09:00
}
2019-06-19 11:34:47 +09:00
debug!(
"next_from_current_with_num : {:?}, {:?}",
&index, self.current
2019-06-19 11:34:47 +09:00
);
2019-06-02 22:03:35 +09:00
}
fn next_all_from_current(&mut self) {
if let Some(current) = self.current.take() {
let mut tmp = Vec::new();
for c in current {
2019-06-23 18:21:59 +09:00
match c {
Value::Object(map) => {
for (_, v) in map {
tmp.push(v)
}
}
Value::Array(vec) => {
for v in vec {
tmp.push(v);
}
}
_ => {}
}
2019-06-02 22:03:35 +09:00
}
self.current = Some(tmp);
}
debug!("next_all_from_current : {:?}", self.current);
}
fn next_from_current_with_str(&mut self, keys: &[String]) {
if let Some(current) = self.current.take() {
let mut tmp = Vec::new();
for c in current {
if let Value::Object(map) = c {
for key in keys {
if let Some(v) = map.get(key) {
tmp.push(v)
}
}
}
}
if tmp.is_empty() {
self.current = Some(vec![&Value::Null]);
} else {
self.current = Some(tmp);
}
}
debug!(
"next_from_current_with_str : {:?}, {:?}",
keys, self.current
);
}
2019-06-02 22:03:35 +09:00
fn all_from_current(&mut self) {
if let Some(current) = self.current.take() {
let mut tmp = Vec::new();
ValueWalker::all(&current, &mut tmp);
2019-06-02 22:03:35 +09:00
self.current = Some(tmp);
}
debug!("all_from_current: {:?}", self.current);
}
fn all_from_current_with_str(&mut self, key: &str) {
if let Some(current) = self.current.take() {
let mut tmp = Vec::new();
ValueWalker::all_with_str(&current, &mut tmp, key, false);
2019-06-02 22:03:35 +09:00
self.current = Some(tmp);
}
debug!("all_from_current_with_str: {}, {:?}", key, self.current);
}
fn all_from_current_with_num(&mut self, index: f64) {
if let Some(current) = self.current.take() {
let mut tmp = Vec::new();
ValueWalker::all_with_num(&current, &mut tmp, index);
self.current = Some(tmp);
}
debug!("all_from_current_with_num: {}, {:?}", index, self.current);
}
2019-06-25 18:58:25 +09:00
fn compute_absolute_path_filter(&mut self, token: &ParseToken) -> bool {
2019-06-02 22:03:35 +09:00
if !self.selectors.is_empty() {
match token {
ParseToken::Absolute | ParseToken::Relative | ParseToken::Filter(_) => {
2019-06-09 22:55:35 +09:00
let selector = self.selectors.pop().unwrap();
if let Some(current) = &selector.current {
let term = current.into();
2019-06-02 22:03:35 +09:00
if let Some(s) = self.selectors.last_mut() {
s.terms.push(Some(term));
} else {
self.terms.push(Some(term));
}
} else {
unreachable!()
}
}
_ => {}
}
}
2019-06-09 22:55:35 +09:00
if let Some(selector) = self.selectors.last_mut() {
selector.visit_token(token);
2019-06-25 18:58:25 +09:00
true
} else {
false
}
}
}
impl<'a, 'b> Selector<'a, 'b> {
fn visit_absolute(&mut self) {
if self.current.is_some() {
let mut selector = Selector::default();
if let Some(value) = self.value {
selector.value = Some(value);
selector.current = Some(vec![value]);
self.selectors.push(selector);
}
2019-06-02 22:03:35 +09:00
return;
}
2019-06-25 18:58:25 +09:00
if let Some(v) = &self.value {
self.current = Some(vec![v]);
}
}
2019-06-25 18:58:25 +09:00
fn visit_relative(&mut self) {
if let Some(ParseToken::Array) = self.tokens.last() {
let array_token = self.tokens.pop();
if let Some(ParseToken::Leaves) = self.tokens.last() {
self.tokens.pop();
self.all_from_current();
}
self.tokens.push(array_token.unwrap());
}
self.new_filter_context();
}
2019-06-02 22:03:35 +09:00
2019-06-25 18:58:25 +09:00
fn visit_array_eof(&mut self) {
if self.is_last_before_token_match(ParseToken::Array) {
if let Some(Some(e)) = self.terms.pop() {
if let ExprTerm::String(key) = e {
self.next_in_filter_with_str(&key);
self.tokens.pop();
return;
}
self.terms.push(Some(e));
}
}
if self.is_last_before_token_match(ParseToken::Leaves) {
self.tokens.pop();
self.tokens.pop();
if let Some(Some(e)) = self.terms.pop() {
if let ExprTerm::Number(n) = &e {
self.all_from_current_with_num(to_f64(n));
self.terms.pop();
return;
}
self.terms.push(Some(e));
}
}
2019-06-25 18:58:25 +09:00
if let Some(Some(e)) = self.terms.pop() {
match e {
ExprTerm::Number(n) => {
self.next_from_current_with_num(to_f64(&n));
2019-06-02 22:03:35 +09:00
}
2019-06-25 18:58:25 +09:00
ExprTerm::String(key) => {
self.next_from_current_with_str(&[key]);
2019-06-23 17:22:58 +09:00
}
ExprTerm::Json(rel, _, v) => {
2019-06-25 18:58:25 +09:00
if v.is_empty() {
self.current = Some(vec![&Value::Null]);
} else if let Some(vec) = rel {
self.current = Some(vec);
2019-06-25 18:58:25 +09:00
} else {
self.current = Some(v);
2019-06-02 22:03:35 +09:00
}
}
2019-06-25 18:58:25 +09:00
ExprTerm::Bool(false) => {
self.current = Some(vec![&Value::Null]);
}
_ => {}
}
}
self.tokens.pop();
}
fn is_last_before_token_match(&mut self, token: ParseToken) -> bool {
if self.tokens.len() > 1 {
return token == self.tokens[self.tokens.len() - 2];
}
false
}
2019-06-25 18:58:25 +09:00
fn visit_all(&mut self) {
if let Some(ParseToken::Array) = self.tokens.last() {
self.tokens.pop();
}
2019-06-02 22:03:35 +09:00
2019-06-25 18:58:25 +09:00
match self.tokens.last() {
Some(ParseToken::Leaves) => {
2019-06-02 22:03:35 +09:00
self.tokens.pop();
2019-06-25 18:58:25 +09:00
self.all_from_current();
2019-06-02 22:03:35 +09:00
}
2019-06-25 18:58:25 +09:00
Some(ParseToken::In) => {
self.tokens.pop();
self.next_all_from_current();
2019-06-23 18:21:59 +09:00
}
2019-06-25 18:58:25 +09:00
_ => {
self.next_all_from_current();
2019-06-02 22:03:35 +09:00
}
2019-06-25 18:58:25 +09:00
}
}
2019-06-02 22:03:35 +09:00
2019-06-25 18:58:25 +09:00
fn visit_key(&mut self, key: &str) {
if let Some(ParseToken::Array) = self.tokens.last() {
self.terms.push(Some(ExprTerm::String(key.to_string())));
return;
}
if let Some(t) = self.tokens.pop() {
if self.terms.is_empty() {
match t {
ParseToken::Leaves => self.all_from_current_with_str(key),
ParseToken::In => self.next_from_current_with_str(&[key.to_string()]),
_ => {}
}
} else {
match t {
ParseToken::Leaves => {
self.all_in_filter_with_str(key);
}
ParseToken::In => {
self.next_in_filter_with_str(key);
2019-06-02 22:03:35 +09:00
}
_ => {}
}
}
2019-06-25 18:58:25 +09:00
}
}
2019-06-18 11:46:24 +09:00
2019-06-25 18:58:25 +09:00
fn visit_keys(&mut self, keys: &[String]) {
if !self.terms.is_empty() {
unimplemented!("keys in filter");
}
2019-06-20 18:55:41 +09:00
2019-06-25 18:58:25 +09:00
if let Some(ParseToken::Array) = self.tokens.pop() {
self.next_from_current_with_str(keys);
} else {
unreachable!();
}
}
2019-06-20 18:55:41 +09:00
2019-06-25 18:58:25 +09:00
fn visit_filter(&mut self, ft: &FilterToken) {
let right = match self.terms.pop() {
Some(Some(right)) => right,
Some(None) => ExprTerm::Json(
None,
2019-06-25 18:58:25 +09:00
None,
match &self.current {
Some(current) => current.to_vec(),
_ => unreachable!(),
},
),
_ => panic!("empty term right"),
};
2019-06-20 18:55:41 +09:00
2019-06-25 18:58:25 +09:00
let left = match self.terms.pop() {
Some(Some(left)) => left,
Some(None) => ExprTerm::Json(
None,
2019-06-25 18:58:25 +09:00
None,
match &self.current {
Some(current) => current.to_vec(),
_ => unreachable!(),
},
),
_ => panic!("empty term left"),
};
2019-06-02 22:03:35 +09:00
2019-06-25 18:58:25 +09:00
let mut ret = None;
match ft {
FilterToken::Equal => left.eq(&right, &mut ret),
FilterToken::NotEqual => left.ne(&right, &mut ret),
FilterToken::Greater => left.gt(&right, &mut ret),
FilterToken::GreaterOrEqual => left.ge(&right, &mut ret),
FilterToken::Little => left.lt(&right, &mut ret),
FilterToken::LittleOrEqual => left.le(&right, &mut ret),
FilterToken::And => left.and(&right, &mut ret),
FilterToken::Or => left.or(&right, &mut ret),
};
if let Some(e) = ret {
self.terms.push(Some(e));
}
}
fn visit_range(&mut self, from: &Option<isize>, to: &Option<isize>, step: &Option<usize>) {
if !self.terms.is_empty() {
unimplemented!("range syntax in filter");
}
if let Some(ParseToken::Array) = self.tokens.pop() {
let mut tmp = Vec::new();
if let Some(current) = &self.current {
for v in current {
if let Value::Array(vec) = v {
let from = if let Some(from) = from {
abs_index(*from, vec.len())
} else {
0
};
let to = if let Some(to) = to {
abs_index(*to, vec.len())
} else {
vec.len()
};
for i in (from..to).step_by(match step {
Some(step) => *step,
_ => 1,
}) {
if let Some(v) = vec.get(i) {
tmp.push(v);
2019-06-02 22:03:35 +09:00
}
}
}
}
}
2019-06-25 18:58:25 +09:00
self.current = Some(tmp);
} else {
unreachable!();
}
}
2019-06-02 22:03:35 +09:00
2019-06-25 18:58:25 +09:00
fn visit_union(&mut self, indices: &[isize]) {
if !self.terms.is_empty() {
unimplemented!("union syntax in filter");
}
if let Some(ParseToken::Array) = self.tokens.pop() {
let mut tmp = Vec::new();
if let Some(current) = &self.current {
for v in current {
if let Value::Array(vec) = v {
for i in indices {
if let Some(v) = vec.get(abs_index(*i, vec.len())) {
tmp.push(v);
2019-06-02 22:03:35 +09:00
}
}
}
}
}
2019-06-25 18:58:25 +09:00
self.current = Some(tmp);
} else {
unreachable!();
}
}
}
impl<'a, 'b> NodeVisitor for Selector<'a, 'b> {
fn visit_token(&mut self, token: &ParseToken) {
debug!("token: {:?}, stack: {:?}", token, self.tokens);
if self.compute_absolute_path_filter(token) {
return;
}
match token {
ParseToken::Absolute => self.visit_absolute(),
ParseToken::Relative => self.visit_relative(),
ParseToken::In | ParseToken::Leaves | ParseToken::Array => {
self.tokens.push(token.clone());
}
ParseToken::ArrayEof => self.visit_array_eof(),
ParseToken::All => self.visit_all(),
ParseToken::Bool(b) => {
self.terms.push(Some(ExprTerm::Bool(*b)));
}
ParseToken::Key(key) => self.visit_key(key),
ParseToken::Keys(keys) => self.visit_keys(keys),
ParseToken::Number(v) => {
self.terms
.push(Some(ExprTerm::Number(Number::from_f64(*v).unwrap())));
}
ParseToken::Filter(ref ft) => self.visit_filter(ft),
ParseToken::Range(from, to, step) => self.visit_range(from, to, step),
ParseToken::Union(indices) => self.visit_union(indices),
2019-06-02 22:03:35 +09:00
ParseToken::Eof => {
debug!("visit_token eof");
}
}
}
}
2019-06-25 18:58:25 +09:00
#[derive(Default)]
2019-06-09 22:55:35 +09:00
pub struct SelectorMut {
path: Option<Node>,
value: Option<Value>,
2019-06-03 21:41:08 +09:00
}
2019-06-02 22:03:35 +09:00
fn replace_value<F: FnMut(Value) -> Option<Value>>(
mut tokens: Vec<String>,
value: &mut Value,
fun: &mut F,
) {
let mut target = value;
2020-02-10 23:53:56 +09:00
let last_index = tokens.len().saturating_sub(1);
for (i, token) in tokens.drain(..).enumerate() {
let target_once = target;
let is_last = i == last_index;
let target_opt = match *target_once {
Value::Object(ref mut map) => {
if is_last {
if let Entry::Occupied(mut e) = map.entry(token) {
let v = e.insert(Value::Null);
2019-08-11 18:07:15 +03:00
if let Some(res) = fun(v) {
e.insert(res);
} else {
e.remove();
}
2019-08-11 18:07:15 +03:00
}
return;
}
map.get_mut(&token)
}
Value::Array(ref mut vec) => {
if let Ok(x) = token.parse::<usize>() {
if is_last {
2019-08-11 18:07:15 +03:00
let v = std::mem::replace(&mut vec[x], Value::Null);
if let Some(res) = fun(v) {
vec[x] = res;
} else {
vec.remove(x);
}
return;
}
vec.get_mut(x)
} else {
None
}
}
_ => None,
};
if let Some(t) = target_opt {
target = t;
} else {
break;
}
}
}
2019-06-09 22:55:35 +09:00
impl SelectorMut {
pub fn new() -> Self {
2019-06-25 18:58:25 +09:00
Self::default()
2019-06-09 22:55:35 +09:00
}
pub fn str_path(&mut self, path: &str) -> Result<&mut Self, JsonPathError> {
2019-06-25 18:58:25 +09:00
self.path = Some(Parser::compile(path).map_err(JsonPathError::Path)?);
2019-06-09 22:55:35 +09:00
Ok(self)
}
pub fn value(&mut self, value: Value) -> &mut Self {
self.value = Some(value);
self
}
pub fn take(&mut self) -> Option<Value> {
self.value.take()
}
fn compute_paths(&self, mut result: Vec<&Value>) -> Vec<Vec<String>> {
2019-06-19 11:34:47 +09:00
fn _walk(
origin: &Value,
target: &mut Vec<&Value>,
tokens: &mut Vec<String>,
2019-06-24 14:12:00 +09:00
visited: &mut HashSet<*const Value>,
visited_order: &mut Vec<Vec<String>>,
2019-06-19 11:34:47 +09:00
) -> bool {
trace!("{:?}, {:?}", target, tokens);
2019-06-09 22:55:35 +09:00
if target.is_empty() {
2019-06-09 22:55:35 +09:00
return true;
}
target.retain(|t| {
if std::ptr::eq(origin, *t) {
2019-06-24 14:12:00 +09:00
if visited.insert(*t) {
visited_order.push(tokens.to_vec());
}
false
} else {
true
}
});
2019-06-09 22:55:35 +09:00
match origin {
2019-06-19 11:34:47 +09:00
Value::Array(vec) => {
for (i, v) in vec.iter().enumerate() {
tokens.push(i.to_string());
2019-06-24 14:12:00 +09:00
if _walk(v, target, tokens, visited, visited_order) {
2019-06-19 11:34:47 +09:00
return true;
}
tokens.pop();
2019-06-09 22:55:35 +09:00
}
2019-06-19 11:34:47 +09:00
}
Value::Object(map) => {
for (k, v) in map {
tokens.push(k.clone());
2019-06-24 14:12:00 +09:00
if _walk(v, target, tokens, visited, visited_order) {
2019-06-19 11:34:47 +09:00
return true;
}
tokens.pop();
2019-06-09 22:55:35 +09:00
}
}
_ => {}
}
2019-06-25 18:58:25 +09:00
false
2019-06-09 22:55:35 +09:00
}
2019-06-24 14:12:00 +09:00
let mut visited = HashSet::new();
let mut visited_order = Vec::new();
2019-06-09 22:55:35 +09:00
if let Some(origin) = &self.value {
let mut tokens = Vec::new();
2019-06-24 14:12:00 +09:00
_walk(
origin,
&mut result,
&mut tokens,
&mut visited,
&mut visited_order,
);
2019-06-09 22:55:35 +09:00
}
2019-06-24 14:12:00 +09:00
visited_order
2019-06-09 22:55:35 +09:00
}
pub fn delete(&mut self) -> Result<&mut Self, JsonPathError> {
self.replace_with(&mut |_| Some(Value::Null))
}
pub fn remove(&mut self) -> Result<&mut Self, JsonPathError> {
self.replace_with(&mut |_| None)
2019-06-09 22:55:35 +09:00
}
fn select(&self) -> Result<Vec<&Value>, JsonPathError> {
if let Some(node) = &self.path {
2019-06-25 18:58:25 +09:00
let mut selector = Selector::default();
selector.compiled_path(&node);
2019-06-09 22:55:35 +09:00
if let Some(value) = &self.value {
selector.value(value);
}
Ok(selector.select()?)
} else {
Err(JsonPathError::EmptyPath)
2019-06-09 22:55:35 +09:00
}
}
2019-06-09 22:55:35 +09:00
pub fn replace_with<F: FnMut(Value) -> Option<Value>>(
2019-06-19 11:34:47 +09:00
&mut self,
fun: &mut F,
) -> Result<&mut Self, JsonPathError> {
let paths = {
let result = self.select()?;
self.compute_paths(result)
};
2019-06-09 22:55:35 +09:00
if let Some(ref mut value) = &mut self.value {
2019-06-09 22:55:35 +09:00
for tokens in paths {
replace_value(tokens, value, fun);
2019-06-09 22:55:35 +09:00
}
}
Ok(self)
}
2019-06-19 11:34:47 +09:00
}
2019-06-26 15:07:49 +09:00
2020-03-22 23:48:24 +09:00
2019-06-26 15:07:49 +09:00
#[cfg(test)]
mod select_inner_tests {
2020-03-22 23:48:24 +09:00
use serde_json::Value;
2019-06-26 15:07:49 +09:00
#[test]
fn to_f64_i64() {
let number = 0_i64;
let v: Value = serde_json::from_str(&format!("{}", number)).unwrap();
if let Value::Number(n) = v {
assert_eq!((super::to_f64(&n) - number as f64).abs() == 0_f64, true);
} else {
panic!();
}
}
#[test]
fn to_f64_f64() {
let number = 0.1_f64;
let v: Value = serde_json::from_str(&format!("{}", number)).unwrap();
if let Value::Number(n) = v {
assert_eq!((super::to_f64(&n) - number).abs() == 0_f64, true);
} else {
panic!();
}
}
#[test]
fn to_f64_u64() {
let number = u64::max_value();
let v: Value = serde_json::from_str(&format!("{}", number)).unwrap();
if let Value::Number(n) = v {
assert_eq!((super::to_f64(&n) - number as f64).abs() == 0_f64, true);
} else {
panic!();
}
}
2020-03-22 23:48:24 +09:00
}