UserRoles.php

<?php

declare(strict_types=1);

namespace Yard\UserRoles;

use Illuminate\Contracts\Foundation\Application;
use Webmozart\Assert\Assert;

class UserRoles
{
    /**
     * Create a new UserRoles instance.
     */
    public function __construct(protected Application $app)
    {
    }

    public function setRoles(): void
    {
        if (is_multisite()) {
            foreach (get_sites(['fields' => 'ids']) as $siteId) {
                switch_to_blog($siteId);
                $this->setRolesForSite();
            }
        } else {
            $this->setRolesForSite();
        }
    }

    private function setRolesForSite(): void
    {
        $prefix = config('user-roles.prefix');
        Assert::stringNotEmpty($prefix);

        $this->removeCurrentCustomRoles();
        $this->addCustomRoles();
    }

    private function removeCurrentCustomRoles(): void
    {
        $prefix = config('user-roles.prefix');
        $currentCustomRoles = \wp_roles()->roles;
        $currentCustomRoles = array_filter(
            array_keys($currentCustomRoles),
            fn (string $role): bool => str_starts_with($role, $prefix)
        );
        foreach($currentCustomRoles as $role) {
            remove_role($role);
        }
    }

    private function addCustomRoles(): void
    {
        $prefix = config('user-roles.prefix');
        $roles = config('user-roles.roles');
        Assert::isArray($roles);

        foreach($roles as $role => $properties) {
            $capabilities = [];
            if (! empty($properties['cap_groups'])) {
                $capGroups = config('user-roles.cap_groups');
                Assert::isArray($capGroups);
                Assert::isArray($properties['cap_groups']);
                foreach ($properties['cap_groups'] as $group) {
                    $groupCaps = $capGroups[$group];
                    Assert::isArray($groupCaps);

                    foreach ($groupCaps as $cap) {
                        $capabilities[$cap] = true;
                    }
                }
            }

            if (! empty($properties['post_type_caps'])) {
                Assert::isArray($properties['post_type_caps']);
                foreach ($properties['post_type_caps'] as $postType) {
                    $postTypeCaps = get_post_type_object($postType)?->cap;
                    Assert::object($postTypeCaps);
                    foreach(array_values((array)$postTypeCaps) as $cap) {
                        $capabilities[$cap] = true;
                    }
                }
            }

            foreach($properties['caps'] as $cap) {
                $capabilities[$cap] = true;
            }

            add_role(
                $prefix . '_' . $role,
                $properties['display_name'],
                $capabilities
            );
        }
    }
}