<?php
/**
 * @package wpml-core
 */

global $wpdb;

global $sitepress_settings;

if ( ! isset( $sitepress_settings ) ) {
	$sitepress_settings = get_option( 'icl_sitepress_settings' );
}

// change icl_translate.field_type size to 160
$sql = "ALTER TABLE {$wpdb->prefix}icl_translate MODIFY COLUMN field_type VARCHAR( 160 ) NOT NULL";
$wpdb->query( $sql );

// Add 'batch_id' column to icl_translation_status
$sql             = $wpdb->prepare(
	"SELECT count(*) FROM information_schema.COLUMNS
     WHERE COLUMN_NAME = 'batch_id'
     and TABLE_NAME = '{$wpdb->prefix}icl_translation_status'AND TABLE_SCHEMA = %s",
	DB_NAME
);
$batch_id_exists = $wpdb->get_var( $sql );
if ( ! $batch_id_exists || ! (int) $batch_id_exists ) {
	$sql = "ALTER TABLE `{$wpdb->prefix}icl_translation_status` ADD batch_id int DEFAULT 0 NOT NULL;";
	$wpdb->query( $sql );
}

// Add 'batch_id' column to icl_string_translations
$sql             = $wpdb->prepare(
	"SELECT count(*) FROM information_schema.COLUMNS
     WHERE COLUMN_NAME = 'batch_id'
     and TABLE_NAME = '{$wpdb->prefix}icl_string_translations' AND TABLE_SCHEMA = %s",
	DB_NAME
);
$batch_id_exists = $wpdb->get_var( $sql );
if ( ! $batch_id_exists || ! (int) $batch_id_exists ) {
	$sql = "ALTER TABLE `{$wpdb->prefix}icl_string_translations` ADD batch_id int DEFAULT -1 NOT NULL;";
	$wpdb->query( $sql );
	require dirname( __FILE__ ) . '/3.2/wpml-upgrade-string-statuses.php';
	update_string_statuses();
	fix_icl_string_status();
}

// Add 'translation_service' column to icl_string_translations
$sql             = $wpdb->prepare(
	"SELECT count(*) FROM information_schema.COLUMNS
     WHERE COLUMN_NAME = 'translation_service'
     and TABLE_NAME = '{$wpdb->prefix}icl_string_translations' AND TABLE_SCHEMA = %s",
	DB_NAME
);
$batch_id_exists = $wpdb->get_var( $sql );
if ( ! $batch_id_exists || ! (int) $batch_id_exists ) {
	$sql = "ALTER TABLE `{$wpdb->prefix}icl_string_translations` ADD translation_service varchar(16) DEFAULT '' NOT NULL;";
	$wpdb->query( $sql );
}

// Add 'icl_translation_batches' table
$sql = "CREATE TABLE IF NOT EXISTS {$wpdb->prefix}icl_translation_batches (
				  `id` int(11) NOT NULL AUTO_INCREMENT,
				  `batch_name` text NOT NULL,
				  `tp_id` int NULL,
				  `ts_url` text NULL,
				  `last_update` DATETIME NULL,
				  PRIMARY KEY (`id`)
				);";
$wpdb->query( $sql );


$res                 = $wpdb->get_results( "SHOW COLUMNS FROM {$wpdb->prefix}icl_strings" );
$icl_strings_columns = array();
foreach ( $res as $row ) {
	$icl_strings_columns[] = $row->Field;
}
if ( ! in_array( 'string_package_id', $icl_strings_columns ) ) {
	$wpdb->query(
		"ALTER TABLE {$wpdb->prefix}icl_strings
        ADD `string_package_id` BIGINT unsigned NULL AFTER value,
        ADD `type` VARCHAR(40) NOT NULL DEFAULT 'LINE' AFTER string_package_id,
        ADD `title` VARCHAR(160) NULL AFTER type,
        ADD INDEX (`string_package_id`)
    "
	);
}

$wpdb->update( $wpdb->prefix . 'postmeta', array( 'meta_key' => '_wpml_original_post_id' ), array( 'meta_key' => 'original_post_id' ) );

$sitepress_settings = get_option( 'icl_sitepress_settings' );

if ( isset( $sitepress_settings['translation-management'] ) ) {
	$updated_tm_settings = false;
	$tm_settings         = $sitepress_settings['translation-management'];
	$tm_setting_keys     = array(
		'custom_fields_translation',
		'custom_fields_readonly_config',
		'custom_fields_translation_custom_readonly',
	);
	foreach ( $tm_setting_keys as $tm_setting_key ) {
		$updated_tm_settings_key = false;
		if ( isset( $tm_settings[ $tm_setting_key ] ) ) {
			$tm_custom_fields_settings = $tm_settings[ $tm_setting_key ];
			if ( array_key_exists( 'original_post_id', $tm_custom_fields_settings ) ) {
				$tm_custom_fields_settings['_wpml_original_post_id'] = $tm_custom_fields_settings['original_post_id'];
				unset( $tm_custom_fields_settings['original_post_id'] );
				$updated_tm_settings_key = true;
			}
			$index = array_search( 'original_post_id', $tm_custom_fields_settings, true );
			if ( $index ) {
				$tm_custom_fields_settings[ $index ] = '_wpml_original_post_id';
				$updated_tm_settings_key             = true;
			}
			if ( $updated_tm_settings_key ) {
				$tm_settings[ $tm_setting_key ] = $tm_custom_fields_settings;
				$updated_tm_settings            = true;
			}
		}
	}
	if ( $updated_tm_settings ) {
		$sitepress_settings['translation-management'] = $tm_settings;
		update_option( 'icl_sitepress_settings', $sitepress_settings );
	}
}