X-Git-Url: http://git.squeep.com/?a=blobdiff_plain;f=lib%2Fpleroma%2Fweb%2Fcommon_api%2Futils.ex;h=4cbbd0c7df6bee3a229f23ab7b15200833125b70;hb=833161b5d21f85e2276cd0cee3e148ecbe6e1f05;hp=0d8b39d70022548f1dae5e768b0265397deb38e2;hpb=50409326a853db7bd9f538f0ddbec805c134920f;p=akkoma diff --git a/lib/pleroma/web/common_api/utils.ex b/lib/pleroma/web/common_api/utils.ex index 0d8b39d70..abdeee947 100644 --- a/lib/pleroma/web/common_api/utils.ex +++ b/lib/pleroma/web/common_api/utils.ex @@ -1,98 +1,214 @@ +# Pleroma: A lightweight social networking server +# Copyright © 2017-2019 Pleroma Authors +# SPDX-License-Identifier: AGPL-3.0-only + defmodule Pleroma.Web.CommonAPI.Utils do - alias Pleroma.{Repo, Object, Formatter, User, Activity} - alias Pleroma.Web.ActivityPub.Utils alias Calendar.Strftime + alias Comeonin.Pbkdf2 + alias Pleroma.Activity + alias Pleroma.Formatter + alias Pleroma.Object + alias Pleroma.Repo + alias Pleroma.User + alias Pleroma.Web + alias Pleroma.Web.Endpoint + alias Pleroma.Web.MediaProxy + alias Pleroma.Web.ActivityPub.Utils # 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 + activity = Repo.get(Activity, id) || Activity.get_create_by_object_ap_id(id) + + activity && + if activity.data["type"] == "Create" do + activity + else + Activity.get_create_by_object_ap_id(activity.data["object"]) + end end + def get_replied_to_activity(""), do: nil + def get_replied_to_activity(id) when not is_nil(id) do Repo.get(Activity, id) end + def get_replied_to_activity(_), do: nil - def attachments_from_ids(ids) do - Enum.map(ids || [], fn (media_id) -> + def attachments_from_ids(data) do + if Map.has_key?(data, "descriptions") do + attachments_from_ids_descs(data["media_ids"], data["descriptions"]) + else + attachments_from_ids_no_descs(data["media_ids"]) + end + end + + def attachments_from_ids_no_descs(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) do - default_to = [ - user.follower_address, - "https://www.w3.org/ns/activitystreams#Public" - ] + def attachments_from_ids_descs(ids, descs_str) do + {_, descs} = Jason.decode(descs_str) + + Enum.map(ids || [], fn media_id -> + Map.put(Repo.get(Object, media_id).data, "name", descs[media_id]) + end) + end + + def to_for_user_and_mentions(user, mentions, inReplyTo, "public") do + mentioned_users = Enum.map(mentions, fn {_, %{ap_id: ap_id}} -> ap_id end) + + to = ["https://www.w3.org/ns/activitystreams#Public" | mentioned_users] + cc = [user.follower_address] + + if inReplyTo do + {Enum.uniq([inReplyTo.data["actor"] | to]), cc} + else + {to, cc} + end + end + + def to_for_user_and_mentions(user, mentions, inReplyTo, "unlisted") do + 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 + 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) - to = default_to ++ Enum.map(mentions, fn ({_, %{ap_id: ap_id}}) -> ap_id end) if inReplyTo do - Enum.uniq([inReplyTo.data["actor"] | to]) + {Enum.uniq([inReplyTo.data["actor"] | mentioned_users]), []} else - to + {mentioned_users, []} end end - def make_content_html(status, mentions, attachments) do + def make_content_html( + status, + mentions, + attachments, + tags, + content_type, + no_attachment_links \\ false + ) do status - |> format_input(mentions) - |> add_attachments(attachments) + |> format_input(mentions, tags, content_type) + |> 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 + def make_context(_), do: Utils.generate_context_id() + + def maybe_add_attachments(text, _attachments, true = _no_links), do: text + + def maybe_add_attachments(text, attachments, _no_links) do + add_attachments(text, attachments) + end def add_attachments(text, attachments) do - attachment_text = Enum.map(attachments, fn - (%{"url" => [%{"href" => href} | _]}) -> - name = URI.decode(Path.basename(href)) - "#{shortname(name)}" - _ -> "" - end) - Enum.join([text | attachment_text], "
\n") + attachment_text = + Enum.map(attachments, fn + %{"url" => [%{"href" => href} | _]} = attachment -> + name = attachment["name"] || URI.decode(Path.basename(href)) + href = MediaProxy.url(href) + "#{shortname(name)}" + + _ -> + "" + end) + + Enum.join([text | attachment_text], "
") end - def format_input(text, mentions) do - HtmlSanitizeEx.strip_tags(text) - |> Formatter.linkify - |> String.replace("\n", "
\n") - |> add_user_links(mentions) + def format_input(text, mentions, tags, format, options \\ []) + + @doc """ + Formatting text to plain text. + """ + def format_input(text, mentions, tags, "text/plain", options) do + text + |> Formatter.html_escape("text/plain") + |> String.replace(~r/\r?\n/, "
") + |> (&{[], &1}).() + |> Formatter.add_links() + |> Formatter.add_user_links(mentions, options[:user_links] || []) + |> Formatter.add_hashtag_links(tags) + |> Formatter.finalize() 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) + @doc """ + Formatting text to html. + """ + def format_input(text, mentions, _tags, "text/html", options) do + text + |> Formatter.html_escape("text/html") + |> (&{[], &1}).() + |> Formatter.add_user_links(mentions, options[:user_links] || []) + |> Formatter.finalize() + 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) + @doc """ + Formatting text to markdown. + """ + def format_input(text, mentions, tags, "text/markdown", options) do + text + |> Formatter.mentions_escape(mentions) + |> Earmark.as_html!() + |> Formatter.html_escape("text/html") + |> (&{[], &1}).() + |> Formatter.add_user_links(mentions, options[:user_links] || []) + |> Formatter.add_hashtag_links(tags) + |> Formatter.finalize() + end + + def add_tag_links(text, tags) do + tags = + tags + |> Enum.sort_by(fn {tag, _} -> -String.length(tag) end) - Enum.reduce(mentions, step_one, fn ({match, %User{ap_id: ap_id}, uuid}, text) -> - short_match = String.split(match, "@") |> tl() |> hd() - String.replace(text, uuid, "@#{short_match}") + Enum.reduce(tags, text, fn {full, tag}, text -> + url = "" + String.replace(text, full, url) end) end - def make_note_data(actor, to, context, content_html, attachments, inReplyTo, tags) do - object = %{ - "type" => "Note", - "to" => to, - "content" => content_html, - "context" => context, - "attachment" => attachments, - "actor" => actor, - "tag" => tags |> Enum.map(fn ({_, tag}) -> tag end) - } + 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) |> Enum.uniq() + } if inReplyTo do object @@ -112,13 +228,31 @@ defmodule Pleroma.Web.CommonAPI.Utils do end def date_to_asctime(date) do - with {:ok, date, _offset} <- date |> DateTime.from_iso8601 do + with {:ok, date, _offset} <- date |> DateTime.from_iso8601() do format_asctime(date) - else _e -> + else + _e -> "" end end + 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 @@ -126,4 +260,66 @@ defmodule Pleroma.Web.CommonAPI.Utils do String.slice(name, 0..30) <> "…" end end + + def confirm_current_password(user, password) do + with %User{local: true} = db_user <- Repo.get(User, user.id), + true <- Pbkdf2.checkpw(password, db_user.password_hash) do + {:ok, db_user} + else + _ -> {:error, "Invalid password."} + end + end + + def emoji_from_profile(%{info: _info} = user) do + (Formatter.get_emoji(user.bio) ++ Formatter.get_emoji(user.name)) + |> Enum.map(fn {shortcode, url} -> + %{ + "type" => "Emoji", + "icon" => %{"type" => "Image", "url" => "#{Endpoint.url()}#{url}"}, + "name" => ":#{shortcode}:" + } + end) + end + + 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: [] end