support: parse schema directive

Author: n9te9

schema/directive.go

@@ -12,9 +12,10 @@ type Location struct {
 }
 
 type Directive struct {
-	Name      []byte
-	Arguments []*DirectiveArgument
-	Locations []*Location
+	Name       []byte
+	Arguments  []*DirectiveArgument
+	Locations  []*Location
+	Extentions []*Directive
 }
 
 type Directives []*Directive
@@ -39,6 +40,7 @@ type DirectiveDefinition struct {
 	Arguments   []*ArgumentDefinition
 	Repeatable  bool
 	Locations   []*Location
+	Extentions  []*DirectiveDefinition
 }
 
 func (d *DirectiveDefinition) IsAllowedApplySchema() bool {

schema/parser.go

@@ -272,7 +272,28 @@ func (p *Parser) parseExtendDefinition(schema *Schema, tokens Tokens, cur int) (
 
 		t.Extentions = append(t.Extentions, unionDefinition)
 	case Scalar:
-		// TODO: Support
+		scalarDefinition, newCur, err := p.parseScalarDefinition(tokens, cur)
+		if err != nil {
+			return nil, 0, err
+		}
+		cur = newCur
+		t := get(schema.Indexes, string(scalarDefinition.Name), scalarDefinition)
+		if t == nil {
+			return nil, 0, fmt.Errorf("%s is not defined", scalarDefinition.Name)
+		}
+
+		t.Extentions = append(t.Extentions, scalarDefinition)
+	case At:
+		directive, newCur, err := p.parseDirectiveDefinition(tokens, cur)
+		if err != nil {
+			return nil, 0, err
+		}
+		cur = newCur
+		t := get(schema.Indexes, string(directive.Name), directive)
+		if t == nil {
+			return nil, 0, fmt.Errorf("%s is not defined", directive.Name)
+		}
+		t.Extentions = append(t.Extentions, directive)
 	}
 
 	return schema, cur, nil

schema/schema.go

@@ -170,7 +170,7 @@ func (o *OperationDefinition) GetFieldByName(name []byte) *FieldDefinition {
 }
 
 type DefinitionType interface {
-	*TypeDefinition | *OperationDefinition | *EnumDefinition | *UnionDefinition | *InterfaceDefinition | *InputDefinition | *ScalarDefinition
+	*TypeDefinition | *OperationDefinition | *EnumDefinition | *UnionDefinition | *InterfaceDefinition | *InputDefinition | *ScalarDefinition | *DirectiveDefinition
 }
 
 type Indexes struct {
@@ -181,6 +181,7 @@ type Indexes struct {
 	InputIndex       map[string]*InputDefinition
 	OperationIndexes map[OperationType]map[string]*OperationDefinition
 	ScalarIndex      map[string]*ScalarDefinition
+	DirectiveIndex   map[string]*DirectiveDefinition
 }
 
 func (i *Indexes) GetTypeDefinition(name string) *TypeDefinition {
@@ -675,6 +676,10 @@ func get[T DefinitionType](indexes *Indexes, key string, t T) T {
 		return any(indexes.InterfaceIndex[key]).(T)
 	case *InputDefinition:
 		return any(indexes.InputIndex[key]).(T)
+	case *ScalarDefinition:
+		return any(indexes.ScalarIndex[key]).(T)
+	case *DirectiveDefinition:
+		return any(indexes.DirectiveIndex[key]).(T)
 	}
 
 	return nil
@@ -683,6 +688,7 @@ func get[T DefinitionType](indexes *Indexes, key string, t T) T {
 type ScalarDefinition struct {
 	Name       []byte
 	Directives []*Directive
+	Extentions []*ScalarDefinition
 }
 
 type SchemaDefinition struct {