dgilperez/validates_zipcode

# frozen_string_literal: true
module ValidatesZipcode
  class Formatter
    WORD_CHAR_AND_DIGIT = /[A-Z0-9]/.freeze

    ZIPCODES_TRANSFORMATIONS = {
      AT: ->(z) { z.scan(/\d/).join },
      ES: :AT,
      CA: ->(z) { z.upcase.scan(WORD_CHAR_AND_DIGIT).insert(3, ' ').join },
      CZ: ->(z) { z.scan(/\d/).insert(3, ' ').join },
      DE: ->(z) {
        digits = z.scan(/\d/)
        result = digits.join
        result.prepend('0') if digits.count < 5
        result
      },
      GB: ->(z) { z.upcase.scan(WORD_CHAR_AND_DIGIT).insert(-4, ' ').join },
      NL: ->(z) { z.upcase.scan(WORD_CHAR_AND_DIGIT).insert(4, ' ').join },
      PL: ->(z) { z.scan(/\d/).insert(2, '-').join },
      SK: :CZ,
      UK: :GB,
      MT: ->(z) { z.upcase.scan(WORD_CHAR_AND_DIGIT).insert(3, ' ').join },
      FK: :GB,
      GS: :GB,
      PN: :GB,
      SH: :GB,
      TC: :GB,
      GI: :GB,
      US: ->(z) {
        digits = z.scan(/\d/)
        digits.insert(5, '-') if digits.count > 5
        digits.join
      },
      PR: :US
    }.freeze

    def initialize(args = {})
      @zipcode        = args.fetch(:zipcode).to_s
      @country_alpha2 = args.fetch(:country_alpha2).to_s.upcase.to_sym
    end

    def format
      transformation = ZIPCODES_TRANSFORMATIONS[@country_alpha2]
      case transformation
      when Proc
        transformation.call(@zipcode)
      when Symbol
        ZIPCODES_TRANSFORMATIONS[transformation].call(@zipcode)
      else
        @zipcode.strip
      end
    end
  end
end