-
Notifications
You must be signed in to change notification settings - Fork 15
Expand file tree
/
Copy pathValidationInterceptor.cs
More file actions
58 lines (50 loc) · 1.83 KB
/
ValidationInterceptor.cs
File metadata and controls
58 lines (50 loc) · 1.83 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
using System.Linq;
using HotChocolate.Configuration;
using HotChocolate.Types.Descriptors;
using HotChocolate.Types.Descriptors.Definitions;
namespace AppAny.HotChocolate.FluentValidation
{
internal sealed class ValidationInterceptor : TypeInterceptor
{
public override void OnBeforeCompleteType(
ITypeCompletionContext completionContext,
DefinitionBase? definition)
{
if (definition is not ObjectTypeDefinition objectTypeDefinition)
{
return;
}
var validationOptions = completionContext.ContextData.GetValidationOptions();
foreach (var objectFieldDefinition in objectTypeDefinition.Fields)
{
var argumentOptions = objectFieldDefinition.Arguments
.Where(argument => argument.ContextData.ShouldValidateArgument())
.Select(argument => argument.ContextData.GetArgumentOptions())
.ToList();
if (argumentOptions is { Count: > 0 })
{
foreach (var options in argumentOptions)
{
options.MergeValidationOptions(validationOptions);
}
objectFieldDefinition.ContextData.CreateObjectFieldOptions();
objectFieldDefinition.MiddlewareDefinitions.Insert(
index: 0,
new FieldMiddlewareDefinition(ValidationDefaults.Middleware));
}
}
}
[Obsolete]
public override void OnAfterCreateSchema(IDescriptorContext descriptorContext, ISchema schema)
{
foreach (var objectField in schema.Types.OfType<IObjectType>().SelectMany(type => type.Fields))
{
var objectOptions = objectField.ContextData.TryGetObjectFieldOptions();
foreach (var argument in objectField.Arguments.Where(arg => arg.ContextData.ShouldValidateArgument()))
{
objectOptions?.Arguments.Add(argument.Name, argument);
}
}
}
}
}