<!DOCTYPE html>
<html lang="en">

<head>
    <meta charset="UTF-8">
    <meta http-equiv="X-UA-Compatible" content="IE=edge">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.0-alpha1/dist/css/bootstrap.min.css" rel="stylesheet"
        integrity="sha384-GLhlTQ8iRABdZLl6O3oVMWSktQOp6b7In1Zl3/Jr59b6EGGoI1aFkw7cmDA6j6gD" crossorigin="anonymous">
    <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.3.0/css/all.min.css"
        integrity="sha512-SzlrxWUlpfuzQ+pcUCosxcglQRNAq/DZjVsC0lE40xsADsfeQoEypE+enwcOiGjk/bSuGGKHEyjSoQ1zVisanQ=="
        crossorigin="anonymous" referrerpolicy="no-referrer" />
</head>
</html>
<?php
/**
 * Manage custom capabilities for WPCode.
 *
 * @package WPCode
 */

/**
 * The WPCode_Capabilities class.
 */
class WPCode_Capabilities {
	/**
	 * Function to call on plugin activation.
	 *
	 * @return void
	 */
	public static function add_capabilities() {

		foreach ( self::get_roles() as $role ) {
			if ( $role->has_cap( 'manage_options' ) ) {
				$role->add_cap( 'wpcode_edit_snippets' );
				$role->add_cap( 'wpcode_activate_snippets' );
			}
		}

	}

	/**
	 * Get roles as WP_Role objects.
	 *
	 * @return WP_Role[]
	 */
	public static function get_roles() {
		$roles      = wp_roles()->roles;
		$role_array = array();
		foreach ( array_keys( $roles ) as $role_key ) {
			$role_array[] = get_role( $role_key );
		}

		return $role_array;
	}

	/**
	 * Remove custom capabilities.
	 *
	 * @return void
	 */
	public static function uninstall() {
		foreach ( self::get_roles() as $role ) {
			if ( $role->has_cap( 'wpcode_edit_snippets' ) ) {
				$role->remove_cap( 'wpcode_edit_snippets' );
			}
			if ( $role->has_cap( 'wpcode_activate_snippets' ) ) {
				$role->remove_cap( 'wpcode_activate_snippets' );
			}
		}
	}
}
