LoneSchemaDefinition.php — WooCommerce 10.9.3

woocommerce / lib / packages / GraphQL / Validator / Rules / LoneSchemaDefinition.php

woocommerce / lib / packages / GraphQL / Validator / Rules Last commit date

LoneSchemaDefinition.php

58 lines

1	<?php declare(strict_types=1);
2
3	namespace Automattic\WooCommerce\Vendor\GraphQL\Validator\Rules;
4
5	use Automattic\WooCommerce\Vendor\GraphQL\Error\Error;
6	use Automattic\WooCommerce\Vendor\GraphQL\Language\AST\NodeKind;
7	use Automattic\WooCommerce\Vendor\GraphQL\Language\AST\SchemaDefinitionNode;
8	use Automattic\WooCommerce\Vendor\GraphQL\Validator\SDLValidationContext;
9
10	/**
11	* Lone schema definition.
12	*
13	* A Automattic\WooCommerce\Vendor\GraphQL document is only valid if it contains only one schema definition.
14	*/
15	class LoneSchemaDefinition extends ValidationRule
16	{
17	public static function schemaDefinitionNotAloneMessage(): string
18	{
19	return 'Must provide only one schema definition.';
20	}
21
22	public static function canNotDefineSchemaWithinExtensionMessage(): string
23	{
24	return 'Cannot define a new schema within a schema extension.';
25	}
26
27	public function getSDLVisitor(SDLValidationContext $context): array
28	{
29	$oldSchema = $context->getSchema();
30	$alreadyDefined = $oldSchema === null
31	? false
32	: (
33	$oldSchema->astNode !== null
34	\|\| $oldSchema->getQueryType() !== null
35	\|\| $oldSchema->getMutationType() !== null
36	\|\| $oldSchema->getSubscriptionType() !== null
37	);
38
39	$schemaDefinitionsCount = 0;
40
41	return [
42	NodeKind::SCHEMA_DEFINITION => static function (SchemaDefinitionNode $node) use ($alreadyDefined, $context, &$schemaDefinitionsCount): void {
43	if ($alreadyDefined) {
44	$context->reportError(new Error(static::canNotDefineSchemaWithinExtensionMessage(), $node));
45
46	return;
47	}
48
49	if ($schemaDefinitionsCount > 0) {
50	$context->reportError(new Error(static::schemaDefinitionNotAloneMessage(), $node));
51	}
52
53	++$schemaDefinitionsCount;
54	},
55	];
56	}
57	}
58