exdjango

defmodule ExDjango.Utils.Baseconv do
  @moduledoc """
  Functions for converting numbers from base 10 integers to base X strings and back again.

  Compatible with django/utils/baseconv.py
  """

  @base10_chars "0123456789"

  def get_chars(base) do
    case base do
      :base2  -> "012"
      :base10 -> @base10_chars
      :base16 -> "0123456789ABCDEF"
      :base36 -> "0123456789abcdefghijklmnopqrstuvwxyz"
      :base56 -> "23456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnpqrstuvwxyz"
      :base62 -> "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
      :base64 -> "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz-_"
    end
  end

  def get_sign(:base64), do: "$"
  def get_sign(_base), do: "-"

  def convert(number, from_digits, to_digits, sign) do
    {neg, number} = case String.starts_with?(number, sign) do
      false -> {false, number}
      true -> {true, String.slice(number, 1..-1)}
    end

    res = case convert_from(String.graphemes(number), from_digits, 0) do
      0 -> String.slice(to_digits, 1..-1)
      x -> convert_to(x, to_digits, "")
    end

    {neg, res}
  end

  def convert_from([], _from_digits, x), do: x
  def convert_from([digit|t], from_digits, x) do
    x = x * String.length(from_digits) + Enum.find_index(String.graphemes(from_digits), fn(x) -> x == digit end)
    convert_from(t, from_digits, x)
  end

  def convert_to(x, _to_digits, res) when x <= 0, do: res
  def convert_to(x, to_digits, res) when x > 0 do
    digit = rem(x, String.length(to_digits))
    res = String.slice(to_digits, digit, 1) <> res
    x = div(x, String.length(to_digits))
    convert_to(x, to_digits, res)
  end

  def encode(base, s), do: encode(get_chars(base), get_sign(base), s)
  def encode(chars, sign, s) when is_integer(s), do: encode(chars, sign, Integer.to_string(s))
  def encode(chars, sign, s) do
    {neg, res} = convert(s, @base10_chars, chars, "-")
    case neg do
      true -> sign <> res
      _ -> res
    end
  end

  def decode(base, s), do: decode(get_chars(base), get_sign(base), s)
  def decode(chars, sign, s) do
    {neg, res} = convert(s, chars, @base10_chars, sign)
    case neg do
      true -> String.to_integer("-" <> res)
      _ -> String.to_integer(res)
    end
  end

end