consider whether first char or not at Array

ynqa · Mar 27, 2024 · 11d11a8 · 11d11a8
1 parent c3f53a3
commit 11d11a8
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>()
                 }