Merge pull request #177 from itchyny/implement-pick

Implement pick/1 filter
MiSawa · Feb 25, 2024 · 1a37b3d · 1a37b3d
2 parents d8f28b4 + 656918f
commit 1a37b3d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/prelude.jq b/prelude.jq
@@ -41,6 +41,7 @@ def nth($n): .[$n];
 
 def del(f): delpaths([path(f)]);
 def setpath($paths; $v): getpath($paths) |= $v;
+def pick(f): . as $v | reduce path(f) as $p (null; setpath($p; $v | getpath($p)));
 
 def to_entries: [keys[] as $key | {$key, value: .[$key]}];
 def from_entries: reduce .[] as $entry ({}; .[$entry.key]=$entry.value);

diff --git a/tests/from_manual/functions.rs b/tests/from_manual/functions.rs
@@ -264,6 +264,19 @@ test!(
     "#
 );
 
+test!(
+    pick1,
+    r#"
+    pick(.x, .y[2], .y[0], .z)
+    "#,
+    r#"
+    {"x":1,"y":[1,2,3]}
+    "#,
+    r#"
+    {"x":1,"y":[1,null,3],"z":null}
+    "#
+);
+
 test!(
     entries1,
     r#"