X-Git-Url: http://git.squeep.com/?a=blobdiff_plain;f=lib%2Fpleroma%2Fweb%2Fpush%2Fpush.ex;h=5259e8e33083c248f38621fcfdc693e63dcb2fc9;hb=92a0210fb03ca3e0aefe769fb6b0ab7bda6e5336;hp=5a873ec19cf3231c4079ef731e71c7525ce983b9;hpb=b0adea30b38713e70039475acdd3defbef57171a;p=akkoma

diff --git a/lib/pleroma/web/push/push.ex b/lib/pleroma/web/push/push.ex
index 5a873ec19..5259e8e33 100644
--- a/lib/pleroma/web/push/push.ex
+++ b/lib/pleroma/web/push/push.ex
@@ -1,116 +1,64 @@
+# Pleroma: A lightweight social networking server
+# Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
+# SPDX-License-Identifier: AGPL-3.0-only
+
 defmodule Pleroma.Web.Push do
   use GenServer
 
-  alias Pleroma.{Repo, User}
-  alias Pleroma.Web.Push.Subscription
+  alias Pleroma.Web.Push.Impl
 
   require Logger
-  import Ecto.Query
-
-  @types ["Create", "Follow", "Announce", "Like"]
 
-  @gcm_api_key nil
+  ##############
+  # Client API #
+  ##############
 
-  def start_link() do
+  def start_link do
     GenServer.start_link(__MODULE__, :ok, name: __MODULE__)
   end
 
-  def init(:ok) do
-    case Application.get_env(:web_push_encryption, :vapid_details) do
-      nil ->
-        Logger.warn(
-          "VAPID key pair is not found. Please, add VAPID configuration to config. Run `mix web_push.gen.keypair` mix task to create a key pair"
-        )
-
-        :ignore
-
-      _ ->
-        {:ok, %{}}
-    end
+  def vapid_config do
+    Application.get_env(:web_push_encryption, :vapid_details, [])
   end
 
-  def send(notification) do
-    if Application.get_env(:web_push_encryption, :vapid_details) do
-      GenServer.cast(Pleroma.Web.Push, {:send, notification})
+  def enabled do
+    case vapid_config() do
+      [] -> false
+      list when is_list(list) -> true
+      _ -> false
     end
   end
 
-  def handle_cast(
-        {:send, %{activity: %{data: %{"type" => type}}, user_id: user_id} = notification},
-        state
-      )
-      when type in @types do
-    actor = User.get_cached_by_ap_id(notification.activity.data["actor"])
-    body = notification |> format(actor) |> Jason.encode!()
-
-    Subscription
-    |> where(user_id: ^user_id)
-    |> Repo.all()
-    |> Enum.each(fn record ->
-      subscription = %{
-        keys: %{
-          p256dh: record.key_p256dh,
-          auth: record.key_auth
-        },
-        endpoint: record.endpoint
-      }
-
-      case WebPushEncryption.send_web_push(body, subscription, @gcm_api_key) do
-        {:ok, %{status_code: code}} when 400 <= code and code < 500 ->
-          Logger.debug("Removing subscription record")
-          Repo.delete!(record)
-          :ok
-
-        {:ok, %{status_code: code}} when 200 <= code and code < 300 ->
-          :ok
-
-        {:ok, %{status_code: code}} ->
-          Logger.error("Web Push Nonification failed with code: #{code}")
-          :error
+  def send(notification),
+    do: GenServer.cast(__MODULE__, {:send, notification})
 
-        _ ->
-          Logger.error("Web Push Nonification failed with unknown error")
-          :error
-      end
-    end)
+  ####################
+  # Server Callbacks #
+  ####################
 
-    {:noreply, state}
-  end
+  @impl true
+  def init(:ok) do
+    if enabled() do
+      {:ok, nil}
+    else
+      Logger.warn("""
+      VAPID key pair is not found. If you wish to enabled web push, please run
 
-  def handle_cast({:send, _}, state) do
-    Logger.warn("Unknown notification type")
-    {:noreply, state}
-  end
+          mix web_push.gen.keypair
 
-  def format(%{activity: %{data: %{"type" => "Create"}}}, actor) do
-    %{
-      title: "New Mention",
-      body: "@#{actor.nickname} has mentiond you",
-      icon: User.avatar_url(actor)
-    }
-  end
+      and add the resulting output to your configuration file.
+      """)
 
-  def format(%{activity: %{data: %{"type" => "Follow"}}}, actor) do
-    %{
-      title: "New Follower",
-      body: "@#{actor.nickname} has followed you",
-      icon: User.avatar_url(actor)
-    }
+      :ignore
+    end
   end
 
-  def format(%{activity: %{data: %{"type" => "Announce"}}}, actor) do
-    %{
-      title: "New Announce",
-      body: "@#{actor.nickname} has announced your post",
-      icon: User.avatar_url(actor)
-    }
-  end
+  @impl true
+  def handle_cast({:send, notification}, state) do
+    if enabled() do
+      Impl.perform_send(notification)
+    end
 
-  def format(%{activity: %{data: %{"type" => "Like"}}}, actor) do
-    %{
-      title: "New Like",
-      body: "@#{actor.nickname} has liked your post",
-      icon: User.avatar_url(actor)
-    }
+    {:noreply, state}
   end
 end