ace_http

defmodule Ace.HTTP1.Worker do
  use GenServer

  def start_link(client, {mod, config}) do
    GenServer.start_link(__MODULE__, {mod, config, client}, [])
  end

  # conn = stream
  def handle_info({client, request = %Raxx.Request{}}, {mod, state, client}) do
    mod.handle_headers(request, state)
    |> normalise_reaction({mod, state, client})
  end
  def handle_info({client, fragment = %Raxx.Fragment{}}, {mod, state, client}) do
    mod.handle_fragment(fragment.data, state)
    |> normalise_reaction({mod, state, client})
    |> case do
      {:noreply, {mod, state, client}} ->
        if fragment.end_stream do
          mod.handle_trailers([], state)
          |> normalise_reaction({mod, state, client})
        else
          {:noreply, {mod, state, client}}
        end
    end
  end
  # DEBT I think that the worker should expect to explicitly receive a tail message
  def handle_info({client, trailer = %Raxx.Trailer{}}, {mod, state, client}) do
    mod.handle_trailers(trailer.headers, state)
    |> normalise_reaction({mod, state, client})
  end

  def handle_info(other, {mod, state, client}) do
    mod.handle_info(other, state)
    |> normalise_reaction({mod, state, client})
  end

  defp normalise_reaction(response = %Raxx.Response{}, {mod, state, client}) do
    send_client(client, response)
    if Raxx.complete?(response) do
      {:stop, :normal, {mod, state, client}}
    else
      {:noreply, {mod, state, client}}
    end
  end
  defp normalise_reaction({parts, new_state}, {mod, _old_state, client}) do
    Enum.each(parts, fn(part) -> send_client(client, part) end)
    {:noreply, {mod, new_state, client}}
  end

  defp send_client(ref = {:http1, pid, _count}, part) do
    send(pid, {ref, part})
  end
end