<?php

/**
 * Class EIG_WP_CLI_Branding
 *
 * This class is instantiated in /inc/cli-init.php
 */
class EIG_WP_CLI_Branding extends EIG_WP_CLI_Command {

	/**
	 * @var string - Stores brand slug
	 * @see EIG_WP_CLI_Loader->brand_aliases & EIG_WP_CLI_Loader->must_use_aliases
	 */
	protected static $option_key = 'mm_brand';

	/**
	 * @var string - Stores SVG of current brand's icon in database.
	 */
	protected static $icon_transient_key = 'mm_icon_hash';

	/**
	 * @var string - Remote location of branding.json config file.
	 */
	protected static $api_branding_file = MM_BASE_URL . 'assets/json/branding.json';

	/**
	 * @var string - User-provided action.
	 */
	protected $action;

	/**
	 * @var string - User-provided brand.
	 */
	protected $brand;

	/**
	 * Manage Endurance Branding and Conditional Connections
	 *
	 * @param array $args
	 * @param array $assoc_args
	 *
	 * @throws \WP_CLI\ExitException
	 */
	public function __invoke( $args, $assoc_args ) {
		/**
		 * Initialize $this->action & $this->brand
		 */
		$this->init_class_params( $args, $assoc_args );

		/**
		 * Route Sub-commands to Methods
		 */
		switch ( $this->action ) {
			case 'update':
				$this->update();
				break;
			case 'remove':
				$this->remove();
				break;
		}
	}

	/**
	 * Initialize $this->action, $this->brand and other class vars.
	 *
	 * @param array $args
	 * @param array $assoc_args
	 */
	protected function init_class_params( $args, $assoc_args ) {
		/**
		 * Validate & Decide $this->action + $this->brand
		 */
		if ( isset( $args[0] ) && 'update' === $args[0] ) {
			$this->action = $args[0];
			if ( ! empty( $args[1] ) ) {
				$this->brand = $args[1];
			} else {
				$this->error( 'No brand identified to update.' );
			}
		} elseif ( isset( $args[0] ) && 'remove' === $args[0] ) {
			$this->action = 'remove';
		} else {
			if ( ! empty( $assoc_args['update'] ) ) {
				$this->action = 'update';
				$this->brand  = $assoc_args['update'];
			} elseif ( isset( $assoc_args['remove'] ) ) {
				$this->action = 'remove';
			} else {
				$this->error( 'No valid action provided.' );
			}
		}
	}

	/**
	 * wp {alias} branding update {brand}
	 *       -- -- -- OR -- -- --
	 * wp {alias} branding --update={brand}
	 *
	 * @throws \WP_CLI\ExitException
	 */
	protected function update() {
		$valid_brands   = $this->get_brands();
		$existing_brand = get_option( static::$option_key, null );

		if ( in_array( $this->brand, $valid_brands ) ) {
			/**
			 * Check if passed brand param matches saved brand
			 */
			if ( $this->brand === $existing_brand ) {
				$this->success( 'Branding already set to "' . $this->brand . '", wiping transients...' );
				delete_transient( static::$icon_transient_key );
				\WP_CLI::halt( 200 );
			} elseif ( update_option( static::$option_key, $this->brand ) ) {
				delete_transient( static::$icon_transient_key );
				$this->success( 'Plugin branding updated to: ' . $this->brand );
			} else {
				$this->error( 'Failed to update plugin branding to: ' . $this->brand );
			}
		} else {
			if ( ! empty( $valid_brands ) ) {
				$lowercase = array_map( 'strtolower', $valid_brands );
				if ( in_array( $this->brand, $lowercase ) ) {
					$this->warning( 'WRONG CASE! Try that string again with the correct case.' );
				}
				$this->colorize_log( 'Valid brands: ' );
				$this->table( $valid_brands, array( '---', 'BRAND SLUG' ) );
			}
			$this->error( 'Didn\'t receive valid brand in subcommand or flag value.' );
		}
	}

	/**
	 * wp {alias} branding remove
	 *       -- -- -- OR -- -- --
	 * wp {alias} branding --remove
	 */
	protected function remove() {
		if ( delete_option( static::$option_key ) ) {
			delete_transient( static::$icon_transient_key );
			$this->success( 'Plugin branding removed successfully.' );
		} else {
			$this->warning( 'No plugin branding found. Try "wp {alias} branding update {brand}"' );
			if ( ! empty( $valid_brands = $this->get_brands() ) ) {
				$this->colorize_log( 'Valid brands: ' );
				$this->table( $valid_brands, array( '---', 'BRAND SLUG' ) );
			}
		}
	}

	/**
	 * Helper to hit Mojo API Cache for JSON Branding Map.
	 *
	 * @return array
	 */
	protected function get_brands() {
		$valid_brands = array();
		$brands       = mm_api_cache( static::$api_branding_file );
		if ( ! is_wp_error( $brands ) && $brands = json_decode( $brands['body'] ) ) {
			$brands = (array) $brands;
			while ( false !== ( $brand = array_search( 'default', $brands ) ) ) {
				unset( $brands[ $brand ] );
			}
			$valid_brands = array_keys( $brands );
		}

		return $valid_brands;
	}
}
