<?php
/**
 * Custom capability registration.
 *
 * @package Abientot
 */

namespace Abientot;

if ( ! defined( 'ABSPATH' ) ) {
	exit;
}

final class Capabilities {
	/**
	 * Abientot capabilities granted to administrators.
	 *
	 * @return string[]
	 */
	public static function all(): array {
		return array(
			'abientot_manage_bookings',
			'abientot_manage_customers',
			'abientot_manage_services',
			'abientot_manage_staff',
			'abientot_manage_locations',
			'abientot_manage_settings',
			'abientot_view_reports',
		);
	}

	/**
	 * Adds all Abientot capabilities to administrator roles.
	 */
	public static function grant_to_administrators(): void {
		$role = get_role( 'administrator' );

		if ( ! $role ) {
			return;
		}

		foreach ( self::all() as $capability ) {
			$role->add_cap( $capability );
		}
	}

	/**
	 * Removes Abientot capabilities from administrator roles.
	 */
	public static function remove_from_administrators(): void {
		$role = get_role( 'administrator' );

		if ( ! $role ) {
			return;
		}

		foreach ( self::all() as $capability ) {
			$role->remove_cap( $capability );
		}
	}
}