From 10cba769c16f84b422df5c07e043d1db2b2ddefe Mon Sep 17 00:00:00 2001
From: Robin Cornett <hello@robincornett.com>
Date: Mon, 24 Jun 2019 12:05:38 -0400
Subject: [PATCH] Add a getter for the widget form class

---
 ...turedimagegenesis-widgets-blocks-fields.php | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/includes/widgets/class-displayfeaturedimagegenesis-widgets-blocks-fields.php b/includes/widgets/class-displayfeaturedimagegenesis-widgets-blocks-fields.php
index be7130b..d92546f 100644
--- a/includes/widgets/class-displayfeaturedimagegenesis-widgets-blocks-fields.php
+++ b/includes/widgets/class-displayfeaturedimagegenesis-widgets-blocks-fields.php
@@ -80,7 +80,7 @@ class DisplayFeaturedImageGenesisWidgetsBlocksFields {
 	 * @return array
 	 */
 	protected function cpt_fields() {
-		$form = new DisplayFeaturedImageGenesisWidgetsForm( $this, array() );
+		$form = $this->get_form_class();
 
 		return array_merge(
 			include 'fields/cpt-post_type.php',
@@ -94,7 +94,7 @@ class DisplayFeaturedImageGenesisWidgetsBlocksFields {
 	 * @return array
 	 */
 	protected function author_fields() {
-		$form = new DisplayFeaturedImageGenesisWidgetsForm( $this, array() );
+		$form = $this->get_form_class();
 		$user = array(
 			array(
 				'method' => 'select',
@@ -179,4 +179,18 @@ class DisplayFeaturedImageGenesisWidgetsBlocksFields {
 
 		return $output;
 	}
+
+	/**
+	 * Get the widget builder form class.
+	 *
+	 * @return \DisplayFeaturedImageGenesisWidgetsForm
+	 */
+	private function get_form_class() {
+		if ( isset( $this->form ) ) {
+			return $this->form;
+		}
+		$this->form = new DisplayFeaturedImageGenesisWidgetsForm( $this, array() );
+
+		return $this->form;
+	}
 }