Notfication: Add validation of notification types

2020-06-06 16:59:08 +02:00 · 2020-06-06 16:59:08 +02:00 · 40fc4e974e
commit 40fc4e974e
parent f4cf4ae16e
1 changed files with 12 additions and 0 deletions
--- a/lib/pleroma/notification.ex
+++ b/lib/pleroma/notification.ex
@ -61,9 +61,21 @@ defmodule Pleroma.Notification do
    |> Repo.aggregate(:count, :id)
  end

+  @notification_types ~w{
+    favourite
+    follow
+    follow_request
+    mention
+    move
+    pleroma:chat_mention
+    pleroma:emoji_reaction
+    reblog
+  }
+
  def changeset(%Notification{} = notification, attrs) do
    notification
    |> cast(attrs, [:seen, :type])
+    |> validate_inclusion(:type, @notification_types)
  end

  @spec last_read_query(User.t()) :: Ecto.Queryable.t()