Changeset 27016 for trunk/src/wp-includes/query.php

Timestamp:

01/22/2014 10:30:36 PM (11 years ago)

Author:

wonderboymusic

Message:

Let is_attachment() accept an $attachment parameter, similar to is_page() and is_single(). Adds Unit Tests for all 3.

Props alex-ye for the initial patch.
Fixes #24257.

File:

• trunk/src/wp-includes/query.php (modified) (3 diffs)

trunk/src/wp-includes/query.php

@@ -176 +176 @@
  * @uses $wp_query
  *
+ * @param mixed $attachment Attachment ID, title, slug, or array of such.
  * @return bool
  */
-function is_attachment() {
+function is_attachment( $attachment = '' ) {
     global $wp_query;
 
@@ -186 +187 @@
     }
 
-    return $wp_query->is_attachment();
+    return $wp_query->is_attachment( $attachment );
 }
 
@@ -3388 +3389 @@
      * @since 3.1.0
      *
+         * @param mixed $attachment Attachment ID, title, slug, or array of such.
      * @return bool
      */
-    function is_attachment() {
-        return (bool) $this->is_attachment;
+    function is_attachment( $attachment = '' ) {
+        if ( ! $this->is_attachment ) {
+            return false;
+        }
+
+        if ( empty( $attachment ) ) {
+            return true;
+        }
+
+        $attachment = (array) $attachment;
+
+        $post_obj = $this->get_queried_object();
+
+        if ( in_array( $post_obj->ID, $attachment ) ) {
+            return true;
+        } elseif ( in_array( $post_obj->post_title, $attachment ) ) {
+            return true;
+        } elseif ( in_array( $post_obj->post_name, $attachment ) ) {
+            return true;
+        }
+        return false;
     }