<?php
namespace NewfoldLabs\WP\Module\Installer\Services;

use NewfoldLabs\WP\Module\Installer\Data\Themes;

/**
 * Class ThemeInstaller.
 */
class ThemeInstaller {

	/**
	 * Install a whitelisted Theme based on the activation status.
	 *
	 * @param string  $theme Theme URL from Themes.php.
	 * @param boolean $activate Whether to activate the theme after install.
	 * @return \WP_REST_Response|\WP_Error
	 */
	public static function install( $theme, $activate ) {
		$theme_list = Themes::get();

		// Checks if the theme slug is an nfd slug.
		if ( self::is_nfd_slug( $theme ) ) {
			// Retrieve the theme stylesheet to determine if it has been already installed.
			$stylesheet = $theme_list['nfd_slugs'][ $theme ]['stylesheet'];
			// Check if the theme already exists.
			if ( ! ( \wp_get_theme( $stylesheet ) )->exists() ) {
				$status = self::install_from_zip(
					$theme_list['nfd_slugs'][ $theme ]['url'],
					$activate,
					$stylesheet
				);
				if ( \is_wp_error( $status ) ) {
					return $status;
				}

				return new \WP_REST_Response(
					array(),
					201
				);
			}

			// If specified then activate the theme even if it already installed.
			if ( $activate && ( ( \wp_get_theme() )->get( 'TextDomain' ) !== $stylesheet ) ) {
				$status = \switch_theme( $stylesheet );
			}
		}

		return new \WP_REST_Response(
			array(),
			201
		);
	}

	/**
	 * Install theme from an custom zip url if not already installed. Activate and switch to the theme, if specified.
	 *
	 * @param string  $url The ZIP URL to install the theme from.
	 * @param boolean $activate Whether to activate the plugin after install.
	 * @param string  $stylesheet Theme Stylesheet Name.
	 * @return \WP_REST_Response|\WP_Error
	 */
	public static function install_from_zip( $url, $activate, $stylesheet ) {
		require_once ABSPATH . 'wp-admin/includes/file.php';
		require_once ABSPATH . 'wp-admin/includes/misc.php';
		require_once ABSPATH . 'wp-admin/includes/class-wp-upgrader.php';

		$skin     = new \WP_Ajax_Upgrader_Skin();
		$upgrader = new \Theme_Upgrader( $skin );
		$result   = $upgrader->install( $url );

		if ( is_wp_error( $result ) ) {
			return $result;
		}

		if ( is_wp_error( $skin->result ) ) {
			return $skin->result;
		}

		if ( $skin->get_errors()->has_errors() ) {
			return new \WP_Error(
				'unable_to_install_theme',
				$skin->get_error_messages(),
				array( 'status' => 500 )
			);
		}

		if ( is_null( $result ) ) {
			// Pass through the error from WP_Filesystem if one was raised.
			if ( $wp_filesystem instanceof \WP_Filesystem_Base
			&& \is_wp_error( $wp_filesystem->errors ) && $wp_filesystem->errors->has_errors()
			) {
				return new \WP_Error(
					'unable_to_connect_to_filesystem',
					$wp_filesystem->errors->get_error_message(),
					array( 'status' => 500 )
				);
			}

			return new \WP_Error(
				'unable_to_connect_to_filesystem',
				__( 'Unable to connect to the filesystem.', 'wp-module-installer' ),
				array( 'status' => 500 )
			);
		}

		// Activate the theme if specified.
		if ( $activate && ( ( \wp_get_theme() )->get( 'TextDomain' ) !== $stylesheet ) ) {
			\switch_theme( $stylesheet );
		}

		return new \WP_REST_Response(
			array(),
			201
		);
	}

	/**
	 * Checks if a given slug is a valid nfd_slug. Ref: includes/Data/Themes.php for nfd_slug.
	 *
	 * @param string $theme Slug of the theme.
	 * @return boolean
	 */
	public static function is_nfd_slug( $theme ) {
		$theme_list = Themes::get();
		if ( isset( $theme_list['nfd_slugs'][ $theme ]['approved'] ) ) {
			return true;
		}
		return false;
	}

	/**
	 * Retrieve Theme Stylesheet Name for a specified theme name and theme type.
	 *
	 * @param mixed $theme Slug of the theme present under includes/Data/Themes.php.
	 * @param mixed $theme_type Type of theme Ref: includes/Data/Themes.php for types of theme slugs.
	 * @return string|boolean
	 */
	public static function get_theme_stylesheet( $theme, $theme_type ) {
		$theme_list = Themes::get();
		return isset( $theme_list[ $theme_type ][ $theme ]['stylesheet'] ) ? $theme_list[ $theme_type ][ $theme ]['stylesheet'] : false;
	}

	/**
	 * Retrieve Theme Type - approved NFD Slug/WP Slug.
	 *
	 * @param string $theme Theme name
	 * @return string Type of theme. Ref: includes/Data/Themes.php for the different types.
	 */
	public static function get_theme_type( $theme ) {
		if ( self::is_nfd_slug( $theme ) ) {
			return 'nfd_slugs';
		}
		return 'wp_slugs';
	}

	/**
	 * Determines if a theme has already been installed.
	 *
	 * @param string $stylesheet The stylesheet of the theme.
	 * @return boolean
	 */
	public static function is_theme_installed( $stylesheet ) {
		return ( \wp_get_theme( $stylesheet ) )->exists();
	}

	/**
	 * Determines if a theme is already active.
	 *
	 * @param string $stylesheet The stylesheet of the theme.
	 * @return boolean
	 */
	public static function is_theme_active( $stylesheet ) {
		return ( ( \wp_get_theme() )->get( 'TextDomain' ) ) === $stylesheet;
	}

	/**
	 * Checks if a theme with the given slug and activation criteria already exists.
	 *
	 * @param string $theme Theme name
	 * @param string $activate Activation Criteria
	 * @return boolean
	 */
	public static function exists( $theme, $activate ) {
		$theme_type       = self::get_theme_type( $theme );
		$theme_stylesheet = self::get_theme_stylesheet( $theme, $theme_type );
		if ( ! self::is_theme_installed( $theme_stylesheet ) ) {
			return false;
		}
		if ( $activate && ! self::is_theme_active( $theme_stylesheet ) ) {
			return false;
		}
		return true;
	}
}
