class-media-library-organizer-tree-view-taxonomy-walker.php 2.95 KB
<?php
/**
 * Tree View Taxonomy Walker class.
 *
 * @package Media_Library_Organizer
 * @author WP Media Library
 */

/**
 * Taxonomy Walker for Tree View
 *
 * @version   1.1.1
 */
class Media_Library_Organizer_Tree_View_Taxonomy_Walker extends Walker_Category {

	/**
	 * Wraps Taxonomy Term counts in a span, that can be styled using CSS.
	 *
	 * @since   1.1.1
	 *
	 * @param   string $output     Passed by reference. Used to append additional content.
	 * @param   object $category   The current term object.
	 * @param   int    $depth      Depth of the term in reference to parents. Default 0.
	 * @param   array  $args       An array of arguments. @see wp_terms_checklist().
	 * @param   int    $id         ID of the current term.
	 */
	public function start_el( &$output, $category, $depth = 0, $args = array(), $id = 0 ) {

		// Get output from parent Walker.
		parent::start_el( $output, $category, $depth, $args, $id );

		// Wrap (n) in a span.
		$output = $this->wrap_count( $output, $category );

		// Change link.
		$output = $this->change_filter_link( $output, $category );

	}

	/**
	 * Wrap the Attachment Count in a <span> for styling
	 *
	 * @since   1.1.1
	 *
	 * @param   string  $output     Output.
	 * @param   WP_Term $term       Term.
	 * @return  string                  Output
	 */
	private function wrap_count( $output, $term ) {

		$output  = str_replace( '</a> (', ' <span class="count" data-term-id="' . $term->term_id . '">', $output );
		$output .= '</span></a>';
		$output  = str_replace( ")\n</span>", '</span>', $output );

		return $output;

	}

	/**
	 * Replace the Term's link with a contextual link, depending on the screen
	 * we're on.
	 *
	 * @since   1.1.1
	 *
	 * @param   string  $output     Output.
	 * @param   WP_Term $term       Term.
	 * @return  string                  Output
	 */
	private function change_filter_link( $output, $term ) {

		// Replace URL with upload.php version.
		$output = str_replace( get_term_link( $term ), $this->get_term_filter_link( $term ), $output );

		// Return.
		return $output;

	}

	/**
	 * Returns the Taxonomy Term's URL for the given Media View in the WordPress Administration
	 *
	 * @since   1.1.1
	 *
	 * @param   WP_Term $term   Taxonomy Term.
	 * @return  string              URL
	 */
	private function get_term_filter_link( $term ) {

		// Build URL arguments.
		$args = array(
			'mode'          => Media_Library_Organizer()->get_class( 'common' )->get_media_view(),
			$term->taxonomy => $term->slug,
		);

		// Define the filters to append to the URL.
		$conditions = array(
			'attachment-filter',
			'm',
			'orderby',
			'order',
		);

		// Sanitize request.
		$request = array_map( 'sanitize_text_field', $_REQUEST ); // phpcs:ignore WordPress.Security.NonceVerification

		foreach ( $conditions as $condition ) {
			if ( isset( $request[ $condition ] ) ) {
				$args[ $condition ] = $request[ $condition ];
			}
		}

		// Build and return the URL.
		$url = add_query_arg( $args, 'upload.php' );

		return $url;

	}

}