<?php
namespace Burst\Admin;

if ( ! defined( 'ABSPATH' ) ) {
	exit;
}

use Burst\Admin\Capability\Capability;
use Burst\Traits\Admin_Helper;
use Burst\Traits\Database_Helper;
use Burst\Traits\Save;
use Burst\Frontend\Goals\Goals;


class Upgrade {
	use Admin_Helper;
	use Database_Helper;
	use Save;

	/**
	 * Constructor
	 */
	public function init(): void {
		add_action( 'init', [ $this, 'check_upgrade' ], 10, 2 );
	}

	/**
	 * Run an upgrade procedure if the version has changed
	 */
	public function check_upgrade(): void {
		if ( ! $this->has_admin_access() ) {
			return;
		}

		$prev_version = get_option( 'burst-current-version', false );
		$new_version  = BURST_VERSION;

		// strip off everything after '#'.
		if ( strpos( $new_version, '#' ) !== false ) {
			$new_version = substr( $new_version, 0, strpos( $new_version, '#' ) );
		}

		if ( $prev_version === $new_version ) {
			return;
		}

		// install the tables, so we can access new columns below if necessary.
		do_action( 'burst_upgrade_before', $prev_version );

		// no upgrade.
		// add burst capabilities.
		if ( $prev_version
			&& version_compare( $prev_version, '1.1.1', '<' )
		) {
			Capability::add_capability( 'view', [ 'administrator', 'editor' ] );
			Capability::add_capability( 'manage' );
		}

		// Version 1.3.5.
		// - Upgrade to new bounce table.
		// - Upgrade to remove `event` and `action` columns from `burst_statistics` table.
		if ( $prev_version
			&& version_compare( $prev_version, '1.4.2.1', '<' ) ) {
			update_option( 'burst_db_upgrade_bounces', true );
			update_option( 'burst_db_upgrade_goals_remove_columns', true );
		}
		if ( $prev_version
			&& version_compare( $prev_version, '1.5.2', '<' ) ) {
			update_option( 'burst_db_upgrade_goals_set_conversion_metric', true );
		}

		if ( $prev_version
			&& version_compare( $prev_version, '1.5.3', '<' ) ) {
			update_option( 'burst_db_upgrade_strip_domain_names_from_entire_page_url', true );
			update_option( 'burst_db_upgrade_empty_referrer_when_current_domain', true );
			update_option( 'burst_db_upgrade_drop_user_agent', true );

			// remove the endpoint file from the old location.
			if ( file_exists( ABSPATH . '/burst-statistics-endpoint.php' ) ) {
				wp_delete_file( ABSPATH . '/burst-statistics-endpoint.php' );
			}
		}

		if ( $prev_version
			&& version_compare( $prev_version, '1.6.1', '<' ) ) {
			// add the admin to the email reports mailing list.
			$mailinglist = burst_get_option( 'email_reports_mailinglist' );
			if ( ! $mailinglist ) {
				$defaults = [
					[
						'email'     => get_option( 'admin_email' ),
						'frequency' => 'monthly',
					],
				];
				$this->update_option( 'email_reports_mailinglist', $defaults );
			}
		}

		if ( $prev_version
			&& version_compare( $prev_version, '1.6.1', '<' ) ) {
			// add the admin to the email reports mailing list.
			$mailinglist = burst_get_option( 'email_reports_mailinglist' );
			if ( ! $mailinglist ) {
				$defaults = [
					[
						'email'     => get_option( 'admin_email' ),
						'frequency' => 'monthly',
					],
				];
				$this->update_option( 'email_reports_mailinglist', $defaults );
			}
		}

		// check if column 'device_id' exists in the table 'burst_statistics'.
		$is_version_upgrade      = $prev_version && version_compare( $prev_version, '1.7.0', '<' );
		$lookup_table_incomplete = version_compare( $prev_version, '1.7.1', '=' ) && ! $this->column_exists( 'burst_statistics', 'device_id' );
		if ( $lookup_table_incomplete || $is_version_upgrade ) {
			update_option( 'burst_last_cron_hit', time(), false );
			// this option is used in the tracking, so should autoload until completed.
			update_option( 'burst_db_upgrade_create_lookup_tables', true, true );
			update_option( 'burst_db_upgrade_init_lookup_ids', true, false );
			update_option( 'burst_db_upgrade_upgrade_lookup_tables', true, false );
			update_option( 'burst_db_upgrade_upgrade_lookup_tables_drop_columns', true, false );

			// for each table separately, for fine grained control.
			update_option( 'burst_db_upgrade_create_lookup_tables_browser', true, false );
			update_option( 'burst_db_upgrade_create_lookup_tables_browser_version', true, false );
			update_option( 'burst_db_upgrade_create_lookup_tables_platform', true, false );
			update_option( 'burst_db_upgrade_create_lookup_tables_device', true, false );
			update_option( 'burst_db_upgrade_upgrade_lookup_tables_browser', true, false );
			update_option( 'burst_db_upgrade_upgrade_lookup_tables_browser_version', true, false );
			update_option( 'burst_db_upgrade_upgrade_lookup_tables_platform', true, false );
			update_option( 'burst_db_upgrade_upgrade_lookup_tables_device', true, false );

			// drop post_meta feature.
			update_option( 'burst_db_upgrade_drop_page_id_column', true, false );

			wp_schedule_single_event( time() + 300, 'burst_upgrade_iteration' );

			$mu_plugin = trailingslashit( WPMU_PLUGIN_DIR ) . 'burst_rest_api_optimizer.php';
			if ( file_exists( $mu_plugin ) ) {
				wp_delete_file( $mu_plugin );
			}
		}

		if ( $prev_version
			&& version_compare( $prev_version, '1.7.3', '<' ) ) {
			wp_clear_scheduled_hook( 'burst_every_5_minutes' );
			update_option( 'burst_db_upgrade_rename_entire_page_url_column', true, false );
			update_option( 'burst_db_upgrade_drop_path_from_parameters_column', true, false );

			wp_schedule_single_event( time() + 300, 'burst_upgrade_iteration' );
		}

		if ( $prev_version && version_compare( $prev_version, '2.0.0', '<' ) ) {
			add_action( 'plugins_loaded', [ $this, 'upgrade_goals' ], 30 );
		}

		// upgrade missing session_ids.
		// in the stats table, find the oldest session_id = 1 that is preceded with a higher sesssion_id.
		if ( $prev_version && version_compare( $prev_version, '2.0.4', '<' ) && ! defined( 'BURST_FREE' ) ) {
			update_option( 'burst_fix_incorrect_bounces', true, false );
			update_option( 'burst_db_upgrade_fix_missing_session_ids', true, false );
			update_option( 'burst_db_upgrade_clean_orphaned_session_ids', true, false );
		}

		// ensure the onboarding doesn't start again if users already had the plugin activated.
		if ( $prev_version && version_compare( $prev_version, '2.1.0', '<' ) ) {
			if ( defined( 'BURST_PRO' ) ) {
				update_option( 'burst_activation_time_pro', time(), false );
			}
		}

		if ( $prev_version && version_compare( $prev_version, '2.2.3', '<' ) ) {
			$mu_plugin = trailingslashit( WPMU_PLUGIN_DIR ) . 'burst_rest_api_optimizer.php';
			if ( file_exists( $mu_plugin ) ) {
				wp_delete_file( $mu_plugin );
			}

			// Remove duplicates before dbDelta adds UNIQUE constraint.
			// Keep only the row with the lowest ID for each (goal_id, statistic_id) pair.
			// this is an old upgrade, moving it here to clean up the install_goal_statistics_table() function and ensure it only runs once.
			if ( $this->table_exists( 'burst_goal_statistics' ) ) {
				global $wpdb;
				$wpdb->query(
					"DELETE gs1 FROM {$wpdb->prefix}burst_goal_statistics gs1
                INNER JOIN {$wpdb->prefix}burst_goal_statistics gs2 
                WHERE gs1.goal_id = gs2.goal_id 
                AND gs1.statistic_id = gs2.statistic_id 
                AND gs1.ID > gs2.ID"
				);
			}
		}

		if ( $prev_version && version_compare( $prev_version, '2.2.6', '<' ) ) {
			update_option( 'burst_db_upgrade_add_page_ids', true, false );
			delete_post_meta_by_key( 'burst_total_pageviews_count' );
		}

		if ( $prev_version && version_compare( $prev_version, '3.0.0', '<' ) ) {
			\Burst\burst_loader()->admin->tasks->add_task( 'ecommerce_integration' );
			burst_reinstall_rest_api_optimizer();
		}

		if ( $prev_version && version_compare( $prev_version, '3.0.1', '<' ) ) {
			update_option( 'burst_is_multi_domain', false );
			$plugin_activated_time = get_option( 'burst_activation_time', 0 );
			$cutoff_date           = strtotime( '2025-11-24 00:00:00' );
			if ( $plugin_activated_time > 0 && $plugin_activated_time < $cutoff_date ) {
				if ( ! defined( 'BURST_PRO' ) ) {
					\Burst\burst_loader()->admin->tasks->undismiss_task( 'bf_notice' );
					\Burst\burst_loader()->admin->tasks->undismiss_task( 'cm_notice' );
				}
			}
		}

        // phpcs:disable
        if ( $prev_version && version_compare( $prev_version, '3.1.0', '<' ) ) {
            global $wpdb;
            $sql = "DROP TABLE IF EXISTS {$wpdb->prefix}burst_summary";
            $wpdb->query( $sql );

            $stats_table = "{$wpdb->prefix}burst_statistics";
            $known_table = "{$wpdb->prefix}burst_known_uids";

            // One-time fill from existing data
            $wpdb->query("
                INSERT INTO $known_table (uid, first_seen, last_seen)
                SELECT uid, MIN(time) as first_seen, MAX(time) as last_seen
                FROM $stats_table
                WHERE time >= UNIX_TIMESTAMP(NOW() - INTERVAL 1 MONTH)
                GROUP BY uid
            ");
        }

        //phpcs:enable
		if ( $prev_version && version_compare( $prev_version, '3.1.0.4', '<' ) ) {
			$plugin_activated_time = get_option( 'burst_activation_time', 0 );
			$cutoff_date           = strtotime( '2025-11-24 00:00:00' );
			if ( $plugin_activated_time > 0 && $plugin_activated_time < $cutoff_date ) {
				// free already dismissed in 3.0.1.
				if ( defined( 'BURST_PRO' ) ) {
					\Burst\burst_loader()->admin->tasks->undismiss_task( 'bf_notice' );
					\Burst\burst_loader()->admin->tasks->undismiss_task( 'cm_notice' );
					\Burst\burst_loader()->admin->tasks->schedule_task_validation();
				}
			}
		}

		if ( $prev_version && version_compare( $prev_version, '3.1.1', '<' ) ) {
			// reinstall with permalink for woocommerce exclusion.
			burst_reinstall_rest_api_optimizer();
		}

		if ( $prev_version && version_compare( $prev_version, '3.1.4', '<' ) ) {
			\Burst\burst_loader()->admin->tasks->add_task( 'filters_in_url' );
			update_option( 'burst_db_upgrade_move_referrers_to_sessions', true, false );
			delete_option( 'burst_pageviews_to_update' );
			// clear old referrers table.
			\Burst\burst_loader()->admin->app->weekly_clear_referrers_table();
		}

		if ( $prev_version && version_compare( $prev_version, '3.2.0', '<' ) ) {
			burst_reinstall_rest_api_optimizer();
			$installed_by = get_option( 'teamupdraft_installation_source_burst-statistics' );
			if ( ! empty( $installed_by ) ) {
				update_site_option( 'teamupdraft_installation_source_burst-statistics', $installed_by );
			}
			update_option( 'burst_db_upgrade_move_reports_to_new_tables', true, false );
			\Burst\burst_loader()->admin->tasks->add_task( 'join-discord' );
		}

		if ( $prev_version && version_compare( $prev_version, '3.2.1', '<' ) ) {
			if ( ! $this->get_option_bool( 'anonymous_usage_data' ) ) {
				\Burst\burst_loader()->admin->tasks->add_task( 'opt-in-sharing' );
			}

			if ( $this->get_option_bool( 'enable_cookieless_tracking' ) && ! $this->get_option_bool( 'enable_turbo_mode' ) ) {
				\Burst\burst_loader()->admin->tasks->add_task( 'turbo_mode_recommended' );
			}
		}

		if ( $prev_version && version_compare( $prev_version, '3.3.0', '<' ) ) {
			update_option( 'burst_db_upgrade_move_columns_to_sessions', true, false );
		}

		$admin = new Admin();
		$admin->run_table_init_hook();
		$admin->create_js_file();
		wp_schedule_single_event( time() + 60, 'burst_upgrade_iteration' );
		do_action( 'burst_upgrade_after', $prev_version );
		update_option( 'burst-current-version', $new_version );
	}

	/**
	 * Run upgrade for goals after 2.0 update
	 */
	public function upgrade_goals(): void {
		// upgrade all goals to use the new selector field.
		$goal_object = new Goals();
		$goals       = $goal_object->get_goals();
		foreach ( $goals as $goal ) {
			if ( $goal->type === 'clicks' || $goal->type === 'views' ) {
				if ( isset( $goal->attribute_value ) && $goal->attribute_value !== '' ) {
					$goal->selector = $goal->attribute === 'id' ? '#' . $goal->attribute_value : '.' . $goal->attribute_value;
				}
			}
			$goal->save();
		}
	}
}
