Add methods for retrieving capability names

Author: sheabunge

php/class-code-snippets.php

@@ -153,6 +153,24 @@ public function current_user_can() {
 		return current_user_can( $this->get_cap() );
 	}
 
+	/**
+	 * Retrieve the name of the capability required to manage sub-site snippets
+	 *
+	 * @return string
+	 */
+	public function get_cap_name() {
+		return apply_filters( 'code_snippets_cap', 'manage_snippets' );
+	}
+
+	/**
+	 * Retrieve the name of the capability required to manage network snippets
+	 *
+	 * @return string
+	 */
+	public function get_network_cap_name() {
+		return apply_filters( 'code_snippets_network_cap', 'manage_network_snippets' );
+	}
+
 	/**
 	 * Get the required capability to perform a certain action on snippets.
 	 * Does not check if the user has this capability or not.
@@ -171,11 +189,11 @@ public function get_cap() {
 			/* If multisite is enabled and the snippet menu is not activated,
 			   restrict snippet operations to super admins only */
 			if ( empty( $menu_perms['snippets'] ) ) {
-				return apply_filters( 'code_snippets_network_cap', 'manage_network_snippets' );
+				return $this->get_network_cap_name();
 			}
 		}
 
-		return apply_filters( 'code_snippets_cap', 'manage_snippets' );
+		return $this->get_cap_name();
 	}
 
 	/**
@@ -190,7 +208,7 @@ function grant_network_cap( $user_id ) {
 		$user = new WP_User( $user_id );
 
 		/* Add the capability */
-		$user->add_cap( apply_filters( 'code_snippets_network_cap', 'manage_network_snippets' ) );
+		$user->add_cap( $this->get_network_cap_name() );
 	}
 
 	/**
@@ -205,7 +223,7 @@ function remove_network_cap( $user_id ) {
 		$user = new WP_User( $user_id );
 
 		/* Remove the capability */
-		$user->remove_cap( apply_filters( 'code_snippets_network_cap', 'manage_network_snippets' ) );
+		$user->remove_cap( $this->get_network_cap_name() );
 	}
 
 	/**

php/upgrade.php

@@ -23,7 +23,7 @@ function code_snippets_upgrader() {
 
 		/* Register capabilities */
 		$role = get_role( apply_filters( 'code_snippets_role', 'administrator' ) );
-		$role->add_cap( apply_filters( 'code_snippets_cap', 'manage_snippets' ) );
+		$role->add_cap( code_snippets()->get_cap_name() );
 	}
 
 	/* Check if we have upgraded from an older version */