teebo 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4d25e0acff23d8791b96d50d72b36d326143a2c1
-  data.tar.gz: deb705f45ee7dfc04e34b6b6021a32640ff08c5a
+  metadata.gz: 0116fd7256c8ebc11aaf32761e425cfe26793979
+  data.tar.gz: 4ad1b311bdd1ca5420f50df2448407245618df95
 SHA512:
-  metadata.gz: bed5a9899872dc9f634824f8614838af9b70c74e4c232041484bd40889376470e8aabadb992854d87343e632839a6c1db4b7773b21db628bfb50cab3a60fda66
-  data.tar.gz: c941cb7f2fc2f4cf7fc6c98b8a2d5a85a8c69aeeb26adf1d74787c914cf6ade5f6ac1e6e585798a394646837c4465ff8ac5f5776192635926517e6cf4cd4a442
+  metadata.gz: 2cf98921da7875e0398d77725c62efb1d41d6d5ca2ef8c336fbd03516504fb607465b3e1d48d5b613b50ca74e4793a150fb4f6e670d8ab24566a2d1c0515a2ad
+  data.tar.gz: 9f02487e929191882ea3ac11c55431f03dc42b944917654b211c90cdeb6790b0a6515dbc8382f75dc4432203b482a8a3263458f7a2bb3606f238ddd56b0bd99a

data/lib/teebo/credit_card.rb

@@ -14,7 +14,7 @@ module Teebo
     #
     # Returns a credit card issuer according to the likelihood that it would be seen in the wild.
     #
-    def get_issuer
+    def generate_issuer
       random_choice = Random.rand
       full_weight = 0
       @cc_issuers.each do |issuer|
@@ -30,12 +30,53 @@ module Teebo
     #
     def generate_number(issuer)
       # TODO: Sample according to realistic distribution - numbers w/long prefixes are prioritized too highly right now.
-      prefix = issuer['iin-prefixes'].sample
+      prefix = issuer['iin-prefixes'].sample.to_s
       length = issuer['lengths'].sample
-      puts prefix, length
+      generated = number_to_digits(length - prefix.length)
+      number = prefix + generated
+      if issuer['validation']
+        last_digit_validation(number)
+      end
+
     end
 
+    #
+    # Gets the sum of the digits of the specified number. Necessary to calculate a credit card's
+    # check digit.
+    #
+    def luhn_sum(number)
+      digits = number.to_s.chars.map(&:to_i)
+      digits.reverse.each_slice(2).map do |x, y|
+        [(x * 2).divmod(10), y || 0]
+      end.flatten.inject(:+)
+    end
 
+    #
+    # A simple implementation of the [Luhn algorithm](http://en.wikipedia.org/wiki/Luhn_algorithm),
+    # which all U.S.-based Credit Card issuers use for the validation check on credit card numbers.
+    #
+    def luhn_algorithm(number)
+      digit_sum = luhn_sum(number)
+      ((10 - digit_sum % 10) % 10).to_s
+    end
 
+    #
+    # Uses the Luhn algorithm to replace the last digit in the generated number with the correct
+    # check digit.
+    #
+    def last_digit_validation(number)
+      trimmed_number = number[0..-2]
+      check_digit = luhn_algorithm(trimmed_number)
+      trimmed_number + check_digit
+    end
+
+    #
+    # Generates a random number with the specified number of digits, padding the beginning with
+    # '0' characters, if necessary.
+    #
+    def number_to_digits(digits)
+      digits = digits.to_i
+      rand(10 ** digits).to_s.rjust(digits,'0')
+    end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: teebo
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Russ Taylor