defmodule Codex.Base.Base64 do
  import Codex

  @capitals 65..90 |> Enum.map(&:binary.encode_unsigned/1)
  @lowers 97..122 |> Enum.map(&:binary.encode_unsigned/1)
  @numbers 48..57 |> Enum.map(&:binary.encode_unsigned/1)

  @num_to_char (@capitals ++ @lowers ++ @numbers ++ ["+", "/"])
               |> Enum.with_index()
               |> Enum.map(fn {char, num} -> {num, char} end)
               |> Enum.into(%{})

  @spec codec() :: Codex.codec(String.t())
  def codec() do
    char_2 = bits(2) |> done() |> pad(4) |> mapping(@num_to_char)
    char_4 = bits(4) |> done() |> pad(2) |> mapping(@num_to_char)

    list(
      choice(
        [
          char(4),
          char(1) ++ [char_2] ++ padding(2),
          char(2) ++ [char_4] ++ padding(1)
        ]
        |> Enum.map(&(&1 |> sequence() |> join()))
      )
    )
    |> join(4)
  end

  defp padding(count), do: List.duplicate(value("="), count)
  defp char(count), do: List.duplicate(bits(6) |> mapping(@num_to_char), count)
end