ShortCodeController.php
3.82 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
<?php
/**
* ShortCodeController.php
*
* The ShortCodeController class file.
*
* PHP versions 5
*
* @author Alexander Schneider <alexanderschneider85@gmail.com>
* @copyright 2008-2017 Alexander Schneider
* @license http://www.gnu.org/licenses/gpl-2.0.html GNU General Public License, version 2
* @version SVN: $id$
* @link http://wordpress.org/extend/plugins/user-access-manager/
*/
declare(strict_types=1);
namespace UserAccessManager\Controller\Frontend;
use UserAccessManager\Config\WordpressConfig;
use UserAccessManager\Controller\Controller;
use UserAccessManager\UserGroup\UserGroupHandler;
use UserAccessManager\UserGroup\UserGroupTypeException;
use UserAccessManager\Wrapper\Php;
use UserAccessManager\Wrapper\Wordpress;
/**
* Class ShortCodeController
*
* @package UserAccessManager\Controller\Frontend
*/
class ShortCodeController extends Controller
{
use LoginControllerTrait;
/**
* @var UserGroupHandler
*/
protected $userGroupHandler;
/**
* ShortCodeController constructor.
* @param Php $php
* @param Wordpress $wordpress
* @param WordpressConfig $wordpressConfig
* @param UserGroupHandler $userGroupHandler
*/
public function __construct(
Php $php,
Wordpress $wordpress,
WordpressConfig $wordpressConfig,
UserGroupHandler $userGroupHandler
) {
parent::__construct($php, $wordpress, $wordpressConfig);
$this->userGroupHandler = $userGroupHandler;
}
/**
* @return Wordpress
*/
protected function getWordpress(): Wordpress
{
return $this->wordpress;
}
/**
* Returns the login bar.
* @return string
*/
public function getLoginFormHtml(): string
{
$loginForm = '';
if ($this->wordpress->isUserLoggedIn() === false) {
$loginForm = $this->getIncludeContents('LoginForm.php');
}
return $this->wordpress->applyFilters('uam_login_form', $loginForm);
}
/**
* Handles the login form short code.
* @return string
*/
public function loginFormShortCode(): string
{
return $this->getLoginFormHtml();
}
/**
* Handles the public short code.
* @param string|array $attributes
* @param string $content
* @return string
*/
public function publicShortCode($attributes, $content = ''): string
{
return ($this->wordpress->isUserLoggedIn() === false) ? $this->wordpress->doShortCode($content) : '';
}
/**
* Returns the user group map from the short code attribute.
* @param array $attributes
* @return array
*/
private function getUserGroupsMapFromAttributes(array $attributes): array
{
$userGroups = (isset($attributes['group']) === true) ? explode(',', $attributes['group']) : [];
return (array) array_flip(array_map('trim', $userGroups));
}
/**
* Handles the private short code.
* @param array $attributes
* @param string $content
* @return string
* @throws UserGroupTypeException
*/
public function privateShortCode(array $attributes, $content = ''): string
{
if ($this->wordpress->isUserLoggedIn() === true) {
$userGroupMap = $this->getUserGroupsMapFromAttributes($attributes);
if ($userGroupMap === []) {
return $this->wordpress->doShortCode($content);
}
$userUserGroups = $this->userGroupHandler->getUserGroupsForUser();
foreach ($userUserGroups as $userGroup) {
if (isset($userGroupMap[$userGroup->getId()])
|| isset($userGroupMap[$userGroup->getName()])
) {
return $this->wordpress->doShortCode($content);
}
}
}
return '';
}
}