module Ruby
  module ShortUrl
    # Original imprementation is here: https://github.com/Alir3z4/python-short_url
    class Encoder
      DEFAULT_ALPHABET = ((0..9).to_a + ('A'..'Z').to_a + ('a'..'z').to_a).join
      DEFAULT_BLOCK_SIZE = 24
      MIN_LENGTH = 5

      def initialize(alphabet: DEFAULT_ALPHABET, block_size: DEFAULT_BLOCK_SIZE)
        @alphabet = alphabet
        @block_size = block_size
        @mask = (1 << block_size) - 1
        @mapping = (0..block_size - 1).to_a
      end

      def encode_url(n, min_length: MIN_LENGTH)
        enbase(encode(n), min_length)
      end

      def decode_url(n)
        decode(debase(n))
      end

      private

      def encode(n)
        (n & ~@mask) | _encode(n & @mask)
      end

      def decode(n)
        (n & ~@mask) | _decode(n & @mask)
      end

      def enbase(x, min_length=MIN_LENGTH)
        result = _enbase(x)
        padding_length = min_length - result.length
        padding_length = 0 if padding_length < 0
        padding = @alphabet[0] * padding_length
        padding.to_s + result.to_s
      end

      def debase(x)
        n = @alphabet.length
        result = 0
        x.split('').reverse.each_with_index do |c, i|
          unless @alphabet.index(c).nil?
            result += @alphabet.index(c) * (n ** i)
          end
        end
        result
      end

      def _encode(n)
        result = 0
        @mapping.reverse.each_with_index do |b, i|
          if (n & (1 << i)) != 0
            result |= (1 << b)
          end
        end
        result
      end

      def _decode(n)
        result = 0
        @mapping.reverse.each_with_index do |b, i|
          if n & (1 << b) != 0
            result |= (1 << i)
          end
        end
        result
      end

      def _enbase(x)
        n = @alphabet.length
        return @alphabet[x] if x < n
        _enbase(x / n) + @alphabet[x % n]
      end
    end
  end
end