mainwp-child/class/class-mainwp-custom-post-type.php

<?php
/**
 * MainWP Custom Post Type
 *
 * This file handles all custom post types.
 */
namespace MainWP\Child;

/**
 * Class MainWP_Custom_Post_Type
 *
 * @package MainWP\Child
 */
class MainWP_Custom_Post_Type {

	/**
	 * @static
	 * @var null Holds the Public static instance of MainWP_Custom_Post_Type.
	 */
	public static $instance = null;

	/** @var array Response array. */
	public static $information = array();

	/** @var string Plugin slug. */
	public $plugin_translate = 'mainwp-child';

	/**
	 * Create a public static instance of MainWP_Custom_Post_Type.
	 *
	 * @return MainWP_Custom_Post_Type|null
	 */
	public static function instance() {
		if ( null == self::$instance ) {
			self::$instance = new self();
		}

		return self::$instance;
	}

	/**
	 * Custom post type action.
	 */
	public function action() {

		/**
		 * Method mainwp_custom_post_type_handle_fatal_error()
		 *
		 * Custom post type fatal error handler.
		 */
		function mainwp_custom_post_type_handle_fatal_error() {
			$error = error_get_last();
			if ( isset( $error['type'] ) && E_ERROR === $error['type'] && isset( $error['message'] ) ) {
				$data = array( 'error' => 'MainWPChild fatal error : ' . $error['message'] . ' Line: ' . $error['line'] . ' File: ' . $error['file'] );
			} else {
				$data = self::$information;
			}

			if ( isset( $_REQUEST['json_result'] ) && $_REQUEST['json_result'] ) {
				$data = wp_json_encode( $data );
			} else {
				$data = serialize( $data ); // phpcs:ignore WordPress.PHP.DiscouragedPHPFunctions
			}
			die( '<mainwp>' . base64_encode( $data ) . '</mainwp>' ); // phpcs:ignore WordPress.PHP.DiscouragedPHPFunctions -- base64_encode function is used for http encode compatibility.
		}

		register_shutdown_function( 'MainWP\Child\mainwp_custom_post_type_handle_fatal_error' );

		/** @var $information @deprecated Unused local variable 'information'. The value of the variable is overwritten immediately. */
		$information = array();
		switch ( $_POST['action'] ) {
			case 'custom_post_type_import':
				$information = $this->import_custom_post();
				break;

			default:
				$information = array( 'error' => 'Unknown action' );

		}

		self::$information = $information;

		exit();
	}

	/**
	 * Import custom post type.
	 *
	 * @return array|string[] $return Response array, or error message on failure.
	 */
	private function import_custom_post() {

		add_filter( 'http_request_host_is_external', '__return_true' );

		if ( ! isset( $_POST['data'] ) || strlen( $_POST['data'] ) < 2 ) {
			return array( 'error' => __( 'Missing data', $this->plugin_translate ) );
		}

		$data = stripslashes( $_POST['data'] );

		$data = json_decode( $data, true );

		if ( empty( $data ) || ! is_array( $data ) || ! isset( $data['post'] ) ) {
			return array( 'error' => __( 'Cannot decode data', $this->plugin_translate ) );
		}
		$edit_id = ( isset( $_POST['post_id'] ) && ! empty( $_POST['post_id'] ) ) ? $_POST['post_id'] : 0;
		$return  = $this->insert_post( $data, $edit_id, $parent_id = 0 );
		if ( isset( $return['success'] ) && 1 == $return['success'] ) {
			if ( isset( $data['product_variation'] ) && is_array( $data['product_variation'] ) ) {
				foreach ( $data['product_variation'] as $product_variation ) {
					$return_variantion = $this->insert_post( $product_variation, 0, $return['post_id'] );
				}
			}
		}
		return $return;
	}

	/**
	 * Search for images inside post content and upload it to Child Site.
	 *
	 * @param string $post_content Post content to search.
	 * @param string $upload_dir Upload directory.
	 * @param bool   $check_image Check if file exists. Default: false.
	 * @return string|string[] Error message or post content string.
	 */
	private function search_images( $post_content, $upload_dir, $check_image = false ) {
		$foundMatches = preg_match_all( '/(<a[^>]+href=\"(.*?)\"[^>]*>)?(<img[^>\/]*src=\"((.*?)(png|gif|jpg|jpeg))\")/ix', $post_content, $matches, PREG_SET_ORDER );
		if ( $foundMatches > 0 ) {
			foreach ( $matches as $match ) {
				$hrefLink = $match[2];
				$imgUrl   = $match[4];

				if ( ! isset( $upload_dir['baseurl'] ) || ( 0 !== strripos( $imgUrl, $upload_dir['baseurl'] ) ) ) {
					continue;
				}

				if ( preg_match( '/-\d{3}x\d{3}\.[a-zA-Z0-9]{3,4}$/', $imgUrl, $imgMatches ) ) {
					$search         = $imgMatches[0];
					$replace        = '.' . $match[6];
					$originalImgUrl = str_replace( $search, $replace, $imgUrl );
				} else {
					$originalImgUrl = $imgUrl;
				}

				try {
					$downloadfile      = MainWP_Utility::upload_image( $originalImgUrl, array(), $check_image );
					$localUrl          = $downloadfile['url'];
					$linkToReplaceWith = dirname( $localUrl );
					if ( '' !== $hrefLink ) {
						$server     = get_option( 'mainwp_child_server' );
						$serverHost = wp_parse_url( $server, PHP_URL_HOST );
						if ( ! empty( $serverHost ) && strpos( $hrefLink, $serverHost ) !== false ) {
							$serverHref        = 'href="' . $serverHost;
							$replaceServerHref = 'href="' . wp_parse_url( $localUrl, PHP_URL_SCHEME ) . '://' . wp_parse_url( $localUrl, PHP_URL_HOST );
							$post_content      = str_replace( $serverHref, $replaceServerHref, $post_content );
						} elseif ( strpos( $hrefLink, 'http' ) !== false ) {
							$lnkToReplace = dirname( $hrefLink );
							if ( 'http:' !== $lnkToReplace && 'https:' !== $lnkToReplace ) {
								$post_content = str_replace( $lnkToReplace, $linkToReplaceWith, $post_content );
							}
						}
					}

					$lnkToReplace = dirname( $imgUrl );
					if ( 'http:' !== $lnkToReplace && 'https:' !== $lnkToReplace ) {
						$post_content = str_replace( $lnkToReplace, $linkToReplaceWith, $post_content );
					}
				} catch ( \Exception $e ) {
					// ok!
				}
			}
		}

		return $post_content;
	}

	/**
	 * Insert data into published post.
	 *
	 * @param string $data Data to insert.
	 * @param int    $edit_id Post ID to edit.
	 * @param int    $parent_id Post parent ID.
	 *
	 * @return array|bool|string[] Response array, true|false, Error message.
	 */
	private function insert_post( $data, $edit_id, $parent_id = 0 ) {
		$data_insert                = array();
		$data_post                  = $data['post'];
		$data_insert['post_author'] = get_current_user_id();

		$data_keys = array(
			'post_date',
			'post_date_gmt',
			'post_content',
			'post_title',
			'post_excerpt',
			'post_status',
			'comment_status',
			'ping_status',
			'post_password',
			'post_name',
			'to_ping',
			'pinged',
			'post_modified',
			'post_modified_gmt',
			'post_content_filtered',
			'menu_order',
			'post_type',
		);

		foreach ( $data_keys as $key ) {
			if ( ! isset( $data_post[ $key ] ) ) {
				return array( 'error' => _( 'Missing', $this->plugin_translate ) . ' ' . $key . ' ' . __( 'inside post data', $this->plugin_translate ) );
			}

			$data_insert[ $key ] = $data_post[ $key ];
		}

		if ( ! in_array( $data_insert['post_type'], get_post_types( array( '_builtin' => false ) ) ) ) {
			return array( 'error' => __( 'Please install', $this->plugin_translate ) . ' ' . $data_insert['post_type'] . ' ' . __( 'on child and try again', $this->plugin_translate ) );
		}

		$is_woocomerce = false;
		if ( ( 'product' == $data_insert['post_type'] || 'product_variation' == $data_insert['post_type'] ) && function_exists( 'wc_product_has_unique_sku' ) ) {
			$is_woocomerce = true;
		}

		$check_image_existed = false;

		if ( ! empty( $edit_id ) ) {
			$old_post_id = (int) $edit_id;
			$old_post    = get_post( $old_post_id, ARRAY_A );
			if ( is_null( $old_post ) ) {
				return array(
					'delete_connection' => 1,
					'error'             => __( 'Cannot get old post. Probably is deleted now. Please try again for create new post', $this->plugin_translate ),
				);
			}

			if ( get_post_status( $old_post_id ) == 'trash' ) {
				return array( 'error' => __( 'This post is inside trash on child website. Please try publish it manually and try again.', $this->plugin_translate ) );
			}
			$check_image_existed = true;
			$data_insert['ID']   = $old_post_id;

			// Remove all previous post meta.
			// Get all unique meta_key.
			foreach ( get_post_meta( $old_post_id ) as $temp_meta_key => $temp_meta_val ) {
				if ( ! delete_post_meta( $old_post_id, $temp_meta_key ) ) {
					return array( 'error' => __( 'Cannot delete old post meta values', $this->plugin_translate ) );
				}
			}

			// Remove all previous taxonomy.
			wp_delete_object_term_relationships( $old_post_id, get_object_taxonomies( $data_insert['post_type'] ) );
		}

		$data_insert['post_content'] = $this->search_images( $data_insert['post_content'], $data['extras']['upload_dir'], $check_image_existed );

		if ( ! empty( $parent_id ) ) {
			$data_insert['post_parent'] = $parent_id;
		}
		$post_id = wp_insert_post( $data_insert, true );
		if ( is_wp_error( $post_id ) ) {
			return array( 'error' => __( 'Error when insert new post:', $this->plugin_translate ) . ' ' . $post_id->get_error_message() );
		}

		// Insert post meta.
		if ( ! empty( $data['postmeta'] ) && is_array( $data['postmeta'] ) ) {
			$ret = $this->insert_postmeta( $post_id, $data, $check_image_existed, $is_woocomerce );
			if ( true !== $ret ) {
				return $ret;
			}
		}

		$ret = $this->insert_custom_data( $post_id, $data );

		if ( true !== $ret ) {
			return $ret;
		}

		return array(
			'success' => 1,
			'post_id' => $post_id,
		);
	}

	/**
	 * Insert custom post data.
	 *
	 * @param int    $post_id Post ID to update.
	 * @param string $data Custom data to add.
	 * @return array|bool|string[] Response array, true|false, Error message.
	 */
	private function insert_custom_data( $post_id, $data ) {

		// MainWP Categories.
		if ( ! empty( $data['categories'] ) && is_array( $data['categories'] ) ) {
			// Contains wp_create_categories.
			include_once ABSPATH . 'wp-admin/includes/taxonomy.php';
			$categories = $data['categories'];
			if ( '0' == $data['post_only_existing'] ) {
				$post_category = wp_create_categories( $categories, $post_id );
			} else {
				$cat_ids = array();
				foreach ( $categories as $cat ) {
					$id = category_exists( $cat );
					if ( $id ) {
						$cat_ids[] = $id;
					}
				}
				if ( count( $cat_ids ) > 0 ) {
					wp_set_post_categories( $post_id, $cat_ids );
				}
			}
		}

		// Insert post terms except categories.
		if ( ! empty( $data['terms'] ) && is_array( $data['terms'] ) ) {
			foreach ( $data['terms'] as $key ) {
				if ( ! taxonomy_exists( $key['taxonomy'] ) ) {
					return array( 'error' => __( 'Missing taxonomy', $this->plugin_translate ) . ' `' . esc_html( $key['taxonomy'] ) . '`' );
				}

				$term = wp_insert_term(
					$key['name'],
					$key['taxonomy'],
					array(
						'description' => $key['description'],
						'slug'        => $key['slug'],
					)
				);

				$term_taxonomy_id = 0;

				if ( is_wp_error( $term ) ) {
					if ( isset( $term->error_data['term_exists'] ) ) {
						$term_taxonomy_id = (int) $term->error_data['term_exists'];
					}
				} else {
					if ( isset( $term['term_taxonomy_id'] ) ) {
						$term_taxonomy_id = (int) $term['term_taxonomy_id'];
					}
				}

				if ( $term_taxonomy_id > 0 ) {
					$term_taxonomy_ids = wp_set_object_terms( $post_id, $term_taxonomy_id, $key['taxonomy'], true );
					if ( is_wp_error( $term_taxonomy_ids ) ) {
						return array( 'error' => __( 'Error when adding taxonomy to post', $this->plugin_translate ) );
					}
				}
			}
		}
		return true;
	}

	/**
	 * Insert post meta.
	 *
	 * @param int    $post_id Post ID to update.
	 * @param string $data Meta datat add.
	 * @param bool   $check_image_existed Whether or not to check if image exists. true|false.
	 * @param bool   $is_woocomerce Whether or not the post is a woocommerce product. true|false.
	 *
	 * @return array|bool|string[] Response array, true|false, Error message.
	 */
	private function insert_postmeta( $post_id, $data, $check_image_existed, $is_woocomerce ) {
		foreach ( $data['postmeta'] as $key ) {
			if ( isset( $key['meta_key'] ) && isset( $key['meta_value'] ) ) {
				$meta_value = $key['meta_value'];
				if ( $is_woocomerce ) {
					if ( '_sku' == $key['meta_key'] ) {
						if ( ! wc_product_has_unique_sku( $post_id, $meta_value ) ) {
							return array( 'error' => __( 'Product SKU must be unique', $this->plugin_translate ) );
						}
					}
					if ( '_product_image_gallery' == $key['meta_key'] ) {
						if ( isset( $data['extras']['woocommerce']['product_images'] ) ) {
							$ret = $this->upload_postmeta_image( $data['extras']['woocommerce']['product_images'], $meta_value, $check_image_existed );
							if ( true !== $ret ) {
								return $ret;
							}
						} else {
							continue;
						}
					}
				}

				if ( '_thumbnail_id' == $key['meta_key'] ) {
					if ( isset( $data['extras']['featured_image'] ) ) {
						try {
							$upload_featured_image = MainWP_Utility::upload_image( $data['extras']['featured_image'], array(), $check_image_existed );

							if ( null !== $upload_featured_image ) {
								$meta_value = $upload_featured_image['id'];
							} else {
								return array( 'error' => __( 'Cannot add featured image', $this->plugin_translate ) );
							}
						} catch ( \Exception $e ) {
							continue;
						}
					} else {
						continue;
					}
				}

				$meta_value = maybe_unserialize( $meta_value );
				if ( add_post_meta( $post_id, $key['meta_key'], $meta_value ) === false ) {
					return array( 'error' => __( 'Error when adding post meta', $this->plugin_translate ) . ' `' . esc_html( $key['meta_key'] ) . '`' );
				}
			}
		}
		return true;
	}

	/**
	 * Upload post meta image.
	 *
	 * @param array               $product_images Woocomerce product images.
	 * @param $meta_value
	 * @param $check_image_existed
	 * @return array|bool Error message array or TRUE on success.
	 */
	private function upload_postmeta_image( $product_images, &$meta_value, $check_image_existed ) {
		$product_image_gallery = array();
		foreach ( $product_images as $product_image ) {
			try {
				$upload_featured_image = MainWP_Utility::upload_image( $product_image, array(), $check_image_existed );

				if ( null !== $upload_featured_image ) {
					$product_image_gallery[] = $upload_featured_image['id'];
				} else {
					return array( 'error' => __( 'Cannot add product image', $this->plugin_translate ) );
				}
			} catch ( \Exception $e ) {
				continue;
			}
		}
		$meta_value = implode( $product_image_gallery, ',' );
		return true;
	}
}