pbuf

defmodule Pbuf.Protoc.Fields.Simple do
  use Pbuf.Protoc.Field

  def new(desc, context) do
    typespec = typespec(desc)
    repeated = is_repeated?(desc)
    packed? = context.version == 3 || (desc.options != nil && desc.options.packed == true)

    {tag, name, type, prefix} = extract_core(desc, packed?)

    {encode_fun, typespec, default} =
    cond do
      repeated == false -> {"field", typespec, default(context, type)}
      packed? == false -> {"repeated_unpacked_field", "[" <> typespec <> "]", "[]"}
      true -> {"repeated_field", "[" <> typespec <> "]", "[]"}
    end

    # length-prefixed types aren't packed, treat the decode as individual
    # fields and let something else worry about merging it
    decode_fun = cond do
      type == :struct -> "struct_field"  # special name largely because a module is an atom and we need to match between the two
      !repeated || type in [:bytes, :string] -> "field" # never packed
      packed? == false -> "repeated_unpacked_field"
      true -> "repeated_field"
    end

    decode_type = case type == :struct do
      false -> ":#{type}"
      true -> module_name(desc)
    end

    post_decode = cond do
      repeated && !packed? -> :repack
      repeated && type in [:bytes, :string, :struct] -> :repack
      true -> get_post_decode(desc, repeated, type)
    end

    encode_fun = case post_decode do
      {:encoder, {encoder, _decode_opts}} -> "
case data.#{name} do
        <<>> -> []
        value -> Encoder.#{encode_fun}(:#{type}, #{encoder}.encode!(value), #{prefix})
      end"
      _ -> "Encoder.#{encode_fun}(:#{type}, data.#{name}, #{prefix})"
    end

    %Field{
      tag: tag,
      name: name,
      prefix: prefix,
      default: default,
      typespec: typespec,
      post_decode: post_decode,
      encode_fun: encode_fun,
      json?: json?(desc, repeated, type),
      decode_fun: "Decoder.#{decode_fun}(#{decode_type}, :#{name}, acc, data)",
    }
  end

  @spec typespec(Protoc.proto_field | atom) :: String.t
  def typespec(%{type_name: t} = desc) when t not in [nil, ""] do
    module_name(desc) <> ".t"
  end

  def typespec(:bool), do: "boolean"
  def typespec(:bytes), do: "binary"
  def typespec(:string), do: "String.t"
  def typespec(:double), do: "number"
  def typespec(:float), do: "number"
  def typespec(:uint32), do: "non_neg_integer"
  def typespec(:uint64), do: "non_neg_integer"
  def typespec(other) when is_atom(other) do
    "integer"
  end
  def typespec(%{type: type}) do
    typespec(internal_type(type))
  end

  def default(%{version: 2}, _), do: "nil"

  def default(_, :bool), do: false
  def default(_, :bytes), do: "<<>>"
  def default(_, :string), do: ~s("")
  def default(_, :double), do: 0.0
  def default(_, :float), do: 0.0
  def default(_, :struct), do: "nil"
  def default(_, _other), do: 0

  defp get_post_decode(%{options: options}, false, type) when options != nil and type in [:bytes, :string] do
    case Map.get(options, :json_field, 0) do
      0 -> :none
      1 -> {:encoder, {Jason, "[]"}}
      2 -> {:encoder, {Jason, "[keys: :atoms]"}}
    end
  end

  defp get_post_decode(_desc, _repeated, _type) do
    :none
  end

  defp json?(%{options: options}, false, _type) when options != nil do
    Map.get(options, :json_field, 0) != -1
  end

  defp json?(_, _, _), do: true
end