timeless_phoenix

if Code.ensure_loaded?(Igniter) do
  defmodule Mix.Tasks.TimelessPhoenix.Install do
    @shortdoc "Installs TimelessPhoenix into your application."
    @moduledoc """
    #{@shortdoc}

    Adds TimelessPhoenix to your supervision tree, configures your Phoenix router
    with all observability dashboard pages, and updates the formatter.

    ## Usage

        mix igniter.install timeless_phoenix
        mix igniter.install timeless_phoenix --storage memory

    ## Options

      * `--storage` — `disk` (default) or `memory`. Disk mode persists logs and
        traces with indexing and retention management. Memory mode keeps them
        in RAM only and loses them on restart. Metrics are always persisted to disk.
      * `--http` — Enable HTTP ingest/query endpoints for metrics, logs, and traces.
      * `--http-metrics` — Enable only the metrics HTTP endpoint.
      * `--http-logs` — Enable only the logs HTTP endpoint.
      * `--http-traces` — Enable only the traces HTTP endpoint.
      * `--metrics-port` — Port for the metrics HTTP endpoint (default 8428).
      * `--logs-port` — Port for the logs HTTP endpoint (default 9428).
      * `--traces-port` — Port for the traces HTTP endpoint (default 10428).

    ## What it does

    1. Adds `{TimelessPhoenix, ...}` to your application's supervision tree
       (logs, traces, and metrics persist to disk by default)
    2. Configures OpenTelemetry to export spans to TimelessTraces
    3. Adds `import TimelessPhoenix.Router` to your Phoenix router
    4. Adds `timeless_phoenix_dashboard "/dashboard"` to your router's browser scope
    5. Adds `:timeless_phoenix` to your `.formatter.exs` import_deps
    6. Removes the default `live_dashboard` route (avoids live_session conflict)
    """

    use Igniter.Mix.Task

    @impl Igniter.Mix.Task
    def info(_argv, _composing_task) do
      %Igniter.Mix.Task.Info{
        group: :timeless_phoenix,
        schema: [
          storage: :string,
          http: :boolean,
          http_metrics: :boolean,
          http_logs: :boolean,
          http_traces: :boolean,
          metrics_port: :integer,
          logs_port: :integer,
          traces_port: :integer
        ],
        defaults: [storage: "disk"],
        required: [],
        positional: [],
        aliases: [],
        composes: [],
        installs: [],
        adds_deps: [],
        example: "mix igniter.install timeless_phoenix --storage memory"
      }
    end

    @impl Igniter.Mix.Task
    def igniter(igniter) do
      storage = igniter.args.options[:storage] || "disk"
      http_opts = resolve_http_opts(igniter.args.options)

      igniter
      |> add_to_supervision_tree(storage, http_opts)
      |> configure_opentelemetry()
      |> setup_router()
      |> remove_default_live_dashboard()
      |> Igniter.Project.Formatter.import_dep(:timeless_phoenix)
    end

    defp resolve_http_opts(options) do
      all? = options[:http] || false

      enabled =
        []
        |> then(fn acc ->
          if all? || options[:http_metrics],
            do: [{:metrics, options[:metrics_port] || 8428} | acc],
            else: acc
        end)
        |> then(fn acc ->
          if all? || options[:http_logs],
            do: [{:logs, options[:logs_port] || 9428} | acc],
            else: acc
        end)
        |> then(fn acc ->
          if all? || options[:http_traces],
            do: [{:traces, options[:traces_port] || 10428} | acc],
            else: acc
        end)
        |> Enum.reverse()

      enabled
    end

    # Adds {TimelessPhoenix, ...} to the application's children list.
    defp add_to_supervision_tree(igniter, storage, http_opts) do
      opts_parts = [~s(data_dir: "priv/observability")]

      opts_parts =
        case storage do
          "memory" ->
            opts_parts ++
              ["timeless_logs: [storage: :memory]", "timeless_traces: [storage: :memory]"]

          _ ->
            opts_parts
        end

      opts_parts =
        case http_opts do
          [] ->
            opts_parts

          entries ->
            http_kw = Enum.map_join(entries, ", ", fn {k, v} -> "#{k}: #{v}" end)
            opts_parts ++ ["http: [#{http_kw}]"]
        end

      opts_string = "[" <> Enum.join(opts_parts, ", ") <> "]"
      child_code = Sourceror.parse_string!(opts_string)

      Igniter.Project.Application.add_new_child(
        igniter,
        {TimelessPhoenix, {:code, child_code}}
      )
    end

    # Adds the import and macro call to the Phoenix router.
    defp setup_router(igniter) do
      case Igniter.Libs.Phoenix.select_router(igniter) do
        {igniter, nil} ->
          Igniter.add_warning(igniter, """
          No Phoenix router found. Add the following manually:

              import TimelessPhoenix.Router

              scope "/" do
                pipe_through :browser
                timeless_phoenix_dashboard "/dashboard"
              end
          """)

        {igniter, router} ->
          igniter
          |> add_router_import(router)
          |> Igniter.Libs.Phoenix.append_to_scope(
            "/",
            """
            timeless_phoenix_dashboard "/dashboard"
            """,
            with_pipelines: [:browser],
            router: router
          )
      end
    end

    # Configures OpenTelemetry to export spans to TimelessTraces.
    # This must be in compile-time config so it takes effect before the OTel app starts.
    defp configure_opentelemetry(igniter) do
      Igniter.Project.Config.configure(
        igniter,
        "config.exs",
        :opentelemetry,
        [:traces_exporter],
        {:code, Sourceror.parse_string!("{TimelessTraces.Exporter, []}")}
      )
    end

    # Removes the default Phoenix LiveDashboard route to avoid live_session conflicts.
    # The default Phoenix generator puts `live_dashboard "/dashboard"` inside
    # `if Application.compile_env(:app, :dev_routes) do ... end` — we remove
    # the `live_dashboard` call since TimelessPhoenix provides its own dashboard.
    defp remove_default_live_dashboard(igniter) do
      case Igniter.Libs.Phoenix.select_router(igniter) do
        {igniter, nil} ->
          igniter

        {igniter, router} ->
          Igniter.Project.Module.find_and_update_module!(igniter, router, fn zipper ->
            # Remove `live_dashboard` calls (our macro provides its own)
            zipper =
              Igniter.Code.Common.remove_all_matches(zipper, fn z ->
                Igniter.Code.Function.function_call?(z, :live_dashboard, :any)
              end)

            # Remove `import Phoenix.LiveDashboard.Router` (now unused)
            zipper =
              Igniter.Code.Common.remove_all_matches(zipper, fn z ->
                Igniter.Code.Function.function_call?(z, :import, 1) &&
                  match?(
                    {:ok,
                     %Sourceror.Zipper{
                       node: {:__aliases__, _, [:Phoenix, :LiveDashboard, :Router]}
                     }},
                    Igniter.Code.Function.move_to_nth_argument(z, 0)
                  )
              end)

            {:ok, zipper}
          end)
      end
    end

    # Adds `import TimelessPhoenix.Router` after `use Phoenix.Router` in the router module.
    defp add_router_import(igniter, router) do
      Igniter.Project.Module.find_and_update_module!(igniter, router, fn zipper ->
        case Igniter.Libs.Phoenix.move_to_router_use(igniter, zipper) do
          {:ok, zipper} ->
            {:ok, Igniter.Code.Common.add_code(zipper, "import TimelessPhoenix.Router")}

          _ ->
            {:ok, zipper}
        end
      end)
    end
  end
else
  defmodule Mix.Tasks.TimelessPhoenix.Install do
    @shortdoc "Installs TimelessPhoenix (requires igniter)."
    @moduledoc @shortdoc
    use Mix.Task

    def run(_argv) do
      Mix.shell().error("""
      The task 'timeless_phoenix.install' requires igniter.
      Please install igniter and try again.

          {:igniter, "~> 0.6", only: [:dev]}

      For more information, see: https://hexdocs.pm/igniter
      """)

      exit({:shutdown, 1})
    end
  end
end