Ticket #47616: 47616.1.diff

src/wp-includes/shortcodes.php

@@ -41 +41 @@
  */
 $shortcode_tags = array();
 
+/**
+ * Container to keep track of currently running shortcodes.
+ *
+ * @since 5.7.0
+ *
+ * @var array
+ * @global array $wp_current_shortcode
+ */
+global $wp_current_shortcode;
+
+if ( ! isset( $wp_current_shortcode ) ) {
+        $wp_current_shortcode = array();
+}
+
 /**
  * Adds a new shortcode.
  *
@@ -159 +173 @@
         return false;
 }
 
+/**
+ * Retrieve the name of the current shortcode being processed.
+ *
+ * @since 5.7.0
+ *
+ * @global array $wp_current_shortcode Stores the list of current shortcodes with the current one last.
+ *
+ * @return string Tag name of the current shortcode.
+ */
+function current_shortcode() {
+        global $wp_current_shortcode;
+        return end( $wp_current_shortcode );
+}
+
+/**
+ * Check if a shortcode is currently being processed.
+ *
+ * This function allows detection for any shortcode currently being executed
+ * (despite not being the most recent shortcode to fire, in the case of
+ * shortcodes called from shortcode callbacks) to be verified.
+ *
+ * @since 5.7.0
+ *
+ * @global array $wp_current_shortcode Current shortcode(s).
+ *
+ * @param null|string $tag Optional. Shortcode tag to check. Defaults to null, which
+ *                         checks if any shortcode is currently being run.
+ * @return bool Whether the shortcode is currently in the stack.
+ */
+function doing_shortcode( $tag = null ) {
+        global $wp_current_shortcode;
+
+        if ( null === $tag ) {
+                return ! empty( $wp_current_shortcode );
+        }
+
+        return in_array( $tag, $wp_current_shortcode );
+}
+
 /**
  * Search content for shortcodes and filter shortcodes through their hooks.
  *
@@ -298 +351 @@
  * @access private
  *
  * @global array $shortcode_tags
+ * @global array $wp_current_shortcode
  *
  * @param array $m Regular expression match array.
  * @return string|false Shortcode output on success, false on failure.
  */
 function do_shortcode_tag( $m ) {
-        global $shortcode_tags;
+        global $shortcode_tags, $wp_current_shortcode;
 
         // Allow [[foo]] syntax for escaping a tag.
         if ( '[' === $m[1] && ']' === $m[6] ) {
@@ -320 +374 @@
                 return $m[0];
         }
 
+        $wp_current_shortcode[] = $tag;
+
         /**
          * Filters whether to call a shortcode callback.
          *
@@ -335 +391 @@
          */
         $return = apply_filters( 'pre_do_shortcode_tag', false, $tag, $attr, $m );
         if ( false !== $return ) {
+                array_pop( $wp_current_shortcode );
                 return $return;
         }
 
@@ -352 +409 @@
          * @param array|string $attr   Shortcode attributes array or empty string.
          * @param array        $m      Regular expression match array.
          */
-        return apply_filters( 'do_shortcode_tag', $output, $tag, $attr, $m );
+        $return = apply_filters( 'do_shortcode_tag', $output, $tag, $attr, $m );
+
+        array_pop( $wp_current_shortcode );
+
+        return $return;
 }
 
 /**