<?php

/**

 *

 *

 * @package WordPress

 * @subpackage Upgrader

 * @since 2.8.0

 */

/**

 *

 * @since 2.8.0

 */

class WP_Upgrader {

	/**

	 * The error/notification strings used to update the user on the progress.

	 *

	 * @since 2.8.0

	 */

	public $strings = array();

	/**

	 * The upgrader skin being used.

	 *

	 * @since 2.8.0

	 */

	public $skin = null;

	/**

	 * The result of the installation.

	 *

	 * it.

	 *

	 * @since 2.8.0

	 * @var WP_Error|array $result {

	 *      @type string $source             The full path to the source the files were installed from.

	 *      @type string $source_files       List of all the files in the source directory.

	 *      @type string $destination_name   The name of the destination folder, or empty if `$destination`

	 *                                       and `$local_destination` are the same.

	 *      @type string $local_destination  The full local path to the destination folder. This is usually

	 *                                       the same as `$destination`.

	 *      @type string $remote_destination The full remote path to the destination folder

	 *                                       (i.e., from `$wp_filesystem`).

	 *      @type bool   $clear_destination  Whether the destination folder was cleared.

	 * }

	 */

	public $result = array();

	/**

	 * The total number of updates being performed.

	 *

	 * Set by the bulk update methods.

	 *

	 * @since 3.0.0

	 * @var int $update_count

	 */

	public $update_count = 0;

	/**

	 * The current update if multiple updates are being performed.

	 *

	 * Used by the bulk update methods, and incremented for each update.

	 *

	 * @since 3.0.0

	 * @var int

	 */

	public $update_current = 0;

	/**

	 * Construct the upgrader with a skin.

	 *

	 * @since 2.8.0

	 *

	 *                               instance.

	 */

	public function __construct( $skin = null ) {

		if ( null == $skin )

			$this->skin = new WP_Upgrader_Skin();

		else

			$this->skin = $skin;

	}

	/**

	 * Initialize the upgrader.

	 *

	 * This will set the relationship between the skin being used and this upgrader,

	 * and also add the generic strings to `WP_Upgrader::$strings`.

	 *

	 * @since 2.8.0

	 */

	public function init() {

		$this->skin->set_upgrader($this);

		$this->generic_strings();

	}

	/**

	 * Add the generic strings to WP_Upgrader::$strings.

	 *

	 * @since 2.8.0

	 */

	public function generic_strings() {

		$this->strings['fs_unavailable'] = __('Could not access filesystem.');

		$this->strings['fs_error'] = __('Filesystem error.');

		/* translators: %s: directory name */

		$this->strings['fs_no_folder'] = __('Unable to locate needed folder (%s).');

		$this->strings['download_failed'] = __('Download failed.');

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

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

		$this->strings['folder_exists'] = __('Destination folder already exists.');

		$this->strings['mkdir_failed'] = __('Could not create directory.');

		$this->strings['incompatible_archive'] = __('The package could not be installed.');

		$this->strings['maintenance_start'] = __('Enabling Maintenance mode…');

		$this->strings['maintenance_end'] = __('Disabling Maintenance mode…');

	}

	/**

	 * Connect to the filesystem.

	 *

	 * @since 2.8.0

	 *

	 * @param array $directories                  Optional. A list of directories. If any of these do

	 *                                            Default empty array.

	 * @param bool  $allow_relaxed_file_ownership Whether to allow relaxed file ownership.

	 *                                            Default false.

	 */

	public function fs_connect( $directories = array(), $allow_relaxed_file_ownership = false ) {

		global $wp_filesystem;

		if ( false === ( $credentials = $this->skin->request_filesystem_credentials( false, $directories[0], $allow_relaxed_file_ownership ) ) ) {

			return false;

		}

		if ( ! WP_Filesystem( $credentials, $directories[0], $allow_relaxed_file_ownership ) ) {

			$error = true;

			if ( is_object($wp_filesystem) && $wp_filesystem->errors->get_error_code() )

				$error = $wp_filesystem->errors;

			// Failed to connect, Error and request again

			$this->skin->request_filesystem_credentials( $error, $directories[0], $allow_relaxed_file_ownership );

			return false;

		}

		if ( ! is_object($wp_filesystem) )

			return new WP_Error('fs_unavailable', $this->strings['fs_unavailable'] );

		if ( is_wp_error($wp_filesystem->errors) && $wp_filesystem->errors->get_error_code() )

			return new WP_Error('fs_error', $this->strings['fs_error'], $wp_filesystem->errors);

		foreach ( (array)$directories as $dir ) {

			switch ( $dir ) {

				case ABSPATH:

					if ( ! $wp_filesystem->abspath() )

						return new WP_Error('fs_no_root_dir', $this->strings['fs_no_root_dir']);

					break;

				case WP_CONTENT_DIR:

					if ( ! $wp_filesystem->wp_content_dir() )

						return new WP_Error('fs_no_content_dir', $this->strings['fs_no_content_dir']);

					break;

				case WP_PLUGIN_DIR:

					if ( ! $wp_filesystem->wp_plugins_dir() )

						return new WP_Error('fs_no_plugins_dir', $this->strings['fs_no_plugins_dir']);

					break;

				case get_theme_root():

					if ( ! $wp_filesystem->wp_themes_dir() )

						return new WP_Error('fs_no_themes_dir', $this->strings['fs_no_themes_dir']);

					break;

				default:

					if ( ! $wp_filesystem->find_folder($dir) )

						return new WP_Error( 'fs_no_folder', sprintf( $this->strings['fs_no_folder'], esc_html( basename( $dir ) ) ) );

					break;

			}

		}

		return true;

	} //end fs_connect();

	/**

	 * Download a package.

	 *

	 * @since 2.8.0

	 *

	 * @param string $package The URI of the package. If this is the full path to an

	 *                        existing local file, it will be returned untouched.

	 */

	public function download_package( $package ) {

		/**

		 *

		 * @since 3.7.0

		 *

		 * @param bool        $reply   Whether to bail without returning the package.

		 *                             Default false.

		 * @param string      $package The package file name.

		 * @param WP_Upgrader $this    The WP_Upgrader instance.

		 */

		$reply = apply_filters( 'upgrader_pre_download', false, $package, $this );

		if ( false !== $reply )

			return $reply;

		if ( ! preg_match('!^(http|https|ftp)://!i', $package) && file_exists($package) ) //Local file or remote?

			return $package; //must be a local file..

		if ( empty($package) )

			return new WP_Error('no_package', $this->strings['no_package']);

		$this->skin->feedback('downloading_package', $package);

		$download_file = download_url($package);

		if ( is_wp_error($download_file) )

			return new WP_Error('download_failed', $this->strings['download_failed'], $download_file->get_error_message());

		return $download_file;

	}

	/**

	 * Unpack a compressed package file.

	 *

	 * @since 2.8.0

	 *

	 * @param string $package        Full path to the package file.

	 * @param bool   $delete_package Optional. Whether to delete the package file after attempting

	 *                               to unpack it. Default true.

	 */

	public function unpack_package( $package, $delete_package = true ) {

		global $wp_filesystem;

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

		$upgrade_folder = $wp_filesystem->wp_content_dir() . 'upgrade/';

		//Clean up contents of upgrade directory beforehand.

		$upgrade_files = $wp_filesystem->dirlist($upgrade_folder);

		if ( !empty($upgrade_files) ) {

			foreach ( $upgrade_files as $file )

				$wp_filesystem->delete($upgrade_folder . $file['name'], true);

		}

		// We need a working directory - Strip off any .tmp or .zip suffixes

		$working_dir = $upgrade_folder . basename( basename( $package, '.tmp' ), '.zip' );

		// Clean up working directory

		if ( $wp_filesystem->is_dir($working_dir) )

			$wp_filesystem->delete($working_dir, true);

		// Unzip package to working directory

		$result = unzip_file( $package, $working_dir );

		// Once extracted, delete the package if required.

		if ( $delete_package )

			unlink($package);

		if ( is_wp_error($result) ) {

			$wp_filesystem->delete($working_dir, true);

			if ( 'incompatible_archive' == $result->get_error_code() ) {

				return new WP_Error( 'incompatible_archive', $this->strings['incompatible_archive'], $result->get_error_data() );

			}

			return $result;

		}

		return $working_dir;

	}

	/**

	 * Install a package.

	 *

	 * Copies the contents of a package form a source directory, and installs them in

	 * a destination directory. Optionally removes the source. It can also optionally

	 * clear out the destination folder if it already exists.

	 *

	 * @since 2.8.0

	 *

	 * @param array|string $args {

	 *     Optional. Array or string of arguments for installing a package. Default empty array.

	 *

	 *     @type string $source                      Required path to the package source. Default empty.