feat(cli): custom enum val directive (#31)

* feat(cli): add new custom directive for enum values

* feat(golang): check for custom enum value

cmd/types.go

@@ -29,6 +29,26 @@ var gqlcTypes = []*ast.TypeDecl{
 			}},
 		}},
 	},
+	{
+		Spec: &ast.TypeDecl_TypeSpec{TypeSpec: &ast.TypeSpec{
+			Name: &ast.Ident{Name: "as"},
+			Type: &ast.TypeSpec_Directive{Directive: &ast.DirectiveType{
+				Locs: []*ast.DirectiveLocation{
+					{Loc: ast.DirectiveLocation_ENUM_VALUE},
+				},
+				Args: &ast.InputValueList{
+					List: []*ast.InputValue{
+						{
+							Name: &ast.Ident{Name: "value"},
+							Type: &ast.InputValue_Ident{
+								Ident: &ast.Ident{Name: "String"},
+							},
+						},
+					},
+				},
+			}},
+		}},
+	},
 }
 
 func init() {

golang/go.go

@@ -445,7 +445,11 @@ func (g *Generator) generateEnum(name string, descr bool, doc *ast.DocGroup, ts
 		g.P('"', v.Name.Name, '"', ": &graphql.EnumValueConfig{")
 		g.In()
 
-		g.P("Value: \"", v.Name.Name, "\",")
+		val := getValue(v.Directives)
+		if val == "" {
+			val = v.Name.Name
+		}
+		g.P("Value: \"", val, "\",")
 
 		if v.Doc != nil && descr {
 			g.printDescr(v.Doc)
@@ -841,3 +845,14 @@ func getResolver(dirs []*ast.DirectiveLit) string {
 	}
 	return ""
 }
+
+func getValue(dirs []*ast.DirectiveLit) string {
+	for _, d := range dirs {
+		if d.Name != "as" {
+			continue
+		}
+
+		return strings.Trim(d.Args.Args[0].Value.(*ast.Arg_BasicLit).BasicLit.Value, "\"")
+	}
+	return ""
+}