dorian-yaml-deep-pluck 0.0.2 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e5e1f5a5b642fb4eeaad36b8e22f545ef1de056ff09eac30c09d101ae0b915c5
-  data.tar.gz: d68874f74257300ac8ea8b24a41a78880a4c3d96b2761420b09d4fd4040d7ba8
+  metadata.gz: 7cd6edffcf6ac8ed7610dd3a746e5232798a0aeb9c19ea140dc426dc86b6b25f
+  data.tar.gz: 40668d00b9b6791e16823f7d38a192a726e369f5ac79b382abc3240cfc9989cd
 SHA512:
-  metadata.gz: 94a206a7e10cf21899ea2993c7373a3fd4f4d7c7dada4b06f43f6b98c8e0141a94b2eedaaa7f302f560e19e3bbb1c8cea5e408cec89bfe16f173a73c3bde5250
-  data.tar.gz: 37da93ab44eb2eed269e1b9a9eaef64824de1147ab73f260e944245635c285bf2ec0a9807fe7d3058d80ea83480cf4b7c2c22b6eb0482a8ff6cae36d4460af3b
+  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.2
+  version: 1.0.0
 platform: ruby
 authors:
 - Dorian Marié
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-08-18 00:00:00.000000000 Z
+date: 2024-08-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: yaml
@@ -24,6 +24,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: dorian-arguments
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: plucks yaml deeply from yaml path
 email: dorian@dorianmarie.com
 executables: