utils.ex 4.73 KB
Newer Older
1
defmodule Pleroma.Web.CommonAPI.Utils do
lain's avatar
lain committed
2 3
  alias Pleroma.{Repo, Object, Formatter, User, Activity}
  alias Pleroma.Web.ActivityPub.Utils
dtluna's avatar
dtluna committed
4
  alias Calendar.Strftime
lain's avatar
lain committed
5

6 7 8
  # This is a hack for twidere.
  def get_by_id_or_ap_id(id) do
    activity = Repo.get(Activity, id) || Activity.get_create_activity_by_object_ap_id(id)
lain's avatar
lain committed
9

10 11 12 13 14 15 16 17 18 19
    if activity.data["type"] == "Create" do
      activity
    else
      Activity.get_create_activity_by_object_ap_id(activity.data["object"])
    end
  end

  def get_replied_to_activity(id) when not is_nil(id) do
    Repo.get(Activity, id)
  end
lain's avatar
lain committed
20

21 22
  def get_replied_to_activity(_), do: nil

lain's avatar
lain committed
23
  def attachments_from_ids(ids) do
lain's avatar
lain committed
24
    Enum.map(ids || [], fn media_id ->
lain's avatar
lain committed
25 26 27 28
      Repo.get(Object, media_id).data
    end)
  end

29 30
  def to_for_user_and_mentions(user, mentions, inReplyTo, "public") do
    to = ["https://www.w3.org/ns/activitystreams#Public"]
31

lain's avatar
lain committed
32
    mentioned_users = Enum.map(mentions, fn {_, %{ap_id: ap_id}} -> ap_id end)
33
    cc = [user.follower_address | mentioned_users]
lain's avatar
lain committed
34

35
    if inReplyTo do
36
      {to, Enum.uniq([inReplyTo.data["actor"] | cc])}
lain's avatar
lain committed
37
    else
38 39 40 41 42 43 44 45 46 47 48 49 50 51 52
      {to, cc}
    end
  end

  def to_for_user_and_mentions(user, mentions, inReplyTo, "unlisted") do
    {to, cc} = to_for_user_and_mentions(user, mentions, inReplyTo, "public")
    {cc, to}
  end

  def to_for_user_and_mentions(user, mentions, inReplyTo, "private") do
    {to, cc} = to_for_user_and_mentions(user, mentions, inReplyTo, "direct")
    {[user.follower_address | to], cc}
  end

  def to_for_user_and_mentions(user, mentions, inReplyTo, "direct") do
lain's avatar
lain committed
53 54
    mentioned_users = Enum.map(mentions, fn {_, %{ap_id: ap_id}} -> ap_id end)

55 56 57 58
    if inReplyTo do
      {Enum.uniq([inReplyTo.data["actor"] | mentioned_users]), []}
    else
      {mentioned_users, []}
lain's avatar
lain committed
59 60 61
    end
  end

eal's avatar
eal committed
62
  def make_content_html(status, mentions, attachments, tags, no_attachment_links \\ false) do
63
    status
64
    |> String.replace("\r", "")
lain's avatar
lain committed
65
    |> format_input(mentions, tags)
eal's avatar
eal committed
66
    |> maybe_add_attachments(attachments, no_attachment_links)
67 68 69
  end

  def make_context(%Activity{data: %{"context" => context}}), do: context
lain's avatar
lain committed
70
  def make_context(_), do: Utils.generate_context_id()
71

eal's avatar
eal committed
72
  def maybe_add_attachments(text, attachments, _no_links = true), do: text
lain's avatar
lain committed
73

eal's avatar
eal committed
74 75 76
  def maybe_add_attachments(text, attachments, _no_links) do
    add_attachments(text, attachments)
  end
lain's avatar
lain committed
77

lain's avatar
lain committed
78
  def add_attachments(text, attachments) do
lain's avatar
lain committed
79 80 81 82 83 84 85 86 87 88
    attachment_text =
      Enum.map(attachments, fn
        %{"url" => [%{"href" => href} | _]} ->
          name = URI.decode(Path.basename(href))
          "<a href=\"#{href}\" class='attachment'>#{shortname(name)}</a>"

        _ ->
          ""
      end)

eal's avatar
eal committed
89
    Enum.join([text | attachment_text], "<br>")
lain's avatar
lain committed
90 91
  end

lain's avatar
lain committed
92
  def format_input(text, mentions, tags) do
eal's avatar
eal committed
93
    text
lain's avatar
lain committed
94
    |> Formatter.html_escape()
eal's avatar
eal committed
95
    |> String.replace("\n", "<br>")
lain's avatar
lain committed
96 97
    |> (&{[], &1}).()
    |> Formatter.add_links()
lain's avatar
lain committed
98 99
    |> Formatter.add_user_links(mentions)
    |> Formatter.add_hashtag_links(tags)
lain's avatar
lain committed
100
    |> Formatter.finalize()
lain's avatar
lain committed
101 102 103
  end

  def add_tag_links(text, tags) do
lain's avatar
lain committed
104 105 106
    tags =
      tags
      |> Enum.sort_by(fn {tag, _} -> -String.length(tag) end)
lain's avatar
lain committed
107

lain's avatar
lain committed
108 109
    Enum.reduce(tags, text, fn {full, tag}, text ->
      url = "#<a href='#{Pleroma.Web.base_url()}/tag/#{tag}' rel='tag'>#{tag}</a>"
lain's avatar
lain committed
110 111
      String.replace(text, full, url)
    end)
lain's avatar
lain committed
112 113
  end

lain's avatar
lain committed
114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135
  def make_note_data(
        actor,
        to,
        context,
        content_html,
        attachments,
        inReplyTo,
        tags,
        cw \\ nil,
        cc \\ []
      ) do
    object = %{
      "type" => "Note",
      "to" => to,
      "cc" => cc,
      "content" => content_html,
      "summary" => cw,
      "context" => context,
      "attachment" => attachments,
      "actor" => actor,
      "tag" => tags |> Enum.map(fn {_, tag} -> tag end)
    }
lain's avatar
lain committed
136 137 138 139 140 141 142 143 144

    if inReplyTo do
      object
      |> Map.put("inReplyTo", inReplyTo.data["object"]["id"])
      |> Map.put("inReplyToStatusId", inReplyTo.id)
    else
      object
    end
  end
dtluna's avatar
dtluna committed
145 146 147 148 149 150 151 152 153 154

  def format_naive_asctime(date) do
    date |> DateTime.from_naive!("Etc/UTC") |> format_asctime
  end

  def format_asctime(date) do
    Strftime.strftime!(date, "%a %b %d %H:%M:%S %z %Y")
  end

  def date_to_asctime(date) do
lain's avatar
lain committed
155
    with {:ok, date, _offset} <- date |> DateTime.from_iso8601() do
dtluna's avatar
dtluna committed
156
      format_asctime(date)
lain's avatar
lain committed
157 158
    else
      _e ->
dtluna's avatar
dtluna committed
159 160 161
        ""
    end
  end
162

163 164
  def to_masto_date(%NaiveDateTime{} = date) do
    date
lain's avatar
lain committed
165
    |> NaiveDateTime.to_iso8601()
166 167 168 169 170 171
    |> String.replace(~r/(\.\d+)?$/, ".000Z", global: false)
  end

  def to_masto_date(date) do
    try do
      date
lain's avatar
lain committed
172 173
      |> NaiveDateTime.from_iso8601!()
      |> NaiveDateTime.to_iso8601()
174 175 176 177 178 179
      |> String.replace(~r/(\.\d+)?$/, ".000Z", global: false)
    rescue
      _e -> ""
    end
  end

180 181 182 183 184 185 186
  defp shortname(name) do
    if String.length(name) < 30 do
      name
    else
      String.slice(name, 0..30) <> "…"
    end
  end
lain's avatar
lain committed
187
end