<?php
/**
 * WP Hotel Booking roles.
 *
 * @version       1.9.6
 * @author        ThimPress
 * @package       WP_Hotel_Booking/Classes
 * @category      Classes
 * @author        Thimpress, leehld
 */

/**
 * Prevent loading this file directly
 */
defined( 'ABSPATH' ) || exit;

if ( ! class_exists( 'WPHB_Roles' ) ) {

	/**
	 * Class WPHB_Roles
	 */
	class WPHB_Roles {

		/**
		 * WPHB_Roles constructor.
		 */
		public function __construct() {
			$this->add_roles();
		}

		/**
		 * Add user roles.
		 */
		public function add_roles() {

			add_role(
				'wphb_hotel_manager',
				__( 'Hotel Manager', 'wp-hotel-booking' ),
				array()
			);

			$room_cap    = 'hb_rooms';
			$booking_cap = 'hb_bookings';

			$hotel_manager = get_role( 'wphb_hotel_manager' );

			// add capability for hotel manager
			$hotel_manager->add_cap( 'read' );
			$hotel_manager->add_cap( 'edit_posts' );
			$hotel_manager->add_cap( 'delete_' . $room_cap );
			$hotel_manager->add_cap( 'edit_posts' );
			$hotel_manager->add_cap( 'publish_' . $room_cap );
			$hotel_manager->add_cap( 'delete_published_' . $room_cap );
			$hotel_manager->add_cap( 'delete_private_' . $room_cap );
			$hotel_manager->add_cap( 'delete_others_' . $room_cap );
			$hotel_manager->add_cap( 'edit_others_' . $room_cap );
			$hotel_manager->add_cap( 'edit_' . $room_cap );
			$hotel_manager->add_cap( 'edit_published_' . $room_cap );
			$hotel_manager->add_cap( 'edit_private_' . $room_cap );
			$hotel_manager->add_cap( 'edit_others_' . $room_cap );

			$hotel_manager->add_cap( 'publish_' . $booking_cap );
			$hotel_manager->add_cap( 'delete_' . $booking_cap );
			$hotel_manager->add_cap( 'delete_published_' . $booking_cap );
			$hotel_manager->add_cap( 'delete_private_' . $booking_cap );
			$hotel_manager->add_cap( 'delete_others_' . $booking_cap );
			$hotel_manager->add_cap( 'edit_others_' . $booking_cap );
			$hotel_manager->add_cap( 'edit_' . $booking_cap );
			$hotel_manager->add_cap( 'edit_published_' . $booking_cap );
			$hotel_manager->add_cap( 'edit_private_' . $booking_cap );
			$hotel_manager->add_cap( 'edit_others_' . $booking_cap );

			$hotel_manager->add_cap( 'upload_files' );
			$hotel_manager->add_cap( 'manage_hb_booking' );

			add_role(
				'wphb_booking_editor',
				__( 'Booking Editor', 'wp-hotel-booking' ),
				array()
			);

			$booking_editor = get_role( 'wphb_booking_editor' );

			// add capability for booking
			$booking_editor->add_cap( 'read' );
			$booking_editor->add_cap( 'edit_posts' );
			$booking_editor->add_cap( 'publish_' . $room_cap );
			$booking_editor->add_cap( 'delete_' . $room_cap );
			$booking_editor->add_cap( 'delete_published_' . $room_cap );
			$booking_editor->add_cap( 'delete_private_' . $room_cap );
			$booking_editor->add_cap( 'delete_others_' . $room_cap );
			$booking_editor->add_cap( 'edit_others_' . $room_cap );
			$booking_editor->add_cap( 'edit_' . $room_cap );
			$booking_editor->add_cap( 'edit_published_' . $room_cap );
			$booking_editor->add_cap( 'edit_private_' . $room_cap );
			$booking_editor->add_cap( 'edit_others_' . $room_cap );

			$booking_editor->add_cap( 'publish_' . $booking_cap );
			$booking_editor->add_cap( 'delete_' . $booking_cap );
			$booking_editor->add_cap( 'delete_published_' . $booking_cap );
			$booking_editor->add_cap( 'delete_private_' . $booking_cap );
			$booking_editor->add_cap( 'delete_others_' . $booking_cap );
			$booking_editor->add_cap( 'edit_others_' . $booking_cap );
			$booking_editor->add_cap( 'edit_' . $booking_cap );
			$booking_editor->add_cap( 'edit_published_' . $booking_cap );
			$booking_editor->add_cap( 'edit_private_' . $booking_cap );
			$booking_editor->add_cap( 'edit_others_' . $booking_cap );

			$booking_editor->add_cap( 'upload_files' );

			$admin = get_role( 'administrator' );

			// add capability for admin
			$admin->add_cap( 'publish_' . $room_cap );
			$admin->add_cap( 'delete_' . $room_cap );
			$admin->add_cap( 'delete_published_' . $room_cap );
			$admin->add_cap( 'delete_private_' . $room_cap );
			$admin->add_cap( 'delete_others_' . $room_cap );
			$admin->add_cap( 'edit_others_' . $room_cap );
			$admin->add_cap( 'edit_' . $room_cap );
			$admin->add_cap( 'edit_published_' . $room_cap );
			$admin->add_cap( 'edit_private_' . $room_cap );
			$admin->add_cap( 'edit_others_' . $room_cap );

			$admin->add_cap( 'publish_' . $booking_cap );
			$admin->add_cap( 'delete_' . $booking_cap );
			$admin->add_cap( 'delete_published_' . $booking_cap );
			$admin->add_cap( 'delete_private_' . $booking_cap );
			$admin->add_cap( 'delete_others_' . $booking_cap );
			$admin->add_cap( 'edit_others_' . $booking_cap );
			$admin->add_cap( 'edit_' . $booking_cap );
			$admin->add_cap( 'edit_published_' . $booking_cap );
			$admin->add_cap( 'edit_private_' . $booking_cap );
			$admin->add_cap( 'edit_others_' . $booking_cap );

			$admin->add_cap( 'manage_hb_booking' );
		}
	}
}

new WPHB_Roles();
