utils.ex 7.77 KB
Newer Older
1
# Pleroma: A lightweight social networking server
kaniini's avatar
kaniini committed
2
# Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
3 4
# SPDX-License-Identifier: AGPL-3.0-only

5
defmodule Pleroma.Web.CommonAPI.Utils do
Maksim's avatar
Maksim committed
6 7 8
  alias Calendar.Strftime
  alias Comeonin.Pbkdf2
  alias Pleroma.{Activity, Formatter, Object, Repo}
Haelwenn's avatar
Haelwenn committed
9
  alias Pleroma.{User, Web}
lain's avatar
lain committed
10
  alias Pleroma.Web.ActivityPub.Utils
11
  alias Pleroma.Web.Endpoint
12
  alias Pleroma.Web.MediaProxy
lain's avatar
lain committed
13

14 15
  # This is a hack for twidere.
  def get_by_id_or_ap_id(id) do
16
    activity = Repo.get(Activity, id) || Activity.get_create_by_object_ap_id(id)
lain's avatar
lain committed
17

18 19 20 21
    activity &&
      if activity.data["type"] == "Create" do
        activity
      else
22
        Activity.get_create_by_object_ap_id(activity.data["object"])
23
      end
24 25
  end

26 27
  def get_replied_to_activity(""), do: nil

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

32 33
  def get_replied_to_activity(_), do: nil

lain's avatar
lain committed
34
  def attachments_from_ids(ids) do
lain's avatar
lain committed
35
    Enum.map(ids || [], fn media_id ->
lain's avatar
lain committed
36 37 38 39
      Repo.get(Object, media_id).data
    end)
  end

40
  def to_for_user_and_mentions(user, mentions, inReplyTo, "public") do
lain's avatar
lain committed
41
    mentioned_users = Enum.map(mentions, fn {_, %{ap_id: ap_id}} -> ap_id end)
42 43 44

    to = ["https://www.w3.org/ns/activitystreams#Public" | mentioned_users]
    cc = [user.follower_address]
lain's avatar
lain committed
45

46
    if inReplyTo do
47
      {Enum.uniq([inReplyTo.data["actor"] | to]), cc}
lain's avatar
lain committed
48
    else
49 50 51 52 53
      {to, cc}
    end
  end

  def to_for_user_and_mentions(user, mentions, inReplyTo, "unlisted") do
54 55 56 57 58 59 60 61 62 63
    mentioned_users = Enum.map(mentions, fn {_, %{ap_id: ap_id}} -> ap_id end)

    to = [user.follower_address | mentioned_users]
    cc = ["https://www.w3.org/ns/activitystreams#Public"]

    if inReplyTo do
      {Enum.uniq([inReplyTo.data["actor"] | to]), cc}
    else
      {to, cc}
    end
64 65 66 67 68 69 70
  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

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

74 75 76 77
    if inReplyTo do
      {Enum.uniq([inReplyTo.data["actor"] | mentioned_users]), []}
    else
      {mentioned_users, []}
lain's avatar
lain committed
78 79 80
    end
  end

81 82 83 84 85 86 87 88
  def make_content_html(
        status,
        mentions,
        attachments,
        tags,
        content_type,
        no_attachment_links \\ false
      ) do
89
    status
90
    |> format_input(mentions, tags, content_type)
eal's avatar
eal committed
91
    |> maybe_add_attachments(attachments, no_attachment_links)
92 93 94
  end

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

97
  def maybe_add_attachments(text, _attachments, _no_links = true), do: text
lain's avatar
lain committed
98

eal's avatar
eal committed
99 100 101
  def maybe_add_attachments(text, attachments, _no_links) do
    add_attachments(text, attachments)
  end
lain's avatar
lain committed
102

lain's avatar
lain committed
103
  def add_attachments(text, attachments) do
lain's avatar
lain committed
104 105
    attachment_text =
      Enum.map(attachments, fn
106 107
        %{"url" => [%{"href" => href} | _]} = attachment ->
          name = attachment["name"] || URI.decode(Path.basename(href))
108
          href = MediaProxy.url(href)
lain's avatar
lain committed
109 110 111 112 113 114
          "<a href=\"#{href}\" class='attachment'>#{shortname(name)}</a>"

        _ ->
          ""
      end)

115
    Enum.join([text | attachment_text], "<br>")
lain's avatar
lain committed
116 117
  end

118 119
  def format_input(text, mentions, tags, format, options \\ [])

Maksim's avatar
Maksim committed
120 121 122
  @doc """
  Formatting text to plain text.
  """
123
  def format_input(text, mentions, tags, "text/plain", options) do
eal's avatar
eal committed
124
    text
125
    |> Formatter.html_escape("text/plain")
126
    |> String.replace(~r/\r?\n/, "<br>")
lain's avatar
lain committed
127 128
    |> (&{[], &1}).()
    |> Formatter.add_links()
129
    |> Formatter.add_user_links(mentions, options[:user_links] || [])
lain's avatar
lain committed
130
    |> Formatter.add_hashtag_links(tags)
lain's avatar
lain committed
131
    |> Formatter.finalize()
lain's avatar
lain committed
132 133
  end

Maksim's avatar
Maksim committed
134 135 136
  @doc """
  Formatting text to html.
  """
137
  def format_input(text, mentions, _tags, "text/html", options) do
138 139 140
    text
    |> Formatter.html_escape("text/html")
    |> (&{[], &1}).()
141
    |> Formatter.add_user_links(mentions, options[:user_links] || [])
142 143 144
    |> Formatter.finalize()
  end

Maksim's avatar
Maksim committed
145 146 147
  @doc """
  Formatting text to markdown.
  """
148
  def format_input(text, mentions, tags, "text/markdown", options) do
149
    text
Maksim's avatar
Maksim committed
150
    |> Formatter.mentions_escape(mentions)
151 152 153
    |> Earmark.as_html!()
    |> Formatter.html_escape("text/html")
    |> (&{[], &1}).()
154
    |> Formatter.add_user_links(mentions, options[:user_links] || [])
155
    |> Formatter.add_hashtag_links(tags)
156 157 158
    |> Formatter.finalize()
  end

lain's avatar
lain committed
159
  def add_tag_links(text, tags) do
lain's avatar
lain committed
160 161 162
    tags =
      tags
      |> Enum.sort_by(fn {tag, _} -> -String.length(tag) end)
lain's avatar
lain committed
163

lain's avatar
lain committed
164
    Enum.reduce(tags, text, fn {full, tag}, text ->
Maksim's avatar
Maksim committed
165
      url = "<a href='#{Web.base_url()}/tag/#{tag}' rel='tag'>##{tag}</a>"
lain's avatar
lain committed
166 167
      String.replace(text, full, url)
    end)
lain's avatar
lain committed
168 169
  end

lain's avatar
lain committed
170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189
  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,
190
      "tag" => tags |> Enum.map(fn {_, tag} -> tag end) |> Enum.uniq()
lain's avatar
lain committed
191
    }
lain's avatar
lain committed
192 193 194 195 196 197 198 199 200

    if inReplyTo do
      object
      |> Map.put("inReplyTo", inReplyTo.data["object"]["id"])
      |> Map.put("inReplyToStatusId", inReplyTo.id)
    else
      object
    end
  end
201 202 203 204 205 206 207 208 209 210

  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
211
    with {:ok, date, _offset} <- date |> DateTime.from_iso8601() do
212
      format_asctime(date)
lain's avatar
lain committed
213 214
    else
      _e ->
215 216 217
        ""
    end
  end
218

219 220
  def to_masto_date(%NaiveDateTime{} = date) do
    date
lain's avatar
lain committed
221
    |> NaiveDateTime.to_iso8601()
222 223 224 225 226 227
    |> String.replace(~r/(\.\d+)?$/, ".000Z", global: false)
  end

  def to_masto_date(date) do
    try do
      date
lain's avatar
lain committed
228 229
      |> NaiveDateTime.from_iso8601!()
      |> NaiveDateTime.to_iso8601()
230 231 232 233 234 235
      |> String.replace(~r/(\.\d+)?$/, ".000Z", global: false)
    rescue
      _e -> ""
    end
  end

236 237 238 239 240 241 242
  defp shortname(name) do
    if String.length(name) < 30 do
      name
    else
      String.slice(name, 0..30) <> "…"
    end
  end
243

244
  def confirm_current_password(user, password) do
245
    with %User{local: true} = db_user <- Repo.get(User, user.id),
246
         true <- Pbkdf2.checkpw(password, db_user.password_hash) do
247 248 249
      {:ok, db_user}
    else
      _ -> {:error, "Invalid password."}
250 251
    end
  end
252

Maksim's avatar
Maksim committed
253
  def emoji_from_profile(%{info: _info} = user) do
254 255 256 257
    (Formatter.get_emoji(user.bio) ++ Formatter.get_emoji(user.name))
    |> Enum.map(fn {shortcode, url} ->
      %{
        "type" => "Emoji",
eal's avatar
eal committed
258
        "icon" => %{"type" => "Image", "url" => "#{Endpoint.url()}#{url}"},
259 260 261 262
        "name" => ":#{shortcode}:"
      }
    end)
  end
263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304

  def maybe_notify_to_recipients(
        recipients,
        %Activity{data: %{"to" => to, "type" => _type}} = _activity
      ) do
    recipients ++ to
  end

  def maybe_notify_mentioned_recipients(
        recipients,
        %Activity{data: %{"to" => _to, "type" => type} = data} = _activity
      )
      when type == "Create" do
    object = Object.normalize(data["object"])

    object_data =
      cond do
        !is_nil(object) ->
          object.data

        is_map(data["object"]) ->
          data["object"]

        true ->
          %{}
      end

    tagged_mentions = maybe_extract_mentions(object_data)

    recipients ++ tagged_mentions
  end

  def maybe_notify_mentioned_recipients(recipients, _), do: recipients

  def maybe_extract_mentions(%{"tag" => tag}) do
    tag
    |> Enum.filter(fn x -> is_map(x) end)
    |> Enum.filter(fn x -> x["type"] == "Mention" end)
    |> Enum.map(fn x -> x["href"] end)
  end

  def maybe_extract_mentions(_), do: []
lain's avatar
lain committed
305
end