<?php

namespace FootyRoom\Queries\Comment;

use FootyRoom\Core\Comment\CommentPolicy;

trait ViewedBy
{
    /**
     * Sets up query according to user role to make sure right data is
     * requested.
     *
     * @param int $userRole
     *
     * @return self
     */
    public function viewedBy($userRole)
    {
        if (property_exists($this, 'unpublished')) {
            $this->unpublished = CommentPolicy::seeUnpublished($userRole);
        }

        if (property_exists($this, 'meta')) {
            $this->meta = CommentPolicy::getVisibleMeta($userRole);
        }

        return $this;
    }
}