argoyle
ee378dc6a3
Introduce types for managing fielded, enum, union, and scalar shared types. Implement functionality for comparing values and creating field sets. Enhance schema extensions by integrating new visitors for enum types.
45 lines
1.3 KiB
Go
45 lines
1.3 KiB
Go
package sdlmerge
|
|
|
|
import (
|
|
"github.com/wundergraph/graphql-go-tools/v2/pkg/ast"
|
|
"github.com/wundergraph/graphql-go-tools/v2/pkg/astvisitor"
|
|
)
|
|
|
|
func newRemoveFieldDefinitionDirective(directives ...string) *removeFieldDefinitionDirective {
|
|
directivesSet := make(map[string]struct{}, len(directives))
|
|
for _, directive := range directives {
|
|
directivesSet[directive] = struct{}{}
|
|
}
|
|
|
|
return &removeFieldDefinitionDirective{
|
|
directives: directivesSet,
|
|
}
|
|
}
|
|
|
|
type removeFieldDefinitionDirective struct {
|
|
operation *ast.Document
|
|
directives map[string]struct{}
|
|
}
|
|
|
|
func (r *removeFieldDefinitionDirective) Register(walker *astvisitor.Walker) {
|
|
walker.RegisterEnterDocumentVisitor(r)
|
|
walker.RegisterEnterFieldDefinitionVisitor(r)
|
|
}
|
|
|
|
func (r *removeFieldDefinitionDirective) EnterDocument(operation, _ *ast.Document) {
|
|
r.operation = operation
|
|
}
|
|
|
|
func (r *removeFieldDefinitionDirective) EnterFieldDefinition(ref int) {
|
|
var refsForDeletion []int
|
|
// select directives for deletion
|
|
for _, directiveRef := range r.operation.FieldDefinitions[ref].Directives.Refs {
|
|
directiveName := r.operation.DirectiveNameString(directiveRef)
|
|
if _, ok := r.directives[directiveName]; ok {
|
|
refsForDeletion = append(refsForDeletion, directiveRef)
|
|
}
|
|
}
|
|
// delete directives
|
|
r.operation.RemoveDirectivesFromNode(ast.Node{Kind: ast.NodeKindFieldDefinition, Ref: ref}, refsForDeletion)
|
|
}
|