From 35e6f47f3852ee0114c100b599c2a6740c598095 Mon Sep 17 00:00:00 2001
From: Oscar Sanchez <oscar.sanchez@10up.com>
Date: Fri, 29 May 2020 21:44:08 -0500
Subject: [PATCH] Adds ability to retrieve avatar with WP_Post object

---
 simple-local-avatars.php | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/simple-local-avatars.php b/simple-local-avatars.php
index a1724912..565b58c5 100644
--- a/simple-local-avatars.php
+++ b/simple-local-avatars.php
@@ -53,10 +53,10 @@ public function __construct() {
 	}
 
 	/**
-	 * Retrieve the local avatar for a user who provided a user ID or email address.
+	 * Retrieve the local avatar for a user who provided a user ID, email address or post/comment object.
 	 *
 	 * @param string $avatar Avatar return by original function
-	 * @param int|string|object $id_or_email A user ID,  email address, or comment object
+	 * @param int|string|object $id_or_email A user ID, email address, or post/comment object
 	 * @param int $size Size of the avatar image
 	 * @param string $default URL to a default image to use if no avatar is available
 	 * @param string $alt Alternative text to use in image tag. Defaults to blank
@@ -69,7 +69,9 @@ public function get_avatar( $avatar = '', $id_or_email = '', $size = 96, $defaul
 			$user_id = $user->ID;
 		elseif ( is_object( $id_or_email ) && ! empty( $id_or_email->user_id ) )
 			$user_id = (int) $id_or_email->user_id;
-		
+		elseif ( $id_or_email instanceof WP_Post && ! empty( $id_or_email->post_author ) )
+			$user_id = (int) $id_or_email->post_author;
+
 		if ( empty( $user_id ) )
 			return $avatar;
 
@@ -592,9 +594,9 @@ function get_simple_local_avatar( $id_or_email, $size = 96, $default = '', $alt
 if ( ! function_exists( 'get_avatar' ) && ( $simple_local_avatars_options = get_option('simple_local_avatars') ) && ! empty( $simple_local_avatars_options['only'] ) ) :
 
 	/**
-	 * Retrieve the avatar for a user who provided a user ID or email address.
+	 * Retrieve the avatar for a user who provided a user ID, post/comment object or email address.
 	 *
-	 * @param int|string|object $id_or_email A user ID,  email address, or comment object
+	 * @param int|string|object $id_or_email A user ID, email address, or post/comment object
 	 * @param int $size Size of the avatar image
 	 * @param string $default URL to a default image to use if no avatar is available
 	 * @param string $alt Alternative text to use in image tag. Defaults to blank