1#@!#!123s

https 1#@!#!123s
: / dom737635 / wp-content / plugins / the-events-calendar / src / Tribe /
Filename : Amalgamator.php
back
<?php
/**
 * Merge pre-3.0 duplicate venues and organizers
 */
use Tribe\Events\Admin\Settings as Plugin_Settings;

class Tribe__Events__Amalgamator {
	private $default_venue = 0;
	private $default_community_venue = 0;
	private $default_organizer = 0;
	private $default_community_organizer = 0;

	/**
	 * constructor
	 */
	public function __construct() {
		$this->default_venue     = (int) tribe_get_option( 'eventsDefaultVenueID', 0 );
		$this->default_organizer = (int) tribe_get_option( 'eventsDefaultOrganizerID', 0 );

		if ( class_exists( 'Tribe__Events__Community__Main' ) ) {
			$community                         = Tribe__Events__Community__Main::instance();
			$this->default_community_venue     = (int) $community->getOption( 'defaultCommunityVenueID', 0 );
			$this->default_community_organizer = (int) $community->getOption( 'defaultCommunityOrganizerID', 0 );
		}
	}

	/**
	 * Merge all duplicate event-related posts
	 *
	 */
	public function merge_duplicates() {
		/**
		 * Whether duplicate organizers should be merged.
		 *
		 * @since 5.14.2
		 *
		 * @param boolean $merge_organizers Whether duplicate organizers should be merged, default true.
		 */
		$merge_organizers = (bool) apply_filters( 'tribe_merge_identical_organizers_enabled', true );

		if ( tribe_is_truthy( $merge_organizers ) ) {
			$this->merge_identical_organizers();
		}

		/**
		 * Whether duplicate venues should be merged.
		 *
		 * @since 5.14.2
		 *
		 * @param boolean $merge_venues Whether duplicate venues should be merged, default true.
		 */
		$merge_venues = (bool) apply_filters( 'tribe_merge_identical_venues_enabled', true );

		if ( tribe_is_truthy( $merge_venues ) ) {
			$this->merge_identical_venues();
		}

		$events = Tribe__Events__Main::instance();
		wp_cache_flush();
	}

	/**
	 * Merge identical organizers
	 *
	 */
	public function merge_identical_organizers() {
		$titles  = $this->get_redundant_titles( Tribe__Events__Main::ORGANIZER_POST_TYPE );
		$buckets = [];

		foreach ( $titles as $t ) {
			$organizer_ids = $this->get_posts_with_title( $t, Tribe__Events__Main::ORGANIZER_POST_TYPE );

			foreach ( $organizer_ids as $id ) {
				$post = get_post( $id );
				$data = [
					'title'             => $post->post_title,
					'status'            => $post->post_status,
					'content'           => $post->post_content,
					'_OrganizerPhone'   => get_post_meta( $id, '_OrganizerPhone', true ),
					'_OrganizerWebsite' => get_post_meta( $id, '_OrganizerWebsite', true ),
					'_OrganizerEmail'   => get_post_meta( $id, '_OrganizerEmail', true ),
				];

				/**
				 * Filter the fields that should be used for comparison when checking for duplicates.
				 *
				 * @since 5.14.2
				 *
				 * @param array<string|string> $data The array of fields with values to be checked.
				 * @param integer              $id   The post id of the organizer being merged.
				 */
				$data = apply_filters( 'tribe_merge_identical_organizers_fields', $data, $id );

				$hash = md5( serialize( $data ) );

				if ( ! isset( $buckets[ $hash ] ) ) {
					$buckets[ $hash ] = [];
				}

				// prioritize organizers with an eventbrite id
				$eventbrite = get_post_meta( $id, '_OrganizerEventBriteID', true );
				if ( empty( $eventbrite ) ) {
					array_push( $buckets[ $hash ], $id );
				} else {
					array_unshift( $buckets[ $hash ], $id );
				}
			}
		}

		foreach ( $buckets as $organizer_ids ) {
			$this->amalgamate_organizers( $organizer_ids );
		}
	}

	/**
	 * Merge identical venues
	 *
	 */
	public function merge_identical_venues() {
		$titles  = $this->get_redundant_titles( Tribe__Events__Main::VENUE_POST_TYPE );
		$buckets = [];

		foreach ( $titles as $t ) {
			$venue_ids = $this->get_posts_with_title( $t, Tribe__Events__Main::VENUE_POST_TYPE );

			foreach ( $venue_ids as $id ) {
				$post = get_post( $id );
				$data = [
					'title'          => $post->post_title,
					'status'         => $post->post_status,
					'content'        => $post->post_content,
					'_VenueAddress'  => get_post_meta( $id, '_VenueAddress', true ),
					'_VenueCity'     => get_post_meta( $id, '_VenueCity', true ),
					'_VenueProvince' => get_post_meta( $id, '_VenueProvince', true ),
					'_VenueState'    => get_post_meta( $id, '_VenueState', true ),
					'_VenueCountry'  => get_post_meta( $id, '_VenueCountry', true ),
					'_VenueZip'      => get_post_meta( $id, '_VenueZip', true ),
					'_VenuePhone'    => get_post_meta( $id, '_VenuePhone', true ),
					'_VenueURL'      => get_post_meta( $id, '_VenueURL', true ),
				];

				/**
				 * Filter the fields that should be used for comparison when checking for duplicates.
				 *
				 * @since 5.14.2
				 *
				 * @param array $data The array of fields with values to be checked.
				 */
				$data = apply_filters( 'tribe_merge_identical_venues_fields', $data );

				$hash = md5( serialize( $data ) );

				if ( ! isset( $buckets[ $hash ] ) ) {
					$buckets[ $hash ] = [];
				}

				// prioritize venues with an eventbrite id
				$eventbrite = get_post_meta( $id, '_VenueEventBriteID', true );
				if ( empty( $eventbrite ) ) {
					array_push( $buckets[ $hash ], $id );
				} else {
					array_unshift( $buckets[ $hash ], $id );
				}
			}
		}

		foreach ( $buckets as $venue_ids ) {
			$this->amalgamate_venues( $venue_ids );
		}
	}

	/**
	 * Get all post titles of the given post type that have duplicates
	 *
	 * @param string $type The post type to query
	 *
	 * @return array
	 */
	private function get_redundant_titles( $type ) {
		global $wpdb;
		$sql    = "SELECT post_title FROM {$wpdb->posts} WHERE post_type=%s GROUP BY post_title HAVING COUNT(*) > 1";
		$sql    = $wpdb->prepare( $sql, $type );
		$titles = $wpdb->get_col( $sql );

		return $titles;
	}

	/**
	 * Find all posts of the given type with the given title
	 *
	 * @param string $title
	 * @param string $type
	 *
	 * @return array
	 */
	private function get_posts_with_title( $title, $type ) {
		global $wpdb;
		$sql   = "SELECT ID FROM {$wpdb->posts} WHERE post_type=%s AND post_title=%s ORDER BY ID ASC";
		$sql   = $wpdb->prepare( $sql, $type, $title );
		$posts = $wpdb->get_col( $sql );

		return $posts;
	}

	/**
	 * Merge all venues in the given list into one post (keeping the first)
	 *
	 * @since 5.14.2 - Change to public method.
	 *
	 * @param array<string|integer> $venue_ids An Array of venue ids to merge.
	 */
	public function amalgamate_venues( $venue_ids ) {
		if ( empty( $venue_ids ) || count( $venue_ids ) < 2 ) {
			return;
		}

		$venue_ids = array_map( function ( $item ) {
			return intval( absint( $item ) );
		}, $venue_ids );

		// Sort the array to get the lowest post id.
		sort( $venue_ids );

		/**
		 * Filter the venue ids that should be kept.
		 *
		 * @since 5.14.2
		 *
		 * @param array<string|integer> An Array of the post ids to keep or an empty array.
		 * @param array<string|integer> $venue_ids An Array of venue ids to merge.
		 */
		$keep = (array) apply_filters( 'tribe_amalgamate_venues_keep_venue', [], $venue_ids );

		// If not an array or empty, run the default venues amalgamate.
		if ( ! is_array( $keep ) || empty( $keep ) ) {
			$this->run_amalgamate_venues( $venue_ids, [] );

			return;
		}

		// Check if any of the venue ids is in $keep.
		$intersect = array_intersect( $keep, $venue_ids );
		if ( empty( $intersect ) ) {
			$this->run_amalgamate_venues( $venue_ids, [] );

			return;
		}

		// Return all the venue ids that match.
		$intersect_keys = array_flip( $intersect );
		$venue_ids = array_filter( $venue_ids, function ( $venue_id ) use ( $intersect_keys )  {
			return ! isset( $intersect_keys[ $venue_id ] );
		} );

		// Sort the array to get the lowest post id.
		sort( $intersect );
		// Get the first and lowest venue id as there can only be one venue that everything is amalgamates to.
		$keep = array_shift( $intersect );

		$this->run_amalgamate_venues( $venue_ids, $keep );
	}

	/**
	 * Run the Venue amalgamation, by default it keeps the lowest venue_id.
	 *
	 * @since 5.14.2
	 *
	 * @param array<string|integer> $venue_ids An Array of venue ids to merge.
	 * @param array<string|integer> $keep      An Array of the post IDs to keep or an empty array.
	 */
	public function run_amalgamate_venues( $venue_ids, $keep = [] ) {
		global $wpdb;

		// If $venue_ids is empty, then return.
		if ( empty( $venue_ids ) ) {
			return;
		}

		// If $keep is empty, then use the first venue id in the array..
		if ( empty( $keep ) ) {
			$keep = array_shift( $venue_ids );
		}

		$merging_ids = array_map( static function ( $id ) use ( $wpdb ) {
			return $wpdb->prepare( '%d', $id );
		}, $venue_ids );

		$sql     = "
			UPDATE {$wpdb->postmeta}
			SET meta_value=%d
			WHERE meta_key=%s
			AND meta_value
			IN(%s)
		";
		$sql     = $wpdb->prepare( $sql, $keep, '_EventVenueID', implode(',', $merging_ids) );
		$wpdb->query( $sql );

		$this->update_default_venues( $keep, $venue_ids );
		$this->delete_posts( $venue_ids );
	}

	/**
	 * Merge all organizers in the given list into one post (keeping the first)
	 *
	 * @param array $organizer_ids
	 *
	 */
	public function amalgamate_organizers( $organizer_ids ) {
		if ( empty( $organizer_ids ) || count( $organizer_ids ) < 2 ) {
			return;
		}

		$organizer_ids = array_map( function ( $item ) {
			return intval( absint( $item ) );
		}, $organizer_ids );

		// Sort the array to get the lowest post id.
		sort( $organizer_ids );

		/**
		 * Filter the organizer IDs that should be kept.
		 *
		 * @since 5.14.2
		 *
		 * @param array<string|integer> An Array of the post IDs to keep or an empty array if not defined.
		 * @param array<string|integer> An Array of organizer ids to merge.
         */
		$keep = (array) apply_filters( 'tribe_amalgamate_organizers_keep_organizer', [], $organizer_ids );

		// If not an array or empty, run the default venues amalgamate.
		if ( ! is_array( $keep ) || empty( $keep ) ) {
			$this->run_amalgamate_organizers( $organizer_ids, [] );

			return;
		}

		// Check if any of the organizer ids is in $keep.
		$intersect = array_intersect( $keep, $organizer_ids );
		if ( empty( $intersect ) ) {
			$this->run_amalgamate_organizers( $organizer_ids, [] );

			return;
		}

		// Return all the organizer ids that match.
		$intersect_keys = array_flip( $intersect );
		$organizer_ids = array_filter( $organizer_ids, function ( $organizer_id ) use ( $intersect_keys )  {
			return ! isset( $intersect_keys[ $organizer_id ] );
		} );

		// Sort the array to get the lowest post id.
		sort( $intersect );
		// Get the first and lowest organizer id as there can only be one organizer that everything is amalgamates to.
		$keep = array_shift( $intersect );

		$this->run_amalgamate_organizers( $organizer_ids, $keep );
	}

	/**
	 * Run the Venue amalgamation, by default it keeps the lowest venue_id.
	 *
	 * @since 5.14.2
	 *
	 * @param array<string|integer> $organizer_ids An Array of organizer ids to merge.
	 * @param array<string|integer> $keep      An Array of the post IDs to keep or an empty array.
	 */
	public function run_amalgamate_organizers( $organizer_ids, $keep = [] ) {
		global $wpdb;

		// If $organizer_ids is empty, then return.
		if ( empty( $organizer_ids ) ) {
			return;
		}

		// If $keep is empty, then use the first venue id in the array..
		if ( empty( $keep ) ) {
			$keep = array_shift( $organizer_ids );
		}

		$merging_ids = array_map( static function ( $id ) use ( $wpdb ) {
			return $wpdb->prepare( '%d', $id );
		}, $organizer_ids );

		$sql     = "
			UPDATE {$wpdb->postmeta}
			SET meta_value=%d
			WHERE meta_key=%s
			AND meta_value
			IN(%s)
		";
		$sql     = $wpdb->prepare( $sql, $keep, '_EventOrganizerID', implode(',', $merging_ids) );
		$wpdb->query( $sql );
		$this->update_default_organizers( $keep, $organizer_ids );
		$this->delete_posts( $organizer_ids );
	}

	/**
	 * If a removed venue is being used as a default, change the default to
	 * its replacement.
	 *
	 * @param int   $keep
	 * @param array $replace
	 *
	 */
	private function update_default_venues( $keep, array $replace ) {
		if ( $this->default_venue && in_array( $this->default_venue, $replace ) ) {
			$events = Tribe__Events__Main::instance();
			$events->setOption( 'eventsDefaultVenueID', $keep );
		}
		if ( $this->default_community_venue && in_array( $this->default_community_venue, $replace ) ) {
			$community = Tribe__Events__Community__Main::instance();
			$community->setOption( 'defaultCommunityVenueID', $keep );
		}
	}

	/**
	 * If a removed organizer is being used as a default, change the default to
	 * its replacement.
	 *
	 * @param int   $keep
	 * @param array $replace
	 *
	 */
	private function update_default_organizers( $keep, array $replace ) {
		if ( $this->default_organizer && in_array( $this->default_organizer, $replace ) ) {
			$events = Tribe__Events__Main::instance();
			$events->setOption( 'eventsDefaultOrganizerID', $keep );
		}
		if ( $this->default_community_organizer && in_array( $this->default_community_organizer, $replace ) ) {
			$community = Tribe__Events__Community__Main::instance();
			$community->setOption( 'defaultCommunityOrganizerID', $keep );
		}
	}

	/**
	 * Delete all the posts given
	 *
	 * @param array $post_ids
	 */
	private function delete_posts( $post_ids ) {
		foreach ( $post_ids as $id ) {
			$force = apply_filters( 'tribe_force_delete_duplicates', true );
			wp_delete_post( $id, $force );
		}
	}

	/**
	 * Make a button to trigger the amalgamation process
	 *
	 * @param string $text
	 *
	 * @return string
	 */
	public static function migration_button( $text = '' ) {
		$text = $text ? $text : __( 'Merge Duplicates', 'the-events-calendar' );

		// Get the base settings page URL.
		$url = tribe( Plugin_Settings::class )->get_url(
			[
				'tab' => 'general-maintenance-tab',
			]
		);

		$url = add_query_arg( [ 'amalgamate' => '1' ], $url );
		$url = wp_nonce_url( $url, 'amalgamate_duplicates' );

		return sprintf( '<a href="%s" class="button-secondary">%s</a>', $url, $text );
	}

	/**
	 * If the migration button is clicked, start working
	 *
	 */
	public static function listen_for_migration_button() {
		if ( empty( $_REQUEST['amalgamate'] ) || ! wp_verify_nonce( $_REQUEST['_wpnonce'], 'amalgamate_duplicates' ) ) {
			return;
		}

		if ( ! current_user_can( 'manage_options' ) ) {
			return;
		}

		$amalgamator = new self();
		$amalgamator->merge_duplicates();

		// Redirect to the TEC general -> maintenance settings tab.
		$url = tribe( Plugin_Settings::class )->get_url(
			[
				'tab' => 'general-maintenance-tab',
			]
		);

		wp_redirect( esc_url_raw( $url ) );
		tribe_exit();
	}
}