Added html generator, fix multiple entry generation duplicating content

Author: aprokopenko

framework/Objects/Post_Type.php

@@ -342,6 +342,6 @@ public function views( $template ) {
 	 * Init FakerPress.
 	 */
 	public function init_faker() {
-		new FakerPress( $this::$ID, $this );
+		new FakerPress( $this );
 	}
 }

framework/Supports/FakerContent.php

@@ -9,7 +9,9 @@
 
 /**
  * Class FakerContent
- * Fakerpress plugin extension which allows to generated faker content for custom fields.
+ * FakerPress plugin extension which allows to generated faker content for custom fields.
+ *
+ * @method static FakerContent instance() Singleton design pattern instance creation.
  */
 class FakerContent {
 	use Singleton;
@@ -21,11 +23,28 @@ class FakerContent {
 	 */
 	public $faker;
 
+	/**
+	 * Faker Press custom providers
+	 *
+	 * @var array
+	 */
+	public $providers = array(
+		'\Faker\Provider\Lorem',
+		'\Faker\Provider\DateTime',
+		'\Faker\Provider\HTML',
+	);
+
 	/**
 	 * FakerContent construct
 	 */
 	protected function __construct() {
 		$this->faker = Factory::create();
+
+		// register custom providers.
+		foreach ( $this->providers as $provider_class ) {
+			$provider = new $provider_class( $this->faker );
+			$this->faker->addProvider( $provider );
+		}
 	}
 
 	/**
@@ -79,6 +98,26 @@ public function text( $max_chars = 200 ) {
 		return TextBase::text( $max_chars );
 	}
 
+
+	/**
+	 * Get fake html.
+	 *
+	 * @param array  $content_size Min/max number of paragraphs to generate.
+	 * @param string $html_tags Allowed html tags.
+	 *
+	 * @return string
+	 */
+	public function html_text(
+		$content_size = [ 5, 10 ],
+		$html_tags = 'p,p,p,p,p,p,p,h2,h3,h4,h5,h6,ul,ol,p,blockquote,img,hr,b,i,a' // increase "p" possibility.
+	) {
+		$tags = $this->faker->html_elements([
+			'qty'      => $content_size,
+			'elements' => explode( ',', $html_tags ),
+		]);
+		return implode( "\n", $tags );
+	}
+
 	/**
 	 * Get fake words.
 	 *
@@ -176,14 +215,23 @@ public function person() {
 	}
 
 	/**
-	 * Get fake name.
+	 * Get fake company name.
 	 *
 	 * @return string
 	 */
 	public function company() {
 		return $this->faker->company;
 	}
 
+	/**
+	 * Get fake job title.
+	 *
+	 * @return string
+	 */
+	public function job_title() {
+		return $this->faker->jobTitle;
+	}
+
 	/**
 	 * Get fake email.
 	 *

framework/Supports/FakerPress.php

@@ -1,6 +1,9 @@
 <?php
 namespace JustCoded\WP\Framework\Supports;
 
+use FakerPress\Module\Post;
+use JustCoded\WP\Framework\Objects\Post_Type;
+
 /**
  * Class FakerPress
  * Fakerpress plugin extension which allows to generated faker content for custom fields.
@@ -12,7 +15,14 @@ class FakerPress {
 	 *
 	 * @var string $ID
 	 */
-	public $ID;
+	protected $ID;
+
+	/**
+	 * Post type definition object.
+	 *
+	 * @var Post_Type
+	 */
+	public $post_type;
 
 	/**
 	 * Post Type faker data.
@@ -24,31 +34,18 @@ class FakerPress {
 	/**
 	 * FakerContent construct.
 	 *
-	 * @param string $id        Post Type ID.
-	 * @param object $post_type Object.
+	 * @param Post_Type $post_type Object.
 	 */
-	public function __construct( $id, $post_type ) {
+	public function __construct( $post_type ) {
 		if ( $this->do_fakerpress() ) {
-			$this->ID   = $id;
-			$this->data = $post_type->faker();
+			$this->ID   = $post_type::$ID;
+			$this->post_type = $post_type;
+
 			add_action( 'wp_insert_post_data', array( $this, 'pre_insert_post' ), 20, 2 );
 			add_action( 'wp_insert_post', array( $this, 'insert_post' ), 10, 3 );
 		}
 	}
 
-	/**
-	 * Fires once a post has been saved.
-	 *
-	 * @param int      $post_id Post ID.
-	 * @param \WP_Post $post    Post object.
-	 * @param bool     $update  Whether this is an existing post being updated or not.
-	 */
-	public function insert_post( $post_id, $post, $update ) {
-		if ( $this->ID === $post->post_type ) {
-			$this->do_save( $post_id, $this->data );
-		}
-	}
-
 	/**
 	 * Pre-save post content.
 	 *
@@ -58,19 +55,35 @@ public function insert_post( $post_id, $post, $update ) {
 	 * @return array
 	 */
 	public function pre_insert_post( $data, $postarr ) {
+		$this->data = $this->post_type->faker();
+
 		if ( $this->ID === $data['post_type'] ) {
-			$faker_data = $this->data;
-			if ( isset( $faker_data['post_title'] ) ) {
-				$data['post_title'] = $faker_data['post_title'];
+			if ( isset( $this->data['post_title'] ) ) {
+				$data['post_title'] = $this->data['post_title'];
+				unset( $this->data['post_title'] );
 			}
-			if ( isset( $faker_data['post_content'] ) ) {
-				$data['post_content'] = $faker_data['post_content'];
+			if ( isset( $this->data['post_content'] ) ) {
+				$data['post_content'] = $this->data['post_content'];
+				unset( $this->data['post_content'] );
 			}
 		}
 
 		return $data;
 	}
 
+	/**
+	 * Fires once a post has been saved.
+	 *
+	 * @param int      $post_id Post ID.
+	 * @param \WP_Post $post    Post object.
+	 * @param bool     $update  Whether this is an existing post being updated or not.
+	 */
+	public function insert_post( $post_id, $post, $update ) {
+		if ( $this->ID === $post->post_type ) {
+			$this->update_meta( $post_id, $this->data );
+		}
+	}
+
 	/**
 	 * Saved faker content.
 	 *
@@ -79,11 +92,12 @@ public function pre_insert_post( $data, $postarr ) {
 	 *
 	 * @return bool
 	 */
-	public function do_save( $post_id, $data ) {
+	public function update_meta( $post_id, $data ) {
 		if ( isset( $data['post_featured_image'] ) ) {
 			set_post_thumbnail( $post_id, $data['post_featured_image'] );
+			unset( $data['post_featured_image'] );
 		}
-		unset( $data['post_featured_image'], $data['post_title'], $data['post_content'] );
+
 		foreach ( $data as $meta_key => $meta_value ) {
 			if ( class_exists( 'acf' ) ) {
 				update_field( $meta_key, $meta_value, $post_id );