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
12 alias Pleroma.Web.OAuth.Token
13 alias Pleroma.Web.Push.Subscription
15 schema "push_subscriptions" do
16 belongs_to(:user, User, type: Pleroma.FlakeId)
17 belongs_to(:token, Token)
18 field(:endpoint, :string)
19 field(:key_p256dh, :string)
20 field(:key_auth, :string)
21 field(:data, :map, default: %{})
26 @supported_alert_types ~w[follow favourite mention reblog]
28 defp alerts(%{"data" => %{"alerts" => alerts}}) do
29 alerts = Map.take(alerts, @supported_alert_types)
38 "endpoint" => endpoint,
39 "keys" => %{"auth" => key_auth, "p256dh" => key_p256dh}
43 Repo.insert(%Subscription{
47 key_auth: ensure_base64_urlsafe(key_auth),
48 key_p256dh: ensure_base64_urlsafe(key_p256dh),
53 def get(%User{id: user_id}, %Token{id: token_id}) do
54 Repo.get_by(Subscription, user_id: user_id, token_id: token_id)
57 def update(user, token, params) do
59 |> change(data: alerts(params))
63 def delete(user, token) do
64 Repo.delete(get(user, token))
67 def delete_if_exists(user, token) do
68 case get(user, token) do
70 sub -> Repo.delete(sub)
74 # Some webpush clients (e.g. iOS Toot!) use an non urlsafe base64 as an encoding for the key.
75 # However, the web push rfs specify to use base64 urlsafe, and the `web_push_encryption` library we use
76 # requires the key to be properly encoded. So we just convert base64 to urlsafe base64.
77 defp ensure_base64_urlsafe(string) do
79 |> String.replace("+", "-")
80 |> String.replace("/", "_")
81 |> String.replace("=", "")