dorian-yaml-deep-pluck 0.0.3 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: df3100bed77c5dd2ea8e3f8d47ab84db36858897a940c89176634b6d5d8faedd
-  data.tar.gz: e90188ff2c32848e7109840de3b1e770258095bbd5ee107d5f4651f4c5106f91
+  metadata.gz: 7cd6edffcf6ac8ed7610dd3a746e5232798a0aeb9c19ea140dc426dc86b6b25f
+  data.tar.gz: 40668d00b9b6791e16823f7d38a192a726e369f5ac79b382abc3240cfc9989cd
 SHA512:
-  metadata.gz: 5479e85db0140fa7d8c9b30468a7f81b78db13bdc6832f2e25adafd356773ea2cfd673f0005f67aa5e287617a7f5881392e4c9f7b0b5851c460cd97742cf4277
-  data.tar.gz: 51b71dc52088d12948dca04066de567b3d69f86a5d5350caed8b9aa318c0086a9bdefb01810000b43ddae51d500674fc639daf4bd011fce009c6f6b4601ce9f4
+  metadata.gz: b9e8fa6892192424a4c18438d03080ed2c9e191479e4e01d5bf89b774aeb8246158ea625f3ac6a3822579da6eb892e9de2e218f9df1ca19751518f8945a00fb8
+  data.tar.gz: 40c180418a7b8c5e3052c4fc2c6129c7884982ada5e028de604bf81212a5de8f67826813f01c1b4d98daa534595925ebd422db75523d988a28c4e9732dd60cca

data/bin/yaml-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 "yaml"
+
+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(
+    yaml: {
+      alias: :y
+    },
+    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.join] 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(YAML.safe_load(input)) do |json|
+      json.is_a?(Hash) || json.is_a?(Array) ? json.dig(*keys) : nil
+    rescue TypeError
+      nil
+    end
+
+  if parsed.options.yaml
+    puts result.to_yaml
+  elsif parsed.options.ruby
+    p result
+  else
+    puts result
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: dorian-yaml-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: yaml