| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669 |
- <?php
- /*******************************************************************************
- * Copyright (c) 2019, Code Atlantic LLC
- ******************************************************************************/
- // Exit if accessed directly
- if ( ! defined( 'ABSPATH' ) ) {
- exit;
- }
- /**
- * Class PUM_Upgrades
- */
- class PUM_Admin_Upgrades {
- /**
- * @var PUM_Admin_Upgrades The one true PUM_Admin_Upgrades
- */
- private static $instance;
- /**
- * @var $upgrade_args
- */
- public $upgrade_args = array();
- public $page = null;
- public $doing_upgrades = false;
- public $required_cap = 'manage_options';
- public $current_routine = null;
- public $next_routine = null;
- /**
- * Initialize the actions needed to process upgrades.
- */
- public static function instance() {
- if ( ! isset( self::$instance ) && ! ( self::$instance instanceof PUM_Admin_Upgrades ) ) {
- self::$instance = new PUM_Admin_Upgrades;
- self::$instance->init();
- }
- return self::$instance;
- }
- /**
- * Initialize the actions needed to process upgrades.
- */
- public function init() {
- $this->update_plugin_version();
- $this->required_cap = apply_filters( 'pum_upgrade_required_cap', 'manage_options' );
- // bail if this plugin data doesn't need updating
- if ( pum_get_db_ver() >= Popup_Maker::$DB_VER ) {
- return;
- }
- add_action( 'admin_menu', array( $this, 'register_pages' ) );
- add_action( 'network_admin_menu', array( $this, 'register_pages' ) );
- add_action( 'admin_init', array( $this, 'process_upgrade_args' ) );
- add_action( 'wp_ajax_pum_trigger_upgrades', array( $this, 'trigger_upgrades' ) );
- add_action( 'admin_notices', array( $this, 'show_upgrade_notices' ) );
- }
- public function update_plugin_version() {
- $current_ver = get_option( 'pum_ver', false );
- if ( ! $current_ver ) {
- $deprecated_ver = get_site_option( 'popmake_version', false );
- $current_ver = $deprecated_ver ? $deprecated_ver : Popup_Maker::$VER;
- add_option( 'pum_ver', Popup_Maker::$VER );
- }
- if ( version_compare( $current_ver, Popup_Maker::$VER, '<' ) ) {
- // Save Upgraded From option
- update_option( 'pum_ver_upgraded_from', $current_ver );
- update_option( 'pum_ver', Popup_Maker::$VER );
- }
- }
- /**
- * Registers the pum-upgrades admin page.
- */
- public function register_pages() {
- global $pum_upgrades_page;
- $parent = null;
- /*
- if ( function_exists( 'is_network_admin' ) && is_network_admin() ) {
- add_menu_page(
- __( 'Popup Maker', 'popup-maker' ),
- __( 'Popup Maker', 'popup-maker' ),
- 'manage_network_plugins',
- 'popup-maker',
- '',
- pum_get_svg_icon( true )
- );
- $parent = 'popup-maker';
- }
- */
- $this->page = add_submenu_page(
- $parent,
- __( 'Popup Maker Upgrades', 'popup-maker' ),
- __( 'Popup Maker Upgrades', 'popup-maker' ),
- $this->required_cap,
- 'pum-upgrades',
- array( $this, 'upgrades_screen' )
- );
- $pum_upgrades_page = $this->page;
- }
- /**
- * Process upgrade args.
- */
- public function process_upgrade_args() {
- $page = isset( $_GET['page'] ) ? $_GET['page'] : '';
- if ( ! ( defined( 'DOING_AJAX' ) && DOING_AJAX && $_REQUEST['action'] == 'pum_trigger_upgrades' ) && $page != 'pum-upgrades' ) {
- return;
- }
- $this->doing_upgrades = true;
- $action = isset( $_REQUEST['pum-upgrade'] ) ? sanitize_text_field( $_REQUEST['pum-upgrade'] ) : $this->get_pum_db_ver() + 1;
- $step = isset( $_REQUEST['step'] ) ? absint( $_REQUEST['step'] ) : 1;
- $total = isset( $_REQUEST['total'] ) ? absint( $_REQUEST['total'] ) : false;
- $custom = isset( $_REQUEST['custom'] ) ? absint( $_REQUEST['custom'] ) : 0;
- $number = isset( $_REQUEST['number'] ) ? absint( $_REQUEST['number'] ) : 100;
- $completed = isset( $_REQUEST['completed'] ) ? absint( $_REQUEST['completed'] ) : false;
- $steps = ceil( $total / $number );
- if ( $step > $steps ) {
- // Prevent a weird case where the estimate was off. Usually only a couple.
- $steps = $step;
- }
- $this->upgrade_args = array(
- 'page' => 'pum-upgrades',
- 'pum-upgrade' => $action,
- 'step' => $step,
- 'total' => $total,
- 'custom' => $custom,
- 'steps' => $steps,
- 'number' => $number,
- 'completed' => $completed,
- );
- update_option( 'pum_doing_upgrade', $this->upgrade_args );
- }
- /**
- * Get upgrade arg.
- *
- * @param string $key
- *
- * @return bool|null
- */
- public function set_arg( $key, $value = null ) {
- $this->upgrade_args[ $key ] = $value;
- if ( $key == 'number' || $key == 'total' ) {
- $this->upgrade_args['steps'] = ceil( $this->upgrade_args['total'] / $this->upgrade_args['number'] );
- }
- if ( $this->upgrade_args['step'] > $this->upgrade_args['steps'] ) {
- // Prevent a weird case where the estimate was off. Usually only a couple.
- $this->upgrade_args['steps'] = $this->upgrade_args['step'];
- } elseif ( $this->upgrade_args['step'] * $this->upgrade_args['steps'] ) {
- update_option( 'pum_doing_upgrade', $this->upgrade_args );
- }
- }
- /**
- * Get upgrade arg.
- *
- * @param string $key
- *
- * @return bool|null
- */
- public function get_arg( $key = null ) {
- if ( ! $key ) {
- return null;
- }
- if ( ! isset( $this->upgrade_args[ $key ] ) ) {
- return false;
- }
- return $this->upgrade_args[ $key ];
- }
- public function get_args() {
- return $this->upgrade_args;
- }
- public function doing_upgrades() {
- return $this->doing_upgrades;
- }
- /**
- * Display Upgrade Notices
- *
- * @return void
- */
- public function show_upgrade_notices() {
- $screen = get_current_screen();
- if ( $screen->id == $this->page ) {
- return; // Don't show notices on the upgrades page
- }
- if ( ! $this->has_upgrades() ) {
- return;
- }
- // Sequential Orders was the first stepped upgrade, so check if we have a stalled upgrade
- $resume_upgrade = $this->maybe_resume_upgrade();
- if ( ! empty( $resume_upgrade ) ) {
- $resume_url = add_query_arg( $resume_upgrade, admin_url( 'index.php' ) );
- printf(
- '<div class="error"><p>' . __( 'Popup Maker needs to complete a database upgrade that was previously started, click <a href="%s">here</a> to resume the upgrade.', 'popup-maker' ) . '</p></div>',
- esc_url( $resume_url )
- );
- } else {
- printf(
- '<div class="error"><p><strong>%s:</strong> <span class="dashicons dashicons-warning" style="color: #dc3232;"></span> %s %s %s</p></div>',
- __( 'Popup Maker', 'popup-maker' ),
- __( 'Important', 'popup-maker' ),
- __( 'Database upgrades required.', 'popup-maker' ),
- sprintf(
- __( 'Please click %shere%s to complete these changes now.', 'popup-maker' ),
- '<a href="' . esc_url( admin_url( 'options.php?page=pum-upgrades' ) ) . '">',
- '</a>'
- )
- );
- }
- }
- /**
- * Triggers all upgrade functions
- *
- * This function is usually triggered via AJAX
- *
- * @return void
- */
- public function trigger_upgrades() {
- if ( ! current_user_can( $this->required_cap ) ) {
- wp_die( __( 'You do not have permission to do upgrades', 'popup-maker' ), __( 'Error', 'popup-maker' ), array( 'response' => 403 ) );
- }
- $deprecated_ver = get_site_option( 'popmake_version', false );
- $current_ver = get_option( 'pum_ver', $deprecated_ver );
- // Save Upgraded From option
- if ( $current_ver ) {
- update_option( 'pum_ver_upgraded_from', $current_ver );
- }
- update_option( 'pum_ver', Popup_Maker::$VER );
- // Process DB Upgrades
- $this->process_upgrades();
- if ( DOING_AJAX ) {
- echo wp_json_encode( array(
- 'complete' => true,
- 'status' => sprintf(
- '<strong>%s</strong><br/>%s',
- __( 'Upgrades have been completed successfully.', 'popup-maker' ),
- sprintf( 'You will automatically be redirected in %s seconds', '<span id="pum-countdown">5</span>' )
- ),
- 'redirect' => admin_url( 'edit.php?post_type=popup' ),
- 'countdown' => 5000,
- ) ); // Let AJAX know that the upgrade is complete
- exit;
- }
- }
- /**
- * Updates the pum_db_ver to the passed $version.
- *
- * If no $version is passed a default value will be established.
- *
- * @param null $version
- */
- public function set_pum_db_ver( $version = null ) {
- if ( $version ) {
- $version = preg_replace( '/[^0-9.].*/', '', $version );
- update_option( 'pum_db_ver', $version );
- return;
- }
- $upgraded_from = get_option( 'pum_ver_upgraded_from', false );
- // this is the current database schema version number
- $current_db_ver = pum_get_db_ver();
- // If no current db version, but prior install detected, set db version correctly.
- if ( ! $current_db_ver ) {
- if ( $upgraded_from ) {
- if ( version_compare( $upgraded_from, '1.3.0', '<' ) ) {
- $current_db_ver = 1;
- } else {
- $current_db_ver = 2;
- }
- } else {
- $current_db_ver = Popup_Maker::$DB_VER;
- }
- add_option( 'pum_db_ver', $current_db_ver );
- }
- }
- /**
- * Gets the pum_db_ver or sets and returns the correct one.
- *
- * @see PUM_Utils_Upgrades::set_pum_db_ver()
- *
- * return $pum_db_ver
- */
- public function get_pum_db_ver() {
- static $pum_db_ver;
- if ( ! isset( $pum_db_ver ) ) {
- // this is the current database schema version number
- $pum_db_ver = pum_get_db_ver();
- }
- if ( ! $pum_db_ver ) {
- $this->set_pum_db_ver();
- $pum_db_ver = pum_get_db_ver();
- }
- return preg_replace( '/[^0-9.].*/', '', $pum_db_ver );
- }
- /**
- * Process upgrades in a stepped succession.
- *
- * Starts with the current version and loops until reaching the target version.
- */
- public function process_upgrades() {
- // this is the target version that we need to reach
- $target_db_ver = Popup_Maker::$DB_VER;
- // this is the current database schema version number
- $current_db_ver = $this->get_pum_db_ver();
- // Run upgrade routine until target version reached.
- while ( $current_db_ver < $target_db_ver ) {
- // increment the current db_ver by one
- $current_db_ver ++;
- $this->current_routine = $current_db_ver;
- $this->next_routine = $current_db_ver == $target_db_ver ? null : $current_db_ver + 1;
- if ( file_exists( POPMAKE_DIR . "includes/admin/upgrades/class-pum-admin-upgrade-routine-{$current_db_ver}.php" ) ) {
- require_once POPMAKE_DIR . "includes/admin/upgrades/class-pum-admin-upgrade-routine-{$current_db_ver}.php";
- $func = "PUM_Admin_Upgrade_Routine_{$current_db_ver}::run";
- if ( is_callable( $func ) ) {
- call_user_func( $func );
- }
- }
- }
- }
- public function current_routine() {
- return $this->current_routine;
- }
- public function next_routine() {
- return $this->next_routine;
- }
- /**
- * Process upgrades in a stepped succession.
- *
- * Starts with the current version and loops until reaching the target version.
- */
- public function get_upgrades() {
- // this is the target version that we need to reach
- $target_db_ver = Popup_Maker::$DB_VER;
- // this is the current database schema version number
- $current_db_ver = $this->get_pum_db_ver();
- $upgrades = array();
- // Run upgrade routine until target version reached.
- while ( $current_db_ver < $target_db_ver ) {
- // increment the current db_ver by one
- $current_db_ver ++;
- if ( file_exists( POPMAKE_DIR . "includes/admin/upgrades/class-pum-admin-upgrade-routine-{$current_db_ver}.php" ) ) {
- require_once POPMAKE_DIR . "includes/admin/upgrades/class-pum-admin-upgrade-routine-{$current_db_ver}.php";
- $func = "PUM_Admin_Upgrade_Routine_{$current_db_ver}::description";
- if ( is_callable( $func ) ) {
- $upgrades[ $current_db_ver ] = call_user_func( $func );
- }
- }
- }
- return $upgrades;
- }
- public function get_upgrade( $version = null ) {
- $upgrades = $this->get_upgrades();
- if ( isset ( $upgrades[ $version ] ) ) {
- return $upgrades[ $version ];
- } else {
- return false;
- }
- }
- /**
- * Returns true if there are unprocessed upgrades.
- *
- * @return bool
- */
- public function has_upgrades() {
- return boolval( count( $this->get_upgrades() ) );
- }
- /**
- * For use when doing 'stepped' upgrade routines, to see if we need to start somewhere in the middle
- *
- * @return mixed When nothing to resume returns false, otherwise starts the upgrade where it left off
- */
- public function maybe_resume_upgrade() {
- $doing_upgrade = get_option( 'pum_doing_upgrade', false );
- if ( empty( $doing_upgrade ) ) {
- return false;
- }
- return $doing_upgrade;
- }
- /**
- * Adds an upgrade action to the completed upgrades array
- *
- * @param string $upgrade_action The action to add to the competed upgrades array
- *
- * @return bool If the function was successfully added
- */
- public function set_upgrade_complete( $upgrade_action = '' ) {
- if ( empty( $upgrade_action ) ) {
- return false;
- }
- $completed_upgrades = $this->get_completed_upgrades();
- $completed_upgrades[] = $upgrade_action;
- // Remove any blanks, and only show uniques
- $completed_upgrades = array_unique( array_values( $completed_upgrades ) );
- return update_option( 'pum_completed_upgrades', $completed_upgrades );
- }
- /**
- * Check if the upgrade routine has been run for a specific action
- *
- * @param string $upgrade_action The upgrade action to check completion for
- *
- * @return bool If the action has been added to the copmleted actions array
- */
- public function has_upgrade_completed( $upgrade_action = '' ) {
- if ( empty( $upgrade_action ) ) {
- return false;
- }
- $completed_upgrades = $this->get_completed_upgrades();
- return in_array( $upgrade_action, $completed_upgrades );
- }
- /**
- * Get's the array of completed upgrade actions
- *
- * @return array The array of completed upgrades
- */
- public function get_completed_upgrades() {
- $completed_upgrades = get_option( 'pum_completed_upgrades' );
- if ( false === $completed_upgrades ) {
- $completed_upgrades = array();
- }
- return $completed_upgrades;
- }
- public function step_up() {
- $step = $this->upgrade_args['step'];
- if ( $step >= $this->upgrade_args['steps'] ) {
- $this->upgrade_args['step'] = $this->upgrade_args['steps'];
- return false;
- }
- $this->upgrade_args['step'] ++;
- return true;
- }
- /**
- * Renders the upgrades screen.
- */
- public function upgrades_screen() { ?>
- <div class="wrap">
- <h2>
- <?php _e( 'Popup Maker - Upgrades', 'popup-maker' ); ?>
- <img src="<?php echo POPMAKE_URL . '/assets/images/admin/loading.gif'; ?>" id="pum-upgrade-loader"/>
- </h2>
- <style>
- #pum-upgrade-status {
- max-height: 300px;
- background: #fff;
- box-shadow: inset 0 1px 1px rgba(0, 0, 0, .5);
- overflow-y: scroll;
- text-overflow: ellipsis;
- padding: 0 1.5em;
- }
- </style>
- <p>
- <?php _e( 'The upgrade process has started, please be patient. This could take several minutes. You will be automatically redirected when the upgrade is finished.', 'popup-maker' ); ?>
- </p>
- <div id="pum-upgrade-status"></div>
- <script type="text/javascript">
- (function ($, document, undefined) {
- var $loader = $('#pum-upgrade-loader').hide(),
- $status_box = $('#pum-upgrade-status'),
- $timer,
- timer = 500;
- function update_status(message) {
- $('<p>')
- .html(message)
- .appendTo($status_box);
- $status_box.animate({
- scrollTop: $status_box.get(0).scrollHeight
- }, {
- duration: 'slow',
- queue: false
- });
- }
- function countdown(timer, callback) {
- var time_left = timer - 1000;
- if (time_left >= 0) {
- setTimeout(function () {
- $timer.text(time_left / 1000);
- countdown(time_left, callback);
- }, 1000);
- } else {
- callback();
- }
- }
- function next_step(args) {
- $loader.show();
- if (args === undefined) {
- args = {};
- }
- $.ajax({
- url: ajaxurl,
- data: $.extend({action: 'pum_trigger_upgrades'}, args),
- type: 'GET',
- dataType: 'json'
- })
- .done(function (response) {
- if (response.status !== undefined) {
- update_status(response.status);
- }
- if (response.complete !== undefined) {
- $loader.hide();
- } else if (response.next !== undefined && typeof response.next === 'object') {
- next_step(response.next);
- }
- if (response.redirect !== undefined) {
- if (response.countdown === undefined) {
- setTimeout(function () {
- document.location.href = response.redirect;
- }, timer);
- } else {
- $timer = $('#pum-countdown');
- countdown(response.countdown, function () {
- document.location.href = response.redirect;
- });
- }
- }
- })
- .fail(function () {
- update_status("<?php _e( 'Upgrade failed, please try again.', 'popup-maker' ); ?>");
- });
- }
- $(document).ready(function () {
- // Trigger upgrades on page load
- next_step(<?php echo wp_json_encode( $this->get_args() ); ?>);
- update_status('<?php printf( '<strong>%s</strong>', $this->get_upgrade( $this->get_arg( 'pum-upgrade' ) ) ); ?>');
- });
- }(jQuery, document));
- </script>
- </div>
- <?php
- }
- }
- PUM_Admin_Upgrades::instance();
|