fixes for revison wipe

Author: selul

classes/Visualizer/Gutenberg/Block.php

@@ -520,7 +520,9 @@ public function update_chart_data( $data ) {
 		}
 
 		if ( $data['id'] && ! is_wp_error( $data['id'] ) ) {
-
+			if ( get_post_type( $data['id'] ) !== Visualizer_Plugin::CPT_VISUALIZER ) {
+				return new WP_Error( 'invalid_post_type', 'Invalid post type.' );
+			}
 			$chart_type = sanitize_text_field( $data['visualizer-chart-type'] );
 			$source_type = sanitize_text_field( $data['visualizer-source'] );
 

classes/Visualizer/Module.php

@@ -402,7 +402,9 @@ private function _getHTML( $rows ) {
 	 */
 	public final function undoRevisions( $chart_id, $restore = false ) {
 		do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, sprintf( 'undoRevisions for %d with%s restore', $chart_id, ( $restore ? '' : 'out' ) ), 'debug', __FILE__, __LINE__ );
-
+		if ( get_post_type( $chart_id ) !== Visualizer_Plugin::CPT_VISUALIZER ) {
+			return false;
+		}
 		$revisions = wp_get_post_revisions( $chart_id, array( 'order' => 'ASC' ) );
 		if ( count( $revisions ) > 1 ) {
 			$revision_ids = array_keys( $revisions );
@@ -416,6 +418,7 @@ public final function undoRevisions( $chart_id, $restore = false ) {
 				// restore to the oldest one i.e. the first one.
 				wp_restore_post_revision( array_shift( $revision_ids ) );
 			}
+			remove_filter( 'wp_revisions_to_keep', '__return_false' );
 
 			// delete all revisions.
 			foreach ( $revision_ids as $id ) {
@@ -431,8 +434,11 @@ public final function undoRevisions( $chart_id, $restore = false ) {
 	 * If existing revisions exist for the chart, restore the earliest version and then create a new revision to initiate editing.
 	 */
 	public final function handleExistingRevisions( $chart_id, $chart ) {
-		do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, sprintf( 'handleExistingRevisions for %d', $chart_id ), 'debug', __FILE__, __LINE__ );
 
+		do_action( 'themeisle_log_event', Visualizer_Plugin::NAME, sprintf( 'handleExistingRevisions for %d', $chart_id ), 'debug', __FILE__, __LINE__ );
+		if ( get_post_type( $chart_id ) !== Visualizer_Plugin::CPT_VISUALIZER ) {
+			return $chart_id;
+		}
 		// undo revisions.
 		$revisions_found    = $this->undoRevisions( $chart_id, true );