agent_session_manager

defmodule ASM.ProviderBackend.Core do
  @moduledoc """
  Backend that runs the shared CLI runtime from `cli_subprocess_core`.
  """

  @behaviour ASM.ProviderBackend

  alias ASM.Error
  alias ASM.Execution
  alias ASM.Options
  alias ASM.Provider
  alias ASM.ProviderBackend.Proxy
  alias ASM.Remote.NodeConnector
  alias ASM.RuntimeAuth
  alias ASM.RuntimeAuth.CodexMaterialization
  alias CliSubprocessCore.ProviderCLI.Error, as: ProviderCLIError
  alias CliSubprocessCore.RecoveryEnvelope
  alias CliSubprocessCore.Session

  @impl true
  def start_run(%{provider: %Provider{} = provider} = config) do
    with {:ok, execution_config} <- fetch_execution_config(config),
         :ok <- validate_approval_posture(execution_config),
         {:ok, session_opts} <- build_session_opts(provider, config, execution_config),
         {:ok, proxy, info} <-
           Proxy.start_link(
             starter: fn subscriber ->
               do_start_run(execution_config, Keyword.put(session_opts, :subscriber, subscriber))
             end,
             runtime_api: Session,
             runtime: Session,
             provider: provider.name,
             lane: :core,
             backend: __MODULE__,
             capabilities: core_capabilities(provider),
             initial_subscribers: initial_subscribers(config)
           ) do
      {:ok, proxy, info}
    else
      {:error, %ProviderCLIError{} = error} ->
        {:error, provider_cli_error(provider.name, error)}

      {:error, %Error{} = error} ->
        {:error, error}

      {:error, reason} ->
        {:error,
         Error.new(:runtime, :runtime, "core backend start failed: #{inspect(reason)}",
           cause: reason
         )}
    end
  end

  @impl true
  def send_input(session, input, opts \\ []) when is_pid(session) do
    Proxy.send_input(session, input, opts)
  end

  @impl true
  def end_input(session) when is_pid(session), do: Proxy.end_input(session)

  @impl true
  def interrupt(session) when is_pid(session), do: Proxy.interrupt(session)

  @impl true
  def close(session) when is_pid(session), do: Proxy.close(session)

  @impl true
  def subscribe(session, pid, ref) when is_pid(session) and is_pid(pid) and is_reference(ref) do
    Proxy.subscribe(session, pid, ref)
  end

  @impl true
  def info(session) when is_pid(session), do: Proxy.info(session)

  defp fetch_execution_config(%{execution_config: %Execution.Config{} = config}),
    do: {:ok, config}

  defp fetch_execution_config(_config) do
    {:error, Error.new(:config_invalid, :config, "missing execution config for core backend")}
  end

  defp do_start_run(%Execution.Config{execution_mode: :local}, session_opts) do
    Session.start_session(session_opts)
  end

  defp do_start_run(
         %Execution.Config{execution_mode: :remote_node, remote: remote_cfg},
         session_opts
       )
       when is_map(remote_cfg) do
    with :ok <- connect_remote(remote_cfg),
         :ok <- preflight_remote(remote_cfg),
         :ok <- maybe_bootstrap_remote(remote_cfg) do
      remote_start(remote_cfg, session_opts)
    end
  end

  defp build_session_opts(provider, config, execution_config) do
    metadata =
      Map.merge(
        %{lane: :core, asm_provider: provider.name},
        Map.get(config, :metadata, %{})
      )

    with {:ok, provider_opts} <-
           Options.finalize_provider_opts(
             provider.name,
             effective_provider_opts(config, execution_config)
           ),
         :ok <-
           RuntimeAuth.authorize_governed_provider_runtime(provider.name, config, provider_opts),
         {:ok, materialization} <-
           authorize_codex_materialization(provider, config, provider_opts) do
      provider_opts =
        provider_opts
        |> Keyword.put(:prompt, Map.fetch!(config, :prompt))
        |> put_codex_materialization_opts(materialization)
        |> maybe_put_cli_path()

      {:ok,
       [
         provider: provider.name,
         profile: provider.core_profile,
         metadata: metadata
       ] ++
         execution_surface_opts(execution_config) ++
         provider_opts}
    end
  end

  defp effective_provider_opts(config, execution_config) when is_map(execution_config) do
    execution_environment = Execution.Config.to_execution_environment(execution_config)

    config
    |> Map.get(:provider_opts, [])
    |> maybe_put(:permission_mode, execution_environment.permission_mode)
    |> maybe_put(
      :provider_permission_mode,
      Map.get(execution_config, :provider_permission_mode)
    )
  end

  defp execution_surface_opts(%Execution.Config{} = execution_config) do
    [execution_surface: Execution.Config.to_execution_surface(execution_config)]
  end

  defp maybe_put_cli_path(provider_opts) do
    case Keyword.get(provider_opts, :cli_path) do
      path when is_binary(path) and path != "" ->
        provider_opts
        |> Keyword.put(:command, path)
        |> Keyword.delete(:cli_path)

      _ ->
        provider_opts
    end
  end

  defp put_codex_materialization_opts(provider_opts, nil), do: provider_opts

  defp put_codex_materialization_opts(provider_opts, %CodexMaterialization{} = materialization) do
    provider_opts
    |> Keyword.put(:command, materialization.command)
    |> Keyword.put(:cwd, materialization.cwd)
    |> Keyword.put(:env, materialization.env)
    |> Keyword.put(:clear_env?, materialization.clear_env?)
  end

  defp authorize_codex_materialization(%Provider{name: :codex}, config, provider_opts) do
    CodexMaterialization.authorize_config(config, provider_opts)
  end

  defp authorize_codex_materialization(%Provider{}, _config, _provider_opts), do: {:ok, nil}

  defp validate_approval_posture(execution_config) when is_map(execution_config) do
    if Execution.Config.to_execution_environment(execution_config).approval_posture == :none do
      {:error,
       Error.new(
         :config_invalid,
         :config,
         "approval_posture :none is not supported for runtime start"
       )}
    else
      :ok
    end
  end

  defp connect_remote(remote_cfg) do
    case NodeConnector.ensure_connected(remote_cfg) do
      :ok ->
        :ok

      {:error, reason} ->
        {:error, remote_error("remote connect failed: #{inspect(reason)}", reason)}
    end
  end

  defp preflight_remote(remote_cfg) do
    case NodeConnector.preflight(remote_cfg) do
      :ok ->
        :ok

      {:error, :remote_not_ready} ->
        if Map.get(remote_cfg, :remote_bootstrap_mode, :require_prestarted) == :ensure_started do
          :ok
        else
          {:error, remote_error("remote preflight failed: :remote_not_ready", :remote_not_ready)}
        end

      {:error, reason} ->
        {:error, remote_error("remote preflight failed: #{inspect(reason)}", reason)}
    end
  end

  defp maybe_bootstrap_remote(remote_cfg) do
    if Map.get(remote_cfg, :remote_bootstrap_mode, :require_prestarted) == :ensure_started do
      case :rpc.call(
             remote_cfg.remote_node,
             Application,
             :ensure_all_started,
             [:agent_session_manager],
             remote_cfg.remote_rpc_timeout_ms
           ) do
        {:ok, _apps} ->
          :ok

        {:error, reason} ->
          {:error, remote_error("remote bootstrap failed: #{inspect(reason)}", reason)}

        {:badrpc, reason} ->
          {:error, remote_error("remote bootstrap failed: #{inspect(reason)}", reason)}

        _other ->
          :ok
      end
    else
      :ok
    end
  end

  defp remote_start(remote_cfg, session_opts) do
    case :rpc.call(
           remote_cfg.remote_node,
           ASM.Remote.BackendStarter,
           :start_core_session,
           [session_opts],
           remote_cfg.remote_rpc_timeout_ms
         ) do
      {:ok, pid, info} when is_pid(pid) ->
        {:ok, pid, info}

      {:error, reason} ->
        {:error, remote_error("remote backend start failed: #{inspect(reason)}", reason)}

      {:badrpc, reason} ->
        {:error, remote_error("remote backend start failed: #{inspect(reason)}", reason)}

      other ->
        {:error, remote_error("remote backend start failed: #{inspect(other)}", other)}
    end
  end

  defp remote_error(message, cause) do
    Error.new(:connection_failed, :runtime, message, cause: cause)
  end

  defp provider_cli_error(provider, %ProviderCLIError{} = error) do
    Error.new(:cli_not_found, :provider, Exception.message(error),
      cause: error,
      provider: provider,
      retryable: false,
      recovery:
        RecoveryEnvelope.from_runtime_failure(%CliSubprocessCore.ProviderCLI.ErrorRuntimeFailure{
          kind: :cli_not_found,
          provider: provider,
          message: Exception.message(error),
          context: %{provider: provider},
          cause: error
        })
    )
  end

  defp core_capabilities(%Provider{core_profile: profile}) when is_atom(profile) do
    if function_exported?(profile, :capabilities, 0) do
      profile.capabilities()
    else
      []
    end
  end

  defp initial_subscribers(config) do
    case {Map.get(config, :subscription_ref), Map.get(config, :subscriber_pid)} do
      {ref, pid} when is_reference(ref) and is_pid(pid) -> %{ref => pid}
      _ -> %{}
    end
  end

  defp maybe_put(provider_opts, _key, nil), do: provider_opts
  defp maybe_put(provider_opts, key, value), do: Keyword.put(provider_opts, key, value)
end