defmodule API.Gateway.WSReverseProxy.AAA do
  @moduledoc """
  Callback module to proxy WebSocket connections to the AAA microservice.
  """

  use API.Gateway.WSReverseProxy.CallbackModule

  alias API.Gateway.Config

  @impl API.Gateway.WSReverseProxy.CallbackModule
  def websocket_endpoint(_req, _opts) do
    Config.aaa_base_url()
    |> sub_scheme()
    |> URI.merge("/aaa/websocket")
  end

  @impl API.Gateway.WSReverseProxy.CallbackModule
  def conn_options(req, _opts) do
    case :cowboy_req.header("cookie", req) do
      :undefined ->
        []

      cookie ->
        [extra_headers: [{"cookie", cookie}]]
    end
  end
end