git.squeep.com Git - akkoma/blob - lib/pleroma/web/channels/user_socket.ex

   1 defmodule Pleroma.Web.UserSocket do
   2   use Phoenix.Socket
   3   alias Pleroma.User
   4   alias Comeonin.Pbkdf2
   5
   6   ## Channels
   7   # channel "room:*", Pleroma.Web.RoomChannel
   8   if Application.get_env(:pleroma, :chat) |> Keyword.get(:enabled) do
   9     channel "chat:*", Pleroma.Web.ChatChannel
  10   end
  11
  12   ## Transports
  13   transport :websocket, Phoenix.Transports.WebSocket
  14   # transport :longpoll, Phoenix.Transports.LongPoll
  15
  16   # Socket params are passed from the client and can
  17   # be used to verify and authenticate a user. After
  18   # verification, you can put default assigns into
  19   # the socket that will be set for all channels, ie
  20   #
  21   #     {:ok, assign(socket, :user_id, verified_user_id)}
  22   #
  23   # To deny connection, return `:error`.
  24   #
  25   # See `Phoenix.Token` documentation for examples in
  26   # performing token verification on connect.
  27   def connect(%{"token" => token}, socket) do
  28     with {:ok, user_id} <- Phoenix.Token.verify(socket, "user socket", token, max_age: 84600),
  29          %User{} = user <- Pleroma.Repo.get(User, user_id) do
  30       {:ok, assign(socket, :user_name, user.nickname)}
  31     else
  32       _e -> :error
  33     end
  34   end
  35
  36   # Socket id's are topics that allow you to identify all sockets for a given user:
  37   #
  38   #     def id(socket), do: "user_socket:#{socket.assigns.user_id}"
  39   #
  40   # Would allow you to broadcast a "disconnect" event and terminate
  41   # all active sockets and channels for a given user:
  42   #
  43   #     Pleroma.Web.Endpoint.broadcast("user_socket:#{user.id}", "disconnect", %{})
  44   #
  45   # Returning `nil` makes this socket anonymous.
  46   def id(_socket), do: nil
  47 end