Changeset 31107 for trunk/src/wp-includes/link-template.php

Timestamp:

01/09/2015 04:42:48 AM (7 years ago)

Author:

pento

Message:

Add get_avatar_url(), for retrieving just the URL of an avatar, rather than the entire <img> tag that get_avatar() produces.

Unlike get_avatar(), get_avatar_url() is not pluggable. It can be extended/or modified through the new filters included.

Fixes #21195.

Props mdawaffe, pento, pathawks, DrewAPicture

File:

• trunk/src/wp-includes/link-template.php (modified) (1 diff)

trunk/src/wp-includes/link-template.php

@@ -3325 +3325 @@
     }
 }
+
+
+/**
+ * Retrieve the avatar URL.
+ *
+ * @since 4.2.0
+ *
+ * @param mixed $id_or_email The Gravatar to retrieve a URL for. Accepts a user_id, gravatar md5 hash,
+ *                           user email, WP_User object, WP_Post object, or comment object.
+ * @param array $args        {
+ *     Optional. Arguments to return instead of the default arguments.
+ *
+ *     @type int    $size           Height and width of the avatar in pixels. Default 96.
+ *     @type string $default        URL for the default image or a default type. Accepts '404' (return
+ *                                  a 404 instead of a default image), 'retro' (8bit), 'monsterid' (monster),
+ *                                  'wavatar' (cartoon face), 'indenticon' (the "quilt"), 'mystery', 'mm',
+ *                                  or 'mysterman' (The Oyster Man), 'blank' (transparent GIF), or
+ *                                  'gravatar_default' (the Gravatar logo). Default is the value of the
+ *                                  'avatar_default' option, with a fallback of 'mystery'.
+ *     @type bool   $force_default  Whether to always show the default image, never the Gravatar. Default false.
+ *     @type string $rating         What rating to display avatars up to. Accepts 'G', 'PG', 'R', 'X', and are
+ *                                  judged in that order. Default is the value of the 'avatar_rating' option.
+ *     @type string $scheme         URL scheme to use. See {@see set_url_scheme()} for accepted values.
+ *                                  Default null.
+ *     @type array  $processed_args When the function returns, the value will be the processed/sanitized $args
+ *                                  plus a "found_avatar" guess. Pass as a reference. Default null.
+ * }
+ *
+ * @return false|string The URL of the avatar we found, or false if we couldn't find an avatar.
+ */
+function get_avatar_url( $id_or_email, $args = null ) {
+    $args = get_avatar_data( $id_or_email, $args );
+    return $args['url'];
+}
+
+/**
+ * Retrieve default data about the avatar.
+ *
+ * @since 4.2.0
+ *
+ * @param mixed $id_or_email The Gravatar to check the data against. Accepts a user_id, gravatar md5 hash,
+ *                           user email, WP_User object, WP_Post object, or comment object.
+ * @param array $args        {
+ *     Optional. Arguments to return instead of the default arguments.
+ *
+ *     @type int    $size           Height and width of the avatar in pixels. Default 96.
+ *     @type string $default        URL for the default image or a default type. Accepts '404' (return
+ *                                  a 404 instead of a default image), 'retro' (8bit), 'monsterid' (monster),
+ *                                  'wavatar' (cartoon face), 'indenticon' (the "quilt"), 'mystery', 'mm',
+ *                                  or 'mysterman' (The Oyster Man), 'blank' (transparent GIF), or
+ *                                  'gravatar_default' (the Gravatar logo). Default is the value of the
+ *                                  'avatar_default' option, with a fallback of 'mystery'.
+ *     @type bool   $force_default  Whether to always show the default image, never the Gravatar. Default false.
+ *     @type string $rating         What rating to display avatars up to. Accepts 'G', 'PG', 'R', 'X', and are
+ *                                  judged in that order. Default is the value of the 'avatar_rating' option.
+ *     @type string $scheme         URL scheme to use. See {@see set_url_scheme()} for accepted values.
+ *                                  Default null.
+ *     @type array  $processed_args When the function returns, the value will be the processed/sanitized $args
+ *                                  plus a "found_avatar" guess. Pass as a reference. Default null.
+ * }
+ *
+ * @return array $processed_args {
+ *     Along with the arguments passed in $args, this will contain a couple of extra arguments.
+ *
+ *     @type bool         $found_avatar True if we were able to find an avatar for this user,
+ *                                      false or not set if we couldn't.
+ *     @type false|string $url          The URL of the avatar we found, or false if we couldn't find an avatar.
+ * }
+ */
+function get_avatar_data( $id_or_email, $args = null ) {
+    $args = wp_parse_args( $args, array(
+        'size'           => 96,
+        'default'        => get_option( 'avatar_default', 'mystery' ),
+        'force_default'  => false,
+        'rating'         => get_option( 'avatar_rating' ),
+        'scheme'         => null,
+        'processed_args' => null, // if used, should be a reference
+    ) );
+
+    if ( is_numeric( $args['size'] ) ) {
+        $args['size'] = absint( $args['size'] );
+        if ( ! $args['size'] ) {
+            $args['size'] = 96;
+        }
+    } else {
+        $args['size'] = 96;
+    }
+
+    if ( empty( $args['default'] ) ) {
+        $args['default'] = 'mystery';
+    }
+
+    switch ( $args['default'] ) {
+        case 'mm' :
+        case 'mystery' :
+        case 'mysteryman' :
+            $args['default'] = 'mm';
+            break;
+        case 'gravatar_default' :
+            $args['default'] = false;
+            break;
+    }
+
+    $args['force_default'] = (bool) $args['force_default'];
+
+    $args['rating'] = strtolower( $args['rating'] );
+
+    $args['found_avatar'] = false;
+
+    /**
+     * Filter whether to retrieve the avatar URL early.
+     *
+     * Passing a non-null value in the 'url' member of the return array will
+     * effectively short circuit {@see get_avatar_data()}, passing the value
+     * through the 'get_avatar_data' filter and returning early.
+     *
+     * @since 4.2.0
+     *
+     * @param array             $args          Arguments passed to get_avatar_data(), after processing.
+     * @param int|object|string $id_or_email   A user ID, email address, or comment object.
+     */
+    $args = apply_filters( 'pre_get_avatar_data', $args, $id_or_email );
+
+    if ( isset( $args['url'] ) && ! is_null( $args['url'] ) ) {
+        /** This filter is documented in src/wp-includes/link-template.php */
+        return apply_filters( 'get_avatar_data', $args, $id_or_email );
+    }
+
+    $email_hash = '';
+    $user = $email = false;
+
+    // Process the user identifier.
+    if ( is_numeric( $id_or_email ) ) {
+        $user = get_user_by( 'id', absint( $id_or_email ) );
+    } elseif ( is_string( $id_or_email ) ) {
+        if ( strpos( $id_or_email, '@md5.gravatar.com' ) ) {
+            // md5 hash
+            list( $email_hash ) = explode( '@', $id_or_email );
+        } else {
+            // email address
+            $email = $id_or_email;
+        }
+    } elseif ( $id_or_email instanceof WP_User ) {
+        // User Object
+        $user = $id_or_email;
+    } elseif ( $id_or_email instanceof WP_Post ) {
+        // Post Object
+        $user = get_user_by( 'id', (int) $id_or_email->post_author );
+    } elseif ( is_object( $id_or_email ) && isset( $id_or_email->comment_ID ) ) {
+        // Comment Object
+
+        /**
+         * Filter the list of allowed comment types for retrieving avatars.
+         *
+         * @since 3.0.0
+         *
+         * @param array $types An array of content types. Default only contains 'comment'.
+         */
+        $allowed_comment_types = apply_filters( 'get_avatar_comment_types', array( 'comment' ) );
+        if ( ! empty( $id_or_email->comment_type ) && ! in_array( $id_or_email->comment_type, (array) $allowed_comment_types ) ) {
+            $args['url'] = false;
+            /** This filter is documented in src/wp-includes/link-template.php */
+            return apply_filters( 'get_avatar_data', $args, $id_or_email );
+        }
+
+        if ( ! empty( $id_or_email->user_id ) ) {
+            $user = get_user_by( 'id', (int) $id_or_email->user_id );
+        }
+        if ( ( ! $user || is_wp_error( $user ) ) && ! empty( $id_or_email->comment_author_email ) ) {
+            $email = $id_or_email->comment_author_email;
+        }
+    }
+
+    if ( ! $email_hash ) {
+        if ( $user ) {
+            $email = $user->user_email;
+        }
+
+        if ( $email ) {
+            $email_hash = md5( strtolower( trim( $email ) ) );
+        }
+    }
+
+    if ( $email_hash ) {
+        $args['found_avatar'] = true;
+        $gravatar_server = hexdec( $email_hash[0] ) % 3;
+    } else {
+        $gravatar_server = rand( 0, 2 );
+    }
+
+    $url_args = array(
+        's' => $args['size'],
+        'd' => $args['default'],
+        'f' => $args['force_default'] ? 'y' : false,
+        'r' => $args['rating'],
+    );
+
+    $url = sprintf( 'http://%d.gravatar.com/avatar/%s', $gravatar_server, $email_hash );
+
+    $url = add_query_arg(
+        rawurlencode_deep( array_filter( $url_args ) ),
+        set_url_scheme( $url, $args['scheme'] )
+    );
+
+    /**
+     * Filter the avatar URL.
+     *
+     * @since 4.2.0
+     *
+     * @param string            $url           The URL of the avatar.
+     * @param int|object|string $id_or_email   A user ID, email address, or comment object.
+     * @param array             $args          Arguments passed to get_avatar_data(), after processing.
+     */
+    $args['url'] = apply_filters( 'get_avatar_url', $url, $id_or_email, $args );
+
+    /**
+     * Filter the avatar data.
+     *
+     * @since 4.2.0
+     *
+     * @param array             $args          Arguments passed to get_avatar_data(), after processing.
+     * @param int|object|string $id_or_email   A user ID, email address, or comment object.
+     */
+    $args = apply_filters( 'get_avatar_data', $args, $id_or_email );
+
+    // Don't return a broken URL if we couldn't find the email hash, and none of the filters returned a different URL.
+    if ( ! $email_hash && $url === $args['url'] ) {
+        $args['url'] = false;
+    }
+
+    return $args;
+}