Add Directives to Build

codegen/build.go

@@ -21,6 +21,7 @@ type Build struct {
 	SubscriptionRoot *Object
 	SchemaRaw        string
 	SchemaFilename   string
+	Directives       []*Directive
 }
 
 type ModelBuild struct {
@@ -84,6 +85,7 @@ func (cfg *Config) bind() (*Build, error) {
 		Imports:        imports.finalize(),
 		SchemaRaw:      cfg.SchemaStr,
 		SchemaFilename: cfg.SchemaFilename,
+		Directives:     cfg.buildDirectives(),
 	}
 
 	if cfg.schema.Query != nil {

codegen/directive.go

@@ -0,0 +1,11 @@
+package codegen
+
+import "strings"
+
+type Directive struct {
+	name string
+}
+
+func (d *Directive) Name() string {
+	return strings.Title(d.name)
+}

codegen/directive_build.go

@@ -0,0 +1,13 @@
+package codegen
+
+func (cfg *Config) buildDirectives() (directives []*Directive) {
+	for name := range cfg.schema.Directives {
+		if name == "skip" || name == "include" || name == "deprecated" {
+			continue
+		}
+		directives = append(directives, &Directive{
+			name: name,
+		})
+	}
+	return directives
+}

codegen/templates/generated.gotpl

@@ -13,6 +13,11 @@ func NewExecutableSchema(resolvers ResolverRoot) graphql.ExecutableSchema {
 	return &executableSchema{resolvers: resolvers}
 }
 
+type Config struct {
+	resolvers ResolverRoot
+	{{ if .Directives }}directive DirectiveRoot{{ end }}
+}
+
 type ResolverRoot interface {
 {{- range $object := .Objects -}}
 	{{ if $object.HasResolvers -}}
@@ -21,6 +26,14 @@ type ResolverRoot interface {
 {{- end }}
 }
 
+{{ if .Directives }}
+type DirectiveRoot struct {
+{{ range $directive := .Directives }}
+	{{$directive.Name}} graphql.FieldMiddleware
+{{ end }}
+}
+{{ end }}
+
 {{- range $object := .Objects -}}
 	{{ if $object.HasResolvers }}
 		type {{$object.GQLType}}Resolver interface {