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