Capability.php — Advanced Access Manager – Access Governance for WordPress 5.0

advanced-access-manager / Application / Core / Object / Capability.php

advanced-access-manager / Application / Core / Object Last commit date

Capability.php

67 lines

1	<?php
2
3	/**
4	* ======================================================================
5	* LICENSE: This file is subject to the terms and conditions defined in *
6	* file 'license.txt', which is part of this source code package. *
7	* ======================================================================
8	*/
9
10	/**
11	* Capability object
12	*
13	* @package AAM
14	* @author Vasyl Martyniuk <vasyl@vasyltech.com>
15	*/
16	class AAM_Core_Object_Capability extends AAM_Core_Object {
17
18	/**
19	* Constructor
20	*
21	* @param AAM_Core_Subject $subject
22	*
23	* @return void
24	*
25	* @access public
26	*/
27	public function __construct(AAM_Core_Subject $subject) {
28	parent::__construct($subject);
29
30	$this->setOption($this->getSubject()->getCapabilities());
31
32	//check if capabilities are overwritten but only for user subject
33	if (is_a($this->getSubject(), 'AAM_Core_Subject_User')) {
34	$caps = get_user_option(
35	AAM_Core_Subject_User::AAM_CAPKEY, $this->getSubject()->getId()
36	);
37	if (!empty($caps)) {
38	$this->setOverwritten(true);
39	}
40	}
41	}
42
43	/**
44	*
45	* @param type $capability
46	* @param type $granted
47	*/
48	public function save($capability, $granted) {
49	if (intval($granted)) {
50	$result = $this->getSubject()->addCapability($capability);
51	} else {
52	$result = $this->getSubject()->removeCapability($capability);
53	}
54
55	return $result;
56	}
57
58	/**
59	*
60	* @param type $capability
61	* @return type
62	*/
63	public function has($capability) {
64	return $this->getSubject()->hasCapability($capability);
65	}
66
67	}