lib/bra_documents/formatter.rb in bra_documents-1.0.0 vs lib/bra_documents/formatter.rb in bra_documents-1.0.1

@@ -1,21 +1,48 @@
 # frozen_string_literal: true
 
 module BraDocuments
   class Formatter
+    NOT_NUMBER = /\D/
     FORMATS = {
       cpf: { pattern: /\A(\d{3})(\d{3})(\d{3})(\d{2})\z/, mask: '%s.%s.%s-%s' },
-      cnpj: { pattern: /\A(\d{2})(\d{3})(\d{3})(\d{4})(\d{2})\z/, mask: '%s.%s.%s/%s-%s'}
+      cnpj: { pattern: /\A(\d{2})(\d{3})(\d{3})(\d{4})(\d{2})\z/, mask: '%s.%s.%s/%s-%s' }
     }.freeze
 
-    # Formats a only numbers CPF or CNPJ in their own mask
-    #
-    # BraDocuments::Formatter.format('86027265892', as: :cpf) # => "860.272.658-92"
-    # BraDocuments::Formatter.format('53855973879456', as: :cnpj) # => "53.855.973/8794-56"
-    def self.format(number, as:)
-      format_data = FORMATS[as]
+    class << self
+      # Formats a only numbers CPF or CNPJ in their own mask
+      #
+      # BraDocuments::Formatter.format('86027265892', as: :cpf) # => "860.272.658-92"
+      # BraDocuments::Formatter.format('53855973879456', as: :cnpj) # => "53.855.973/8794-56"
+      def format(number, as:)
+        raise ArgumentError, "\"#{number.inspect}\" must be a String." unless number.is_a?(String)
+        unless known_format?(as)
+          raise ArgumentError, "Format \"#{as}\" is not know. Known formats: #{known_formats.join(', ')}."
+        end
 
-      Kernel.format(format_data[:mask], *format_data[:pattern].match(number).captures)
+        format_data = FORMATS[as]
+
+        Kernel.format(format_data[:mask], *format_data[:pattern].match(number).captures)
+      end
+
+      # Formats removing all not number caracters from string.
+      #
+      # BraDocuments::Formatter.raw('860.272.658-9') # => "286027265892"
+      # BraDocuments::Formatter.format('53.855.973/8794-56') # => "53855973879456"
+      def raw(number)
+        raise ArgumentError, "\"#{number.inspect}\" must be a String." unless number.is_a?(String)
+
+        number.gsub(NOT_NUMBER, '')
+      end
+
+      private
+
+      def known_formats
+        @known_formats ||= FORMATS.keys
+      end
+
+      def known_format?(format)
+        known_formats.include?(format)
+      end
     end
   end
 end
-