<?php namespace Elementor\Modules\GlobalClasses\Database\Migrations; use Elementor\Core\Database\Base_Migration; class Add_Capabilities extends Base_Migration { const UPDATE_CLASS = 'elementor_global_classes_update_class'; const REMOVE_CSS_CLASS = 'elementor_global_classes_remove_class'; const APPLY_CSS_CLASS = 'elementor_global_classes_apply_class'; public function up() { $capabilities = [ self::UPDATE_CLASS => [ 'administrator' ], self::REMOVE_CSS_CLASS => [ 'administrator', 'editor', 'author', 'contributor', 'shop_manager' ], self::APPLY_CSS_CLASS => [ 'administrator', 'editor', 'author', 'contributor', 'shop_manager' ], ]; foreach ( $capabilities as $capability => $roles ) { foreach ( $roles as $role_name ) { $role = get_role( $role_name ); if ( $role ) { $role->add_cap( $capability ); } } } } }