package validate

import (
	"github.com/99designs/gqlgen/plugin/modelgen"
	"github.com/vektah/gqlparser/v2/ast"
)

func FieldHook(td *ast.Definition, fd *ast.FieldDefinition, f *modelgen.Field) (*modelgen.Field, error) {
	if f, err := modelgen.DefaultFieldMutateHook(td, fd, f); err != nil {
		return f, err
	}
	c := fd.Directives.ForName("validation_constraint")
	if c != nil {
		formatConstraint := c.Arguments.ForName("format")
		if formatConstraint != nil {
			f.Tag += " validate:" + formatConstraint.Value.String()
		}
	}
	return f, nil
}