42 lines
805 B
PHP
42 lines
805 B
PHP
<?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 );
|
|
}
|
|
}
|
|
}
|
|
}
|