enhance(directives): Add support for passing multiple roles to @role (

Fixes #49)
Log1x · Jul 29, 2023 · 36b076b · 36b076b
1 parent dbbb0e7
commit 36b076b
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/Directives/WordPress.php b/src/Directives/WordPress.php
@@ -431,8 +431,15 @@
 
     'role' => function ($expression) {
         $expression = Util::parse($expression);
+        $condition = [];
 
-        return "<?php if (is_user_logged_in() && in_array(strtolower({$expression->get(0)}), (array) wp_get_current_user()->roles)) : ?>"; // phpcs:ignore
+        foreach ($expression as $value) {
+            $condition[] = "&& in_array(strtolower({$value}), (array) wp_get_current_user()->roles)";
+        }
+
+        $conditions = implode(' ', $condition);
+
+        return "<?php if (is_user_logged_in() {$conditions}) : ?>"; // phpcs:ignore
     },
 
     'endrole' => function () {