defmodule Codex.Base.Base32 do
  import Codex

  @capitals 65..90 |> Enum.map(&:binary.encode_unsigned/1)
  @numbers 50..55 |> Enum.map(&:binary.encode_unsigned/1)

  @char_list (@capitals ++ @numbers) |> Enum.with_index()

  @char_to_num @char_list |> Enum.into(%{})
  @num_to_char @char_list |> Enum.map(fn {char, num} -> {num, char} end) |> Enum.into(%{})

  @spec codec() :: Codex.codec(String.t())
  def codec() do
    char_4 = bits(4) |> done() |> pad(1) |> index_to_char()
    char_3 = bits(3) |> done() |> pad(2) |> index_to_char()
    char_2 = bits(2) |> done() |> pad(3) |> index_to_char()
    char_1 = bits(1) |> done() |> pad(4) |> index_to_char()

    list(
      choice(
        [
          char(8),
          char(1) ++ [char_3] ++ padding(6),
          char(3) ++ [char_1] ++ padding(4),
          char(4) ++ [char_4] ++ padding(3),
          char(6) ++ [char_2] ++ padding(1)
        ]
        |> Enum.map(&(&1 |> sequence() |> join()))
      )
    )
    |> join(8)
  end

  defp padding(count), do: List.duplicate(value("="), count)
  defp char(count), do: List.duplicate(bits(5) |> index_to_char(), count)

  defp index_to_char(index),
    do: convert(index, &Map.fetch!(@num_to_char, &1), &Map.fetch!(@char_to_num, &1))
end