diff --git a/includes/widgets/class-displayfeaturedimagegenesis-widgets-form.php b/includes/widgets/class-displayfeaturedimagegenesis-widgets-form.php index 0d62a2b..5eb6898 100644 --- a/includes/widgets/class-displayfeaturedimagegenesis-widgets-form.php +++ b/includes/widgets/class-displayfeaturedimagegenesis-widgets-form.php @@ -26,6 +26,28 @@ class DisplayFeaturedImageGenesisWidgetsForm { $this->instance = $instance; } + /** + * Get the public registered post types on the site. + * + * @return mixed + */ + public function get_post_types() { + $args = array( + 'public' => true, + '_builtin' => false, + 'has_archive' => true, + ); + $output = 'objects'; + $post_types = get_post_types( $args, $output ); + + $options['post'] = __( 'Posts', 'display-featured-image-genesis' ); + foreach ( $post_types as $post_type ) { + $options[ $post_type->name ] = $post_type->label; + } + + return $options; + } + /** * @return array */ diff --git a/includes/widgets/displayfeaturedimagegenesis-cpt-archive-widget.php b/includes/widgets/displayfeaturedimagegenesis-cpt-archive-widget.php index e69da59..c6e2656 100644 --- a/includes/widgets/displayfeaturedimagegenesis-cpt-archive-widget.php +++ b/includes/widgets/displayfeaturedimagegenesis-cpt-archive-widget.php @@ -163,26 +163,4 @@ class Display_Featured_Image_Genesis_Widget_CPT extends WP_Widget { 'archive' => include 'fields/archive.php', ) ); } - - /** - * Get the public registered post types on the site. - * - * @return mixed - */ - protected function get_post_types() { - $args = array( - 'public' => true, - '_builtin' => false, - 'has_archive' => true, - ); - $output = 'objects'; - $post_types = get_post_types( $args, $output ); - - $options['post'] = __( 'Posts', 'display-featured-image-genesis' ); - foreach ( $post_types as $post_type ) { - $options[ $post_type->name ] = $post_type->label; - } - - return $options; - } } diff --git a/includes/widgets/fields/cpt-post_type.php b/includes/widgets/fields/cpt-post_type.php index cf0ab6f..99d567b 100644 --- a/includes/widgets/fields/cpt-post_type.php +++ b/includes/widgets/fields/cpt-post_type.php @@ -10,7 +10,7 @@ return array( 'id' => 'post_type', 'label' => __( 'Post Type:', 'display-featured-image-genesis' ), 'flex' => true, - 'choices' => $this->get_post_types(), + 'choices' => $form->get_post_types(), ), ), );