Add post type extractor.

kuno1 · May 27, 2020 · fd473cf · fd473cf
1 parent b3413f7
commit fd473cf
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/Kunoichi/ThemeCustomizer/CustomizerSetting.php b/src/Kunoichi/ThemeCustomizer/CustomizerSetting.php
@@ -127,4 +127,25 @@ protected function get_section() {
 			return $this->camelize( get_called_class() );
 		}
 	}
+
+	/**
+	 * Get post type for multiple choices.
+	 *
+	 * @param array    $query
+	 * @param string[] $exclude
+	 * @return array
+	 */
+	protected function get_post_types( $query = [], $exclude = [] ) {
+		if ( ! $query ) {
+			$query = [ 'public' => true ];
+		}
+		$post_types = [];
+		foreach ( get_post_types( $query, OBJECT ) as $post_type ) {
+			if ( in_array( $post_type->name, $exclude ) ) {
+				continue;
+			}
+			$post_types[ $post_type->name ] = $post_type->label;
+		}
+		return $post_types;
+	}
 }