From 243bfff7acfb752bb6836b43329421e840254935 Mon Sep 17 00:00:00 2001
From: Robin Cornett <hello@robincornett.com>
Date: Wed, 26 Jun 2019 08:58:38 -0400
Subject: [PATCH] Update block output loading

---
 ...aturedimagegenesis-widgets-blocks-output.php | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/includes/widgets/class-displayfeaturedimagegenesis-widgets-blocks-output.php b/includes/widgets/class-displayfeaturedimagegenesis-widgets-blocks-output.php
index 8b5605c..ae6efdf 100644
--- a/includes/widgets/class-displayfeaturedimagegenesis-widgets-blocks-output.php
+++ b/includes/widgets/class-displayfeaturedimagegenesis-widgets-blocks-output.php
@@ -24,9 +24,9 @@ class DisplayFeaturedImageGenesisWidgetsBlocksOutput {
 		if ( ! $post_type ) {
 			return '';
 		}
-
 		$classes = $this->get_block_classes( $atts, $block_id );
-		include_once plugin_dir_path( dirname( __FILE__ ) ) . 'output/class-displayfeaturedimagegenesis-output-cpt.php';
+		$this->include_output_file( $block_id );
+
 		ob_start();
 		echo '<div class="' . esc_attr( $classes ) . '">';
 		new DisplayFeaturedImageGenesisOutputCPT( $atts, array(), $post_type );
@@ -45,11 +45,11 @@ class DisplayFeaturedImageGenesisWidgetsBlocksOutput {
 	public function render_author( $atts ) {
 		$block_id = 'author';
 		$atts     = $this->update_attributes( $atts, $block_id );
-		include_once plugin_dir_path( dirname( __FILE__ ) ) . 'output/class-displayfeaturedimagegenesis-output-author.php';
 		if ( empty( $atts['user'] ) ) {
 			return '';
 		}
 		$classes = $this->get_block_classes( $atts, $block_id );
+		$this->include_output_file( $block_id );
 
 		ob_start();
 		echo '<div class="' . esc_attr( $classes ) . '">';
@@ -74,7 +74,7 @@ class DisplayFeaturedImageGenesisWidgetsBlocksOutput {
 			return '';
 		}
 		$classes = $this->get_block_classes( $atts, $block_id );
-		include_once plugin_dir_path( dirname( __FILE__ ) ) . 'output/class-displayfeaturedimagegenesis-output-term.php';
+		$this->include_output_file( $block_id );
 
 		ob_start();
 		echo '<div class="' . esc_attr( $classes ) . '">';
@@ -119,4 +119,13 @@ class DisplayFeaturedImageGenesisWidgetsBlocksOutput {
 
 		return implode( ' ', $classes );
 	}
+
+	/**
+	 * Load the block output file.
+	 *
+	 * @param $block_id
+	 */
+	private function include_output_file( $block_id ) {
+		include_once plugin_dir_path( dirname( __FILE__ ) ) . "output/class-displayfeaturedimagegenesis-output-{$block_id}.php";
+	}
 }