lib/prom_ex/plugins/application.ex

defmodule PromEx.Plugins.Application do
  @moduledoc """
  This plugin captures metrics regarding your application, and its dependencies. Specifically,
  it captures the versions of your application and the application dependencies and also
  how many modules each dependency is bringing into the project.

  This plugin supports the following options:
  - `otp_app`: This is a REQUIRED option and is the name of you application in snake case (e.g. :my_cool_app).

  - `deps`: This option is OPTIONAL and defines what dependencies the plugin should track. A value of `:all`
    means that PromEx will fetch details on all application dependencies. A list of dependency names like
    `[:phoenix, :ecto, :unplug]` means that PromEx will only fetch details regarding those dependencies.

  - `git_sha_mfa`: This option is OPTIONAL and defines an MFA that will be called in order to fetch the
    application's Git SHA at the time of deployment. By default, an Application Plugin function will be called
    and will attempt to read the GIT_SHA environment variable to populate the value.

  - `git_author_mfa`: This option is OPTIONAL and defines an MFA that will be called in order to fetch the
    application's last Git commit author at the time of deployment. By default, an Application Plugin function
    will be called and will attempt to read the GIT_AUTHOR environment variable to populate the value.

  - `metric_prefix`: This option is OPTIONAL and is used to override the default metric prefix of
    `[otp_app, :prom_ex, :application]`. If this changes you will also want to set `application_metric_prefix`
    in your `dashboard_assigns` to the snakecase version of your prefix, the default
    `application_metric_prefix` is `{otp_app}_prom_ex_application`.

  - `duration_unit`: This is an OPTIONAL option and is a `Telemetry.Metrics.time_unit()`. It can be one of:
    `:second | :millisecond | :microsecond | :nanosecond`. It is `:millisecond` by default.

  This plugin exposes the following metric groups:
  - `:application_versions_manual_metrics`

  To use plugin in your application, add the following to your application supervision tree:
  ```
  def start(_type, _args) do
    children = [
      ...
      {
        PromEx,
        plugins: [
          {PromEx.Plugins.Application, [otp_app: :my_cool_app]},
          ...
        ],
        delay_manual_start: :no_delay
      }
    ]

    opts = [strategy: :one_for_one, name: WebApp.Supervisor]
    Supervisor.start_link(children, opts)
  end
  ```

  This plugin exposes manual metrics so be sure to configure the PromEx `:delay_manual_start` as required.
  """

  use PromEx.Plugin

  @filter_out_apps [
    :iex,
    :inets,
    :logger,
    :runtime_tools,
    :ssl,
    :crypto
  ]

  @impl true
  def manual_metrics(opts) do
    otp_app = Keyword.fetch!(opts, :otp_app)
    deps = Keyword.get(opts, :deps, :all)
    git_sha_mfa = Keyword.get(opts, :git_sha_mfa, {__MODULE__, :git_sha, []})
    git_author_mfa = Keyword.get(opts, :git_author_mfa, {__MODULE__, :git_author, []})
    metric_prefix = Keyword.get(opts, :metric_prefix, PromEx.metric_prefix(otp_app, :application))

    [
      Manual.build(
        :application_versions_manual_metrics,
        {__MODULE__, :apps_running, [otp_app, deps, git_sha_mfa, git_author_mfa]},
        [
          # Capture information regarding the primary application (i.e the user's application)
          last_value(
            metric_prefix ++ [:primary, :info],
            event_name: [otp_app | [:application, :primary, :info]],
            description: "Information regarding the primary application.",
            measurement: :status,
            tags: [:name, :version, :modules]
          ),

          # Capture information regarding the application dependencies (i.e the user's libs)
          last_value(
            metric_prefix ++ [:dependency, :info],
            event_name: [otp_app | [:application, :dependency, :info]],
            description: "Information regarding the application's dependencies.",
            measurement: :status,
            tags: [:name, :version, :modules]
          ),

          # Capture application Git SHA using user provided MFA
          last_value(
            metric_prefix ++ [:git_sha, :info],
            event_name: [otp_app | [:application, :git_sha, :info]],
            description: "The application's Git SHA at the time of deployment.",
            measurement: :status,
            tags: [:sha]
          ),

          # Capture application Git author using user provided MFA
          last_value(
            metric_prefix ++ [:git_author, :info],
            event_name: [otp_app | [:application, :git_author, :info]],
            description: "The application's author of the last Git commit at the time of deployment.",
            measurement: :status,
            tags: [:author]
          )
        ]
      )
    ]
  end

  @impl true
  def polling_metrics(opts) do
    otp_app = Keyword.fetch!(opts, :otp_app)
    poll_rate = Keyword.get(opts, :poll_rate, 5_000)
    metric_prefix = Keyword.get(opts, :metric_prefix, PromEx.metric_prefix(otp_app, :application))
    duration_unit = Keyword.get(opts, :duration_unit, :millisecond)
    duration_unit_plural = PromEx.Utils.make_plural_atom(duration_unit)

    Polling.build(
      :application_time_polling_metrics,
      poll_rate,
      {__MODULE__, :execute_time_metrics, []},
      [
        last_value(
          metric_prefix ++ [:uptime, duration_unit_plural, :count],
          event_name: [:prom_ex, :plugin, :application, :uptime, :count],
          description:
            "The total number of wall clock #{duration_unit_plural} that have passed since the application started.",
          measurement: :count,
          unit: duration_unit
        )
      ]
    )
  end

  @doc false
  def execute_time_metrics do
    {wall_clock_time, _} = :erlang.statistics(:wall_clock)
    :telemetry.execute([:prom_ex, :plugin, :application, :uptime, :count], %{count: wall_clock_time})
  end

  @doc false
  def git_sha do
    case System.fetch_env("GIT_SHA") do
      {:ok, git_sha} ->
        git_sha

      :error ->
        "Git SHA not available"
    end
  end

  @doc false
  def git_author do
    case System.fetch_env("GIT_AUTHOR") do
      {:ok, git_sha} ->
        git_sha

      :error ->
        "Git author not available"
    end
  end

  @doc false
  def apps_running(otp_app, deps, git_sha_mfa, git_author_mfa) do
    # Loop through all of the dependencies
    filtered_deps =
      if deps == :all do
        otp_app
        |> Application.spec()
        |> Keyword.get(:applications)
        |> Enum.reject(fn
          app when app in @filter_out_apps -> true
          _ -> false
        end)
      else
        deps
      end

    filtered_deps
    |> Enum.each(fn app ->
      :telemetry.execute(
        [otp_app | [:application, :dependency, :info]],
        %{status: 1},
        %{
          name: app,
          version: Application.spec(app)[:vsn],
          modules: length(Application.spec(app)[:modules])
        }
      )
    end)

    # Emit primary app details
    :telemetry.execute(
      [otp_app | [:application, :primary, :info]],
      %{status: 1},
      %{
        name: otp_app,
        version: Application.spec(otp_app)[:vsn],
        modules: length(Application.spec(otp_app)[:modules])
      }
    )

    # Publish Git SHA data
    {module, function, args} = git_sha_mfa
    git_sha = apply(module, function, args)

    :telemetry.execute(
      [otp_app | [:application, :git_sha, :info]],
      %{status: 1},
      %{sha: git_sha}
    )

    # Publish Git author data
    {module, function, args} = git_author_mfa
    git_author = apply(module, function, args)

    :telemetry.execute(
      [otp_app | [:application, :git_author, :info]],
      %{status: 1},
      %{author: git_author}
    )
  end
end