<?php

class WPML_Upgrade_Schema {

	/** @var wpdb $wpdb */
	private $wpdb;

	public function __construct( wpdb $wpdb ) {
		$this->wpdb = $wpdb;
	}

	/**
	 * @param string $table_name
	 *
	 * @return bool
	 */
	public function does_table_exist( $table_name ) {
		return (bool) count( $this->wpdb->get_results( "SHOW TABLES LIKE '{$this->wpdb->prefix}{$table_name}'" ) );
	}

	/**
	 * @param string $table_name
	 * @param string $column_name
	 *
	 * @return bool
	 */
	public function does_column_exist( $table_name, $column_name ) {
		return (bool) count( $this->wpdb->get_results( "SHOW COLUMNS FROM {$this->wpdb->prefix}{$table_name} LIKE '{$column_name}'" ) );
	}

	/**
	 * @param string $table_name
	 * @param string $column_name
	 * @param string $attribute_string
	 *
	 * @return false|int
	 */
	public function add_column( $table_name, $column_name, $attribute_string ) {
		return $this->wpdb->query( "ALTER TABLE {$this->wpdb->prefix}{$table_name} ADD `{$column_name}` {$attribute_string}" );
	}

	/**
	 * @param string $table_name
	 * @param string $column_name
	 * @param string $attribute_string
	 *
	 * @return false|int
	 */
	public function modify_column( $table_name, $column_name, $attribute_string ) {
		return $this->wpdb->query( "ALTER TABLE {$this->wpdb->prefix}{$table_name} MODIFY COLUMN `{$column_name}` {$attribute_string}" );
	}

	/**
	 * @param string $table_name
	 * @param string $column_name
	 *
	 * @return null|string
	 */
	public function get_column_collation( $table_name, $column_name ) {
		return $this->wpdb->get_var(
			"SELECT COLLATION_NAME FROM INFORMATION_SCHEMA.COLUMNS
			 WHERE TABLE_SCHEMA = '{$this->wpdb->dbname}'
			 	AND TABLE_NAME = '{$this->wpdb->prefix}{$table_name}'
			 	AND COLUMN_NAME = '{$column_name}'"
		);
	}

	/**
	 * @param string $table_name
	 *
	 * @return string|null
	 */
	public function get_table_collation( $table_name ) {
		$table_data = $this->wpdb->get_row(
			$this->wpdb->prepare( 'SHOW TABLE status LIKE %s', $table_name )
		);

		if ( isset( $table_data->Collation ) ) {
			return $table_data->Collation;
		}

		return null;
	}

	/**
	 * @return string|null
	 */
	public function get_default_collate() {
		$posts_table_collate = $this->get_table_collation( $this->wpdb->posts );

		if ( $posts_table_collate ) {
			return $posts_table_collate;
		} elseif ( ! empty( $this->wpdb->collate ) ) {
			return $this->wpdb->collate;
		}

		return null;
	}

	public function get_wpdb() {
		return $this->wpdb;
	}
}