lazymaru

defmodule LazyHelper.Response do
  defmacro __using__(_) do
    quote do
      import Plug.Conn
      import unquote(__MODULE__)
    end
  end


  defmacro assigns do
    quote do
      var!(conn).assigns
    end
  end

  defmacro assign(key, value) do
    quote do
      var!(conn) = var!(conn)
   |> Plug.Conn.assign(unquote(key), unquote(value))
    end
  end


  defmacro headers do
    quote do
      var!(conn).req_headers |> Enum.into %{}
    end
  end

  defmacro header(key, nil) do
    quote do
      var!(conn) = var!(conn)
   |> Plug.Conn.delete_resp_header(unquote(key))
    end
  end

  defmacro header(key, value) do
    quote do
      var!(conn) = var!(conn)
   |> Plug.Conn.put_resp_header(unquote(key), unquote(value))
    end
  end


  defmacro content_type(value) do
    quote do
      var!(conn) = var!(conn)
   |> Plug.Conn.put_resp_content_type(unquote(value))
    end
  end


  defmacro json(reply, code \\ 200) do
    quote do
      content_type "application/json"
      var!(conn) |> send_resp(unquote(code), unquote(reply) |> JSON.encode!)
    end
  end


  defmacro html(reply, code \\ 200) do
    quote do
      content_type "text/html"
      var!(conn) |> send_resp(unquote(code), unquote(reply))
    end
  end


  defmacro text(reply, code \\ 200) do
    quote do
      content_type "text/plain"
      var!(conn) |> send_resp(unquote(code), unquote(reply))
    end
  end
end