<?php

/**

 *

 * @package WordPress

 * @since 3.1.0

 * @access private

 */

class WP_MS_Themes_List_Table extends WP_List_Table {

	public $site_id;

	public $is_site_themes;

	private $has_items;

	/**

	 * Constructor.

	 *

	 * @since 3.1.0

	 *

	 * @see WP_List_Table::__construct() for more information on default arguments.

	 *

	 * @param array $args An associative array of arguments.

	 */

	public function __construct( $args = array() ) {

		global $status, $page;

		parent::__construct( array(

			'plural' => 'themes',

			'screen' => isset( $args['screen'] ) ? $args['screen'] : null,

		) );

		$status = isset( $_REQUEST['theme_status'] ) ? $_REQUEST['theme_status'] : 'all';

		if ( !in_array( $status, array( 'all', 'enabled', 'disabled', 'upgrade', 'search', 'broken' ) ) )

			$status = 'all';

		$page = $this->get_pagenum();

		if ( $this->is_site_themes )

			$this->site_id = isset( $_REQUEST['id'] ) ? intval( $_REQUEST['id'] ) : 0;

	}

	protected function get_table_classes() {

		// todo: remove and add CSS for .themes

		return array( 'widefat', 'plugins' );

	}

	public function ajax_user_can() {

		if ( $this->is_site_themes )

			return current_user_can( 'manage_sites' );

		else

			return current_user_can( 'manage_network_themes' );

	}

	public function prepare_items() {

		global $status, $totals, $page, $orderby, $order, $s;

		wp_reset_vars( array( 'orderby', 'order', 's' ) );

		$themes = array(

			/**

			 * themes list table.

			 *

			 * @since 3.1.0

			 *

			 * @param array $all An array of WP_Theme objects to display in the list table.

			 */

			'all' => apply_filters( 'all_themes', wp_get_themes() ),

			'search' => array(),

			'enabled' => array(),

			'disabled' => array(),

			'upgrade' => array(),

			'broken' => $this->is_site_themes ? array() : wp_get_themes( array( 'errors' => true ) ),

		);

		if ( $this->is_site_themes ) {

			$themes_per_page = $this->get_items_per_page( 'site_themes_network_per_page' );

			$allowed_where = 'site';

		} else {

			$themes_per_page = $this->get_items_per_page( 'themes_network_per_page' );

			$allowed_where = 'network';

		}

		$maybe_update = current_user_can( 'update_themes' ) && ! $this->is_site_themes && $current = get_site_transient( 'update_themes' );

		foreach ( (array) $themes['all'] as $key => $theme ) {

			if ( $this->is_site_themes && $theme->is_allowed( 'network' ) ) {

				unset( $themes['all'][ $key ] );

				continue;

			}

			if ( $maybe_update && isset( $current->response[ $key ] ) ) {

				$themes['all'][ $key ]->update = true;

				$themes['upgrade'][ $key ] = $themes['all'][ $key ];

			}

			$filter = $theme->is_allowed( $allowed_where, $this->site_id ) ? 'enabled' : 'disabled';

			$themes[ $filter ][ $key ] = $themes['all'][ $key ];

		}

		if ( $s ) {

			$status = 'search';

			$themes['search'] = array_filter( array_merge( $themes['all'], $themes['broken'] ), array( $this, '_search_callback' ) );

		}

		$totals = array();

		foreach ( $themes as $type => $list )

			$totals[ $type ] = count( $list );

		if ( empty( $themes[ $status ] ) && !in_array( $status, array( 'all', 'search' ) ) )

			$status = 'all';

		$this->items = $themes[ $status ];

		WP_Theme::sort_by_name( $this->items );

		$this->has_items = ! empty( $themes['all'] );

		$total_this_page = $totals[ $status ];

		if ( $orderby ) {

			$orderby = ucfirst( $orderby );

			$order = strtoupper( $order );

					$this->items = array_reverse( $this->items );

			} else {

				uasort( $this->items, array( $this, '_order_callback' ) );

			}

		}

		$start = ( $page - 1 ) * $themes_per_page;

		if ( $total_this_page > $themes_per_page )

			$this->items = array_slice( $this->items, $start, $themes_per_page, true );

		$this->set_pagination_args( array(

			'total_items' => $total_this_page,

			'per_page' => $themes_per_page,

		) );

	}

	/**

	 * @staticvar string $term

	 * @param WP_Theme $theme

	 * @return bool

	 */

	public function _search_callback( $theme ) {

		if ( is_null( $term ) )

			$term = wp_unslash( $_REQUEST['s'] );

		foreach ( array( 'Name', 'Description', 'Author', 'Author', 'AuthorURI' ) as $field ) {

			// Don't mark up; Do translate.

			if ( false !== stripos( $theme->display( $field, false, true ), $term ) )

				return true;

		}

		if ( false !== stripos( $theme->get_stylesheet(), $term ) )

			return true;

		if ( false !== stripos( $theme->get_template(), $term ) )

			return true;

		return false;

	}

	// Not used by any core columns.

	/**

	 * @global string $orderby

	 * @global string $order

	 * @param array $theme_a

	 * @param array $theme_b

	 * @return int

	 */

	public function _order_callback( $theme_a, $theme_b ) {

		global $orderby, $order;

		$a = $theme_a[ $orderby ];

		$b = $theme_b[ $orderby ];

		if ( $a == $b )

			return 0;

			return ( $a < $b ) ? 1 : -1;

		else

			return ( $a < $b ) ? -1 : 1;

	}

	public function no_items() {

			_e( 'No themes found.' );

			_e( 'You do not appear to have any themes available at this time.' );

	}

	public function get_columns() {

		return array(

			'cb'          => '<input type="checkbox" />',

			'name'        => __( 'Theme' ),

			'description' => __( 'Description' ),

		);

	}

	protected function get_sortable_columns() {

		return array(

			'name'         => 'name',

		);

	}

	protected function get_views() {

		global $totals, $status;

		$status_links = array();

		foreach ( $totals as $type => $count ) {

			if ( !$count )

				continue;

			switch ( $type ) {

				case 'all':

					$text = _nx( 'All <span class="count">(%s)</span>', 'All <span class="count">(%s)</span>', $count, 'themes' );

					break;

				case 'enabled':

					$text = _n( 'Enabled <span class="count">(%s)</span>', 'Enabled <span class="count">(%s)</span>', $count );

					break;

				case 'disabled':

					$text = _n( 'Disabled <span class="count">(%s)</span>', 'Disabled <span class="count">(%s)</span>', $count );

					break;

				case 'upgrade':

					$text = _n( 'Update Available <span class="count">(%s)</span>', 'Update Available <span class="count">(%s)</span>', $count );

					break;

				case 'broken' :

					$text = _n( 'Broken <span class="count">(%s)</span>', 'Broken <span class="count">(%s)</span>', $count );

					break;

			}

			if ( $this->is_site_themes )

				$url = 'site-themes.php?id=' . $this->site_id;

			else

				$url = 'themes.php';

			if ( 'search' != $type ) {

					esc_url( add_query_arg('theme_status', $type, $url) ),

					sprintf( $text, number_format_i18n( $count ) )

				);

			}

		}

		return $status_links;

	}

	protected function get_bulk_actions() {

		global $status;

		$actions = array();

		if ( 'enabled' != $status )

			$actions['enable-selected'] = $this->is_site_themes ? __( 'Enable' ) : __( 'Network Enable' );

		if ( 'disabled' != $status )

			$actions['disable-selected'] = $this->is_site_themes ? __( 'Disable' ) : __( 'Network Disable' );

		if ( ! $this->is_site_themes ) {

			if ( current_user_can( 'update_themes' ) )

				$actions['update-selected'] = __( 'Update' );

			if ( current_user_can( 'delete_themes' ) )

				$actions['delete-selected'] = __( 'Delete' );

		}

		return $actions;

	}

	public function display_rows() {

		foreach ( $this->items as $theme )

			$this->single_row( $theme );

	}

	/**

	 * @global string $status

	 * @global string $s

	 */

		$context = $status;

		if ( $this->is_site_themes ) {

			$url = "site-themes.php?id={$this->site_id}&";

			$allowed = $theme->is_allowed( 'site', $this->site_id );

		} else {

			$url = 'themes.php?';

			$allowed = $theme->is_allowed( 'network' );

		}

		// Pre-order.

		$actions = array(

			'enable' => '',

			'disable' => '',

			'delete' => ''

		);

		$stylesheet = $theme->get_stylesheet();

		$theme_key = urlencode( $stylesheet );

		if ( ! $allowed ) {