Pleroma/lib/pleroma/web/mastodon_api/mastodon_socket.ex

82 lines
1.9 KiB
Elixir
Raw Normal View History

2017-11-11 14:59:25 +01:00
defmodule Pleroma.Web.MastodonAPI.MastodonSocket do
use Phoenix.Socket
2017-11-16 13:48:58 +01:00
alias Pleroma.Web.OAuth.Token
alias Pleroma.{User, Repo}
2018-03-30 15:01:53 +02:00
transport(
2018-12-09 10:12:48 +01:00
:websocket,
2018-03-30 15:01:53 +02:00
Phoenix.Transports.WebSocket.Raw,
# We never receive data.
timeout: :infinity
)
2017-11-11 14:59:25 +01:00
2018-12-10 07:39:57 +01:00
@spec connect(params :: map(), Phoenix.Socket.t()) :: {:ok, Phoenix.Socket.t()} | :error
def connect(%{"access_token" => token} = params, socket) do
with %Token{user_id: user_id} <- Repo.get_by(Token, token: token),
2017-11-16 13:48:58 +01:00
%User{} = user <- Repo.get(User, user_id),
stream
when stream in [
"public",
"public:local",
"public:media",
"public:local:media",
"user",
"direct",
"list",
"hashtag"
] <- params["stream"] do
2018-11-11 04:33:14 +01:00
topic =
case stream do
"hashtag" -> "hashtag:#{params["tag"]}"
"list" -> "list:#{params["list"]}"
_ -> stream
end
2018-05-30 15:33:37 +02:00
2018-03-30 15:01:53 +02:00
socket =
socket
2018-05-30 15:33:37 +02:00
|> assign(:topic, topic)
2018-03-30 15:01:53 +02:00
|> assign(:user, user)
2018-11-11 04:33:14 +01:00
Pleroma.Web.Streamer.add_socket(topic, socket)
{:ok, socket}
else
2017-11-16 13:48:58 +01:00
_e -> :error
end
2017-11-11 14:59:25 +01:00
end
def connect(%{"stream" => stream} = params, socket)
when stream in ["public", "public:local", "hashtag"] do
topic =
case stream do
"hashtag" -> "hashtag:#{params["tag"]}"
_ -> stream
end
2018-12-10 07:39:57 +01:00
socket =
socket
|> assign(:topic, topic)
Pleroma.Web.Streamer.add_socket(topic, socket)
{:ok, socket}
end
2018-12-10 07:39:57 +01:00
def connect(_params, _socket), do: :error
2017-11-19 02:22:07 +01:00
def id(_), do: nil
2017-11-11 14:59:25 +01:00
2017-11-19 02:22:07 +01:00
def handle(:text, message, _state) do
2018-03-30 15:01:53 +02:00
# | :ok
# | state
# | {:text, message}
# | {:text, message, state}
# | {:close, "Goodbye!"}
2017-11-11 14:59:25 +01:00
{:text, message}
end
2017-11-19 02:22:07 +01:00
def handle(:closed, _, %{socket: socket}) do
topic = socket.assigns[:topic]
Pleroma.Web.Streamer.remove_socket(topic, socket)
2017-11-11 14:59:25 +01:00
end
end