Feature: Save google profile picture of the user during account creation

src/Plugin.php

@@ -112,6 +112,8 @@ public function run(): void {
 		add_action( 'init', [ $this, 'load_translations' ] );
 
 		add_action( 'plugin_action_links_' . plugin_basename( $this->path ) . '/login-with-google.php', [ $this, 'add_plugin_action_links' ] );
+
+		add_action( 'get_avatar_url', [ $this, 'return_avatar_url' ], 10, 3 );
 	}
 
 	/**
@@ -162,4 +164,48 @@ public function add_plugin_action_links( $actions ) {
 
 		return array_merge( $new_actions, $actions );
 	}
+
+	/**
+	 * Return the stored profile picture during the account creation.
+	 *
+	 * @param string $url The URL of the avatar.
+	 * @param mixed  $id_or_email The avatar to retrieve. Accepts a user ID, Gravatar SHA-256 or MD5 hash, user email, WP_User object, WP_Post object, or WP_Comment object.
+	 * @param array  $args Arguments passed to get_avatar_data() , after processing.
+	 *
+	 * @return string The URL of the avatar.
+	 */
+	public function return_avatar_url( $url, $id_or_email, $args ): string {
+		/**
+		 * Filter to bypass the use of saved profile picture for avatar.
+		 *
+		 * @since n.e.x.t
+		 *
+		 * @param boolean $use_saved_profile_picture_for_avatar Whether to bypass the use the saved profile picture for avatar or not.
+		 */
+		$use_avatar_url = apply_filters( 'rtcamp.google_use_saved_profile_picture_for_avatar', true );
+
+		if ( ! $use_avatar_url ) {
+			return $url;
+		}
+
+		$wp_user = null;
+		if ( is_int( $id_or_email ) ) {
+			$wp_user = get_user_by( 'id', $id_or_email );
+		} elseif ( is_string( $id_or_email ) && is_email( $id_or_email ) ) {
+			$wp_user = get_user_by( 'email', $id_or_email );
+		}
+
+		if ( $wp_user ) {
+			$width  = isset( $args['width'] ) ? absint( $args['width'] ) : 64;
+			$height = isset( $args['height'] ) ? absint( $args['height'] ) : 64;
+
+			$profile_picture_id = get_user_meta( $wp_user->ID, 'rtlwg_profile_picture_id', true );
+
+			if ( ! empty( $profile_picture_id ) ) {
+				$url = wp_get_attachment_image_url( $profile_picture_id, [ $width, $height ] );
+			}
+		}
+
+		return $url;
+	}
 }

src/Utils/Authenticator.php

@@ -112,6 +112,21 @@ public function register( stdClass $user ): ?WP_User {
 					]
 				);
 
+				/**
+				 * Filter to bypass the profile picture saving process.
+				 *
+				 * @since n.e.x.t
+				 *
+				 * @param boolean $save Whether to save profile picture or not.
+				 * @param int $uid WP User ID.
+				 * @param \stdClass User object return by Google.
+				 */
+				$save_profile_picture = apply_filters( 'rtcamp.google_save_user_profile_picture', true, $uid, $user );
+
+				if ( $save_profile_picture ) {
+					$this->save_user_profile_picture( $uid, $user );
+				}
+
 				/**
 				 * Fires once the user has been registered successfully.
 				 */
@@ -179,4 +194,57 @@ private function can_register_with_email( string $email ): bool {
 
 		return in_array( $email_parts[1], $whitelisted_domains, true );
 	}
+
+	/**
+	 * Save user profile picture.
+	 *
+	 * @param int $user_id WP User ID.
+	 * @param \stdClass $user User object returned by google.
+	 * @return void
+	 */
+	private function save_user_profile_picture( $user_id, $user ): void {
+		global $wp_filesystem;
+
+		if ( is_null( $wp_filesystem ) ) {
+			require_once ABSPATH . '/wp-admin/includes/file.php';
+			WP_Filesystem();
+		}
+
+		require_once ABSPATH . 'wp-admin/includes/media.php';
+		require_once ABSPATH . 'wp-admin/includes/file.php';
+		require_once ABSPATH . 'wp-admin/includes/image.php';
+
+		// Using larger image size. By default, profile picture has 96 width size with cropped.
+		$profile_picture_url = str_replace( '=s96-c', '', $user->picture );
+
+		$profile_picture_filename = download_url( $profile_picture_url );
+
+		if ( str_ends_with( $profile_picture_filename, '.tmp' ) ) {
+			$profile_picture_mime_type = wp_get_image_mime( $profile_picture_filename );
+
+			$mime_types = wp_get_mime_types();
+			foreach ( $mime_types as $ext => $mime_type ) {
+				if ( $profile_picture_mime_type === $mime_type ) {
+					$profile_picture_extension = current( explode( '|', $ext ) );
+					break;
+				}
+			}
+
+			$new_profile_picture_filename = str_replace( '.tmp', ".{$profile_picture_extension}", $profile_picture_filename );
+			$wp_filesystem->move( $profile_picture_filename, $new_profile_picture_filename, true );
+
+			$profile_picture_filename = $new_profile_picture_filename;
+		}
+
+		$file_array = array(
+			'name'     => basename( $profile_picture_filename ),
+			'tmp_name' => $profile_picture_filename,
+		);
+
+		$attachment_id = media_handle_sideload( $file_array );
+
+		if ( is_int( $attachment_id ) ) {
+			update_user_meta( $user_id, 'rtlwg_profile_picture_id', $attachment_id );
+		}
+	}
 }