Add option to change the shortcode namespace

sitepilot · Mar 31, 2022 · 8209f17 · 8209f17
1 parent f5b2c7a
commit 8209f17
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/framework/Traits/HasShortcodes.php b/framework/Traits/HasShortcodes.php
@@ -4,12 +4,19 @@
 
 trait HasShortcodes
 {
+ /**
+ * The shortcode namespace.
+ * 
+ * @var ?string
+ */
+ protected $shortcode_namespace = null;
+
  /**
  * Add namespaced shortcode.
  *
  * @param callable|string $callback
  */
- public function add_shortcode(string $tag, $callback): void
+ public function add_shortcode(string $tag, $callback, $namespaced = true): void
  {
  if (is_string($callback)) {
  $callback = [$this, $callback];
@@ -21,6 +28,11 @@ public function add_shortcode(string $tag, $callback): void
  $app = $this;
  }
 
- add_shortcode($app->namespace($tag, '_'), $callback);
+ add_shortcode(
+ $this->shortcode_namespace
+ ? $this->shortcode_namespace . '_' . $tag
+ : $app->namespace($tag, '_'),
+ $callback
+ );
  }
 }