teebo 0.0.5 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0116fd7256c8ebc11aaf32761e425cfe26793979
-  data.tar.gz: 4ad1b311bdd1ca5420f50df2448407245618df95
+  metadata.gz: f974479fae12561857d5bf06a7d8dca7ccdb9ec0
+  data.tar.gz: 625cc67d205696380258f0609b5bf9d629a0d815
 SHA512:
-  metadata.gz: 2cf98921da7875e0398d77725c62efb1d41d6d5ca2ef8c336fbd03516504fb607465b3e1d48d5b613b50ca74e4793a150fb4f6e670d8ab24566a2d1c0515a2ad
-  data.tar.gz: 9f02487e929191882ea3ac11c55431f03dc42b944917654b211c90cdeb6790b0a6515dbc8382f75dc4432203b482a8a3263458f7a2bb3606f238ddd56b0bd99a
+  metadata.gz: ba28733ba9e1d07ce99788a6256dba6e963a82f2502f3b097f3471c3a7b0aa628fe15c7f7670cdafcf2748730ed29e76ab554b83814714a6080ca7d79fb0af3f
+  data.tar.gz: 27330412ec7063ebf78042caa44236bf57860ea4a2bac435ee42700406876bd52fcc5c897773b23d6a116fc63cbb1aad0560c0e40086064d92de08248f9b5882

data/lib/data/en-us.yml

@@ -9,7 +9,7 @@ credit-card-issuers:
       - 15
     validation: yes
     probability: .092
-    cvc-length: 4
+    cvv-length: 4
   - 
     name: Discover Card
     iin-prefixes: [ 6011, 622126, 622127, 622128, 622129, 622130, 622131, 622132, 622133, 622134, 622135, 622136, 622137, 622138, 622139, 622140, 622141, 622142, 622143, 622144, 622145, 622146, 622147, 622148, 622149, 622150, 622151, 622152, 622153, 622154, 622155, 622156, 622157, 622158, 622159, 622160, 622161, 622162, 622163, 622164, 622165, 622166, 622167, 622168, 622169, 622170, 622171, 622172, 622173, 622174, 622175, 622176, 622177, 622178, 622179, 622180, 622181, 622182, 622183, 622184, 622185, 622186, 622187, 622188, 622189, 622190, 622191, 622192, 622193, 622194, 622195, 622196, 622197, 622198, 622199, 622200, 6222, 6223, 6224, 6225, 6226, 6227, 6228, 622901, 622902, 622903, 622904, 622905, 622906, 622907, 622908, 622909, 622910, 622911, 622912, 622913, 622914, 622915, 622916, 622917, 622918, 622919, 622920, 622921, 622922, 622923, 622924, 622925, 644, 645, 646, 647, 648, 649, 65 ]
@@ -17,7 +17,7 @@ credit-card-issuers:
       - 16
     validation: yes
     probability: .107
-    cvc-length: 3
+    cvv-length: 3
   - 
     name: MasterCard
     iin-prefixes: [ 51, 52, 53, 54, 55 ]
@@ -25,7 +25,7 @@ credit-card-issuers:
       - 16
     validation: yes
     probability: .313
-    cvc-length: 3
+    cvv-length: 3
   - 
     name: Visa
     iin-prefixes:
@@ -35,4 +35,4 @@ credit-card-issuers:
       - 16
     validation: yes
     probability: .488
-    cvc-length: 3
+    cvv-length: 3

data/lib/teebo.rb

@@ -5,9 +5,12 @@ require 'randexp'
 module Teebo
 
   class TeeboGenerator
-    def initialize
-      @db_connection = Teebo::DatabaseHandler.new
-      @yaml_mapping = YAML::load(File.open('lib/data/en-us.yml'))
+    def self.db_connection
+      @db_connection ||= Teebo::DatabaseHandler.new
+    end
+
+    def self.yaml_mapping
+      @yaml_mapping ||= YAML::load(File.open('lib/data/en-us.yml'))
     end
   end
 end

data/lib/teebo/credit_card.rb

@@ -6,18 +6,29 @@ module Teebo
   #
   class CreditCard < TeeboGenerator
 
-    def initialize
-      super
-      @cc_issuers = @yaml_mapping['credit-card-issuers']
+    def self.issuers
+      @cc_issuers ||= yaml_mapping['credit-card-issuers']
+    end
+
+    def self.generate_card(issuer=nil)
+      if issuer.nil?
+        issuer = generate_issuer
+      end
+      [
+          'issuer' => issuer['name'],
+          'number' => generate_number(issuer),
+          'cvv' => generate_cvv_code(issuer),
+          'expiration' => generate_expiration_date
+      ]
     end
 
     #
     # Returns a credit card issuer according to the likelihood that it would be seen in the wild.
     #
-    def generate_issuer
+    def self.generate_issuer
       random_choice = Random.rand
       full_weight = 0
-      @cc_issuers.each do |issuer|
+      issuers.each do |issuer|
         full_weight += issuer['probability']
         if random_choice < full_weight
           return issuer
@@ -28,23 +39,22 @@ module Teebo
     #
     # Generates a credit card number according to the pattern specified in the 'issuer' passed in.
     #
-    def generate_number(issuer)
+    def self.generate_number(issuer)
       # TODO: Sample according to realistic distribution - numbers w/long prefixes are prioritized too highly right now.
       prefix = issuer['iin-prefixes'].sample.to_s
       length = issuer['lengths'].sample
-      generated = number_to_digits(length - prefix.length)
+      generated = number_with_length(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.
+    # Gets the sum of the digits of the specified number, with every other digit doubled. Necessary
+    # to calculate a credit card's check digit.
     #
-    def luhn_sum(number)
+    def self.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]
@@ -55,7 +65,7 @@ module Teebo
     # 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)
+    def self.luhn_algorithm(number)
       digit_sum = luhn_sum(number)
       ((10 - digit_sum % 10) % 10).to_s
     end
@@ -64,7 +74,7 @@ module Teebo
     # Uses the Luhn algorithm to replace the last digit in the generated number with the correct
     # check digit.
     #
-    def last_digit_validation(number)
+    def self.last_digit_validation(number)
       trimmed_number = number[0..-2]
       check_digit = luhn_algorithm(trimmed_number)
       trimmed_number + check_digit
@@ -74,9 +84,32 @@ module Teebo
     # 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')
+    def self.number_with_length(length)
+      length = length.to_i
+      rand(10 ** length).to_s.rjust(length,'0')
+    end
+
+    #
+    # Generates a Credit Card CVV code. This simply returns a random number of the length used by
+    # the specified 'issuer' object.
+    #
+    def self.generate_cvv_code(issuer)
+      number_with_length(issuer['cvv-length'])
+    end
+
+    #
+    # Generates an expiration date for a credit card. These dates are, somewhat arbitrarily, simply
+    # some time in the next 5 years.
+    #
+    def self.generate_expiration_date
+      current_year = Time.now.year - 2000
+      month = rand(1...13)
+      if month <= Time.now.month
+        current_year += 1
+      end
+      year = rand(current_year...(current_year + 5))
+      month = month.to_s.rjust(2,'0')
+      "#{month}/#{year}"
     end
   end
 end

data/lib/teebo/name.rb

@@ -12,11 +12,11 @@ module Teebo
     # Picks a random first & last name, selecting a random gender if it's not
     # specified.
     #
-    def name (sex=nil)
+    def self.generate_name (sex=nil)
       if sex.nil?
         sex = %w(M F).sample
       end
-      given_name(sex) + ' ' + surname
+      generate_given_name(sex) + ' ' + generate_surname
     end
 
     #
@@ -26,38 +26,38 @@ module Teebo
     # simply be another given name of the same gender is almost certainly less
     # than 100%.
     #
-    def full_name(sex=nil)
+    def self.generate_full_name(sex=nil)
       # TODO: Take into account different probabilities of different types of middle names.
       if sex.nil?
         sex = %w(M F).sample
       end
-      given_name(sex) + ' ' + given_name(sex) + ' ' + surname
+      generate_given_name(sex) + ' ' + generate_given_name(sex) + ' ' + generate_surname
     end
 
     #
     # Finds the total count for the number of names in the database.
     #
-    def sum_count(sex)
-      @db_connection.get_sum(GIVEN_NAMES_TABLE, 'count', {column: 'sex', condition: sex})
+    def self.sum_count(sex)
+      db_connection.get_sum(GIVEN_NAMES_TABLE, 'count', {column: 'sex', condition: sex})
     end
 
     #
     # Selects a random (weighted) given name from the database.
     #
-    def given_name(sex)
+    def self.generate_given_name(sex)
       count = sum_count(sex)
       selection = rand(count)
-      @db_connection.get_row_for_count(GIVEN_NAMES_TABLE, 'count_to', selection,
+      db_connection.get_row_for_count(GIVEN_NAMES_TABLE, 'count_to', selection,
                                 {column: 'sex', condition: sex})['name']
     end
 
     #
     # Selects a random (weighted) surname from the database.
     #
-    def surname
-      count = @db_connection.get_sum(SURNAMES_TABLE, 'count')
+    def self.generate_surname
+      count = db_connection.get_sum(SURNAMES_TABLE, 'count')
       selection = rand(count)
-      @db_connection.get_row_for_count(SURNAMES_TABLE, 'count_to', selection)['name']
+      db_connection.get_row_for_count(SURNAMES_TABLE, 'count_to', selection)['name']
     end
   end
 end

data/lib/teebo/number.rb

@@ -9,7 +9,7 @@ module Teebo
     # A basic implementation of the Box-Muller transform. Adapted from an answer by antonakos on
     # Stack Overflow here: http://stackoverflow.com/questions/5825680
     #
-    def normal_dist(mean, std_deviation, rand = lambda { Kernel.rand })
+    def self.normal_dist(mean, std_deviation, rand = lambda { Kernel.rand })
       theta = 2 * Math::PI * rand.call
       rho = Math.sqrt(-2 * Math.log(1 - rand.call))
       scale = std_deviation * rho
@@ -20,7 +20,7 @@ module Teebo
     # Generates a number according to Benford's law, meaning that it is more indicative of numbers
     # encountered in real life.
     #
-    def benford_dist(upper_bound, decimals=0)
+    def self.benford_dist(upper_bound, decimals=0)
       (upper_bound**Random.rand).round(decimals)
     end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: teebo
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.0.6
 platform: ruby
 authors:
 - Russ Taylor
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-09-12 00:00:00.000000000 Z
+date: 2014-10-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sqlite3