compox

defmodule Compox.Docker.ConnectionCheck do
  @moduledoc false

  # XXX: use back off?
  @default_sleep 500

  @doc """
  Waits the current process and checks the given service until the container is
  up.

  This function receives a _service name_ and uses the `container_upchecks`
  config to ensure the container meets the requirements to be consider
  available.
  """
  @spec maybe_do_upcheck(service_name :: String.t()) :: :ok
  def maybe_do_upcheck(service_name) do
    :compox
    |> Application.get_env(:container_upchecks)
    |> Keyword.get(String.to_atom(service_name))
    |> case do
      nil -> :ok
      upcheck_fun -> do_upcheck(upcheck_fun, service_name)
    end
  end

  @doc """
  Waits the current process un the container is running.
  """
  @spec wait_until_up(container_id :: String.t()) :: :ok
  def wait_until_up(container_id) do
    container_id
    |> Compox.Docker.API.Containers.running?()
    |> case do
      true ->
        Process.sleep(@default_sleep)
        wait_until_up(container_id)

      _ ->
        :ok
    end
  end

  @doc """
  Waits the current process un the container is not running.
  """
  @spec wait_until_down(container_id :: String.t()) :: :ok
  def wait_until_down(container_id) do
    container_id
    |> Compox.Docker.API.Containers.running?()
    |> case do
      true ->
        Process.sleep(500)
        wait_until_down(container_id)

      _ ->
        :ok
    end
  end

  #
  # Private functions
  #

  @spec do_upcheck(upcheck_fun :: function(), service_name :: String.t()) :: :ok
  defp do_upcheck(upcheck_fun, service_name) do
    case upcheck_fun.() do
      :ok ->
        :ok

       _error ->
        Mix.shell().info("[compox] waiting for #{service_name}...")
        Process.sleep(@default_sleep)
        do_upcheck(upcheck_fun, service_name)
    end
  end
end