<?php
namespace ElementorPro\Modules\Notes\Database\Migrations;

use ElementorPro\Core\Database\Base_Migration;
use ElementorPro\Modules\Notes\User\Capabilities;

if ( ! defined( 'ABSPATH' ) ) {
	exit; // Exit if accessed directly.
}

class Add_Capabilities extends Base_Migration {
	/**
	 * @inheritDoc
	 */
	public function up() {
		$admin = get_role( 'administrator' );

		if ( $admin instanceof \WP_Role ) {
			foreach ( Capabilities::all() as $capability ) {
				$admin->add_cap( $capability );
			}
		}
	}

	/**
	 * @inheritDoc
	 */
	public function down() {
		$roles = array_values( wp_roles()->role_objects );

		foreach ( $roles as $role ) {
			if ( ! ( $role instanceof \WP_Role ) ) {
				continue;
			}

			foreach ( Capabilities::all() as $cap ) {
				$role->remove_cap( $cap );
			}
		}
	}
}