first commit

wp-content/plugins/brizy/admin/post/abstract-monitor.php

@@ -0,0 +1,70 @@
+<?php
+
+
+abstract class Brizy_Admin_Post_AbstractMonitor {
+
+	/**
+	 * @var string
+	 */
+	protected $postType;
+
+	/**
+	 * @var string[]
+	 */
+	protected $postMetaKeys;
+
+	/**
+	 * @param $postId
+	 * @param $postType
+	 *
+	 * @return mixed
+	 */
+	abstract public function shouldStoreMetaRevision( $postId, $postType );
+
+	/**
+	 * Brizy_Admin_Post_AbstractMonitor constructor.
+	 *
+	 * @param $postType
+	 * @param array $postMetaKeys
+	 */
+	public function __construct( $postType, $postMetaKeys = array() ) {
+		$this->postType = $postType;
+		$this->postMetaKeys = $postMetaKeys;
+	}
+
+	/**
+	 * @return mixed
+	 */
+	public function getPostType() {
+		return $this->postType;
+	}
+
+	/**
+	 * @param $postType
+	 *
+	 * @return $this
+	 */
+	public function setPostType( $postType ) {
+		$this->postType = $postType;
+
+		return $this;
+	}
+
+	/**
+	 * @return string[]
+	 */
+	public function getPostMetaKeys() {
+		return $this->postMetaKeys;
+	}
+
+	/**
+	 * @param string[] $postMetaKeys
+	 *
+	 * @return Brizy_Admin_Post_AbstractMonitor
+	 */
+	public function setPostMetaKeys( $postMetaKeys ) {
+		$this->postMetaKeys = $postMetaKeys;
+
+		return $this;
+	}
+}

wp-content/plugins/brizy/admin/post/brizy-posts-monitor.php

@@ -0,0 +1,38 @@
+<?php
+
+class Brizy_Admin_Post_BrizyPostsMonitor extends Brizy_Admin_Post_AbstractMonitor {
+
+	/**
+	 * Brizy_Admin_Post_BrizyPostsMonitor constructor.
+	 */
+	public function __construct() {
+		parent::__construct( null, array(
+			'brizy',
+			'brizy-post-compiler-version',
+			'brizy-post-editor-version',
+			'brizy_post_uid',
+			'brizy-rules',
+			'brizy_data_version'
+		) );
+	}
+
+	/**
+	 * @param $postId
+	 * @param $postType
+	 *
+	 * @return bool|mixed
+	 */
+	public function shouldStoreMetaRevision( $postId, $postType ) {
+		return $this->isUsingBrizy( $postId );
+	}
+
+	/**
+	 * @param $postId
+	 *
+	 * @return bool
+	 */
+	protected function isUsingBrizy( $postId ) {
+		$get_post_meta = get_post_meta( $postId, Brizy_Editor_Storage_Post::META_KEY, true );
+		return ! ! $get_post_meta;
+	}
+}

wp-content/plugins/brizy/admin/post/post-type-monitor.php

@@ -0,0 +1,8 @@
+<?php
+
+class Brizy_Admin_Post_PostTypeMonitor extends Brizy_Admin_Post_AbstractMonitor {
+
+	public function shouldStoreMetaRevision( $postId, $postType ) {
+		return $this->postType == $postType;
+	}
+}

wp-content/plugins/brizy/admin/post/project-post-monitor.php

@@ -0,0 +1,22 @@
+<?php
+
+
+class Brizy_Admin_Post_ProjectPostMonitor extends Brizy_Admin_Post_AbstractMonitor {
+
+	/**
+	 * Brizy_Admin_Post_ProjectPostMonitor constructor.
+	 */
+	public function __construct() {
+		parent::__construct( 'brizy-project', array( 'brizy-project','brizy_data_version' ) );
+	}
+
+	/**
+	 * @param $postId
+	 * @param $postType
+	 *
+	 * @return bool|mixed
+	 */
+	public function shouldStoreMetaRevision( $postId, $postType ) {
+		return $this->postType == $postType;
+	}
+}

wp-content/plugins/brizy/admin/post/revision-manager.php

@@ -0,0 +1,263 @@
+<?php
+
+
+/**
+ * Class Brizy_Admin_Post_RevisionManager
+ */
+class Brizy_Admin_Post_RevisionManager {
+
+	/**
+	 * @var Brizy_Admin_Post_AbstractMonitor[]
+	 */
+	protected $monitors = array();
+
+
+	/**
+	 * Brizy_Admin_Post_RevisionManager constructor.
+	 */
+	public function __construct() {
+
+		add_action( 'wp_restore_post_revision', array( $this, 'restorePostRevisionMeta' ), 11, 2 );
+
+		if ( ! defined( 'WP_POST_REVISIONS' ) || ( defined( 'WP_POST_REVISIONS' ) && WP_POST_REVISIONS !== false ) ) {
+			add_action( 'save_post', array( $this, 'savePost' ), 11, 2 );
+		}
+
+
+	}
+
+	/**
+	 * @return Brizy_Admin_Post_AbstractMonitor[]
+	 */
+	public function getMonitors() {
+		return $this->monitors;
+	}
+
+	/**
+	 * @param Brizy_Admin_Post_AbstractMonitor[] $monitors
+	 *
+	 * @return Brizy_Admin_Post_RevisionManager
+	 */
+	public function setMonitors( $monitors ) {
+		$this->monitors = $monitors;
+
+		return $this;
+	}
+
+	/**
+	 * @param Brizy_Admin_Post_AbstractMonitor $monitor
+	 *
+	 * @return Brizy_Admin_Post_RevisionManager
+	 */
+	public function addMonitor( $monitor ) {
+		$this->monitors[] = $monitor;
+
+		return $this;
+	}
+
+	/**
+	 * @param $postId
+	 * @param $type
+	 *
+	 * @return bool
+	 */
+	public function getMatchingMonitor( $postId, $type ) {
+		foreach ( $this->monitors as $monitor ) {
+			if ( $monitor->shouldStoreMetaRevision( $postId, $type ) ) {
+				return $monitor;
+			}
+		}
+
+		return false;
+	}
+
+	/**
+	 * @param $postId
+	 * @param $post
+	 */
+	public function savePost( $postId, $post, $update = false ) {
+		$postParentId = wp_is_post_revision( $postId );
+		$postType     = get_post_type( $post->post_parent );
+
+		// ignore all posts that are not watched
+		if ( $postParentId && $monitor = $this->getMatchingMonitor( $postParentId, $postType ) ) {
+			// copy meta data only for existing posts
+			$this->copyMetaDataToRevision( $postParentId, $postId, $monitor );
+		}
+	}
+
+	/**
+	 * @param int $postId
+	 * @param int $revisionId
+	 */
+	public function restorePostRevisionMeta( $postId, $revisionId ) {
+		$postType = get_post_type( $postId );
+
+		// ignore all posts that are not watched
+		if ( $monitor = $this->getMatchingMonitor( $postId, $postType ) ) {
+			// copy meta data only for existing posts
+			$this->restoreRevisionMetaDataToPost( $postId, $revisionId, $monitor );
+		}
+
+	}
+
+	/**
+	 * This method will add all meta data entries from $post to $revision post
+	 *
+	 * @param int $post
+	 * @param int $revision
+	 * @param Brizy_Admin_Post_AbstractMonitor $monitor
+	 */
+	private function copyMetaDataToRevision( $post, $revision, $monitor ) {
+
+		global $wpdb;
+		$tablePostMeta    = $wpdb->postmeta;
+		$meta_key_count   = count( $monitor->getPostMetaKeys() );
+		$meta_keys_params = rtrim( str_repeat( '%s,', $meta_key_count ), ',' );
+		$params           = array( (int) $revision, (int) $post );
+
+		$this->cleanMetaData( $post, $revision, $monitor );
+
+		$query = "INSERT INTO {$tablePostMeta} (post_Id, meta_key, meta_value) 
+									SELECT %d, meta_key, meta_value 
+									FROM {$tablePostMeta} 
+									WHERE post_id=%d";
+
+		if ( $meta_key_count > 0 ) {
+			$query  .= " and meta_key IN ({$meta_keys_params})";
+			$params = array_merge( $params, $monitor->getPostMetaKeys() );
+		} else {
+			$query .= " and meta_key NOT IN ('_edit_last','_edit_lock','_thumbnail_id','_wp_attached_file','_wp_attachment_metadata')";
+		}
+
+		$wpdb->query( $wpdb->prepare( $query, $params ) );
+
+		if ( $meta_key_count > 0 && false !== array_search( 'brizy', $monitor->getPostMetaKeys() ) ) {
+
+			$storage = Brizy_Editor_Storage_Post::instance( $revision );
+			$data    = $storage->get( Brizy_Editor_Post::BRIZY_POST );
+
+			$data['compiled_html']      = '';
+			$data['compiled_html_body'] = '';
+			$data['compiled_html_head'] = '';
+
+			$storage->set( Brizy_Editor_Post::BRIZY_POST, $data );
+
+			$wpdb->update(
+				$wpdb->posts,
+				[ 'post_content' => '<div class="brz-root__container"><!-- version:' . time() . ' --></div>' ],
+				[ 'ID' => $revision ],
+                [ '%s' ]
+			);
+		}
+	}
+
+	private function cleanMetaData( $post, $revision, $monitor ) {
+		global $wpdb;
+		$params           = array( (int) $revision );
+		$meta_key_count   = count( $monitor->getPostMetaKeys() );
+		$meta_keys_params = rtrim( str_repeat( '%s,', $meta_key_count ), ',' );
+		$tablePostMeta    = "{$wpdb->prefix}postmeta";
+		$query            = "DELETE FROM {$tablePostMeta} WHERE post_id=%d and meta_key IN ({$meta_keys_params})";
+		$params           = array_merge( $params, $monitor->getPostMetaKeys() );
+		$wpdb->query( $wpdb->prepare( $query, $params ) );
+	}
+
+	/**
+	 * This method will add all meta data entries from $revision to $post
+	 *
+	 * @param int $post
+	 * @param int $revision
+	 * @param Brizy_Admin_Post_AbstractMonitor $monitor
+	 */
+	private function restoreRevisionMetaDataToPost( $post, $revision, $monitor ) {
+		global $wpdb;
+		try {
+
+			$tablePostMeta = "{$wpdb->prefix}postmeta";
+
+			$wpdb->query( "START TRANSACTION" );
+
+			$revisionMetaValues = $this->getRevisionMetaValues( $revision, $monitor );
+
+			if ( $revisionMetaValues === false ) {
+				throw new Exception();
+			}
+
+			foreach ( $revisionMetaValues as $meta ) {
+
+				$wpdb->query( $wpdb->prepare( "DELETE FROM {$wpdb->prefix}postmeta WHERE post_id=%d and meta_key=%s", $post, $meta->meta_key ) );
+				$res = $wpdb->query( $wpdb->prepare( "INSERT INTO {$tablePostMeta} (post_id,meta_key,meta_value) VALUES (%d,%s,%s)", $post, $meta->meta_key, $meta->meta_value ) );
+
+				if ( $res === false ) {
+					throw new Exception();
+				}
+			}
+
+			Brizy_Editor_Post::cleanClassCache();
+			$brizyPost = Brizy_Editor_Post::get( $post );
+
+			if ( $brizyPost->uses_editor() ) {
+				$brizyPost->set_needs_compile( true );
+				$brizyPost->saveStorage();
+			}
+
+			$wpdb->query( 'COMMIT' );
+		} catch ( Exception $e ) {
+			$wpdb->query( 'ROLLBACK' );
+		}
+
+	}
+
+	/**
+	 * @param int $post
+	 * @param Brizy_Admin_Post_AbstractMonitor $monitor
+	 */
+	private function cleanMetakeys( $post, $monitor ) {
+		global $wpdb;
+
+		$meta_key_count   = count( $monitor->getPostMetaKeys() );
+		$params           = array( (int) $post );
+		$meta_keys_params = rtrim( str_repeat( '%s,', $meta_key_count ), ',' );
+
+		$query = "DELETE FROM {$wpdb->prefix}postmeta WHERE post_id=%d";
+
+		if ( $meta_key_count > 0 ) {
+			$query  .= " and meta_key IN ({$meta_keys_params})";
+			$params = array_merge( $params, $monitor->getPostMetaKeys() );
+		} else {
+			$query .= " and meta_key NOT IN ('_edit_last','_edit_lock','_thumbnail_id','_wp_attached_file','_wp_attachment_metadata')";
+		}
+
+		$wpdb->query( $wpdb->prepare( $query, $params ) );
+	}
+
+	/**
+	 * @param $revision
+	 * @param $monitor
+	 *
+	 * @return array|null|object
+	 */
+	private function getRevisionMetaValues( $revision, $monitor ) {
+		global $wpdb;
+
+		$meta_key_count   = count( $monitor->getPostMetaKeys() );
+		$params           = array( (int) $revision );
+		$meta_keys_params = rtrim( str_repeat( '%s,', $meta_key_count ), ',' );
+
+		$query = "SELECT * FROM {$wpdb->prefix}postmeta WHERE post_id=%d";
+
+		if ( $meta_key_count > 0 ) {
+			$query  .= " and meta_key IN ({$meta_keys_params})";
+			$params = array_merge( $params, $monitor->getPostMetaKeys() );
+		} else {
+			$query .= " and meta_key NOT IN ('_edit_last','_edit_lock','_thumbnail_id','_wp_attached_file','_wp_attachment_metadata')";
+		}
+
+		$query = $wpdb->prepare( $query, $params );
+
+		return $wpdb->get_results( $query );
+	}
+
+
+}