utils.ex 4.58 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 9 10 11 12 13 14 15 16 17 18 19 20
  # 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)
    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
  def get_replied_to_activity(_), do: nil

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

27 28 29 30 31 32 33 34 35
  def to_for_user_and_mentions(user, mentions, inReplyTo) do
    default_to = [
      user.follower_address,
      "https://www.w3.org/ns/activitystreams#Public"
    ]

    to = default_to ++ Enum.map(mentions, fn ({_, %{ap_id: ap_id}}) -> ap_id end)
    if inReplyTo do
      Enum.uniq([inReplyTo.data["actor"] | to])
lain's avatar
lain committed
36
    else
37
      to
lain's avatar
lain committed
38 39 40
    end
  end

eal's avatar
eal committed
41
  def make_content_html(status, mentions, attachments, tags, no_attachment_links \\ false) do
42
    status
lain's avatar
lain committed
43
    |> format_input(mentions, tags)
eal's avatar
eal committed
44
    |> maybe_add_attachments(attachments, no_attachment_links)
45 46 47 48 49
  end

  def make_context(%Activity{data: %{"context" => context}}), do: context
  def make_context(_), do: Utils.generate_context_id

eal's avatar
eal committed
50 51 52 53
  def maybe_add_attachments(text, attachments, _no_links = true), do: text
  def maybe_add_attachments(text, attachments, _no_links) do
    add_attachments(text, attachments)
  end
lain's avatar
lain committed
54 55 56
  def add_attachments(text, attachments) do
    attachment_text = Enum.map(attachments, fn
      (%{"url" => [%{"href" => href} | _]}) ->
lain's avatar
lain committed
57 58
        name = URI.decode(Path.basename(href))
        "<a href=\"#{href}\" class='attachment'>#{shortname(name)}</a>"
lain's avatar
lain committed
59 60
      _ -> ""
    end)
eal's avatar
eal committed
61
    Enum.join([text | attachment_text], "<br>")
lain's avatar
lain committed
62 63
  end

Thog's avatar
Thog committed
64
  def format_input(text, mentions, _tags) do
eal's avatar
eal committed
65 66
    text
    |> Formatter.html_escape
lain's avatar
lain committed
67
    |> Formatter.linkify
eal's avatar
eal committed
68
    |> String.replace("\n", "<br>")
lain's avatar
lain committed
69
    |> add_user_links(mentions)
70
    # |> add_tag_links(tags)
lain's avatar
lain committed
71 72 73 74 75 76 77 78 79 80
  end

  def add_tag_links(text, tags) do
    tags = tags
    |> Enum.sort_by(fn ({tag, _}) -> -String.length(tag) end)

    Enum.reduce(tags, text, fn({full, tag}, text) ->
      url = "#<a href='#{Pleroma.Web.base_url}/tag/#{tag}' rel='tag'>#{tag}</a>"
      String.replace(text, full, url)
    end)
lain's avatar
lain committed
81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96
  end

  def add_user_links(text, mentions) do
    mentions = mentions
    |> Enum.sort_by(fn ({name, _}) -> -String.length(name) end)
    |> Enum.map(fn({name, user}) -> {name, user, Ecto.UUID.generate} end)

    # This replaces the mention with a unique reference first so it doesn't
    # contain parts of other replaced mentions. There probably is a better
    # solution for this...
    step_one = mentions
    |> Enum.reduce(text, fn ({match, _user, uuid}, text) ->
      String.replace(text, match, uuid)
    end)

    Enum.reduce(mentions, step_one, fn ({match, %User{ap_id: ap_id}, uuid}, text) ->
lain's avatar
lain committed
97
      short_match = String.split(match, "@") |> tl() |> hd()
eal's avatar
eal committed
98
      String.replace(text, uuid, "<span><a href='#{ap_id}'>@<span>#{short_match}</span></a></span>")
lain's avatar
lain committed
99 100 101
    end)
  end

lain's avatar
lain committed
102
  def make_note_data(actor, to, context, content_html, attachments, inReplyTo, tags, cw \\ nil) do
lain's avatar
lain committed
103 104 105 106
      object = %{
        "type" => "Note",
        "to" => to,
        "content" => content_html,
lain's avatar
lain committed
107
        "summary" => cw,
lain's avatar
lain committed
108 109
        "context" => context,
        "attachment" => attachments,
lain's avatar
lain committed
110 111
        "actor" => actor,
        "tag" => tags |> Enum.map(fn ({_, tag}) -> tag end)
lain's avatar
lain committed
112 113 114 115 116 117 118 119 120 121
      }

    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
122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137

  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
    with {:ok, date, _offset} <- date |> DateTime.from_iso8601 do
      format_asctime(date)
    else _e ->
        ""
    end
  end
138

139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155
  def to_masto_date(%NaiveDateTime{} = date) do
    date
    |> NaiveDateTime.to_iso8601
    |> String.replace(~r/(\.\d+)?$/, ".000Z", global: false)
  end

  def to_masto_date(date) do
    try do
      date
      |> NaiveDateTime.from_iso8601!
      |> NaiveDateTime.to_iso8601
      |> String.replace(~r/(\.\d+)?$/, ".000Z", global: false)
    rescue
      _e -> ""
    end
  end

156 157 158 159 160 161 162
  defp shortname(name) do
    if String.length(name) < 30 do
      name
    else
      String.slice(name, 0..30) <> "…"
    end
  end
lain's avatar
lain committed
163
end