genql: quick hack at variables

generate/generate.go

@@ -26,22 +26,31 @@ type TemplateParams struct {
 	// The name of the package into which to generate the operation-helpers.
 	PackageName string
 	// The list of operations for which to generate code.
-	Operations []OperationParams
+	Operations []Operation
 }
 
-type OperationParams struct {
+type Operation struct {
+	// The type of the operation (query, mutation, or subscription).
+	Type ast.Operation
+	// The name of the operation, from GraphQL.
+	Name string
+	// The documentation for the operation, from GraphQL.
+	Doc string
+	// The body of the operation to send.
+	Body string
+	// The arguments to the operation.
+	Args []Argument
+
 	// The type-name for the operation's response type.
 	ResponseName string
 	// The body of the operation's response type (e.g. struct { ... }).
 	ResponseType string
-	// The type of the operation (query, mutation, or subscription).
-	OperationType ast.Operation
-	// The name of the operation, from GraphQL.
-	OperationName string
-	// The documentation for the operation, from GraphQL.
-	OperationDoc string
-	// The body of the operation to send.
-	Operation string
+}
+
+type Argument struct {
+	GoName      string
+	GoType      string
+	GraphQLName string
 }
 
 func Generate(specFilename, schemaFilename, generatedFilename string) error {
@@ -98,7 +107,7 @@ func Generate(specFilename, schemaFilename, generatedFilename string) error {
 	packageName := "example"
 
 	// TODO: this should probably get factored out
-	operations := make([]OperationParams, len(document.Operations))
+	operations := make([]Operation, len(document.Operations))
 	for i, operation := range document.Operations {
 		// TODO: we may have to actually get the precise query text, in case we
 		// want to be hashing it or something like that.  Although maybe
@@ -110,20 +119,30 @@ func Generate(specFilename, schemaFilename, generatedFilename string) error {
 			Operations: ast.OperationList{operation},
 			// TODO: handle fragments
 		})
-		operations[i] = OperationParams{
-			OperationType: operation.Operation,
-			OperationName: operation.Name,
+
+		args := make([]Argument, len(operation.VariableDefinitions))
+		for i, arg := range operation.VariableDefinitions {
+			args[i] = Argument{
+				GraphQLName: arg.Variable,
+				GoName:      arg.Variable, // TODO: normalize this to go-style
+				GoType:      typeForInputType(arg.Type, schema),
+				// TODO: figure out what to do about defaults
+			}
+		}
+		operations[i] = Operation{
+			Type: operation.Operation,
+			Name: operation.Name,
 			// TODO: this is actually awkward, because GraphQL doesn't allow
 			// for docstrings on queries (only schemas).  So we have to extract
 			// the comment, or omit doc-comments for now.
-			OperationDoc: "TODO",
+			Doc: "TODO",
+			// The newline just makes it format a little nicer
+			Body: "\n" + builder.String(),
+			Args: args,
 
 			// TODO: configure ResponseName format
 			ResponseName: operation.Name + "Response",
-			ResponseType: typeFor(operation, schema),
-
-			// The newline just makes it format a little nicer
-			Operation: "\n" + builder.String(),
+			ResponseType: typeForOperation(operation, schema),
 		}
 	}
 

generate/operation.go.tmpl

@@ -9,10 +9,17 @@ import (
 {{range .Operations}}
 type {{.ResponseName}} = {{.ResponseType}}
 
-// {{.OperationDoc}}
-func {{.OperationName}}(ctx context.Context, client *graphql.Client) (*{{.ResponseName}}, error) {
+// {{.Doc}}
+func {{.Name}}(ctx context.Context, client *graphql.Client{{range .Args}}, {{.GoName}} {{.GoType}}{{end}}) (*{{.ResponseName}}, error) {
+    {{if .Args}}
+    variables := map[string]interface{}{
+        {{range .Args}}
+        "{{.GraphQLName}}": {{.GoName}},
+        {{end}}
+    }
+    {{end}}
 	var retval {{.ResponseName}}
-    err := client.MakeRequest(ctx, `{{.Operation}}`, &retval)
+    err := client.MakeRequest(ctx, `{{.Body}}`, &retval, {{if .Args}}variables{{else}}nil{{end}})
     return &retval, err
 }
 {{end}}

generate/types.go

@@ -7,7 +7,7 @@ import (
 	"github.com/vektah/gqlparser/ast"
 )
 
-func typeFor(operation *ast.OperationDefinition, schema *ast.Schema) string {
+func typeForOperation(operation *ast.OperationDefinition, schema *ast.Schema) string {
 	var builder strings.Builder
 
 	writeSelectionSetStruct(&builder, operation.SelectionSet, schema)
@@ -15,6 +15,16 @@ func typeFor(operation *ast.OperationDefinition, schema *ast.Schema) string {
 	return builder.String()
 }
 
+func typeForInputType(typ *ast.Type, schema *ast.Schema) string {
+	var builder strings.Builder
+
+	// TODO: handle non-scalar types (by passing ...something... as the
+	// SelectionSet?)
+	writeType(&builder, typ, nil, schema)
+
+	return builder.String()
+}
+
 func writeSelectionSetStruct(builder *strings.Builder, selectionSet ast.SelectionSet, schema *ast.Schema) {
 	builder.WriteString("struct {\n")
 	for _, selection := range selectionSet {