definject

defmodule Definject.Impl do
  @moduledoc false
  @uninjectable Application.get_env(:definject, :uninjectable, [
                  PlaceholderToSuppressCompileWarning
                ])

  def inject_function(%{head: head, body: body, env: %Macro.Env{} = env}) do
    injected_head = head_with_deps(%{head: head})

    {_, env_modified?} =
      Macro.prewalk(body, false, fn ast, env_modified? ->
        {ast, env_modified? || modifies_env?(ast)}
      end)

    if env_modified? do
      quote do
        raise "Cannot import/require/use inside definject. Move it to module level."
      end
    else
      {injected_body, captures} = inject_remote_calls_recursively(body, env)

      # `quote` with dynamic `context` requires Elixir 1.10+
      quote do
        def unquote(injected_head) do
          Definject.Check.validate_deps(unquote(captures), deps)

          unquote(injected_body)
        end
      end
    end
  end

  def inject_remote_calls_recursively(body, env) do
    body
    |> Macro.prewalk(fn ast ->
      if expandable?(ast) do
        Macro.expand(ast, env)
      else
        ast
      end
    end)
    |> Macro.prewalk(&mark_remote_capture/1)
    |> Macro.postwalk([], fn ast, captures ->
      %{ast: ast, captures: new_captures} = inject_remote_call(ast)
      {ast, new_captures ++ captures}
    end)
  end

  defp expandable?({:@, _, _}), do: false
  defp expandable?(_), do: true

  defp modifies_env?({name, _, _}) when name in [:import, :require, :use], do: true
  defp modifies_env?(_), do: false

  def head_with_deps(%{head: {name, meta, context}}) when not is_list(context) do
    # Normalize function head.
    # def some do: nil end   ->   def some(), do: nil end
    head_with_deps(%{head: {name, meta, []}})
  end

  def head_with_deps(%{head: {name, meta, params}}) when is_list(params) do
    deps =
      quote do
        %{} = deps \\ %{}
      end

    {name, meta, params ++ [deps]}
  end

  #
  # This function is not feasible because`no_parens: true` is only available from Elixir 1.10.
  # When we can require Elixir 1.10, uncomment below and delete `mark_remote_capture`.
  #
  # def inject_remote_call({{:., _, [_remote_mod, _name]}, [{:no_parens, true} | _], _args} = ast) do
  #   # nested captures via & are not allowed
  #   %{ast: ast, captures: []}
  # end
  #

  def inject_remote_call(
        {{:., _, [_remote_mod, _name]}, [{:remote_capture, true} | _], _args} = ast
      ) do
    %{ast: ast, captures: []}
  end

  def inject_remote_call({{:., _, [remote_mod, name]}, _, args} = _ast)
      when remote_mod not in @uninjectable and is_atom(name) and is_list(args) do
    arity = Enum.count(args)
    capture = function_capture_ast(remote_mod, name, arity)

    ast =
      quote do
        (deps[unquote(capture)] || unquote(capture)).(unquote_splicing(args))
      end

    %{ast: ast, captures: [capture]}
  end

  def inject_remote_call(ast) do
    %{ast: ast, captures: []}
  end

  def mark_remote_capture(
        {:&, c1,
         [
           {:/, c2,
            [
              {{:., c3, [remote_mod, name]}, c4, []},
              arity
            ]}
         ]}
      ) do
    {:&, c1,
     [
       {:/, c2,
        [
          {{:., c3, [remote_mod, name]}, [{:remote_capture, true} | c4], []},
          arity
        ]}
     ]}
  end

  def mark_remote_capture(ast) do
    ast
  end

  def surround_by_fn({{:&, _, [capture]}, v}) do
    {:/, _, [mf, a]} = capture
    {mf, _, []} = mf
    {:., _, [m, f]} = mf

    capture = function_capture_ast(m, f, a)
    const_fn = {:fn, [], [{:->, [], [Macro.generate_arguments(a, __MODULE__), v]}]}

    {capture, const_fn}
  end

  def surround_by_fn({:strict, _} = orig) do
    orig
  end

  defp function_capture_ast(remote_mod, name, arity) do
    mf = {{:., [], [remote_mod, name]}, [], []}
    mfa = {:/, [], [mf, arity]}
    {:&, [], [mfa]}
  end
end