Shortpixel improvement #254

Author: Md. Alimuzzaman Alim

lib/classes/compatibility/shortpixel.php

@@ -22,7 +22,8 @@ class ShortPixel extends ICompatibility {
             public function module_init($sm){
                 add_action( 'shortpixel_image_optimised', array($this, 'shortpixel_image_optimised') );
                 add_filter( 'get_attached_file', array($this, 'fix_missing_file'), 10, 2 );
-                add_action( 'admin_action_shortpixel_restore_backup', array($this, 'handleRestoreBackup'), 999 );
+                add_action( 'shortpixel_before_restore_image', array($this, 'sync_backup_file') );
+                add_action( 'shortpixel_after_restore_image', array($this, 'handleRestoreBackup') );
                 add_action( 'admin_enqueue_scripts', array( $this, 'shortPixelJS') );
                 // Sync from sync tab
                 add_action( 'sm:synced::image', array( $this, 'sync_backup_file'), 10, 2 );
@@ -141,7 +142,7 @@ public function shortpixel_image_optimised($id){
             /**
              * Sync backup image
              */
-            public function sync_backup_file($id, $metadata){
+            public function sync_backup_file($id, $metadata = null){
 
                 /* Get metadata in case if method is called directly. */
                 if( empty($metadata) ) {
@@ -211,8 +212,7 @@ public function returnSubDir($file){
             /**
              * Sync images after shortpixel restore them from backup.
              */
-            public function handleRestoreBackup(){
-                $attachmentID = intval($_GET['attachment_ID']);
+            public function handleRestoreBackup($attachmentID){
                 $metadata = wp_get_attachment_metadata( $attachmentID );
                 $this->add_media( $metadata, $attachmentID );
             }