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(%{"access_token" => token} = params, socket) do
  15     with %Token{user_id: user_id} <- Repo.get_by(Token, token: token),
  16          %User{} = user <- Repo.get(User, user_id),
  17          stream
  18          when stream in [
  19                 "public",
  20                 "public:local",
  21                 "public:media",
  22                 "public:local:media",
  23                 "user",
  24                 "direct",
  25                 "list",
  26                 "hashtag"
  27               ] <- params["stream"] do
  28       topic =
  29         case stream do
  30           "hashtag" -> "hashtag:#{params["tag"]}"
  31           "list" -> "list:#{params["list"]}"
  32           _ -> stream
  33         end
  34
  35       socket =
  36         socket
  37         |> assign(:topic, topic)
  38         |> assign(:user, user)
  39
  40       Pleroma.Web.Streamer.add_socket(topic, socket)
  41       {:ok, socket}
  42     else
  43       _e -> :error
  44     end
  45   end
  46
  47   def connect(%{"stream" => stream} = params, socket)
  48       when stream in ["public", "public:local", "hashtag"] do
  49     topic =
  50       case stream do
  51         "hashtag" -> "hashtag:#{params["tag"]}"
  52         _ -> stream
  53       end
  54
  55     with socket =
  56            socket
  57            |> assign(:topic, topic) do
  58       Pleroma.Web.Streamer.add_socket(topic, socket)
  59       {:ok, socket}
  60     else
  61       _e -> :error
  62     end
  63   end
  64
  65   def id(_), do: nil
  66
  67   def handle(:text, message, _state) do
  68     # | :ok
  69     # | state
  70     # | {:text, message}
  71     # | {:text, message, state}
  72     # | {:close, "Goodbye!"}
  73     {:text, message}
  74   end
  75
  76   def handle(:closed, _, %{socket: socket}) do
  77     topic = socket.assigns[:topic]
  78     Pleroma.Web.Streamer.remove_socket(topic, socket)
  79   end
  80 end