Merge pull request #34 from ynqa/v0.2.x/fix-suggestion-at-array

fix: consider whether first char or not at Array
ynqa · Mar 28, 2024 · 489421d · 489421d
2 parents c3f53a3 + 11d11a8
commit 489421d
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/jnv.rs b/src/jnv.rs
@@ -56,15 +56,22 @@ impl Jnv {
                 } else {
                     segments
                         .iter()
-                        .map(|segment| match segment {
+                        .enumerate()
+                        .map(|(i, segment)| match segment {
                             JsonPathSegment::Key(key) => {
                                 if key.contains('.') || key.contains('-') || key.contains('@') {
                                     format!(".\"{}\"", key)
                                 } else {
                                     format!(".{}", key)
                                 }
                             }
-                            JsonPathSegment::Index(index) => format!("[{}]", index),
+                            JsonPathSegment::Index(index) => {
+                                if i == 0 {
+                                    format!(".[{}]", index)
+                                } else {
+                                    format!("[{}]", index)
+                                }
+                            }
                         })
                         .collect::<String>()
                 }