dorian-json-deep-pluck 0.0.3 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: eea5e13e775270018c05e67ea7439122f46a9a4aa478a742bfdf3e20c79ce5bb
-  data.tar.gz: 939e8b8909546f0bf159f517aad90c1cfa42d37e7914e4274fbd60ad810ebd1a
+  metadata.gz: eee71f7f0e4215b908a761376f1aa294d1344bf77cbbb011fe392979b7a201ee
+  data.tar.gz: 31924103dcd5d8b10735d97aad93ab3ccac9ee82dea33fb7cc585513ffbf5de4
 SHA512:
-  metadata.gz: 109d6b6942a2629dfe0f8b4dce78ef006b528e928d857506cafdb7037c13ca51f16067cc52906ca60ceeafd5e57f2830433ec0b7b96105f0110c8ac3f643af2d
-  data.tar.gz: efa5ae9f2d2c286f7130e0c56f1467a3d663a5f5a5b7de7ba72f56bf55620910ba1107aa19db5c78383191ea4562975978da07f2326289723320836e05d4ea7a
+  metadata.gz: 1df0e66e4bc07a6af4d7a707ace5dcdefb82d0b04662948682fe4ab14d077d35871259b56c0c49666aa7eae876d37fe43980a81725606824ce132a1126d30627
+  data.tar.gz: fcb8f49d81c54de227e70aed5638a1beb3bc9271ba28df078f499cdbf32f6137ef4f474cfc2cb6b1c505710284794558a3420e61a63b23806ec0459627bbd721

data/bin/json-deep-pluck

@@ -1,12 +1,71 @@
 #!/usr/bin/env ruby
 # frozen_string_literal: true
 
-if ARGV.size != 1 || ARGV[0] == "--help" || ARGV[0] == "-h"
-  puts "USAGE: ... | each CODE"
-  exit
+require "dorian/arguments"
+require "json"
+
+def deep_pluck(json, &block)
+  if json.is_a?(Hash)
+    results =
+      json
+        .map do |key, value|
+          [deep_pluck(key, &block), deep_pluck(value, &block)]
+        end
+        .flatten(1)
+
+    results << yield(json)
+
+    results.compact.empty? ? nil : results.compact
+  elsif json.is_a?(Array)
+    results = json.map { |element| deep_pluck(element, &block) }
+
+    results << yield(json)
+
+    results.compact.empty? ? nil : results.compact
+  else
+    nil
+  end
 end
 
-$stdin.each_line do |it|
-  it.strip!
-  eval(ARGV.first)
+parsed =
+  Dorian::Arguments.parse(
+    json: {
+      alias: :j
+    },
+    ruby: {
+      alias: :r
+    },
+    version: {
+      alias: :v
+    },
+    help: {
+      alias: :h
+    }
+  )
+
+abort parsed.help if parsed.options.help
+
+if parsed.options.version
+  abort File.read(File.expand_path("../../VERSION", __FILE__))
+end
+
+inputs = parsed.files.map { |file| File.read(file) }
+inputs = [$stdin.each_line.to_a.map(&:strip).join("\n")] if inputs.empty?
+keys = parsed.arguments.map { |key| key.to_i.to_s == key ? key.to_i : key }
+
+inputs.each do |input|
+  result =
+    deep_pluck(JSON.parse(input)) do |json|
+      json.is_a?(Hash) || json.is_a?(Array) ? json.dig(*keys) : nil
+    rescue TypeError
+      nil
+    end
+
+  if parsed.options.json
+    puts JSON.pretty_generate(result)
+  elsif parsed.options.ruby
+    p result
+  else
+    puts result
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: dorian-json-deep-pluck
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 1.0.0
 platform: ruby
 authors:
 - Dorian Marié
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-08-22 00:00:00.000000000 Z
+date: 2024-08-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: json