<?php

/**

 * Upgrade API: Theme_Upgrader class

 *

 * @package WordPress

 * @subpackage Upgrader

 * @since 4.6.0

 */

/**

 * Core class used for upgrading/installing themes.

 *

 * It is designed to upgrade/install themes from a local zip, remote zip URL,

 * or uploaded zip file.

 *

 * @since 2.8.0

 * @since 4.6.0 Moved to its own file from wp-admin/includes/class-wp-upgrader.php.

 *

 * @see WP_Upgrader

 */

class Theme_Upgrader extends WP_Upgrader {

	/**

	 * Result of the theme upgrade offer.

	 *

	 * @since 2.8.0

	 * @var array|WP_Error $result

	 * @see WP_Upgrader::$result

	 */

	public $result;

	/**

	 * Whether multiple themes are being upgraded/installed in bulk.

	 *

	 * @since 2.9.0

	 * @var bool $bulk

	 */

	public $bulk = false;

	/**

	 * Initialize the upgrade strings.

	 *

	 * @since 2.8.0

	 */

	public function upgrade_strings() {

		$this->strings['up_to_date'] = __( 'The theme is at the latest version.' );

		$this->strings['no_package'] = __( 'Update package not available.' );

		$this->strings['downloading_package'] = sprintf( __( 'Downloading update from %s&#8230;' ), '<span class="code">%s</span>' );

		$this->strings['unpack_package']      = __( 'Unpacking the update…' );

		$this->strings['remove_old']          = __( 'Removing the old version of the theme…' );

		$this->strings['remove_old_failed']   = __( 'Could not remove the old theme.' );

		$this->strings['process_failed']      = __( 'Theme update failed.' );

		$this->strings['process_success']     = __( 'Theme updated successfully.' );

	}

	/**

	 * Initialize the installation strings.

	 *

	 * @since 2.8.0

	 */

	public function install_strings() {

		$this->strings['no_package'] = __( 'Installation package not available.' );

		$this->strings['downloading_package'] = sprintf( __( 'Downloading installation package from %s&#8230;' ), '<span class="code">%s</span>' );

		$this->strings['unpack_package']      = __( 'Unpacking the package…' );

		$this->strings['installing_package']  = __( 'Installing the theme…' );

		$this->strings['no_files']            = __( 'The theme contains no files.' );

		$this->strings['process_failed']      = __( 'Theme installation failed.' );

		$this->strings['process_success']     = __( 'Theme installed successfully.' );

		$this->strings['process_success_specific'] = __( 'Successfully installed the theme <strong>%1$s %2$s</strong>.' );

		$this->strings['parent_theme_search']      = __( 'This theme requires a parent theme. Checking if it is installed…' );

		$this->strings['parent_theme_prepare_install'] = __( 'Preparing to install <strong>%1$s %2$s</strong>&#8230;' );

		$this->strings['parent_theme_currently_installed'] = __( 'The parent theme, <strong>%1$s %2$s</strong>, is currently installed.' );

		$this->strings['parent_theme_install_success'] = __( 'Successfully installed the parent theme, <strong>%1$s %2$s</strong>.' );

		$this->strings['parent_theme_not_found'] = sprintf( __( '<strong>The parent theme could not be found.</strong> You will need to install the parent theme, %s, before you can use this child theme.' ), '<strong>%s</strong>' );

	}

	/**

	 * Check if a child theme is being installed and we need to install its parent.

	 *

	 * Hooked to the {@see 'upgrader_post_install'} filter by Theme_Upgrader::install().

	 *

	 * @since 3.4.0

	 *

	 * @param bool  $install_result

	 * @param array $hook_extra

	 * @param array $child_result

	 */

	public function check_parent_theme_filter( $install_result, $hook_extra, $child_result ) {

		$theme_info = $this->theme_info();

		if ( ! $theme_info->parent() ) {

			return $install_result;

		}

		$this->skin->feedback( 'parent_theme_search' );

		if ( ! $theme_info->parent()->errors() ) {

			$this->skin->feedback( 'parent_theme_currently_installed', $theme_info->parent()->display( 'Name' ), $theme_info->parent()->display( 'Version' ) );

			// We already have the theme, fall through.

			return $install_result;

		}

		// We don't have the parent theme, let's install it.

		$api = themes_api(

			'theme_information',

			array(

				'slug'   => $theme_info->get( 'Template' ),

				'fields' => array(

					'sections' => false,

					'tags'     => false,

				),

			)

		if ( ! $api || is_wp_error( $api ) ) {

			$this->skin->feedback( 'parent_theme_not_found', $theme_info->get( 'Template' ) );

			add_filter( 'install_theme_complete_actions', array( $this, 'hide_activate_preview_actions' ) );

			return $install_result;

		}

		// Backup required data we're going to override:

		$child_api             = $this->skin->api;

		$child_success_message = $this->strings['process_success'];

		$this->skin->feedback( 'parent_theme_prepare_install', $api->name, $api->version );

		add_filter( 'install_theme_complete_actions', '__return_false', 999 ); // Don't show any actions after installing the theme.

		$parent_result = $this->run(

			array(

				'package'           => $api->download_link,

				'destination'       => get_theme_root(),

				'clear_working'     => true,

			)

		);

		if ( is_wp_error( $parent_result ) ) {

			add_filter( 'install_theme_complete_actions', array( $this, 'hide_activate_preview_actions' ) );

		}

		remove_filter( 'install_theme_complete_actions', '__return_false', 999 );

		$this->result                     = $child_result;

		$this->skin->api                  = $child_api;

		$this->strings['process_success'] = $child_success_message;

		return $install_result;

	}

	/**

	 * Don't display the activate and preview actions to the user.

	 *

	 * Hooked to the {@see 'install_theme_complete_actions'} filter by

	 * Theme_Upgrader::check_parent_theme_filter() when installing

	 * a child theme and installing the parent theme fails.

	 *

	 * @since 3.4.0

	 *

	 * @param array $actions Preview actions.

	 * @return array

	 */

	public function hide_activate_preview_actions( $actions ) {

		unset( $actions['activate'], $actions['preview'] );

		return $actions;

	}

	/**

	 * Install a theme package.

	 *

	 * @since 2.8.0

	 * @since 3.7.0 The `$args` parameter was added, making clearing the update cache optional.

	 *

	 * @param string $package The full local path or URI of the package.

	 * @param array  $args {

	 *     Optional. Other arguments for installing a theme package. Default empty array.

	 *

	 *     @type bool $clear_update_cache Whether to clear the updates cache if successful.

	 *                                    Default true.

	 * }

	 *

	 * @return bool|WP_Error True if the installation was successful, false or a WP_Error object otherwise.

	 */

	public function install( $package, $args = array() ) {

		$defaults    = array(

			'clear_update_cache' => true,

		);

		$parsed_args = wp_parse_args( $args, $defaults );

		$this->init();

		$this->install_strings();

		add_filter( 'upgrader_source_selection', array( $this, 'check_package' ) );

		add_filter( 'upgrader_post_install', array( $this, 'check_parent_theme_filter' ), 10, 3 );

		if ( $parsed_args['clear_update_cache'] ) {

			// Clear cache so wp_update_themes() knows about the new theme.

			add_action( 'upgrader_process_complete', 'wp_clean_themes_cache', 9, 0 );

		}

		$this->run(

			array(

				'package'           => $package,

				'destination'       => get_theme_root(),

				'clear_working'     => true,

				'hook_extra'        => array(

					'type'   => 'theme',

					'action' => 'install',

				),

			)

		);

		remove_action( 'upgrader_process_complete', 'wp_clean_themes_cache', 9 );

		remove_filter( 'upgrader_source_selection', array( $this, 'check_package' ) );

		remove_filter( 'upgrader_post_install', array( $this, 'check_parent_theme_filter' ) );

		if ( ! $this->result || is_wp_error( $this->result ) ) {

			return $this->result;

		}

		wp_clean_themes_cache( $parsed_args['clear_update_cache'] );

		return true;

	}

	/**

	 * Upgrade a theme.

	 *

	 * @since 2.8.0

	 * @since 3.7.0 The `$args` parameter was added, making clearing the update cache optional.

	 *

	 * @param string $theme The theme slug.

	 * @param array  $args {

	 *     Optional. Other arguments for upgrading a theme. Default empty array.

	 *

	 *     @type bool $clear_update_cache Whether to clear the update cache if successful.

	 *                                    Default true.

	 * }

	 * @return bool|WP_Error True if the upgrade was successful, false or a WP_Error object otherwise.

	 */

	public function upgrade( $theme, $args = array() ) {

		$defaults    = array(

			'clear_update_cache' => true,

		);

		$parsed_args = wp_parse_args( $args, $defaults );

		$this->init();

		$this->upgrade_strings();

		// Is an update available?

		$current = get_site_transient( 'update_themes' );

		if ( ! isset( $current->response[ $theme ] ) ) {

			$this->skin->before();

			$this->skin->set_result( false );

			$this->skin->error( 'up_to_date' );

			$this->skin->after();

			return false;

		}

		$r = $current->response[ $theme ];

		add_filter( 'upgrader_pre_install', array( $this, 'current_before' ), 10, 2 );

		add_filter( 'upgrader_post_install', array( $this, 'current_after' ), 10, 2 );

		add_filter( 'upgrader_clear_destination', array( $this, 'delete_old_theme' ), 10, 4 );

		if ( $parsed_args['clear_update_cache'] ) {

			// Clear cache so wp_update_themes() knows about the new theme.

			add_action( 'upgrader_process_complete', 'wp_clean_themes_cache', 9, 0 );

		}

		$this->run(

			array(

				'package'           => $r['package'],

				'destination'       => get_theme_root( $theme ),

				'clear_destination' => true,

				'clear_working'     => true,

				'hook_extra'        => array(

					'theme'  => $theme,

					'type'   => 'theme',

					'action' => 'update',

				),

			)

		);

		remove_action( 'upgrader_process_complete', 'wp_clean_themes_cache', 9 );

		remove_filter( 'upgrader_pre_install', array( $this, 'current_before' ) );

		remove_filter( 'upgrader_post_install', array( $this, 'current_after' ) );

		remove_filter( 'upgrader_clear_destination', array( $this, 'delete_old_theme' ) );

		if ( ! $this->result || is_wp_error( $this->result ) ) {

			return $this->result;

		}

		wp_clean_themes_cache( $parsed_args['clear_update_cache'] );

		return true;

	}

	/**

	 * Upgrade several themes at once.

	 *

	 * @since 3.0.0

	 * @since 3.7.0 The `$args` parameter was added, making clearing the update cache optional.

	 *

	 * @param string[] $themes Array of the theme slugs.

	 * @param array    $args {

	 *     Optional. Other arguments for upgrading several themes at once. Default empty array.

	 *

	 *     @type bool $clear_update_cache Whether to clear the update cache if successful.

	 *                                    Default true.

	 * }

	 * @return array[]|false An array of results, or false if unable to connect to the filesystem.

	 */

	public function bulk_upgrade( $themes, $args = array() ) {

		$defaults    = array(

			'clear_update_cache' => true,

		);

		$parsed_args = wp_parse_args( $args, $defaults );

		$this->init();

		$this->bulk = true;

		$this->upgrade_strings();

		$current = get_site_transient( 'update_themes' );

		add_filter( 'upgrader_pre_install', array( $this, 'current_before' ), 10, 2 );

		add_filter( 'upgrader_post_install', array( $this, 'current_after' ), 10, 2 );

		add_filter( 'upgrader_clear_destination', array( $this, 'delete_old_theme' ), 10, 4 );

		$this->skin->header();

		$res = $this->fs_connect( array( WP_CONTENT_DIR ) );

		if ( ! $res ) {

			$this->skin->footer();

			return false;

		}