Update TypeExtension definition

graphql-go · May 20, 2019 · a9564f8 · a9564f8
1 parent e5f1be4
commit a9564f8
Show file tree

Hide file tree

Showing 5 changed files with 384 additions and 122 deletions.
diff --git a/go.mod b/go.mod
diff --git a/language/ast/definitions.go b/language/ast/definitions.go
@@ -148,44 +148,6 @@ func (vd *VariableDefinition) GetLoc() *Location {
 	return vd.Loc
 }
 
-// TypeExtensionDefinition implements Node, Definition
-type TypeExtensionDefinition struct {
-	Kind       string
-	Loc        *Location
-	Definition *ObjectDefinition
-}
-
-func NewTypeExtensionDefinition(def *TypeExtensionDefinition) *TypeExtensionDefinition {
-	if def == nil {
-		def = &TypeExtensionDefinition{}
-	}
-	return &TypeExtensionDefinition{
-		Kind:       kinds.TypeExtensionDefinition,
-		Loc:        def.Loc,
-		Definition: def.Definition,
-	}
-}
-
-func (def *TypeExtensionDefinition) GetKind() string {
-	return def.Kind
-}
-
-func (def *TypeExtensionDefinition) GetLoc() *Location {
-	return def.Loc
-}
-
-func (def *TypeExtensionDefinition) GetVariableDefinitions() []*VariableDefinition {
-	return []*VariableDefinition{}
-}
-
-func (def *TypeExtensionDefinition) GetSelectionSet() *SelectionSet {
-	return &SelectionSet{}
-}
-
-func (def *TypeExtensionDefinition) GetOperation() string {
-	return ""
-}
-
 // DirectiveDefinition implements Node, Definition
 type DirectiveDefinition struct {
 	Kind        string

diff --git a/language/ast/type_extensions.go b/language/ast/type_extensions.go
@@ -0,0 +1,53 @@
+package ast
+
+import "github.com/graphql-go/graphql/language/kinds"
+
+type TypeExtension interface {
+	GetKind() string
+	GetLoc() *Location
+}
+
+var _ TypeExtension = (*ScalarDefinition)(nil)
+var _ TypeExtension = (*ObjectDefinition)(nil)
+var _ TypeExtension = (*InterfaceDefinition)(nil)
+var _ TypeExtension = (*UnionDefinition)(nil)
+var _ TypeExtension = (*EnumDefinition)(nil)
+var _ TypeExtension = (*InputObjectDefinition)(nil)
+
+// TypeExtensionDefinition implements Node, Definition
+type TypeExtensionDefinition struct {
+	Kind       string
+	Loc        *Location
+	Definition TypeExtension
+}
+
+func NewTypeExtensionDefinition(def *TypeExtensionDefinition) *TypeExtensionDefinition {
+	if def == nil {
+		def = &TypeExtensionDefinition{}
+	}
+	return &TypeExtensionDefinition{
+		Kind:       kinds.TypeExtensionDefinition,
+		Loc:        def.Loc,
+		Definition: def.Definition,
+	}
+}
+
+func (def *TypeExtensionDefinition) GetKind() string {
+	return def.Kind
+}
+
+func (def *TypeExtensionDefinition) GetLoc() *Location {
+	return def.Loc
+}
+
+func (def *TypeExtensionDefinition) GetVariableDefinitions() []*VariableDefinition {
+	return []*VariableDefinition{}
+}
+
+func (def *TypeExtensionDefinition) GetSelectionSet() *SelectionSet {
+	return &SelectionSet{}
+}
+
+func (def *TypeExtensionDefinition) GetOperation() string {
+	return ""
+}