add-capabilities.php
30 lines
| 1 | <?php |
| 2 | |
| 3 | namespace Elementor\Modules\GlobalClasses\Database\Migrations; |
| 4 | |
| 5 | use Elementor\Core\Database\Base_Migration; |
| 6 | |
| 7 | class Add_Capabilities extends Base_Migration { |
| 8 | const UPDATE_CLASS = 'elementor_global_classes_update_class'; |
| 9 | const REMOVE_CSS_CLASS = 'elementor_global_classes_remove_class'; |
| 10 | const APPLY_CSS_CLASS = 'elementor_global_classes_apply_class'; |
| 11 | |
| 12 | public function up() { |
| 13 | $capabilities = [ |
| 14 | self::UPDATE_CLASS => [ 'administrator' ], |
| 15 | self::REMOVE_CSS_CLASS => [ 'administrator', 'editor', 'author', 'contributor', 'shop_manager' ], |
| 16 | self::APPLY_CSS_CLASS => [ 'administrator', 'editor', 'author', 'contributor', 'shop_manager' ], |
| 17 | ]; |
| 18 | |
| 19 | foreach ( $capabilities as $capability => $roles ) { |
| 20 | foreach ( $roles as $role_name ) { |
| 21 | $role = get_role( $role_name ); |
| 22 | |
| 23 | if ( $role ) { |
| 24 | $role->add_cap( $capability ); |
| 25 | } |
| 26 | } |
| 27 | } |
| 28 | } |
| 29 | } |
| 30 |