<?php
function icl_upgrade_2_0_0_steps( $step, $stepper ) {
	global $wpdb, $sitepress, $wp_post_types, $sitepress_settings;

	if ( ! isset( $sitepress ) ) {
		$sitepress = new SitePress();
	}
	$TranslationManagement = new TranslationManagement();

	$default_language = $sitepress->get_default_language();

	define( 'ICL_TM_DISABLE_ALL_NOTIFICATIONS', true ); // make sure no notifications are being sent
	ini_set( 'max_execution_time', '300' );

	$post_types = array_keys( $wp_post_types );
	foreach ( $post_types as $pt ) {
		$types[] = 'post_' . $pt;
	}

	$temp_upgrade_data = get_option(
		'icl_temp_upgrade_data',
		array(
			'step'   => 0,
			'offset' => 0,
		)
	);

	switch ( $step ) {

		case 1:
			// if the tables are missing, call the plugin activation routine
			$table_name = $wpdb->prefix . 'icl_translation_status';
			if ( $wpdb->get_var( "SHOW TABLES LIKE '{$table_name}'" ) != $table_name ) {
				icl_sitepress_activate();
			}

			$wpdb->query( "ALTER TABLE `{$wpdb->prefix}icl_translations` CHANGE `element_type` `element_type` VARCHAR( 32 ) NOT NULL DEFAULT 'post_post'" );
			$wpdb->query( "ALTER TABLE `{$wpdb->prefix}icl_translations` CHANGE `element_id` `element_id` BIGINT( 20 ) NULL DEFAULT NULL " );

			// fix source_language_code
			// all source documents must have null
			if ( isset( $types ) ) {
				$wpdb->query(
					$wpdb->prepare(
						"UPDATE {$wpdb->prefix}icl_translations SET source_language_code = NULL
					WHERE element_type IN('" . join( "','", $types ) . "') AND source_language_code = '' AND language_code=%s",
						$default_language
					)
				);
				// get translated documents with missing source language
				$res = $wpdb->get_results(
					$wpdb->prepare(
						"
					SELECT translation_id, trid, language_code
					FROM {$wpdb->prefix}icl_translations
					WHERE (source_language_code = '' OR source_language_code IS NULL)
						AND element_type IN('" . join( "','", $types ) . "')
						AND language_code <> %s
						",
						$default_language
					)
				);
				foreach ( $res as $row ) {
					$wpdb->query( $wpdb->prepare( "UPDATE {$wpdb->prefix}icl_translations SET source_language_code = %s WHERE translation_id=%d", $default_language, $row->translation_id ) );
				}
			}

			$temp_upgrade_data['step'] = 2;
			update_option( 'icl_temp_upgrade_data', $temp_upgrade_data );

			return array( 'message' => __( 'Processing translations...', 'sitepress' ) );
			break;

		case 2:
			$limit      = 100;
			$offset     = $temp_upgrade_data['offset'];
			$processing = false;

			// loop existing translations
			if ( isset( $types ) ) {
				$res = $wpdb->get_results(
					$wpdb->prepare(
						"SELECT * FROM {$wpdb->prefix}icl_translations
                                 WHERE element_type IN(" . wpml_prepare_in( $types ) . ' )
                                    AND source_language_code IS NULL LIMIT %d  OFFSET %d',
						array( $limit, $offset )
					)
				);
				foreach ( $res as $row ) {
					$processing = true;
					// grab translations
					$translations = $sitepress->get_element_translations( $row->trid, $row->element_type );

					$md5        = 0;
					$table_name = $wpdb->prefix . 'icl_node';
					if ( $wpdb->get_var( "SHOW TABLES LIKE '{$table_name}'" ) == $table_name ) {
						list($md5, $links_fixed) = $wpdb->get_row(
							$wpdb->prepare(
								"
							SELECT md5, links_fixed FROM {$wpdb->prefix}icl_node
							WHERE nid = %d
						",
								$row->element_id
							),
							ARRAY_N
						);
					}
					if ( ! $md5 ) {
						$md5 = $TranslationManagement->post_md5( $row->element_id );
					}

					$translation_package = $TranslationManagement->create_translation_package( $row->element_id );

					foreach ( $translations as $lang => $t ) {
						if ( ! $t->original ) {

							// determine service and status
							$service      = 'local';
							$needs_update = 0;

							list($rid, $status, $current_md5) = $wpdb->get_row(
								$wpdb->prepare(
									"
								SELECT c.rid, n.status , c.md5
								FROM {$wpdb->prefix}icl_content_status c
									JOIN {$wpdb->prefix}icl_core_status n ON c.rid = n.rid
								WHERE c.nid = %d AND target = %s
								ORDER BY rid DESC
								LIMIT 1
							",
									$row->element_id,
									$lang
								),
								ARRAY_N
							);

							$translator_id = false;
							if ( $rid ) {
								if ( $current_md5 != $md5 ) {
									$needs_update = 1;
								}
								if ( $status == 3 ) {
									$status = 10;
								} else {
									$status = 2;
								}
								$service = 'icanlocalize';

								foreach ( $sitepress_settings['icl_lang_status'] as $lpair ) {
									if ( $lpair['from'] == $row->language_code && $lpair['to'] == $lang && isset( $lpair['translators'][0]['id'] ) ) {
										$translator_id = $lpair['translators'][0]['id'];
										break;
									}
								}
							} else {
								$status                              = 10;
								$translator_id                       = $wpdb->get_var( $wpdb->prepare( "SELECT post_author FROM {$wpdb->posts} WHERE ID=%d", $t->element_id ) );
								$tlp                                 = get_user_meta( $translator_id, $wpdb->prefix . 'language_pairs', true );
								$tlp[ $row->language_code ][ $lang ] = 1;
								if ( method_exists( $TranslationManagement, 'edit_translator' ) ) {
									$TranslationManagement->edit_translator( $translator_id, $tlp );
								}
							}

							// add translation_status record
							list($newrid) = $TranslationManagement->update_translation_status(
								array(
									'translation_id'      => $t->translation_id,
									'status'              => $status,
									'translator_id'       => $translator_id,
									'needs_update'        => $needs_update,
									'md5'                 => $md5,
									'translation_service' => $service,
									'translation_package' => serialize( $translation_package ),
									'links_fixed'         => intval( isset( $links_fixed ) ? $links_fixed : 0 ),
								)
							);

							$job_id = wpml_tm_add_translation_job( $newrid, $translator_id, $translation_package, [] );
							if ( $job_id && $status == 10 ) {
								do_action( 'wpml_save_job_fields_from_post', $job_id );
							}
						}
					}
				}
			}
			if ( $processing ) {
				update_option(
					'icl_temp_upgrade_data',
					array(
						'step'   => 2,
						'offset' => intval( $offset + 100 ),
					)
				);
				$stepper->setNextStep( 2 );
			} else {
				update_option(
					'icl_temp_upgrade_data',
					array(
						'step'   => 3,
						'offset' => 99999999999999999999,
					)
				);
			}
			$message = $processing ? __( 'Processing translations...', 'sitepress' ) : __( 'Finalizing upgrade...', 'sitepress' );
			return array( 'message' => $message );
			break;

		case 3:
			// removing the plugins text table; importing data into a SitePress setting
			$results = $wpdb->get_results( "SELECT * FROM {$wpdb->prefix}icl_plugins_texts" );
			if ( ! empty( $results ) ) {
				foreach ( $results as $row ) {
					$cft[ $row->attribute_name ] = $row->translate + 1;
				}
				if ( isset( $cft ) ) {
					$iclsettings['translation-management']['custom_fields_translation'] = $cft;
					$sitepress->save_settings( $iclsettings );
				}

				$wpdb->query( "DROP TABLE {$wpdb->prefix}icl_plugins_texts" );
			}

			$iclsettings['language_selector_initialized'] = 1;

			if ( get_option( '_force_mp_post_http' ) ) {
				$iclsettings['troubleshooting_options']['http_communication'] = intval( get_option( '_force_mp_post_http' ) );
				delete_option( '_force_mp_post_http' );
			}

			$sitepress->save_settings( $iclsettings );

			$iclsettings['migrated_2_0_0'] = 1;
			$sitepress->save_settings( $iclsettings );
			delete_option( 'icl_temp_upgrade_data' );
			return array(
				'message'   => __( 'Done', 'sitepress' ),
				'completed' => 1,
			);
			break;

		default:
			return array(
				'error' => __( 'Missing step', 'sitepress' ),
				'stop'  => 1,
			);
	}
}

// $iclsettings defined in upgrade.php
if ( empty( $iclsettings['migrated_2_0_0'] ) ) {
	wp_enqueue_script( 'icl-stepper', ICL_PLUGIN_URL . '/inc/upgrade-functions/2.0.0/stepper.js', array( 'jquery' ) );
	add_filter( 'admin_notices', 'icl_migrate_2_0_0' );
	add_action( 'icl_ajx_custom_call', 'icl_ajx_upgrade_2_0_0', 1, 2 );
}


function icl_migrate_2_0_0() {
	$ajax_action      = 'wpml_upgrade_2_0_0';
	$ajax_action_none = wp_create_nonce( $ajax_action );
	$link             = 'index.php?icl_ajx_action=' . $ajax_action . '&nonce=' . $ajax_action_none;
	$txt              = get_option( 'icl_temp_upgrade_data', false ) ? __( 'Resume Upgrade Process', 'sitepress' ) : __( 'Run Upgrade Process', 'sitepress' );
	echo '<div class="message error" id="icl-migrate"><p><strong>' . __( 'WPML requires database upgrade', 'sitepress' ) . '</strong></p>'
			. '<p>' . __( 'This normally takes a few seconds, but may last up to several minutes of very large databases.', 'sitepress' ) . '</p>'
			. '<p><a href="' . $link . '" style="" id="icl-migrate-start">' . $txt . '</a></p>'
			. '<div id="icl-migrate-progress" style="display:none; margin: 10px 0 20px 0;">'
			. '</div></div>';
}

function icl_ajx_upgrade_2_0_0( $call, $request ) {
	if ( $call == 'wpml_upgrade_2_0_0' ) {
		$error     = 0;
		$completed = 0;
		$stop      = 0;
		$message   = __( 'Starting the upgrade process...', 'sitepress' );
		include_once WPML_PLUGIN_PATH . '/inc/upgrade-functions/2.0.0/stepper.php';
		include_once WPML_PLUGIN_PATH . '/inc/upgrade-functions/upgrade-2.0.0.php';
		$temp_upgrade_data = get_option(
			'icl_temp_upgrade_data',
			array(
				'step'   => 0,
				'offset' => 0,
			)
		);
		$step              = isset( $request['step'] ) ? $request['step'] : $temp_upgrade_data['step'];
		$migration         = new Icl_Stepper( $step );
		$migration->registerSteps(
			'icl_upgrade_2_0_0_steps',
			'icl_upgrade_2_0_0_steps',
			'icl_upgrade_2_0_0_steps'
		);
		if ( isset( $request['init'] ) ) {
			echo json_encode(
				array(
					'error'   => $error,
					'output'  => $migration->render(),
					'step'    => $migration->getNextStep(),
					'message' => __( 'Creating new tables...', 'sitepress' ),
					'stop'    => $stop,
				)
			);
			exit;
		}
		$data = $migration->init();
		@extract( $data, EXTR_OVERWRITE );
		echo json_encode(
			array(
				'error'     => $error,
				'completed' => $completed,
				'message'   => $message,
				'step'      => $migration->getNextStep(),
				'barWidth'  => $migration->barWidth(),
				'stop'      => $stop,
			)
		);
	}
}