1 # Pleroma: A lightweight social networking server
2 # Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
3 # SPDX-License-Identifier: AGPL-3.0-only
5 defmodule Pleroma.Web.Push.Subscription do
8 alias Pleroma.{Repo, User}
9 alias Pleroma.Web.OAuth.Token
10 alias Pleroma.Web.Push.Subscription
12 schema "push_subscriptions" do
13 belongs_to(:user, User)
14 belongs_to(:token, Token)
15 field(:endpoint, :string)
16 field(:key_p256dh, :string)
17 field(:key_auth, :string)
18 field(:data, :map, default: %{})
23 @supported_alert_types ~w[follow favourite mention reblog]
25 defp alerts(%{"data" => %{"alerts" => alerts}}) do
26 alerts = Map.take(alerts, @supported_alert_types)
35 "endpoint" => endpoint,
36 "keys" => %{"auth" => key_auth, "p256dh" => key_p256dh}
40 Repo.insert(%Subscription{
44 key_auth: ensure_base64_urlsafe(key_auth),
45 key_p256dh: ensure_base64_urlsafe(key_p256dh),
50 def get(%User{id: user_id}, %Token{id: token_id}) do
51 Repo.get_by(Subscription, user_id: user_id, token_id: token_id)
54 def update(user, token, params) do
56 |> change(data: alerts(params))
60 def delete(user, token) do
61 Repo.delete(get(user, token))
64 def delete_if_exists(user, token) do
65 case get(user, token) do
67 sub -> Repo.delete(sub)
71 # Some webpush clients (e.g. iOS Toot!) use an non urlsafe base64 as an encoding for the key.
72 # However, the web push rfs specify to use base64 urlsafe, and the `web_push_encryption` library we use
73 # requires the key to be properly encoded. So we just convert base64 to urlsafe base64.
74 defp ensure_base64_urlsafe(string) do
76 |> String.replace("+", "-")
77 |> String.replace("/", "_")
78 |> String.replace("=", "")