wp-cli-bbpress/components/engagement.php

<?php
namespace bbPress\CLI\Command;

use WP_CLI;

/**
 * Manage bbPress Engagements.
 *
 * @since 1.0.0
 */
class Engagement extends bbPressCommand {

	/**
	 * Add a topic to user's engagements.
	 *
	 * ## OPTIONS
	 *
	 * --user-id=<user>
	 * : Identifier for the user. Accepts either a user_login or a numeric ID.
	 *
	 * --topic-id=<topic-id>
	 * : Identifier for the topic.
	 *
	 * ## EXAMPLES
	 *
	 *    $ wp bbp engagement add --user-id=5465 --topic-id=65476
	 *    Success: Engagement successfully added.
	 *
	 *    $ wp bbp engagement create --user-id=user_test --topic-id=354354
	 *    Success: Engagement successfully added.
	 *
	 * @alias create
	 */
	public function add( $args, $assoc_args ) {
		$user = $this->get_user_id_from_identifier( $assoc_args['user-id'] );

		// Check if topic exists.
		$topic_id = $assoc_args['topic-id'];
		if ( ! bbp_is_topic( $topic_id ) ) {
			\WP_CLI::error( 'No topic found by that ID.' );
		}

		// True if added.
		if ( bbp_add_user_engagement( $user->ID, $topic_id ) ) {
			\WP_CLI::success( 'Engagement successfully added.' );
		} else {
			\WP_CLI::error( 'Could not add the engagement.' );
		}
	}

	/**
	 * Remove a topic from user's engagements.
	 *
	 * ## OPTIONS
	 *
	 * --user-id=<user>
	 * : Identifier for the user. Accepts either a user_login or a numeric ID.
	 *
	 * --topic-id=<topic-id>
	 * : Identifier for the topic.
	 *
	 * [--yes]
	 * : Answer yes to the confirmation message.
	 *
	 * ## EXAMPLES
	 *
	 *    $ wp bbp engagement remove --user-id=5465 --topic-id=65476
	 *    Success: Engagement successfully removed.
	 *
	 *    $ wp bbp engagement remove --user-id=user_test --topic-id=28468
	 *    Success: Engagement successfully removed.
	 *
	 * @alias delete
	 */
	public function remove( $args, $assoc_args ) {
		$user = $this->get_user_id_from_identifier( $assoc_args['user-id'] );

		// Check if topic exists.
		$topic_id = $assoc_args['topic-id'];
		if ( ! bbp_is_topic( $topic_id ) ) {
			\WP_CLI::error( 'No topic found by that ID.' );
		}

		\WP_CLI::confirm( 'Are you sure you want to remove this topic from the user\'s engagement list?', $assoc_args );

		// True if removed.
		if ( bbp_remove_user_engagement( $user->ID, $topic_id ) ) {
			\WP_CLI::success( 'Engagement successfully removed.' );
		} else {
			\WP_CLI::error( 'Could not remove engagement.' );
		}
	}

	/**
	 * List the users who have engaged in a topic.
	 *
	 * ## OPTIONS
	 *
	 * <topic-id>
	 * : Identifier for the topic.
	 *
	 * [--format=<format>]
	 * : Render output in a particular format.
	 * ---
	 * default: count
	 * options:
	 *   - ids
	 *   - count
	 *   - json
	 *   - haml
	 * ---
	 *
	 * ## EXAMPLES
	 *
	 *     $ wp bbp subscription list_users 242
	 *     3
	 *
	 *     $ wp bbp subscription list_users 45765 --format=ids
	 *     54564 465465 65465
	 *
	 * @subcommand list-users
	 */
	public function list_users( $args, $assoc_args ) {
		$formatter = $this->get_formatter( $assoc_args );

		$topic_id = $args[0];

		// Check if topic exists.
		if ( ! bbp_is_topic( $topic_id ) ) {
			\WP_CLI::error( 'No topic found by that ID.' );
		}

		$ids = bbp_get_topic_engagements( $topic_id );

		if ( ! $ids ) {
			\WP_CLI::error( 'Could not find any users.' );
		}

		if ( 'ids' === $formatter->format ) {
			echo implode( ' ', $ids ); // WPCS: XSS ok.
		} elseif ( 'count' === $formatter->format ) {
			$formatter->display_items( $ids );
		}
	}

	/**
	 * List user's topics engagements.
	 *
	 * ## OPTIONS
	 *
	 * <user>
	 * : Identifier for the user. Accepts either a user_login or a numeric ID.
	 *
	 * [--format=<format>]
	 * : Render output in a particular format.
	 * ---
	 * default: table
	 * options:
	 *   - table
	 *   - ids
	 *   - count
	 *   - json
	 *   - haml
	 * ---
	 *
	 * ## EXAMPLES
	 *
	 *     $ wp bbp favorite list 456
	 *     $ wp bbp favorite list 456546 --format=ids
	 *
	 * @subcommand list
	 */
	public function _list( $args, $assoc_args ) {
		$formatter = $this->get_formatter( $assoc_args );
		$user      = $this->get_user_id_from_identifier( $args[0] );
		$topics    = bbp_get_user_engagements( $this->user_args( $user->ID ) );

		if ( 'ids' === $formatter->format ) {
			echo implode( ' ', wp_list_pluck( $topics->posts, 'ID' ) ); // WPCS: XSS ok.
		} elseif ( 'count' === $formatter->format ) {
			$formatter->display_items( $topics->found_posts );
		} else {
			$topics = array_map( function( $post ) {
				$post->url = get_permalink( $post->ID );
				return $post;
			}, $topics->posts );
			$formatter->display_items( $topics->posts );
		}
	}

	/**
	 * Recalculate all of the users who have engaged in a topic.
	 *
	 * ## OPTIONS
	 *
	 * <topic-id>
	 * : Identifier of the topic to recalculate.
	 *
	 * ## EXAMPLE
	 *
	 *    $ wp bbp engagement recalculate 132
	 *    Success: Engagement successfully recalculated.
	 */
	public function recalculate( $args, $assoc_args ) {
		// Check if topic exists.
		$topic_id = $args[0];
		if ( ! bbp_is_topic( $topic_id ) ) {
			\WP_CLI::error( 'No topic found by that ID.' );
		}

		if ( bbp_recalculate_topic_engagements( $topic_id, true ) ) {
			\WP_CLI::success( 'Engagements successfully recalculated.' );
		} else {
			\WP_CLI::error( 'Could not recalculate engagements.' );
		}
	}
}