diff --git a/Gemfile b/Gemfile index cf76ba4..421f62d 100644 --- a/Gemfile +++ b/Gemfile @@ -10,5 +10,7 @@ gem 'simplecov', '~> 0.7.1', group: :test, require: nil # A Ruby implementation of the Coveralls API. gem 'coveralls', require: false +gem 'ruby-version', '~> 0.4.3' + # Specify your gem's dependencies in cpf_utils.gemspec gemspec diff --git a/lib/cpf_utils/cpf.rb b/lib/cpf_utils/cpf.rb index 3910124..756aed1 100644 --- a/lib/cpf_utils/cpf.rb +++ b/lib/cpf_utils/cpf.rb @@ -8,13 +8,23 @@ class Cpf def initialize(numbers) if numbers.is_a? String numbers = numbers.split('') - elsif numbers.is_a? Fixnum - numbers = numbers.to_s.split('') + elsif Ruby::Version < '2.4.0' + if numbers.is_a? Fixnum + numbers = split_number(numbers) + end + elsif Ruby::Version >= '2.4.0' + if numbers.is_a? Integer + numbers = split_number(numbers) + end end @numbers = numbers end + def split_number (numbers) + numbers.to_s.split('') + end + # Gera o CPF propriamente dito def generate_cpf @numbers << first_digit