Leonidas\Library\System\Model\Abstracts\Post\ValidatesPostTypeTrait

Complexity

Total Complexity

12

Size/Duplication

Total Lines	44
Duplicated Lines	0 %

Test Coverage

Coverage

0%

Importance

Changes	1
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	assertPostType()	0	9	2
B	assertPostTypeOnQuery()	0	25	10

<?php

namespace Leonidas\Library\System\Model\Abstracts\Post;

use InvalidArgumentException;
use Stringable;
use WP_Post;
use WP_Query;

The type WP_Query was not found. Maybe you did not declare it correctly or list all dependencies?

trait ValidatesPostTypeTrait
{
    /**
     * @throws InvalidArgumentException
     */
    protected function assertPostType(WP_Post $post, string $postType): self
    {
        if ($postType !== $actual = $post->post_type) {
            throw new InvalidArgumentException(
                "The post type of the post must be \"{$postType}\", but it is \"{$actual}.\"",
            );
        }

        return $this;
    }

    /**
     * @throws InvalidArgumentException
     */
    protected function assertPostTypeOnQuery(WP_Query $query, string $postType): self
    {
        $actual = $query->get('post_type')
            ?: $query->get_queried_object()->post_type
            ?? $postType;

        if (is_array($actual) && count($actual) === 1) {
            $actual = $actual[0];
        } elseif (!is_string($actual)) {
            $actual = null;
        }

        if ($postType !== $actual) {
            if (is_string($actual) || $actual instanceof Stringable) {
                $message = "Query post_type value must be \"{$postType}\", but it is \"{$actual}\".";
            } elseif (is_array($actual) && in_array($postType, $actual, true)) {
                $message = "post_type value for query must be exclusively for \"{$postType}\".";
            } else {
                $message = "Query post_type value must be \"{$postType}\".";
            }

            throw new InvalidArgumentException($message);
        }

        return $this;
    }
}