git.squeep.com Git - akkoma/blob - lib/pleroma/web/ostatus/ostatus_controller.ex

   1 defmodule Pleroma.Web.OStatus.OStatusController do
   2   use Pleroma.Web, :controller
   3
   4   alias Pleroma.{User, Activity}
   5   alias Pleroma.Web.OStatus.{FeedRepresenter, ActivityRepresenter}
   6   alias Pleroma.Repo
   7   alias Pleroma.Web.{OStatus, Federator}
   8   alias Pleroma.Web.XML
   9   import Ecto.Query
  10
  11   def feed_redirect(conn, %{"nickname" => nickname}) do
  12     user = User.get_cached_by_nickname(nickname)
  13
  14     case get_format(conn) do
  15       "html" -> Fallback.RedirectController.redirector(conn, nil)
  16       _ -> redirect conn, external: OStatus.feed_path(user)
  17     end
  18   end
  19
  20   def feed(conn, %{"nickname" => nickname}) do
  21     user = User.get_cached_by_nickname(nickname)
  22     query = from activity in Activity,
  23       where: fragment("?->>'actor' = ?", activity.data, ^user.ap_id),
  24       limit: 20,
  25       order_by: [desc: :id]
  26
  27     activities = query
  28     |> Repo.all
  29
  30     response = user
  31     |> FeedRepresenter.to_simple_form(activities, [user])
  32     |> :xmerl.export_simple(:xmerl_xml)
  33     |> to_string
  34
  35     conn
  36     |> put_resp_content_type("application/atom+xml")
  37     |> send_resp(200, response)
  38   end
  39
  40   defp decode_or_retry(body) do
  41     with {:ok, magic_key} <- Pleroma.Web.Salmon.fetch_magic_key(body),
  42          {:ok, doc} <- Pleroma.Web.Salmon.decode_and_validate(magic_key, body) do
  43       {:ok, doc}
  44     else
  45       _e ->
  46         with [decoded | _] <- Pleroma.Web.Salmon.decode(body),
  47              doc <- XML.parse_document(decoded),
  48              uri when not is_nil(uri) <- XML.string_from_xpath("/entry/author[1]/uri", doc),
  49              {:ok, _} <- Pleroma.Web.OStatus.make_user(uri, true),
  50              {:ok, magic_key} <- Pleroma.Web.Salmon.fetch_magic_key(body),
  51              {:ok, doc} <- Pleroma.Web.Salmon.decode_and_validate(magic_key, body) do
  52           {:ok, doc}
  53         end
  54     end
  55   end
  56
  57   def salmon_incoming(conn, _) do
  58     {:ok, body, _conn} = read_body(conn)
  59     {:ok, doc} = decode_or_retry(body)
  60
  61     Federator.enqueue(:incoming_doc, doc)
  62
  63     conn
  64     |> send_resp(200, "")
  65   end
  66
  67   def object(conn, %{"uuid" => uuid}) do
  68     with id <- o_status_url(conn, :object, uuid),
  69          %Activity{} = activity <- Activity.get_create_activity_by_object_ap_id(id),
  70          %User{} = user <- User.get_cached_by_ap_id(activity.data["actor"]) do
  71       case get_format(conn) do
  72         "html" -> redirect(conn, to: "/notice/#{activity.id}")
  73         _ -> represent_activity(conn, activity, user)
  74       end
  75     end
  76   end
  77
  78   def activity(conn, %{"uuid" => uuid}) do
  79     with id <- o_status_url(conn, :activity, uuid),
  80          %Activity{} = activity <- Activity.get_by_ap_id(id),
  81          %User{} = user <- User.get_cached_by_ap_id(activity.data["actor"]) do
  82       case get_format(conn) do
  83         "html" -> redirect(conn, to: "/notice/#{activity.id}")
  84         _ -> represent_activity(conn, activity, user)
  85       end
  86     end
  87   end
  88
  89   defp represent_activity(conn, activity, user) do
  90     response = activity
  91     |> ActivityRepresenter.to_simple_form(user, true)
  92     |> ActivityRepresenter.wrap_with_entry
  93     |> :xmerl.export_simple(:xmerl_xml)
  94     |> to_string
  95
  96     conn
  97     |> put_resp_content_type("application/atom+xml")
  98     |> send_resp(200, response)
  99   end
 100 end