git.squeep.com Git - akkoma/blob - lib/pleroma/web/websub/websub_controller.ex

   1 defmodule Pleroma.Web.Websub.WebsubController do
   2   use Pleroma.Web, :controller
   3   alias Pleroma.Web.Websub.WebsubServerSubscription
   4   alias Pleroma.{Repo, User}
   5   alias Pleroma.Web.OStatus
   6   def websub_subscription_request(conn, %{"nickname" => nickname} = params) do
   7     user = User.get_cached_by_nickname(nickname)
   8
   9     with {:ok, topic} <- valid_topic(params, user),
  10          {:ok, lease_time} <- lease_time(params),
  11          secret <- params["hub.secret"]
  12     do
  13       data = %{
  14         state: "requested",
  15         topic: topic,
  16         secret: secret
  17       }
  18
  19       change = Ecto.Changeset.change(%WebsubServerSubscription{}, data)
  20       websub = Repo.insert!(change)
  21
  22       change = Ecto.Changeset.change(websub, %{valid_until: NaiveDateTime.add(websub.inserted_at, lease_time)})
  23       websub = Repo.update!(change)
  24
  25       conn
  26       |> send_resp(202, "Accepted")
  27     else {:error, reason} ->
  28       conn
  29       |> send_resp(500, reason)
  30     end
  31   end
  32
  33   defp lease_time(%{"hub.lease_seconds" => lease_seconds}) do
  34     {:ok, lease_seconds}
  35   end
  36
  37   defp lease_time(_) do
  38     {:ok, 60 * 60 * 24 * 3} # three days
  39   end
  40
  41   defp valid_topic(%{"hub.topic" => topic}, user) do
  42     if topic == OStatus.feed_path(user) do
  43       {:ok, topic}
  44     else
  45       {:error, "Wrong topic requested, expected #{OStatus.feed_path(user)}, got #{topic}"}
  46     end
  47   end
  48 end