dorian-csv-uniq 0.1.1 → 1.0.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 840b7e2f1b21b6216b7b4ffbd81fc734e89875451b50a4a413b1c7b648d156a1
-  data.tar.gz: b05e8907b44c6851ed2a99a0cf4c72e1f85d16646ca4596d33a801223138805e
+  metadata.gz: 8890562977177594a6271149092412127fbf019dbd43bcd0b011f6333f910541
+  data.tar.gz: 564679dfac50c4a80084ef502964d01dadafd36e709c73c9422b734e249f7061
 SHA512:
-  metadata.gz: fdf422f2c703a41bb5929c7ffa93c9e23a6820a716064be569430c734e0cb75e0275142d6819d5a757e958f9416ddc4a2b9a8b96da133fa320944985f8e49830
-  data.tar.gz: 51e1d4fc419c4742caf117e699a5fab950f2381153f84d5126f9ee666072ddadd34d347744786cc8273a0174bd7d29039af71f9bb326ae17d81ccc7617b83341
+  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.1
+  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: csv
@@ -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: returns a csv by unique keys, can be index or name
 email: dorian@dorianmarie.com
 executables: