package generate

import (
	"fmt"
	"strings"

	"github.com/vektah/gqlparser/ast"
)

func typeForOperation(operation *ast.OperationDefinition, schema *ast.Schema) (string, error) {
	var builder strings.Builder
	err := writeSelectionSetStruct(&builder, operation.SelectionSet, schema)
	return builder.String(), err
}

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) error {
	builder.WriteString("struct {\n")
	for _, selection := range selectionSet {
		switch selection := selection.(type) {
		case *ast.Field:
			var jsonName string
			if selection.Alias != "" {
				jsonName = selection.Alias
			} else {
				jsonName = selection.Name
			}
			// We need an exportable name for JSON-marshaling.
			goName := strings.Title(jsonName)

			builder.WriteString(goName)
			builder.WriteRune(' ')

			writeType(builder, selection.Definition.Type, selection.SelectionSet, schema)

			if jsonName != goName {
				builder.WriteString("`json:\"")
				builder.WriteString(jsonName)
				builder.WriteString("\"`")
			}
			builder.WriteRune('\n')

		case *ast.FragmentSpread, *ast.InlineFragment:
			return fmt.Errorf("not implemented: %T", selection)
		default:
			return fmt.Errorf("invalid selection type: %v", selection)
		}
	}
	builder.WriteString("}")
	return nil
}

var graphQLNameToGoName = map[string]string{
	"Int":     "int", // TODO: technically int32 is always enough, use that?
	"Float":   "float64",
	"String":  "string",
	"Boolean": "bool",
	"ID":      "string", // TODO: named type for IDs?
}

func writeType(builder *strings.Builder, typ *ast.Type, selectionSet ast.SelectionSet, schema *ast.Schema) error {
	// gqlgen does slightly different things here since it defines names for
	// all the intermediate types, but its implementation may be useful to crib
	// from:
	// https://github.com/99designs/gqlgen/blob/master/plugin/modelgen/models.go#L113
	// TODO: or maybe we should do that?
	if typ.Elem != nil {
		// Type is a list.
		builder.WriteString("[]")
		typ = typ.Elem
	} else if !typ.NonNull { // no need for pointer if we have a list
		builder.WriteString("*")
	}

	if selectionSet != nil {
		return writeSelectionSetStruct(builder, selectionSet, schema)
	}

	// TODO: handle enums better.  (do unions need special handling?)
	goName := graphQLNameToGoName[typ.Name()]
	if goName == "" {
		return fmt.Errorf("unknown scalar name: %s", typ.Name())
	}
	builder.WriteString(goName)
	return nil
}