wordpress-activitypub/includes/collection/class-followers.php

<?php
namespace Activitypub\Collection;

use WP_Error;
use WP_Term_Query;
use Activitypub\Webfinger;
use Activitypub\Model\Activity;
use Activitypub\Model\Follower;

use function Activitypub\safe_remote_get;
use function Activitypub\safe_remote_post;

/**
 * ActivityPub Followers Collection
 *
 * @author Matthias Pfefferle
 */
class Followers {
	const TAXONOMY = 'activitypub-followers';

	/**
	 * Register WordPress hooks/actions and register Taxonomy
	 *
	 * @return void
	 */
	public static function init() {
		// register "followers" taxonomy
		self::register_taxonomy();

		\add_action( 'activitypub_inbox_follow', array( self::class, 'handle_follow_request' ), 10, 2 );
		\add_action( 'activitypub_inbox_undo', array( self::class, 'handle_undo_request' ), 10, 2 );

		\add_action( 'activitypub_followers_post_follow', array( self::class, 'send_follow_response' ), 10, 4 );
	}

	/**
	 * Register the "Followers" Taxonomy
	 *
	 * @return void
	 */
	public static function register_taxonomy() {
		$args = array(
			'labels'            => array(
				'name'          => _x( 'Followers', 'taxonomy general name', 'activitypub' ),
				'singular_name' => _x( 'Followers', 'taxonomy singular name', 'activitypub' ),
				'menu_name'     => __( 'Followers', 'activitypub' ),
			),
			'hierarchical'      => false,
			'show_ui'           => false,
			'show_in_menu'      => false,
			'show_in_nav_menus' => false,
			'show_admin_column' => false,
			'query_var'         => false,
			'rewrite'           => false,
			'public'            => false,
			'capabilities'      => array(
				'edit_terms' => null,
			),
		);

		register_taxonomy( self::TAXONOMY, 'user', $args );
		register_taxonomy_for_object_type( self::TAXONOMY, 'user' );

		register_term_meta(
			self::TAXONOMY,
			'name',
			array(
				'type'              => 'string',
				'single'            => true,
				//'sanitize_callback' => array( self::class, 'validate_displayname' ),
			)
		);

		register_term_meta(
			self::TAXONOMY,
			'username',
			array(
				'type'              => 'string',
				'single'            => true,
				//'sanitize_callback' => array( self::class, 'validate_username' ),
			)
		);

		register_term_meta(
			self::TAXONOMY,
			'avatar',
			array(
				'type'              => 'string',
				'single'            => true,
				//'sanitize_callback' => array( self::class, 'validate_avatar' ),
			)
		);

		register_term_meta(
			self::TAXONOMY,
			'inbox',
			array(
				'type'              => 'string',
				'single'            => true,
				//'sanitize_callback' => array( self::class, 'validate_inbox' ),
			)
		);

		register_term_meta(
			self::TAXONOMY,
			'updated_at',
			array(
				'type'              => 'string',
				'single'            => true,
				//'sanitize_callback' => array( self::class, 'validate_updated_at' ),
			)
		);

		do_action( 'activitypub_after_register_taxonomy' );
	}

	/**
	 * Handle the "Follow" Request
	 *
	 * @param array $object    The JSON "Follow" Activity
	 * @param int   $user_id The ID of the WordPress User
	 * @return void
	 */
	public static function handle_follow_request( $object, $user_id ) {
		// save follower
		$follower = self::add_follower( $user_id, $object['actor'] );

		do_action( 'activitypub_followers_post_follow', $object['actor'], $object, $user_id, $follower );
	}

	/**
	 * Handles "Unfollow" requests
	 *
	 * @param  array $object  The JSON "Undo" Activity
	 * @param  int   $user_id The ID of the WordPress User
	 */
	public static function handle_undo_request( $object, $user_id ) {
		if (
			isset( $object['object'] ) &&
			isset( $object['object']['type'] ) &&
			'Follow' === $object['object']['type']
		) {
			self::remove_follower( $user_id, $object['actor'] );
		}
	}

	/**
	 * Add a new Follower
	 *
	 * @param int    $user_id The WordPress user
	 * @param string $actor   The Actor URL
	 * @return array|WP_Error The Follower (WP_Term array) or an WP_Error
	 */
	public static function add_follower( $user_id, $actor ) {
		$follower = new Follower( $actor );
		$follower->upsert();

		$result = wp_set_object_terms( $user_id, $follower->get_actor(), self::TAXONOMY, true );

		if ( is_wp_error( $result ) ) {
			return $result;
		} else {
			return $follower;
		}
	}

	/**
	 * Remove a Follower
	 *
	 * @param  int           $user_id The WordPress user_id
	 * @param  string        $actor   The Actor URL
	 * @return bool|WP_Error          True on success, false or WP_Error on failure.
	 */
	public static function remove_follower( $user_id, $actor ) {
		return wp_remove_object_terms( $user_id, $actor, self::TAXONOMY );
	}

	/**
	 * Remove a Follower
	 *
	 * @param  string                      $actor The Actor URL
	 * @return \Activitypub\Model\Follower        The Follower object
	 */
	public static function get_follower( $actor ) {
		$term = get_term_by( 'name', $actor, self::TAXONOMY );

		return new Follower( $term->name );
	}

	/**
	 * Send Accept response
	 *
	 * @param string                     $actor    The Actor URL
	 * @param array                      $object   The Activity object
	 * @param int                        $user_id  The WordPress user_id
	 * @param Activitypub\Model\Follower $follower The Follower object
	 * @return void
	 */
	public static function send_follow_response( $actor, $object, $user_id, $follower ) {
		//if ( is_wp_error( $follower ) ) {
			// @todo send error message
		//}

		// get inbox
		$inbox = $follower->get_inbox();

		// send "Accept" activity
		$activity = new Activity( 'Accept' );
		$activity->set_object( $object );
		$activity->set_actor( \get_author_posts_url( $user_id ) );
		$activity->set_to( $actor );
		$activity->set_id( \get_author_posts_url( $user_id ) . '#follow-' . \preg_replace( '~^https?://~', '', $actor ) );

		$activity = $activity->to_simple_json();
		$response = safe_remote_post( $inbox, $activity, $user_id );
	}

	/**
	 * Get the Followers of a given user
	 *
	 * @param  int    $user_id The WordPress user_id
	 * @param  string $output  The output format, supported ARRAY_N, OBJECT and ACTIVITYPUB_OBJECT
	 * @param  int    $number  Limts the result
	 * @param  int    $offset  Offset
	 * @return array           The Term list of Followers, the format depends on $output
	 */
	public static function get_followers( $user_id, $output = ARRAY_N, $number = null, $offset = null ) {
		//self::migrate_followers( $user_id );

		$terms = new WP_Term_Query(
			array(
				'taxonomy'   => self::TAXONOMY,
				'hide_empty' => false,
				'object_ids' => $user_id,
				'number'     => $number,
				'offset'     => $offset,
			)
		);

		$items = array();

		// change output format
		switch ( $output ) {
			case ACTIVITYPUB_OBJECT:
				foreach ( $terms->get_terms() as $follower ) {
					$items[] = new Follower( $follower->name ); // phpcs:ignore
				}
				return $items;
			case OBJECT:
				return $terms->get_terms();
			case ARRAY_N:
			default:
				foreach ( $terms->get_terms() as $follower ) {
					$items[] = $follower->name; // phpcs:ignore
				}
				return $items;
		}
	}

	/**
	 * Count the total number of followers
	 *
	 * @param  int $user_id The WordPress user
	 * @return int          The number of Followers
	 */
	public static function count_followers( $user_id ) {
		return count( self::get_followers( $user_id ) );
	}

	public static function get_inboxes( $user_id ) {
		// get all Followers of a WordPress user
		$terms = new WP_Term_Query(
			array(
				'taxonomy'   => self::TAXONOMY,
				'hide_empty' => false,
				'object_ids' => $user_id,
				'fields'     => 'ids',
			)
		);

		$terms = $terms->get_terms();

		global $wpdb;
		$results = $wpdb->get_col(
			$wpdb->prepare(
				"SELECT DISTINCT meta_value FROM {$wpdb->termmeta}
				WHERE term_id IN (" . implode( ', ', array_fill( 0, count( $terms ), '%d' ) ) . ")
				AND meta_key = 'shared_inbox'
				AND meta_value IS NOT NULL",
				$terms
			)
		);

		return array_filter( $results );
	}

	/**
	 * Undocumented function
	 *
	 * @param [type] $user_id
	 * @return void
	 */
	public static function migrate_followers( $user_id ) {
		$followes = get_user_meta( $user_id, 'activitypub_followers', true );

		if ( $followes ) {
			foreach ( $followes as $follower ) {
				self::add_follower( $user_id, $follower );
			}
		}
	}
}