akkoma/lib/pleroma/web/push/impl.ex

198 lines
5.6 KiB
Elixir
Raw Permalink Normal View History

2019-03-06 13:20:12 +00:00
# Pleroma: A lightweight social networking server
# Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
2019-03-06 13:20:12 +00:00
# SPDX-License-Identifier: AGPL-3.0-only
defmodule Pleroma.Web.Push.Impl do
@moduledoc "The module represents implementation push web notification"
alias Pleroma.Activity
alias Pleroma.Notification
2019-03-06 13:20:12 +00:00
alias Pleroma.Object
alias Pleroma.Repo
alias Pleroma.User
2019-03-06 13:20:12 +00:00
alias Pleroma.Web.Metadata.Utils
alias Pleroma.Web.Push.Subscription
2019-03-06 13:20:12 +00:00
require Logger
import Ecto.Query
@types ["Create", "Follow", "Announce", "Like", "Move", "EmojiReact", "Update"]
2019-03-06 13:20:12 +00:00
@doc "Performs sending notifications for user subscriptions"
2020-10-01 10:32:11 +00:00
@spec perform(Notification.t()) :: list(any) | :error | {:error, :unknown_type}
def perform(
%{
activity: %{data: %{"type" => activity_type}} = activity,
user: %User{id: user_id}
} = notification
2019-03-11 18:32:21 +00:00
)
2019-03-06 13:20:12 +00:00
when activity_type in @types do
actor = User.get_cached_by_ap_id(notification.activity.data["actor"])
2019-03-06 13:20:12 +00:00
mastodon_type = notification.type
2019-03-06 13:20:12 +00:00
gcm_api_key = Application.get_env(:web_push_encryption, :gcm_api_key)
avatar_url = User.avatar_url(actor)
object = Object.normalize(activity, fetch: false)
user = User.get_cached_by_id(user_id)
direct_conversation_id = Activity.direct_conversation_id(activity, user)
2019-03-06 13:20:12 +00:00
for subscription <- fetch_subscriptions(user_id),
Subscription.enabled?(subscription, mastodon_type) do
2019-03-06 13:20:12 +00:00
%{
access_token: subscription.token.token,
notification_id: notification.id,
notification_type: mastodon_type,
2019-03-06 13:20:12 +00:00
icon: avatar_url,
preferred_locale: "en",
pleroma: %{
activity_id: notification.activity.id,
direct_conversation_id: direct_conversation_id
}
2019-03-06 13:20:12 +00:00
}
|> Map.merge(build_content(notification, actor, object, mastodon_type))
2019-03-06 13:20:12 +00:00
|> Jason.encode!()
|> push_message(build_sub(subscription), gcm_api_key, subscription)
end
|> (&{:ok, &1}).()
2019-03-06 13:20:12 +00:00
end
def perform(_) do
2019-03-06 13:20:12 +00:00
Logger.warn("Unknown notification type")
{:error, :unknown_type}
2019-03-06 13:20:12 +00:00
end
@doc "Push message to web"
def push_message(body, sub, api_key, subscription) do
case WebPushEncryption.send_web_push(body, sub, api_key) do
2020-10-01 10:32:11 +00:00
{:ok, %{status: code}} when code in 400..499 ->
2019-03-06 13:20:12 +00:00
Logger.debug("Removing subscription record")
Repo.delete!(subscription)
:ok
2020-10-01 10:32:11 +00:00
{:ok, %{status: code}} when code in 200..299 ->
2019-03-06 13:20:12 +00:00
:ok
2020-10-01 10:32:11 +00:00
{:ok, %{status: code}} ->
2019-03-06 13:20:12 +00:00
Logger.error("Web Push Notification failed with code: #{code}")
:error
2020-10-01 10:32:11 +00:00
error ->
Logger.error("Web Push Notification failed with #{inspect(error)}")
2019-03-06 13:20:12 +00:00
:error
end
end
@doc "Gets user subscriptions"
def fetch_subscriptions(user_id) do
2019-03-06 13:20:12 +00:00
Subscription
|> where(user_id: ^user_id)
|> preload(:token)
|> Repo.all()
end
def build_sub(subscription) do
%{
keys: %{
p256dh: subscription.key_p256dh,
auth: subscription.key_auth
},
endpoint: subscription.endpoint
}
end
def build_content(notification, actor, object, mastodon_type \\ nil)
def build_content(
%{
user: %{notification_settings: %{hide_notification_contents: true}}
} = notification,
_actor,
_object,
mastodon_type
) do
%{body: format_title(notification, mastodon_type)}
end
def build_content(notification, actor, object, mastodon_type) do
mastodon_type = mastodon_type || notification.type
%{
title: format_title(notification, mastodon_type),
body: format_body(notification, actor, object, mastodon_type)
}
end
def format_body(activity, actor, object, mastodon_type \\ nil)
2019-03-06 13:20:12 +00:00
def format_body(
%{activity: %{data: %{"type" => "Create"}}},
actor,
%{data: %{"content" => content}},
_mastodon_type
2019-03-06 13:20:12 +00:00
) do
"@#{actor.nickname}: #{Utils.scrub_html_and_truncate(content, 80)}"
end
def format_body(
%{activity: %{data: %{"type" => "Announce"}}},
actor,
%{data: %{"content" => content}},
_mastodon_type
2019-03-06 13:20:12 +00:00
) do
"@#{actor.nickname} repeated: #{Utils.scrub_html_and_truncate(content, 80)}"
end
def format_body(
%{activity: %{data: %{"type" => "EmojiReact", "content" => content}}},
actor,
_object,
_mastodon_type
) do
"@#{actor.nickname} reacted with #{content}"
end
2019-03-06 13:20:12 +00:00
def format_body(
%{activity: %{data: %{"type" => type}}} = notification,
actor,
_object,
mastodon_type
2019-03-06 13:20:12 +00:00
)
when type in ["Follow", "Like"] do
mastodon_type = mastodon_type || notification.type
case mastodon_type do
"follow" -> "@#{actor.nickname} has followed you"
"follow_request" -> "@#{actor.nickname} has requested to follow you"
"favourite" -> "@#{actor.nickname} has favorited your post"
2019-03-06 13:20:12 +00:00
end
end
def format_body(
%{activity: %{data: %{"type" => "Update"}}},
actor,
_object,
_mastodon_type
) do
"@#{actor.nickname} edited a status"
end
def format_title(activity, mastodon_type \\ nil)
def format_title(%{activity: %{data: %{"directMessage" => true}}}, _mastodon_type) do
"New Direct Message"
end
def format_title(%{type: type}, mastodon_type) do
case mastodon_type || type do
"mention" -> "New Mention"
"follow" -> "New Follower"
"follow_request" -> "New Follow Request"
"reblog" -> "New Repeat"
"favourite" -> "New Favorite"
"update" -> "New Update"
"pleroma:emoji_reaction" -> "New Reaction"
type -> "New #{String.capitalize(type || "event")}"
2019-03-06 13:20:12 +00:00
end
end
end