dorian-csv-uniq 0.1.2 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b0f43afd50cb8ba7012f5ca77ce42426533ac1118e2eb4f0b102e97a73e1cc80
-  data.tar.gz: 5893e31eed4f2eccfc77302d6f286b190e11c8df2588a84dfb17be23d0abb081
+  metadata.gz: 8890562977177594a6271149092412127fbf019dbd43bcd0b011f6333f910541
+  data.tar.gz: 564679dfac50c4a80084ef502964d01dadafd36e709c73c9422b734e249f7061
 SHA512:
-  metadata.gz: 9bec710af2bf383307649cee46a0433cdf8bc74341b602c7c78199d0c63e8e5be9d917b4947d7d3c66b30e5a10dc697ecf238c5d89f3458f8852cbd2d2865d60
-  data.tar.gz: a4196cd5709e61c9341db83ce5f181c99505a83edaed194a0e21f489e39aea2676ac2972d6b24bbf1e637b5525684dd96231ff7cee85200c67b27b746cf18786
+  metadata.gz: 1018b2ffbf3518c1862de7c47b289f44bf23dfa869343e0c361b15e7a6756ddbfb8a0e4c7ac82274c4f19aaf88af76e767069142d90594f2b45c1d16e5205da9
+  data.tar.gz: 7b66353370765cba05dc18ab12d436d71b541ec008f6dfb67c1986a78dc69b806735937d71be08510d0446d0e802971a545e6adf4d8dccab7134833f8d1be792

data/bin/csv-uniq

@@ -2,29 +2,60 @@
 # frozen_string_literal: true
 
 require "csv"
+require "dorian/arguments"
 
-if ARGV.size < 1 || ARGV[0] == "--help" || ARGV[0] == "-h"
-  puts "USAGE: ... | csv-uniq [KEY...] [-h|--headers]"
-  exit
+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
 
-HEADERS = ARGV.detect { |argv| argv == "-h" || argv == "--headers" }
-KEYS = (ARGV - [HEADERS]).map { |key| HEADERS ? key : key.to_i }
+keys = parsed.arguments
+keys = nil if keys.empty?
+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|
-      input =
-        CSV.parse(
-          $stdin.each_line.to_a.map(&:strip).join("\n"),
-          headers: !!HEADERS
-        )
-
-      output << input.headers if HEADERS
-
-      input
-        .uniq { |row| KEYS.map { |key| row[key] } }
-        .each { |row| output << row }
-    end
-    .strip
+  CSV.generate do |output|
+    rows =
+      inputs.map.with_index do |input, index|
+        input = CSV.parse(input, headers:)
+
+        if parsed.options.print_headers && index.zero?
+          output << (keys || input.headers).map do |key|
+            if input.headers.include?(key)
+              key
+            elsif key.to_i.to_s == key
+              input.headers[key.to_i]
+            else
+              key
+            end
+          end
+        end
+
+        input.map do |row|
+          (
+            keys || (headers && input.headers) || (0...input.first.size)
+          ).map { |key| row[key.to_i.to_s == key ? key.to_i : key] }
+        end
+      end
+
+    rows.flatten(1).uniq.each { |row| output << row }
+  end
 )

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: dorian-csv-uniq
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  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: csv