group_by_match_type 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5202086856b91a26669f161ea8edc6982fd6dd8fc6e62e375e4bd334727de16c
-  data.tar.gz: 4b3c5911103ed26e80ffb6ddff7bc77dd3c9b83d7ba19ab0a787c652def1f407
+  metadata.gz: 1e02ae439383bee301cbfe85d02b557dc55883b5259bd11b701030f2d5b7a36d
+  data.tar.gz: b2cc916743083fb97352d194fe64713c4738bf4603be7f4ca5ff2ef786998ce0
 SHA512:
-  metadata.gz: f5167852ae2269c04129ad5a859af81a2e2addaf90db7db1ca72cbf417df91dac9327f263de05e9a8151e9be8dbd587551693b69875558caf8dec4cdb16ffa30
-  data.tar.gz: 27809eb8566ecc18b886fa35fa300725323db17bb51bda0153c72c99df36ba4acb675a644213a1f756421d95da98700774d67f2855501b1ecf173a00871b1fa5
+  metadata.gz: 59ab1c266cffceb1598be59aed1dcdc532a27a252f544d80028791947fb4f94ca24cd0c77b1b56b582ebe62b014205736f7f6afd169974d1cbb11ae066747da5
+  data.tar.gz: 5a8676a87c866a8c740e71253833024f3d048306601385e48f982e7a193fdb1ff0542d2377695dbd0f9afd49c25c225f5250d6cc715ba56db8221c450ee88dac

data/exe/group_by_match_type

@@ -1,4 +1,5 @@
 #!/usr/bin/env ruby
+# frozen_string_literal: true
 
 require 'group_by_match_type'
 require 'group_by_match_type/cli'

data/lib/group_by_match_type/cli.rb

@@ -1,9 +1,11 @@
+# frozen_string_literal: true
+
 require 'optparse'
 require_relative 'matcher'
 
 module GroupByMatchType
   class CLI
-    VALID_TYPES = %w[same_email same_phone same_email_or_phone]
+    VALID_TYPES = %w[same_email same_phone same_email_or_phone].freeze
 
     def self.start(args)
       if args.empty? || args.include?('-h') || args.include?('--help')

data/lib/group_by_match_type/matcher.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'csv'
 require_relative 'union_find'
 
@@ -28,11 +30,11 @@ module GroupByMatchType
       # Determine output file path
       output_file ||= @input_file.sub(/\.csv$/, '_grouped.csv')
       CSV.open(output_file, 'w') do |csv|
-        csv << rows.headers + ['group_id']
+        csv << ['group_id'] + rows.headers
         rows.each_with_index do |row, index|
           root = @union_find.find(index.to_s)
           group_id = group_id_map[root]
-          csv << row.fields + [group_id]
+          csv << [group_id] + row.fields
         end
       end
 

data/lib/group_by_match_type/union_find.rb

@@ -1,26 +1,29 @@
+# frozen_string_literal: true
+
 module GroupByMatchType
   class UnionFind
     def initialize
       @parent = {}
     end
 
-    def find(x)
-      @parent[x] = find(@parent[x]) if @parent[x] && @parent[x] != x
-      @parent[x] ||= x
+    def find(item)
+      @parent[item] = find(@parent[item]) if @parent[item] && @parent[item] != item
+      @parent[item] ||= item
     end
 
-    def union(x, y)
-      root_x = find(x)
-      root_y = find(y)
-      @parent[root_x] = root_y unless root_x == root_y
+    def union(source_item, target_item)
+      source_root = find(source_item)
+      target_root = find(target_item)
+      @parent[source_root] = target_root unless source_root == target_root
     end
 
     def find_or_create(items)
       return nil if items.empty?
+
       items.map!(&:to_s)
       roots = items.map { |item| find(item) }
       main_root = roots.first
-      roots[1..].each { |r| union(main_root, r) }
+      roots[1..].each { |root| union(main_root, root) }
       main_root
     end
   end

data/lib/group_by_match_type/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module GroupByMatchType
-  VERSION = '0.1.0'
+  VERSION = '0.2.0'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: group_by_match_type
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Justin Boltz