git.squeep.com Git - akkoma/blob - lib/pleroma/web/mastodon_api/mastodon_socket.ex

   1 defmodule Pleroma.Web.MastodonAPI.MastodonSocket do
   2   use Phoenix.Socket
   3
   4   alias Pleroma.Web.OAuth.Token
   5   alias Pleroma.{User, Repo}
   6
   7   transport(
   8     :streaming,
   9     Phoenix.Transports.WebSocket.Raw,
  10     # We never receive data.
  11     timeout: :infinity
  12   )
  13
  14   def connect(params, socket) do
  15     with token when not is_nil(token) <- params["access_token"],
  16          %Token{user_id: user_id} <- Repo.get_by(Token, token: token),
  17          %User{} = user <- Repo.get(User, user_id),
  18          stream
  19          when stream in [
  20                 "public",
  21                 "public:local",
  22                 "public:media",
  23                 "public:local:media",
  24                 "user",
  25                 "direct",
  26                 "list",
  27                 "hashtag"
  28               ] <- params["stream"] do
  29       topic = if stream == "list", do: "list:#{params["list"]}", else: stream
  30       socket_stream = if stream == "hashtag", do: "hashtag:#{params["tag"]}", else: stream
  31
  32       socket =
  33         socket
  34         |> assign(:topic, topic)
  35         |> assign(:user, user)
  36
  37       Pleroma.Web.Streamer.add_socket(socket_stream, socket)
  38       {:ok, socket}
  39     else
  40       _e -> :error
  41     end
  42   end
  43
  44   def id(_), do: nil
  45
  46   def handle(:text, message, _state) do
  47     # | :ok
  48     # | state
  49     # | {:text, message}
  50     # | {:text, message, state}
  51     # | {:close, "Goodbye!"}
  52     {:text, message}
  53   end
  54
  55   def handle(:closed, _, %{socket: socket}) do
  56     topic = socket.assigns[:topic]
  57     Pleroma.Web.Streamer.remove_socket(topic, socket)
  58   end
  59 end