product = $product_object;
}
if ( $this->product->can_rollback() ) {
$this->show_link();
$this->add_hooks();
}
}
/**
* Add js scripts for themes rollback.
*/
public function add_footer() {
$screen = get_current_screen();
if ( ! isset( $screen->parent_file ) ) {
return;
}
if ( $screen->parent_file !== 'themes.php' ) {
return;
}
if ( $this->product->get_type() === 'plugin' ) {
return;
}
$version = $this->product->get_rollback();
?>
product->get_key() . '_rollback', array( $this, 'start_rollback' ) );
add_action( 'admin_footer', array( $this, 'add_footer' ) );
}
/**
* If product can be rolled back, show the link to rollback.
*/
private function show_link() {
add_filter(
'plugin_action_links_' . plugin_basename( $this->product->get_basefile() ), array(
$this,
'add_rollback_link',
)
);
}
/**
* Show the rollback links in the plugin page.
*
* @return array The links.
*/
public function add_rollback_link( $links ) {
$version = $this->product->get_rollback();
$links[] = '' . sprintf( apply_filters( $this->product->get_key() . '_rollback_label', 'Rollback to v%s' ), $version['version'] ) . '';
return $links;
}
/**
* Start the rollback operation.
*/
public function start_rollback() {
if ( ! isset( $_GET['_wpnonce'] ) || ! wp_verify_nonce( $_GET['_wpnonce'], $this->product->get_key() . '_rollback' ) ) {
wp_nonce_ays( '' );
}
if ( $this->product->get_type() === 'plugin' ) {
$this->start_rollback_plugin();
} elseif ( $this->product->get_type() === 'theme' ) {
$this->start_rollback_theme();
}
}
/**
* Alter links and remove duplicate customize message.
*
* @param array $links Array of old links.
*
* @return mixed Array of links.
*/
public function alter_links_theme_upgrade( $links ) {
if ( isset( $links['preview'] ) ) {
$links['preview'] = str_replace( 'Customize', '', $links['preview'] );
}
return $links;
}
/**
* Start the rollback operation for the theme.
*/
private function start_rollback_theme() {
add_filter( 'update_theme_complete_actions', array( $this, 'alter_links_theme_upgrade' ) );
$rollback = $this->product->get_rollback();
$transient = get_site_transient( 'update_themes' );
$folder = $this->product->get_slug();
$version = $rollback['version'];
$temp_array = array(
'new_version' => $version,
'package' => $rollback['url'],
);
$transient->response[ $folder . '/style.css' ] = $temp_array;
set_site_transient( 'update_themes', $transient );
$transient = get_transient( $this->product->get_key() . '_warning_rollback' );
if ( false === $transient ) {
set_transient( $this->product->get_key() . '_warning_rollback', 'in progress', 30 );
require_once( ABSPATH . 'wp-admin/includes/class-wp-upgrader.php' );
$title = sprintf( apply_filters( $this->product->get_key() . '_rollback_message', 'Rolling back %s to v%s' ), $this->product->get_name(), $version );
$theme = $folder . '/style.css';
$nonce = 'upgrade-theme_' . $theme;
$url = 'update.php?action=upgrade-theme&theme=' . urlencode( $theme );
$upgrader = new Theme_Upgrader( new Theme_Upgrader_Skin( compact( 'title', 'nonce', 'url', 'theme' ) ) );
$upgrader->upgrade( $theme );
delete_transient( $this->product->get_key() . '_warning_rollback' );
wp_die(
'', $title, array(
'response' => 200,
)
);
}
}
/**
* Start the rollback operation for the plugin.
*/
private function start_rollback_plugin() {
$rollback = $this->product->get_rollback();
$plugin_transient = get_site_transient( 'update_plugins' );
$plugin_folder = $this->product->get_slug();
$plugin_file = $this->product->get_file();
$version = $rollback['version'];
$temp_array = array(
'slug' => $plugin_folder,
'new_version' => $version,
'package' => $rollback['url'],
);
$temp_object = (object) $temp_array;
$plugin_transient->response[ $plugin_folder . '/' . $plugin_file ] = $temp_object;
set_site_transient( 'update_plugins', $plugin_transient );
$transient = get_transient( $this->product->get_key() . '_warning_rollback' );
if ( false === $transient ) {
set_transient( $this->product->get_key() . '_warning_rollback', 'in progress', 30 );
require_once( ABSPATH . 'wp-admin/includes/class-wp-upgrader.php' );
$title = sprintf( apply_filters( $this->product->get_key() . '_rollback_message', 'Rolling back %s to v%s' ), $this->product->get_name(), $version );
$plugin = $plugin_folder . '/' . $plugin_file;
$nonce = 'upgrade-plugin_' . $plugin;
$url = 'update.php?action=upgrade-plugin&plugin=' . urlencode( $plugin );
$upgrader_skin = new Plugin_Upgrader_Skin( compact( 'title', 'nonce', 'url', 'plugin' ) );
$upgrader = new Plugin_Upgrader( $upgrader_skin );
$upgrader->upgrade( $plugin );
delete_transient( $this->product->get_key() . '_warning_rollback' );
wp_die(
'', $title, array(
'response' => 200,
)
);
}
}
}
endif;