Add storage of favourited statuses

Author: toolstack

includes/class-mastodon-api.php

@@ -20,6 +20,15 @@
 class Mastodon_API {
 	const ACTIVITYPUB_USERNAME_REGEXP = '(?:([A-Za-z0-9_-]+)@((?:[A-Za-z0-9_-]+\.)+[A-Za-z]+))';
 	const VERSION = ENABLE_MASTODON_APPS_VERSION;
+
+	/**
+	 * The default reaction for favouriting a status.
+	 *
+	 * 2b50 = star
+	 * 2764 = heart
+	 */
+	const DEFAULT_REACT = '2b50';
+
 	/**
 	 * The OAuth handler.
 	 *
@@ -282,7 +291,7 @@ public function add_rest_routes() {
 			'api/v1/favourites',
 			array(
 				'methods'             => 'GET',
-				'callback'            => '__return_empty_array',
+				'callback'            => array( $this, 'api_favourites'),
 				'permission_callback' => array( $this, 'logged_in_permission' ),
 			)
 		);
@@ -1083,9 +1092,9 @@ function ( $user_id ) {
 				'url'                    => null,
 				'replies_count'          => 0,
 				'reblogs_count'          => 0,
-				'favourites_count'       => 0,
+				'favourites_count'       => $this->get_post_favourite_count( $post->ID ),
 				'edited_at'              => null,
-				'favourited'             => false,
+				'favourited'             => $this->check_if_status_favourited( $user_id, $post->ID ),
 				'reblogged'              => $reblogged,
 				'reblogged_by'           => $reblogged_by,
 				'muted'                  => false,
@@ -1676,33 +1685,68 @@ public function api_get_post_context( $request ) {
 	}
 
 	public function api_favourite_post( $request ) {
+		$user_id = get_current_user_id();
 		$post_id = $request->get_param( 'post_id' );
 		if ( ! $post_id ) {
 			return false;
 		}
 
 		$post_id = $this->maybe_get_remapped_reblog_id( $post_id );
 
-		// 2b50 = star
-		// 2764 = heart
-		do_action( 'mastodon_api_react', $post_id, '2b50' );
+		do_action( 'mastodon_api_react', $post_id, SELF::DEFAULT_REACT );
+
+		$favourite_posts = get_user_meta( $user_id, 'enable-mastodon-apps-favourite-posts', true );
+
+		if ( $favourite_posts == false ) {
+			$favourite_posts = array();
+		} else {
+			$favourite_posts = unserialize( $favourite_posts );
+		}
+
+		$favourite_posts[$post_id] = SELF::DEFAULT_REACT;
+
+		$result = update_user_meta( $user_id, 'enable-mastodon-apps-favourite-posts', serialize( $favourite_posts ) );
+
+		$favourite_count = get_post_meta( $post_id, 'enable-mastodon-apps-favourite-count', true );
+
+		$favourite_count++;
+
+		update_post_meta( $post_id, 'enable-mastodon-apps-favourite-count', $favourite_count );
 
 		$post = get_post( $post_id );
 
 		return $this->get_status_array( $post );
 	}
 
 	public function api_unfavourite_post( $request ) {
+		$user_id = get_current_user_id();
 		$post_id = $request->get_param( 'post_id' );
 		if ( ! $post_id ) {
 			return false;
 		}
 
 		$post_id = $this->maybe_get_remapped_reblog_id( $post_id );
 
-		// 2b50 = star
-		// 2764 = heart
-		do_action( 'mastodon_api_unreact', $post_id, '2b50' );
+		do_action( 'mastodon_api_unreact', $post_id, SELF::DEFAULT_REACT );
+
+		$favourite_posts = get_user_meta( $user_id, 'enable-mastodon-apps-favourite-posts', true );
+
+		if ( $favourite_posts == false ) {
+			$favourite_posts = array();
+		} else {
+			$favourite_posts = unserialize( $favourite_posts );
+		}
+
+		unset( $favourite_posts[$post_id] );
+
+		update_user_meta( $user_id, 'enable-mastodon-apps-favourite-posts', serialize( $favourite_posts ) );
+
+		$favourite_count = get_post_meta( $post_id, 'enable-mastodon-apps-favourite-count', true );
+
+		$favourite_count--;
+		if ( $favourite_count < 0 ) { $favourite_count = 0; }
+
+		update_post_meta( $post_id, 'enable-mastodon-apps-favourite-count', $favourite_count );
 
 		$post = get_post( $post_id );
 
@@ -1811,9 +1855,9 @@ public function convert_activity_to_status( $activity, $user_id ) {
 			'muted'                  => false,
 			'replies_count'          => 0, // could be fetched.
 			'reblogs_count'          => 0,
-			'favourites_count'       => 0,
+			'favourites_count'       => $this->get_post_favourite_count( $post->ID ),
 			'edited_at'              => null,
-			'favourited'             => false,
+			'favourited'             => $this->check_if_status_favourited( $user_id, $id_parts ),
 			'reblogged'              => false,
 			'muted'                  => false,
 			'bookmarked'             => false,
@@ -2845,4 +2889,47 @@ public function api_instance_v2() {
 
 		return apply_filters( 'mastodon_api_instance_v2', $ret );
 	}
+	private function check_if_status_favourited( $user_id, $post_id ) {
+		$favourite_posts = get_user_meta( $user_id, 'enable-mastodon-apps-favourite-posts', true );
+
+		if ( $favourite_posts == false ) {
+			$favourite_posts = array();
+		} else {
+			$favourite_posts = unserialize( $favourite_posts );
+		}
+
+		return array_key_exists( $post_id, $favourite_posts );
+	}
+
+	private function get_post_favourite_count( $post_id ) {
+		return get_post_meta( $post_id, 'enable-mastodon-apps-favourite-count', true );
+	}
+
+	public function api_favourites( $request ) {
+		$user_id = get_current_user_id();
+
+		$favourite_posts = get_user_meta( $user_id, 'enable-mastodon-apps-favourite-posts', true );
+
+		if ( $favourite_posts == false ) {
+			$favourite_posts = array();
+		} else {
+			$favourite_posts = unserialize( $favourite_posts );
+		}
+
+		$args = $this->get_posts_query_args( $request );
+		if ( empty( $args ) ) {
+			return array();
+		}
+		$args = apply_filters( 'mastodon_api_timelines_args', $args, $request );
+
+		$ret = array();
+		foreach( $favourite_posts as $post_id => $value ) {
+			$post = get_post( $post_id, 'OBJECT' );
+			if( is_object( $post ) ) {
+				$ret[] = $this->get_status_array( $post );
+			}
+		}
+
+		return $ret;
+	}
 }