require "correios/version"
module Correios

  # Algoritmo para cálculo do dígito verificador do número de registro:
  #  1. Aplicar os fatores de ponderação sobre os dígitos do número seqüencial do registro utilizando os seguintes algarismos: 8 6 4 2 3 5 9 7;
  #  2. Calcular a soma destes números;
  #  3. Dividir esta soma por 11;
  #  4. Caso o resto seja igual a 0 ou 1, utilizar como dígito de controle, respectivamente, 5 ou 0;
  #  5. Caso contrário, subtrair o resto de 11, sendo o algarismo obtido correspondente ao dígito verificador. EXEMPLO:
  #                Se o número seqüencial for 47312482, aplicando a fórmula teremos: (4*8) + (7*6) + (3*4) + (1*2) + (2*3) + (4*5) + (8*9) + (2*7) = 200 Continuando, 200 / 11 = 18 com resto = 2.
  #                O dígito verificador será dado por: DV = 11 - 2 = 9.
  #
  class SROValidator

    WEIGHTING_FACTORS = [8, 6, 4, 2, 3, 5, 9, 7].freeze

    CONTROL_DIGIT = {
      0 => 5,
      1 => 0
    }.freeze

    attr_reader :sro, :sanitized_sro, :numbers, :verification_digit

    # Initializer the SROValidator
    #
    # @param [String] sro Tracking Code like: LX473124829BR
    # @param [String] suffix_match = 'BR' the regex match suffix (BR as default)
    # @return [self] SROValidator
    def initialize(sro, suffix_match = 'BR')
      @sro = sro
      @sanitized_sro = sro.to_s.strip.upcase
      @sanitized_sro =~ /^[A-Z|a-z]{2}([0-9]{8})([0-9])#{suffix_match}$/
      @numbers = ($1 || "")
      @verification_digit = $2.to_i
    end

    # Return if the SRO is valid or not
    #
    # @return [True] true when the 9th digit matchs the validation algorithm
    # @return [False] false otherwise
    def valid?
      result == verification_digit
    end

    private

    def result
      CONTROL_DIGIT[mod] || (11-mod)
    end

    def mod
      @mod ||= weighted_mean % 11
    end

    def weighted_mean
      sro_array.zip(WEIGHTING_FACTORS)
        .map { |a| a.inject(:*) }
        .inject(:+) || Float::INFINITY
    end

    def sro_array
      numbers.split("").map(&:to_i)
    end

  end
end