dorian-csv-pluck 0.0.1 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7c32747d725ade98f10c9b5826d01375930493d04588348a16f8206cc3b7b6da
-  data.tar.gz: 0140e9cf48ea81dcf779c7de3899b5712b28ace0bcbcb65f8667c5ebbb48be0d
+  metadata.gz: 477b6e2ff01bdab85d414d4bafd17c7bfb81e407fa60d547faa5f5cb7a96dcd4
+  data.tar.gz: eae7ee7a5791b18720f29b979a9f7c7a5600e8310ee58a196c3d0af88f778674
 SHA512:
-  metadata.gz: 1c9775aaeb861b80a72db7381477ff6012a505ecef8a527a544febba642f3f99596dfa4b3b02bfd06ee0953deed50d27f3ddc3dfb04a87d376b0caa32a22fed8
-  data.tar.gz: e5726ae2463ed6070a2131372b16f8bb07633a8943259a363844876f2f3cfc3b4e892c9649b3603116c52311b93642d3b7bb2ca6b15d074a0ad2f57068f34f7a
+  metadata.gz: 9242c5f47be5886c2cdaa93a4b87d78f3473ce738ceeb3bf692840860d4d2cc82dc92e666be13d97871a2ea18c24277646bb8da51673c4e90aa5226940b8d368
+  data.tar.gz: d86a84b08ea909b575d416617cd1821f0294227ca23bcc1bb50e664b531e2f3f6f4e6fa9466dc19943b8630e4b39de263dffc1f606726e03ddc61393439ec84d

data/bin/csv-pluck

@@ -2,14 +2,54 @@
 # frozen_string_literal: true
 
 require "csv"
+require "dorian/arguments"
 
-if ARGV.size != 1 || ARGV.first == "--help" || ARGV.first == "-h"
-  abort "UASGE: ... | csv-pluck KEY"
+parsed =
+  Dorian::Arguments.parse(
+    headers: :boolean,
+    print_headers: { alias: :p },
+    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
 
-KEY = ARGV.first
-HEADERS = KEY !~ /^[0-9]+$/
+keys = parsed.arguments
+inputs = parsed.files.map { |file| File.read(file) }
+inputs = [$stdin.each_line.to_a.map(&:strip).join("\n")] if inputs.empty?
+headers = parsed.options.headers || parsed.options.print_headers
+headers ||= keys.any? { |key| key.to_i.to_s != key }
+
+puts(
+  CSV.generate do |output|
+    inputs.each.with_index do |input, index|
+      input = CSV.parse(input, headers:)
+
+      if parsed.options.print_headers && index.zero?
+        output << keys.map do |key|
+          if input.headers.include?(key)
+            key
+          elsif key.to_s =~ /^[0-9]+$/
+            input.headers[key.to_i]
+          else
+            key
+          end
+        end
+      end
 
-CSV
-  .parse($stdin.each_line.to_a.join, headers: HEADERS)
-  .map { |row| puts row[HEADERS ? KEY : KEY.to_i] }
+      input.each do |row|
+        output << keys.map do |key|
+          row[key.to_s =~ /^[0-9]+$/ ? key.to_i : key]
+        end
+      end
+    end
+  end
+)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: dorian-csv-pluck
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.3
 platform: ruby
 authors:
 - Dorian Marié
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-08-16 00:00:00.000000000 Z
+date: 2024-08-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: csv
@@ -25,7 +25,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: activesupport
+  name: dorian-to_struct
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -38,10 +38,21 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description: |-
-  Evaluates some code pluck lines of the csv input
-
-  e.g. `cat users.csv | csv-pluck --headers "puts it.first[:name]"`
+- !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 each line of the csv input
 email: dorian@dorianmarie.com
 executables:
 - csv-pluck
@@ -72,5 +83,5 @@ requirements: []
 rubygems_version: 3.5.11
 signing_key:
 specification_version: 4
-summary: Evaluates some code pluck lines of the csv input
+summary: plucks each line of the csv input
 test_files: []