class-wpml-st-upgrade-db-strings-add-translation-priority-field.php 1.58 KB
<?php

class WPML_ST_Upgrade_DB_Strings_Add_Translation_Priority_Field implements IWPML_St_Upgrade_Command {
	/** @var wpdb */
	private $wpdb;

	/**
	 * @param wpdb $wpdb
	 */
	public function __construct( wpdb $wpdb ) {
		$this->wpdb = $wpdb;
	}

	public function run() {
		$result = null;

		$table_name = $this->wpdb->prefix . 'icl_strings';
		/** @var array<int, object> $results */
		$results = $this->wpdb->get_results( "SHOW TABLES LIKE '{$table_name}'" );
		if ( 0 !== count( $results ) ) {
			$sql = "SHOW FIELDS FROM  {$table_name} WHERE FIELD = 'translation_priority'";
			/** @var array<int, object> $s_results */
			$s_results = $this->wpdb->get_results( $sql );
			if ( 0 === count( $s_results ) ) {
				$sql = "ALTER TABLE {$this->wpdb->prefix}icl_strings 
				ADD COLUMN `translation_priority` varchar(160) NOT NULL";

				$result = false !== $this->wpdb->query( $sql );
			}

			if ( false !== $result ) {
				$sql = "SHOW KEYS FROM  {$table_name} WHERE Key_name='icl_strings_translation_priority'";
				/** @var array<int, object> $results */
				$results = $this->wpdb->get_results( $sql );
				if ( 0 === count( $results ) ) {
					$sql = "
					ALTER TABLE {$this->wpdb->prefix}icl_strings 
					ADD INDEX `icl_strings_translation_priority` ( `translation_priority` ASC )
					";

					$result = false !== $this->wpdb->query( $sql );
				} else {
					$result = true;
				}
			}
		}

		return (bool) $result;
	}

	public function run_ajax() {
		return $this->run();
	}

	public function run_frontend() {
		return $this->run();
	}

	public static function get_command_id() {
		return __CLASS__;
	}
}