defmodule LightningWeb.WorkerSocket do
  use Phoenix.Socket

  alias Lightning.Workers
  # A Socket handler
  #
  # It's possible to control the websocket connection and
  # assign values that can be accessed by your channel topics.

  ## Channels
  # See the [`Channels guide`](https://hexdocs.pm/phoenix/channels.html)
  # for further details.

  channel "worker:queue", LightningWeb.WorkerChannel
  channel "run:*", LightningWeb.RunChannel

  # Socket params are passed from the client and can
  # be used to verify and authenticate a user. After
  # verification, you can put default assigns into
  # the socket that will be set for all channels, ie
  #
  #     {:ok, assign(socket, :user_id, verified_user_id)}
  #
  # To deny connection, return `:error` or `{:error, term}`. To control the
  # response the client receives in that case, [define an error handler in the
  # websocket
  # configuration](https://hexdocs.pm/phoenix/Phoenix.Endpoint.html#socket/3-websocket-configuration).
  #
  # See `Phoenix.Token` documentation for examples in
  # performing token verification on connect.
  @impl true
  def connect(params, socket, _connect_info) do
    with %{"token" => token} <- params,
         {:ok, claims} <- Workers.verify_worker_token(token) do
      {:ok,
       socket
       |> assign(
         claims: claims,
         token: token,
         worker_version: params["worker_version"],
         api_version: params["api_version"]
       )}
    else
      _ ->
        {:error, :unauthorized}
    end
  end

  # Socket id's are topics that allow you to identify all sockets for a given user:
  #
  #     def id(socket), do: "user_socket:#{socket.assigns.user_id}"
  #
  # Would allow you to broadcast a "disconnect" event and terminate
  # all active sockets and channels for a given user:
  #
  #     Elixir.LightningWeb.Endpoint.broadcast("user_socket:#{user.id}", "disconnect", %{})
  #
  # Returning `nil` makes this socket anonymous.
  @impl true
  def id(_socket), do: nil

  def handle_error(conn, :unauthorized) do
    Plug.Conn.send_resp(conn, 401, "Unauthorized")
  end
end
