1 defmodule Pleroma.Web.Push.Subscription do
4 alias Pleroma.{Repo, User}
5 alias Pleroma.Web.OAuth.Token
6 alias Pleroma.Web.Push.Subscription
8 schema "push_subscriptions" do
9 belongs_to(:user, User)
10 belongs_to(:token, Token)
11 field(:endpoint, :string)
12 field(:key_p256dh, :string)
13 field(:key_auth, :string)
14 field(:data, :map, default: %{})
19 @supported_alert_types ~w[follow favourite mention reblog]
21 defp alerts(%{"data" => %{"alerts" => alerts}}) do
22 alerts = Map.take(alerts, @supported_alert_types)
31 "endpoint" => endpoint,
32 "keys" => %{"auth" => key_auth, "p256dh" => key_p256dh}
36 Repo.insert(%Subscription{
40 key_auth: ensure_base64_urlsafe(key_auth),
41 key_p256dh: ensure_base64_urlsafe(key_p256dh),
46 def get(%User{id: user_id}, %Token{id: token_id}) do
47 Repo.get_by(Subscription, user_id: user_id, token_id: token_id)
50 def update(user, token, params) do
52 |> change(data: alerts(params))
56 def delete(user, token) do
57 Repo.delete(get(user, token))
60 def delete_if_exists(user, token) do
61 case get(user, token) do
63 sub -> Repo.delete(sub)
67 # Some webpush clients (e.g. iOS Toot!) use an non urlsafe base64 as an encoding for the key.
68 # However, the web push rfs specify to use base64 urlsafe, and the `web_push_encryption` library we use
69 # requires the key to be properly encoded. So we just convert base64 to urlsafe base64.
70 defp ensure_base64_urlsafe(string) do
72 |> String.replace("+", "-")
73 |> String.replace("/", "_")
74 |> String.replace("=", "")