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.
62 lines
1.9 KiB
Go
62 lines
1.9 KiB
Go
package sdlmerge
|
|
|
|
import (
|
|
"github.com/wundergraph/graphql-go-tools/v2/pkg/ast"
|
|
"github.com/wundergraph/graphql-go-tools/v2/pkg/astvisitor"
|
|
"github.com/wundergraph/graphql-go-tools/v2/pkg/operationreport"
|
|
)
|
|
|
|
type collectEntitiesVisitor struct {
|
|
*astvisitor.Walker
|
|
document *ast.Document
|
|
collectedEntities entitySet
|
|
}
|
|
|
|
func newCollectEntitiesVisitor(collectedEntities entitySet) *collectEntitiesVisitor {
|
|
return &collectEntitiesVisitor{
|
|
collectedEntities: collectedEntities,
|
|
}
|
|
}
|
|
|
|
func (c *collectEntitiesVisitor) Register(walker *astvisitor.Walker) {
|
|
c.Walker = walker
|
|
walker.RegisterEnterDocumentVisitor(c)
|
|
walker.RegisterEnterInterfaceTypeDefinitionVisitor(c)
|
|
walker.RegisterEnterObjectTypeDefinitionVisitor(c)
|
|
}
|
|
|
|
func (c *collectEntitiesVisitor) EnterDocument(operation, _ *ast.Document) {
|
|
c.document = operation
|
|
}
|
|
|
|
func (c *collectEntitiesVisitor) EnterInterfaceTypeDefinition(ref int) {
|
|
interfaceType := c.document.InterfaceTypeDefinitions[ref]
|
|
name := c.document.InterfaceTypeDefinitionNameString(ref)
|
|
if err := c.resolvePotentialEntity(name, interfaceType.Directives.Refs); err != nil {
|
|
c.StopWithExternalErr(*err)
|
|
}
|
|
}
|
|
|
|
func (c *collectEntitiesVisitor) EnterObjectTypeDefinition(ref int) {
|
|
objectType := c.document.ObjectTypeDefinitions[ref]
|
|
name := c.document.ObjectTypeDefinitionNameString(ref)
|
|
if err := c.resolvePotentialEntity(name, objectType.Directives.Refs); err != nil {
|
|
c.StopWithExternalErr(*err)
|
|
}
|
|
}
|
|
|
|
func (c *collectEntitiesVisitor) resolvePotentialEntity(name string, directiveRefs []int) *operationreport.ExternalError {
|
|
if _, exists := c.collectedEntities[name]; exists {
|
|
err := operationreport.ErrEntitiesMustNotBeDuplicated(name)
|
|
return &err
|
|
}
|
|
for _, directiveRef := range directiveRefs {
|
|
if c.document.DirectiveNameString(directiveRef) != "key" {
|
|
continue
|
|
}
|
|
c.collectedEntities[name] = struct{}{}
|
|
return nil
|
|
}
|
|
return nil
|
|
}
|