From beb5329e42c14da5613b8cd567c9b1aeb21216bf Mon Sep 17 00:00:00 2001
From: Franz Liedke <franz@develophp.org>
Date: Fri, 7 Feb 2020 11:56:13 +0100
Subject: [PATCH] Fix flagging check for all post types

Refs #18.
---
 extensions/flags/src/Listener/AddFlagsApiAttributes.php | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/extensions/flags/src/Listener/AddFlagsApiAttributes.php b/extensions/flags/src/Listener/AddFlagsApiAttributes.php
index b48032d98..6fbab4bf9 100755
--- a/extensions/flags/src/Listener/AddFlagsApiAttributes.php
+++ b/extensions/flags/src/Listener/AddFlagsApiAttributes.php
@@ -14,7 +14,7 @@ use Flarum\Api\Serializer\CurrentUserSerializer;
 use Flarum\Api\Serializer\ForumSerializer;
 use Flarum\Api\Serializer\PostSerializer;
 use Flarum\Flags\Flag;
-use Flarum\Post\CommentPost;
+use Flarum\Post\Post;
 use Flarum\Settings\SettingsRepositoryInterface;
 use Flarum\User\User;
 
@@ -78,12 +78,7 @@ class AddFlagsApiAttributes
         return $query->distinct()->count('flags.post_id');
     }
 
-    /**
-     * @param User $actor
-     * @param CommentPost $post
-     * @return bool
-     */
-    protected function checkFlagOwnPostSetting(User $actor, CommentPost $post): bool
+    protected function checkFlagOwnPostSetting(User $actor, Post $post): bool
     {
         if ($actor->id === $post->user_id) {
             // If $actor is the post author, check to see if the setting is enabled