distraction.party/lib/pleroma/web/common_api/utils.ex

182 lines
5.3 KiB
Elixir
Raw Normal View History

defmodule Pleroma.Web.CommonAPI.Utils do
2017-05-17 16:00:20 +00:00
alias Pleroma.{Repo, Object, Formatter, User, Activity}
alias Pleroma.Web.ActivityPub.Utils
2017-06-19 21:12:37 +00:00
alias Calendar.Strftime
2017-05-17 16:00:20 +00:00
# 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
2017-05-17 16:00:20 +00:00
def attachments_from_ids(ids) do
Enum.map(ids || [], fn (media_id) ->
Repo.get(Object, media_id).data
end)
end
def to_for_user_and_mentions(user, mentions, inReplyTo, "public") do
to = ["https://www.w3.org/ns/activitystreams#Public"]
mentioned_users = Enum.map(mentions, fn ({_, %{ap_id: ap_id}}) -> ap_id end)
cc = [user.follower_address | mentioned_users]
if inReplyTo do
{to, Enum.uniq([inReplyTo.data["actor"] | cc])}
2017-08-28 17:17:38 +00:00
else
{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
mentioned_users = Enum.map(mentions, fn ({_, %{ap_id: ap_id}}) -> ap_id end)
if inReplyTo do
{Enum.uniq([inReplyTo.data["actor"] | mentioned_users]), []}
else
{mentioned_users, []}
2017-08-28 17:17:38 +00:00
end
end
2017-12-07 18:44:09 +00:00
def make_content_html(status, mentions, attachments, tags, no_attachment_links \\ false) do
status
2017-09-17 13:21:44 +00:00
|> format_input(mentions, tags)
2017-12-07 18:44:09 +00:00
|> maybe_add_attachments(attachments, no_attachment_links)
end
def make_context(%Activity{data: %{"context" => context}}), do: context
def make_context(_), do: Utils.generate_context_id
2017-12-07 18:44:09 +00:00
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
2017-05-17 16:00:20 +00:00
def add_attachments(text, attachments) do
attachment_text = Enum.map(attachments, fn
(%{"url" => [%{"href" => href} | _]}) ->
2017-08-28 17:17:38 +00:00
name = URI.decode(Path.basename(href))
"<a href=\"#{href}\" class='attachment'>#{shortname(name)}</a>"
2017-05-17 16:00:20 +00:00
_ -> ""
end)
Enum.join([text | attachment_text], "<br>")
2017-05-17 16:00:20 +00:00
end
2017-11-19 01:22:07 +00:00
def format_input(text, mentions, _tags) do
2017-12-07 19:34:25 +00:00
text
|> Formatter.html_escape
2017-05-17 16:00:20 +00:00
|> Formatter.linkify
|> String.replace("\n", "<br>")
2017-05-17 16:00:20 +00:00
|> add_user_links(mentions)
# |> add_tag_links(tags)
2017-09-17 13:21:44 +00:00
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)
2017-05-17 16:00:20 +00:00
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) ->
2017-05-30 12:26:31 +00:00
short_match = String.split(match, "@") |> tl() |> hd()
String.replace(text, uuid, "<span><a href='#{ap_id}'>@<span>#{short_match}</span></a></span>")
2017-05-17 16:00:20 +00:00
end)
end
def make_note_data(actor, to, context, content_html, attachments, inReplyTo, tags, cw \\ nil, cc \\ []) do
2017-05-17 16:00:20 +00:00
object = %{
"type" => "Note",
"to" => to,
"cc" => cc,
2017-05-17 16:00:20 +00:00
"content" => content_html,
2017-10-31 16:30:46 +00:00
"summary" => cw,
2017-05-17 16:00:20 +00:00
"context" => context,
"attachment" => attachments,
2017-05-18 13:16:49 +00:00
"actor" => actor,
"tag" => tags |> Enum.map(fn ({_, tag}) -> tag end)
2017-05-17 16:00:20 +00:00
}
if inReplyTo do
object
|> Map.put("inReplyTo", inReplyTo.data["object"]["id"])
|> Map.put("inReplyToStatusId", inReplyTo.id)
else
object
end
end
2017-06-19 21:12:37 +00:00
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
2017-09-15 15:50:47 +00:00
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
defp shortname(name) do
if String.length(name) < 30 do
name
else
String.slice(name, 0..30) <> ""
end
end
2017-05-17 16:00:20 +00:00
end