wp-graphql / src / Data / Loader / PostTypeLoader.php

PostTypeLoader.php in WPGraphQL 2.22.1, at src/Data/Loader/PostTypeLoader.php

50 lines 876 B

Raw Download Zip

1	<?php
2	namespace WPGraphQL\Data\Loader;
3
4	use WPGraphQL\Model\PostType;
5
6	/**
7	* Class PostTypeLoader
8	*
9	* @package WPGraphQL\Data\Loader
10	*/
11	class PostTypeLoader extends AbstractDataLoader {
12
13	/**
14	* {@inheritDoc}
15	*
16	* @param mixed\|\WP_Post_Type $entry The Post Type Object
17	*
18	* @return \WPGraphQL\Model\PostType
19	*/
20	protected function get_model( $entry, $key ) {
21	return new PostType( $entry );
22	}
23
24	/**
25	* {@inheritDoc}
26	*
27	* @param string[] $keys
28	* @return array<string,\WP_Post_Type\|null>
29	*/
30	public function loadKeys( array $keys ) {
31	$post_types = \WPGraphQL::get_allowed_post_types( 'objects' );
32
33	if ( empty( $post_types ) ) {
34	return [];
35	}
36
37	$loaded = [];
38
39	foreach ( $keys as $key ) {
40	if ( isset( $post_types[ $key ] ) ) {
41	$loaded[ $key ] = $post_types[ $key ];
42	} else {
43	$loaded[ $key ] = null;
44	}
45	}
46
47	return $loaded;
48	}
49	}
50