akkoma/lib/pleroma/web/websub/websub_controller.ex

85 lines
2.3 KiB
Elixir
Raw Normal View History

2017-04-20 15:47:33 +00:00
defmodule Pleroma.Web.Websub.WebsubController do
use Pleroma.Web, :controller
2017-04-28 13:45:10 +00:00
alias Pleroma.{Repo, User}
alias Pleroma.Web.{Websub, Federator}
2017-04-28 13:45:10 +00:00
alias Pleroma.Web.Websub.WebsubClientSubscription
require Logger
2018-11-05 14:19:03 +00:00
plug(
Pleroma.Web.FederatingPlug
when action in [
:websub_subscription_request,
:websub_subscription_confirmation,
:websub_incoming
]
)
2017-04-20 15:47:33 +00:00
def websub_subscription_request(conn, %{"nickname" => nickname} = params) do
user = User.get_cached_by_nickname(nickname)
2018-03-30 13:01:53 +00:00
with {:ok, _websub} <- Websub.incoming_subscription_request(user, params) do
2017-04-20 15:47:33 +00:00
conn
|> send_resp(202, "Accepted")
2018-03-30 13:01:53 +00:00
else
{:error, reason} ->
conn
|> send_resp(500, reason)
2017-04-20 15:47:33 +00:00
end
end
2017-05-10 16:45:55 +00:00
# TODO: Extract this into the Websub module
2018-03-30 13:01:53 +00:00
def websub_subscription_confirmation(
conn,
%{
"id" => id,
"hub.mode" => "subscribe",
"hub.challenge" => challenge,
"hub.topic" => topic
} = params
) do
2018-03-19 17:41:04 +00:00
Logger.debug("Got WebSub confirmation")
2017-08-02 10:03:45 +00:00
Logger.debug(inspect(params))
2018-03-30 13:01:53 +00:00
lease_seconds =
if params["hub.lease_seconds"] do
String.to_integer(params["hub.lease_seconds"])
else
# Guess 3 days
60 * 60 * 24 * 3
end
with %WebsubClientSubscription{} = websub <-
Repo.get_by(WebsubClientSubscription, id: id, topic: topic) do
valid_until = NaiveDateTime.add(NaiveDateTime.utc_now(), lease_seconds)
2017-05-10 16:45:55 +00:00
change = Ecto.Changeset.change(websub, %{state: "accepted", valid_until: valid_until})
2017-04-28 13:45:10 +00:00
{:ok, _websub} = Repo.update(change)
2018-03-30 13:01:53 +00:00
2017-04-28 13:45:10 +00:00
conn
|> send_resp(200, challenge)
2018-03-30 13:01:53 +00:00
else
_e ->
conn
|> send_resp(500, "Error")
2017-04-28 13:45:10 +00:00
end
end
def websub_incoming(conn, %{"id" => id}) do
with "sha1=" <> signature <- hd(get_req_header(conn, "x-hub-signature")),
2017-05-03 18:06:20 +00:00
signature <- String.downcase(signature),
2017-04-28 13:45:10 +00:00
%WebsubClientSubscription{} = websub <- Repo.get(WebsubClientSubscription, id),
{:ok, body, _conn} = read_body(conn),
^signature <- Websub.sign(websub.secret, body) do
Federator.enqueue(:incoming_doc, body)
2018-03-30 13:01:53 +00:00
2017-04-28 13:45:10 +00:00
conn
|> send_resp(200, "OK")
2018-03-30 13:01:53 +00:00
else
_e ->
Logger.debug("Can't handle incoming subscription post")
conn
|> send_resp(500, "Error")
2017-04-28 13:45:10 +00:00
end
end
2017-04-20 15:47:33 +00:00
end