convert.go

package guts

import (
	"context"
	"fmt"
	"go/constant"
	"go/token"
	"go/types"
	"log/slog"
	"reflect"
	"regexp"
	"sort"
	"strings"

	"github.com/fatih/structtag"
	"golang.org/x/tools/go/packages"
	"golang.org/x/xerrors"

	"github.com/coder/guts/bindings"
)

type TypeOverride func() bindings.ExpressionType

// GoParser takes in Golang packages, and can convert them to the intermediate
// typescript representation. The intermediate representation is closely
// aligned with the typescript AST.
type GoParser struct {
	Pkgs      map[string]*packages.Package
	Reference map[string]bool
	Prefix    map[string]string

	// referencedTypes is a map of all types that are referenced by the generated
	// packages. This is to generated referenced types on demand.
	// map[package][type]generated
	referencedTypes *referencedTypes

	// typeOverrides can override any field type with a custom type.
	// This needs to be a producer function, as the AST is mutated directly,
	// and we cannot have shared references.
	// Eg: "time.Time" -> "string"
	typeOverrides map[string]TypeOverride
	config        *packages.Config
	fileSet       *token.FileSet
}

// NewGolangParser returns a new GoParser object.
// This object is responsible for converting Go types into the intermediate
// typescript AST representation.
// All configuration of the GoParser should be done before calling
// 'ToTypescript'.
// For usage, see 'ExampleGeneration' in convert_test.go.
func NewGolangParser() (*GoParser, error) {
	fileSet := token.NewFileSet()
	config := &packages.Config{
		// Just accept the fact we need these flags for what we want. Feel free to add
		// more, it'll just increase the time it takes to parse.
		Mode: packages.NeedTypes | packages.NeedName | packages.NeedTypesInfo |
			packages.NeedTypesSizes | packages.NeedSyntax | packages.NeedDeps,
		Tests: false,
		Fset:  fileSet,
		//Dir:     "/home/steven/go/src/github.com/coder/guts",
		Context: context.Background(),
	}

	return &GoParser{
		fileSet:         fileSet,
		config:          config,
		Pkgs:            make(map[string]*packages.Package),
		Reference:       map[string]bool{},
		referencedTypes: newReferencedTypes(),
		Prefix:          map[string]string{},
		typeOverrides: map[string]TypeOverride{
			// Some hard coded defaults
			"error": func() bindings.ExpressionType {
				return ptr(bindings.KeywordString)
			},
		},
	}, nil
}

// IncludeCustomDeclaration is an advanced form of IncludeCustom.
func (p *GoParser) IncludeCustomDeclaration(mappings map[string]TypeOverride) {
	for k, v := range mappings {
		p.typeOverrides[k] = v
	}
}

type GolangType = string

// IncludeCustom takes in a remapping of golang types.
// Both the key and value of the map should be valid golang types.
// The key is the type to override, and the value is the new type.
// Typescript will be generated with the new type.
//
// Only named types can be overridden.
// Examples:
// "github.com/your/repo/pkg.ExampleType": "string"
// "time.Time": "string"
func (p *GoParser) IncludeCustom(mappings map[GolangType]GolangType) error {
	for k, v := range mappings {
		// Make sure it parses
		_, err := parseExpression(v)
		if err != nil {
			return fmt.Errorf("failed to parse expression %s: %w", v, err)
		}

		v := v
		p.typeOverrides[k] = func() bindings.ExpressionType {
			exp, err := parseExpression(v)
			if err != nil {
				return ptr(bindings.KeywordUnknown)
			}
			return exp
		}
	}
	return nil
}

// IncludeGenerate parses a directory and adds the parsed package to the list of packages.
// These package's types will be generated.
func (p *GoParser) IncludeGenerate(directory string) error {
	return p.include(directory, "", false)
}

// IncludeGenerateWithPrefix will include a prefix to all output generated types.
func (p *GoParser) IncludeGenerateWithPrefix(directory string, prefix string) error {
	return p.include(directory, prefix, false)
}

// IncludeReference only generates types if they are referenced from the generated packages.
// This is useful for only generating a subset of the types that are being used.
func (p *GoParser) IncludeReference(directory string, prefix string) error {
	return p.include(directory, prefix, true)
}

func (p *GoParser) include(directory string, prefix string, reference bool) error {
	pkgs, err := packages.Load(p.config, directory)
	if err != nil {
		return fmt.Errorf("failed to parse directory %s: %w", directory, err)
	}

	for _, v := range pkgs {
		if _, ok := p.Pkgs[v.PkgPath]; ok {
			return fmt.Errorf("package %s already exists", v.PkgPath)
		}
		p.Pkgs[v.PkgPath] = v
		p.Reference[v.PkgPath] = reference
		p.Prefix[v.PkgPath] = prefix
		if len(v.Errors) > 0 {
			for _, e := range v.Errors {
				slog.Error(
					parsePackageError(e),
					slog.String("error", e.Error()),
					slog.String("pkg", v.PkgPath),
					slog.String("directory", directory),
				)
			}
		}
	}
	return nil
}

// ToTypescript translates the Go types into the intermediate typescript AST
// The returned typescript object can be mutated before serializing.
func (p *GoParser) ToTypescript() (*Typescript, error) {
	typescript := &Typescript{
		typescriptNodes: make(map[string]*typescriptNode),
		parsed:          p,
	}

	// Parse all go types to the typescript AST
	err := typescript.parseGolangIdentifiers()
	if err != nil {
		return nil, err
	}

	// Apply any post-processing mutations to the nodes.
	for key, node := range typescript.typescriptNodes {
		newNode, err := node.applyMutations()
		if err != nil {
			return nil, fmt.Errorf("node %q: %w", key, err)
		}
		typescript.typescriptNodes[key] = &newNode
	}

	return typescript, nil
}

type Typescript struct {
	// typescriptNodes is a map of typescript nodes that are generated from the
	// parsed go code. All names should be unique. If non-unique names exist, that
	// means packages contain the same named types.
	// TODO: the key "string" should be replaced with "Identifier"
	typescriptNodes map[string]*typescriptNode
	parsed          *GoParser
	// Do not allow calling serialize more than once.
	// The call affects the state.
	serialized bool
}

func (ts *Typescript) parseGolangIdentifiers() error {
	// Look for comments that indicate to ignore a type for typescript generation.
	// Comment format to skip typescript generation: `@typescript-ignore <ignored_type>`
	ignoreRegex := regexp.MustCompile("@typescript-ignore[:]?(?P<ignored_types>.*)")

	refPkgs := make([]*packages.Package, 0, len(ts.parsed.Pkgs))
	genPkgs := make([]*packages.Package, 0, len(ts.parsed.Pkgs))
	for _, pkg := range ts.parsed.Pkgs {
		if ts.parsed.Reference[pkg.PkgPath] {
			refPkgs = append(refPkgs, pkg)
			continue
		}
		genPkgs = append(genPkgs, pkg)
	}

	// always do gen packages first to know the references
	for _, pkg := range append(genPkgs, refPkgs...) {
		skippedTypes := make(map[string]struct{})
		for _, file := range pkg.Syntax {
			for _, comment := range file.Comments {
				for _, line := range comment.List {
					text := line.Text
					matches := ignoreRegex.FindStringSubmatch(text)
					ignored := ignoreRegex.SubexpIndex("ignored_types")
					if len(matches) >= ignored && matches[ignored] != "" {
						arr := strings.Split(matches[ignored], ",")
						for _, s := range arr {
							skippedTypes[strings.TrimSpace(s)] = struct{}{}
						}
					}
				}
			}
		}

		allIdents := pkg.Types.Scope().Names()
		for _, ident := range allIdents {
			if _, ok := skippedTypes[ident]; ok {
				continue
			}

			obj := pkg.Types.Scope().Lookup(ident)

			if ts.parsed.Reference[pkg.PkgPath] {
				if !ts.parsed.referencedTypes.IsReferenced(obj) {
					continue
				}
			}
			if ts.parsed.referencedTypes.IsGenerated(obj) {
				continue
			}

			err := ts.parse(obj)
			if err != nil {
				return fmt.Errorf("parse object %q in %q: %w", ident, pkg.PkgPath, err)
			}

			ts.parsed.referencedTypes.MarkGenerated(obj)
		}

		// As long as references other things, we have to keep going.
		err := ts.parsed.referencedTypes.Remaining(func(obj types.Object) error {
			err := ts.parse(obj)
			if err != nil {
				return fmt.Errorf("parse referenced object %q: %w", pkg.PkgPath, err)
			}
			ts.parsed.referencedTypes.MarkGenerated(obj)
			return nil
		})
		if err != nil {
			return fmt.Errorf("generated referenced types: %w", err)
		}

	}
	return nil
}

func (ts *Typescript) ReplaceNode(key string, node bindings.Node) {
	ts.typescriptNodes[key] = &typescriptNode{
		Node: node,
	}
}

func (ts *Typescript) Node(key string) (bindings.Node, bool) {
	v, ok := ts.typescriptNodes[key]
	if !ok {
		return nil, false
	}
	return v.Node, true
}

func (ts *Typescript) SetNode(key string, node bindings.Node) error {
	if _, ok := ts.typescriptNodes[key]; ok {
		return fmt.Errorf("node %q already exists", key)
	}
	ts.ReplaceNode(key, node)
	return nil
}

func (ts *Typescript) setNode(key string, node typescriptNode) error {
	if _, ok := ts.typescriptNodes[key]; ok {
		return fmt.Errorf("node %q already exists", key)
	}
	ts.typescriptNodes[key] = &node
	return nil
}

func (ts *Typescript) updateNode(key string, update func(n *typescriptNode)) {
	v, ok := ts.typescriptNodes[key]
	if !ok {
		v = &typescriptNode{}
		ts.typescriptNodes[key] = v
	}
	update(v)
}

func (ts *Typescript) ApplyMutations(muts ...func(typescript *Typescript)) {
	for _, mut := range muts {
		mut(ts)
	}
}

// ForEach iterates through all the nodes in the typescript AST.
func (ts *Typescript) ForEach(node func(key string, node bindings.Node)) {
	for k, v := range ts.typescriptNodes {
		node(k, v.Node)
	}
}

// Serialize will serialize the typescript AST to typescript code.
// It sorts all types alphabetically.
func (ts *Typescript) Serialize() (string, error) {
	return ts.SerializeInOrder(func(order map[string]bindings.Node) []bindings.Node {
		names := make([]string, 0, len(order))
		for k := range order {
			names = append(names, k)
		}
		sort.Strings(names)

		nodes := make([]bindings.Node, 0, len(names))
		for _, k := range names {
			nodes = append(nodes, order[k])
		}

		return nodes
	})
}

func (ts *Typescript) SerializeInOrder(sort func(nodes map[string]bindings.Node) []bindings.Node) (string, error) {
	if ts.serialized {
		return "", fmt.Errorf("already serialized, create a new TS object to serialize again")
	}
	// Even if it fails, do not allow calling this function again.
	ts.serialized = true

	vm, err := bindings.New()
	if err != nil {
		return "", fmt.Errorf("failed to create typescript bindings: %w", err)
	}

	nodes := make(map[string]bindings.Node)
	for k, v := range ts.typescriptNodes {
		nodes[k] = v.Node
	}
	order := sort(nodes)

	var str strings.Builder
	str.WriteString("// Code generated by 'guts'. DO NOT EDIT.\n\n")

	for k, v := range order {
		obj, err := vm.ToTypescriptNode(v)
		if err != nil {
			return "", fmt.Errorf("convert node %q: %w", k, err)
		}

		text, err := vm.SerializeToTypescript(obj)
		if err != nil {
			return "", fmt.Errorf("serialize to typescript: %w", err)
		}
		str.WriteString(text + "\n\n")
	}
	return str.String(), nil

}

func (ts *Typescript) parse(obj types.Object) error {
	objectIdentifier := ts.parsed.Identifier(obj)

	switch obj := obj.(type) {
	// All named types are type declarations
	case *types.TypeName:
		var rhs types.Type
		switch typedObj := obj.Type().(type) {
		case *types.Named:
			rhs = typedObj.Underlying()
		case *types.Alias:
			rhs = typedObj.Underlying()
		default:
			// Fall the type through... this should be ok
			rhs = typedObj
		}

		switch underNamed := rhs.(type) {
		case *types.Struct:
			// type <Name> struct
			// Structs are obvious.
			node, err := ts.buildStruct(obj, underNamed)
			if err != nil {
				return xerrors.Errorf("generate %q: %w", objectIdentifier.Ref(), err)
			}
			return ts.setNode(objectIdentifier.Ref(), typescriptNode{
				Node: node,
			})
		case *types.Basic:
			// type <Name> string
			// These are enums. Store to expand later.
			rhs, err := ts.typescriptType(underNamed)
			if err != nil {
				return xerrors.Errorf("generate basic %q: %w", objectIdentifier.Ref(), err)
			}

			// If this has 'const's, then it is an enum. The enum code will
			// patch this value to be more specific.
			ts.updateNode(objectIdentifier.Ref(), func(n *typescriptNode) {
				n.Node = &bindings.Alias{
					Name:       objectIdentifier,
					Modifiers:  []bindings.Modifier{},
					Type:       rhs.Value,
					Parameters: rhs.TypeParameters,
					Source:     ts.location(obj),
				}
			})
			return nil
		case *types.Map, *types.Array, *types.Slice:
			// Declared maps that are not structs are still valid codersdk objects.
			// Handle them custom by calling 'typescriptType' directly instead of
			// iterating through each struct field.bindings.Union()
			// These types support no json/typescript tags.
			// These are **NOT** enums, as a map in Go would never be used for an enum.
			ty, err := ts.typescriptType(obj.Type().Underlying())
			if err != nil {
				return xerrors.Errorf("(map) generate %q: %w", objectIdentifier.Ref(), err)
			}

			return ts.setNode(objectIdentifier.Ref(), typescriptNode{
				Node: &bindings.Alias{
					Name:       objectIdentifier,
					Modifiers:  []bindings.Modifier{},
					Type:       ty.Value,
					Parameters: ty.TypeParameters,
					Source:     ts.location(obj),
				},
			})
		case *types.Interface:
			// Interfaces are used as generics. Non-generic interfaces are
			// not supported.
			if underNamed.NumEmbeddeds() == 1 {
				union, ok := underNamed.EmbeddedType(0).(*types.Union)
				if !ok {
					// If the underlying is not a union, but has 1 type. It's
					// just that one type.
					union = types.NewUnion([]*types.Term{
						// Set the tilde to true to support underlying.
						// Doesn't actually affect our generation.
						types.NewTerm(true, underNamed.EmbeddedType(0)),
					})
				}

				block, err := ts.buildUnion(obj, union)
				if err != nil {
					return xerrors.Errorf("generate union %q: %w", objectIdentifier.Ref(), err)
				}
				return ts.setNode(objectIdentifier.Ref(), typescriptNode{
					Node: block,
				})
			}

			if underNamed.NumEmbeddeds() == 0 {
				// type <Name> interface{}
				// TODO: We could convert the function signatures to typescript.
				return nil
			}

			return xerrors.Errorf("interface %q is not a union, has %d embeds and unsupported", objectIdentifier.Ref(), underNamed.NumEmbeddeds())
		case *types.Signature:
			// Ignore named functions.
			return nil
		default:
			// If you hit this error, you added a new unsupported named type.
			// The easiest way to solve this is add a new case above with
			// your type and a TODO to implement it.
			return xerrors.Errorf("unsupported named type %q", underNamed.String())
		}
	case *types.Var:
		// TODO: Are any enums var declarations? This is also codersdk.Me.
		return nil // Maybe we should treat these like consts?
	case *types.Const:
		// Names are very likely enums
		named, ok := obj.Type().(*types.Named)
		if !ok {
			// It could be a raw const value to generate.
			if _, ok := obj.Type().(*types.Basic); ok {
				cnst, err := ts.constantDeclaration(obj)
				if err != nil {
					return xerrors.Errorf("basic const %q: %w", objectIdentifier.Ref(), err)
				}
				return ts.setNode(objectIdentifier.Ref(), typescriptNode{
					Node: cnst,
				})
			}
			return xerrors.Errorf("const %q is not a named type", objectIdentifier.Ref())
		}

		// Treat it as an enum.
		enumObjName := ts.parsed.Identifier(named.Obj())

		switch named.Underlying().(type) {
		case *types.Basic:
		default:
			return xerrors.Errorf("const %q is not a basic type, enums only support basic", objectIdentifier.Ref())
		}

		// Grab the value of the constant. This is the enum value.
		constValue, err := ts.constantValue(obj)
		if err != nil {
			return xerrors.Errorf("const %q: %w", objectIdentifier.Ref(), err)
		}

		// This is a little hacky, but we need to add the enum to the Alias
		// type. However, the order types are parsed is not guaranteed, so we
		// add the enum to the Alias as a post-processing step.
		ts.updateNode(enumObjName.Ref(), func(n *typescriptNode) {
			n.AddEnum(constValue)
		})
		return nil
	case *types.Func:
		// Noop
		return nil
	default:
		return xerrors.Errorf("unsupported object type %T", obj)
	}

	return xerrors.Errorf("should never hit this, obj with type %T", obj)
}

func (ts *Typescript) constantDeclaration(obj *types.Const) (*bindings.VariableStatement, error) {
	val, err := ts.constantValue(obj)
	if err != nil {
		return &bindings.VariableStatement{}, err
	}

	return &bindings.VariableStatement{
		Modifiers: []bindings.Modifier{},
		Declarations: &bindings.VariableDeclarationList{
			Declarations: []*bindings.VariableDeclaration{
				{
					Name:            ts.parsed.Identifier(obj),
					ExclamationMark: false,
					Initializer:     val,
				},
			},
			Flags: bindings.NodeFlagsConstant,
		},
		Source: ts.location(obj),
	}, nil
}

func (ts *Typescript) constantValue(obj *types.Const) (*bindings.LiteralType, error) {
	var constValue bindings.LiteralType
	switch obj.Val().Kind() {
	case constant.String:
		constValue.Value = constant.StringVal(obj.Val())
	case constant.Int:
		// TODO: might want to check this
		constValue.Value, _ = constant.Int64Val(obj.Val())
	case constant.Float:
		constValue.Value, _ = constant.Float64Val(obj.Val())
	case constant.Bool:
		constValue.Value = constant.BoolVal(obj.Val())
	default:
		return &bindings.LiteralType{}, xerrors.Errorf("const %q is not a supported basic type, enums only support basic", obj.Name())
	}
	return &constValue, nil
}

// buildStruct just prints the typescript def for a type.
// Generic type parameters are inferred from the type and inferred.
func (ts *Typescript) buildStruct(obj types.Object, st *types.Struct) (*bindings.Interface, error) {
	tsi := &bindings.Interface{
		Name:       ts.parsed.Identifier(obj),
		Modifiers:  []bindings.Modifier{},
		Fields:     []*bindings.PropertySignature{},
		Parameters: []*bindings.TypeParameter{},  // Generics
		Heritage:   []*bindings.HeritageClause{}, // Extends
		Source:     ts.location(obj),
	}

	// Handle named embedded structs in the codersdk package via extension.
	// This is inheritance.
	// TODO: Maybe this could be done inline in the main for loop?
	var extends []parsedType
	for i := 0; i < st.NumFields(); i++ {
		field := st.Field(i)
		tag := reflect.StructTag(st.Tag(i))
		// Adding a json struct tag causes the json package to consider
		// the field unembedded.
		if field.Embedded() && tag.Get("json") == "" {
			// TODO: Generic args
			heritage, err := ts.typescriptType(field.Type())
			if err != nil {
				return tsi, xerrors.Errorf("heritage type: %w", err)
			}
			extends = append(extends, heritage)
		}
	}

	if len(extends) > 0 {
		var heritages []bindings.ExpressionType
		for _, heritage := range extends {
			heritages = append(heritages, heritage.Value)
		}
		tsi.Heritage = append(tsi.Heritage, bindings.HeritageClauseExtends(heritages...))
	}

	if _, ok := obj.(*types.TypeName); ok {
		var typeParamed interface{ TypeParams() *types.TypeParamList }
		switch typedObj := obj.Type().(type) {
		case *types.Named:
			typeParamed = typedObj
		//case *types.Alias:
		// Generic Go alias types are introduced in Go 1.23. When updated to
		// Go 1.23, this code can be uncommented
		//typeParamed = typedObj
		default:
			return tsi, xerrors.Errorf("not supported type %T for %q to parse type parameters", obj.Type(), obj.Name())
		}

		// This code is usually redundant, as we infer generics from the
		// child usage. However, if the field is unused, then this comes in
		// handy.
		// Note: Maybe we can remove all generic values bubbling up in favor
		// of this?
		// Note: Maybe do not even need this, as it includes unused generics.
		typeParameters, err := ts.typeParametersParameters(typeParamed)
		if err != nil {
			return tsi, xerrors.Errorf("type parameters: %w", err)
		}
		tsi.Parameters = typeParameters
	}

	// Iterate through the fields of the struct.
	for i := 0; i < st.NumFields(); i++ {
		field := st.Field(i)
		tag := reflect.StructTag(st.Tag(i))
		tags, err := structtag.Parse(string(tag))
		if err != nil {
			panic("invalid struct tags on type " + obj.String())
		}

		if field.Embedded() && tag.Get("json") == "" {
			// Heritage was done above
			// TODO: should do it here
			continue
		}

		if !field.Exported() {
			// Skip unexported fields
			continue
		}

		// Create a new field in the intermediate typescript representation.
		tsField := &bindings.PropertySignature{
			Name:          field.Name(),
			Modifiers:     []bindings.Modifier{},
			QuestionToken: false,
			Type:          nil,
		}

		// Use the json name if present
		jsonTag, err := tags.Get("json")
		if err == nil {
			if jsonTag.Name == "-" {
				// Completely ignore this field.
				continue
			}
			// Empty tags are ignored.
			if jsonTag.Name != "" {
				tsField.Name = jsonTag.Name
			}
			if len(jsonTag.Options) > 0 && jsonTag.Options[0] == "omitempty" {
				tsField.QuestionToken = true
			}
		}

		// Infer the type.
		tsType, err := ts.typescriptType(field.Type())
		if err != nil {
			return tsi, xerrors.Errorf("typescript type: %w", err)
		}
		tsField.Type = tsType.Value
		tsi.Parameters = append(tsi.Parameters, tsType.TypeParameters...)
		tsField.FieldComments = tsType.RaisedComments

		// Some tag support
		// TODO: Add more tag support?
		typescriptTag, err := tags.Get("typescript")
		if err == nil {
			if typescriptTag.Name == "-" {
				// Completely ignore this field.
				continue
			}
		}

		tsi.Fields = append(tsi.Fields, tsField)
	}

	simple, err := bindings.Simplify(tsi.Parameters)
	if err != nil {
		return tsi, xerrors.Errorf("simplify generics: %w", err)
	}
	tsi.Parameters = simple
	return tsi, nil
}

type parsedType struct {
	// Value is the typescript type of the passed in go type.
	Value bindings.ExpressionType
	// TypeParameters are any generic types that are used in the Value.
	TypeParameters []*bindings.TypeParameter
	// RaisedComments exists to add comments to the first parent that is willing
	// to accept them. It is for formatting purposes.
	RaisedComments []string
}

func simpleParsedType(et bindings.ExpressionType) parsedType {
	return parsedType{
		Value: et,
	}
}

func (p parsedType) WithComments(comments ...string) parsedType {
	p.RaisedComments = append(p.RaisedComments, comments...)
	return p
}

// TODO: Return comments?
func (ts *Typescript) typescriptType(ty types.Type) (parsedType, error) {
	switch ty := ty.(type) {
	case *types.Signature:
		// TODO: Handle functions better
		return simpleParsedType(ptr(bindings.KeywordUnknown)).
			WithComments("Function type detected, and unsupported. Leaving the type as unknown"), nil
	case *types.Basic:
		bs := ty
		// All basic literals (string, bool, int, etc).
		switch {
		case bs.Info()&types.IsNumeric > 0:
			return simpleParsedType(ptr(bindings.KeywordNumber)), nil
		case bs.Info()&types.IsBoolean > 0:
			return simpleParsedType(ptr(bindings.KeywordBoolean)), nil
		case bs.Kind() == types.Byte:
			// TODO: @emyrk What is a byte for typescript? A string? A uint8?
			// TODO: Comment
			//return bindings.PrependComment("This is a byte in golang", bindings.Literal(bindings.KeywordNumber)), nil
			return simpleParsedType(ptr(bindings.KeywordNumber)), nil
		case bs.Kind() == types.String, bs.Kind() == types.Rune:
			return simpleParsedType(ptr(bindings.KeywordString)), nil
		case bs.Kind() == types.Invalid:
			// TODO: Investigate why this happens
			return simpleParsedType(ptr(bindings.KeywordAny)).WithComments("Invalid type, using 'any'. Might be a reference to any external package"), nil
		default:
			return parsedType{}, xerrors.Errorf("unsupported basic type %q", bs.String())
		}
	case *types.Struct:
		// This handles anonymous structs. This should never happen really.
		// If you require this, either change your datastructures, or implement
		// anonymous structs here.
		// Such as:
		//  type Name struct {
		//	  Embedded struct {
		//		  Field string `json:"field"`
		//	  }
		//  }
		// TODO: Comment: indentedComment("Embedded anonymous struct, please fix by naming it"),
		parsed := simpleParsedType(ptr(bindings.KeywordUnknown))
		parsed.RaisedComments = append(parsed.RaisedComments, "embedded anonymous struct, please fix by naming it")
		return parsed, nil
	case *types.Map:
		// Record is reference type with 2 type parameters.
		// map[string][string] -> Record<string, string>

		m := ty
		keyType, err := ts.typescriptType(m.Key())
		if err != nil {
			return parsedType{}, xerrors.Errorf("map key: %w", err)
		}
		valueType, err := ts.typescriptType(m.Elem())
		if err != nil {
			return parsedType{}, xerrors.Errorf("map key: %w", err)
		}

		tp, err := bindings.Simplify(append(keyType.TypeParameters, valueType.TypeParameters...))
		if err != nil {
			return parsedType{}, xerrors.Errorf("simplify generics in map: %w", err)
		}
		parsed := parsedType{
			Value:          RecordReference(keyType.Value, valueType.Value),
			TypeParameters: tp,
			RaisedComments: append(keyType.RaisedComments, valueType.RaisedComments...),
		}
		return parsed, nil
	case *types.Slice, *types.Array:
		// Slice/Arrays are pretty much the same.
		type hasElem interface {
			Elem() types.Type
		}

		arr, _ := ty.(hasElem)
		switch {
		// When type checking here, just use the string. You can cast it
		// to a types.Basic and get the kind if you want too :shrug:
		case arr.Elem().String() == "byte":
			// All byte arrays are strings on the typescript.
			// Is this ok?
			return simpleParsedType(bindings.Array(ptr(bindings.KeywordString))), nil
		default:
			// By default, just do an array of the underlying type.
			underlying, err := ts.typescriptType(arr.Elem())
			if err != nil {
				return parsedType{}, xerrors.Errorf("array: %w", err)
			}
			//genValue := ""
			//
			//if underlying.GenericValue != "" {
			//	genValue = "Readonly<Array<" + underlying.GenericValue + ">>"
			//}
			return parsedType{
				Value:          bindings.Array(underlying.Value),
				TypeParameters: underlying.TypeParameters,
				RaisedComments: underlying.RaisedComments,
			}, nil
			//return TypescriptType{
			//	ValueType:     "Readonly<Array<" + underlying.ValueType + ">>",
			//	GenericValue:  genValue,
			//	AboveTypeLine: underlying.AboveTypeLine,
			//	GenericTypes:  underlying.GenericTypes,
			//}, nil
		}
	case *types.Named:
		n := ty

		// These are external named types that we handle uniquely.
		// This is unfortunate, but our current code assumes all defined
		// types are enums, but these are really just basic primitives.
		// We would need to add more logic to determine this, but for now
		// just hard code them.
		// TODO: Allow comments here
		custom, ok := ts.parsed.typeOverrides[n.String()]
		if ok {
			return parsedType{
				Value: custom(),
			}, nil
		}

		// If it is not a custom mapping, we should assume the type is
		// defined elsewhere. We want to know where and what that definition
		// is, such that we can raise up any type parameters.
		ref, ok := ts.parsed.lookupNamedReference(n)
		if ok {
			if ref.Pkg().Path() != n.Obj().Pkg().Path() {
				slog.Info("found external type", slog.String("name", ref.Name()), slog.String("ext_pkg", ref.Pkg().Path()))
			}

			args, err := ts.typeParametersArgs(n)
			if err != nil {
				return parsedType{}, xerrors.Errorf("type parameter arguments: %w", err)
			}

			parsed := parsedType{}
			exprArgs := make([]bindings.ExpressionType, 0, len(args))
			for _, arg := range args {
				exprArgs = append(exprArgs, arg.Value)
				parsed.TypeParameters = append(parsed.TypeParameters, arg.TypeParameters...)
				parsed.RaisedComments = append(parsed.RaisedComments, arg.RaisedComments...)
			}
			parsed.Value = bindings.Reference(ts.parsed.Identifier(ref), exprArgs...)

			return parsed, nil
		}

		// If it's a struct, just use the name of the struct type
		if _, ok := n.Underlying().(*types.Struct); ok {
			// This struct comes from an external package that we did not parse.
			// We can introspect it, but then it acts as an anonymous struct
			// embed. Structs should be flat in their fields, so just return a
			// reference with a comment.
			return simpleParsedType(ptr(bindings.KeywordUnknown)).WithComments(
				// '.Include(<pkg_path>, false)' to include this type
				fmt.Sprintf("external type %q, to include this type the package must be explicitly included in the parsing", n.String())), nil
		}

		// Defer to the underlying type.
		ts, err := ts.typescriptType(ty.Underlying())
		if err != nil {
			return parsedType{}, xerrors.Errorf("named underlying: %w", err)
		}

		return ts.WithComments(fmt.Sprintf("this is likely an enum in an external package %q", n.String())), nil
	case *types.Pointer:
		// Dereference pointers.
		pt := ty
		resp, err := ts.typescriptType(pt.Elem())
		if err != nil {
			return parsedType{}, xerrors.Errorf("pointer: %w", err)
		}

		// Golang pointers can json marshal to 'null' if they are nil
		resp.Value = bindings.Union(resp.Value, &bindings.Null{})
		return resp, nil
	case *types.Interface:
		// only handle the empty interface (interface{}) for now
		intf := ty
		if intf.Empty() {
			// This field is 'interface{}'. We can't infer any type from 'interface{}'
			// so just use "unknown" as the type.
			parsed := simpleParsedType(ptr(bindings.KeywordUnknown))
			parsed.RaisedComments = append(parsed.RaisedComments, "empty interface{} type, falling back to unknown")
			return parsed, nil
			//return TypescriptType{
			//	AboveTypeLine: indentedComment("empty interface{} type, falling back to unknown"),
			//	ValueType:     "unknown",
			//}, nil
		}

		if intf.NumEmbeddeds() == 1 {
			parsedI, err := ts.typescriptType(intf.EmbeddedType(0))
			if err != nil {
				return parsedType{}, xerrors.Errorf("parse interface: %w", err)
			}
			return parsedI, nil
		}

		// Interfaces are difficult to determine the JSON type, so just return
		// an 'unknown'.
		parsed := simpleParsedType(ptr(bindings.KeywordUnknown))
		parsed.RaisedComments = append(parsed.RaisedComments, "interface type, falling back to unknown")
		return parsed, nil
	case *types.TypeParam:
		_, ok := ty.Underlying().(*types.Interface)
		if !ok {
			// If it's not an interface, it is likely a usage of generics that
			// we have not hit yet. Feel free to add support for it.
			return parsedType{}, xerrors.New("type param must be an interface")
		}

		// type Foo[T any] struct {
		name := ts.parsed.Identifier(ty.Obj()) // T
		generic := ty.Constraint()             // generic

		// We don't mess with multiple packages, so just trim the package path
		// from the name.
		pkgPath := ty.Obj().Pkg().Path()
		constraintName := strings.TrimPrefix(generic.String(), pkgPath+".")

		// Any is the default
		var constraintNode bindings.ExpressionType
		switch constraintName {
		case "comparable":
			// TODO: Generate this on demand.
			constraintNode = bindings.Reference(builtInComparable)
			ts.includeComparable()
		case "any":
			constraintNode = ptr(bindings.KeywordAny)
		default:
			parsedGeneric, err := ts.typescriptType(generic)
			if err != nil {
				return parsedType{}, xerrors.Errorf("type param %q: %w", generic.String(), err)
			}

			// TODO: Raise comments and generics?
			constraintNode = parsedGeneric.Value
		}

		return parsedType{
			Value: bindings.Reference(name),
			TypeParameters: []*bindings.TypeParameter{
				{
					Name:      name,
					Modifiers: []bindings.Modifier{},
					// All generics in Golang have some type of constraint (even if it's 'any').
					// TODO: if the constraint is 'any', we should probably not bother with the type
					// It is redundant.
					Type:        constraintNode,
					DefaultType: nil,
				},
			},
		}, nil
	case *types.Alias:
		// TODO: Verify this is correct.
		return ts.typescriptType(ty.Underlying())
	case *types.Union:
		allTypes := make([]bindings.ExpressionType, 0, ty.Len())
		for i := 0; i < ty.Len(); i++ {
			constraintType, err := ts.typescriptType(ty.Term(i).Type())
			if err != nil {
				return parsedType{}, xerrors.Errorf("union %q: %w", ty.String(), err)
			}
			allTypes = append(allTypes, constraintType.Value)
		}
		return parsedType{
			Value:          bindings.Union(allTypes...),
			TypeParameters: nil,
			RaisedComments: nil,
		}, nil
	}

	// These are all the other types we need to support.
	return parsedType{}, xerrors.Errorf("unknown type: %s", ty.String())
}

// buildStruct just prints the typescript def for a type.
func (ts *Typescript) buildUnion(obj types.Object, st *types.Union) (*bindings.Alias, error) {
	alias := &bindings.Alias{
		Name:       ts.parsed.Identifier(obj),
		Modifiers:  []bindings.Modifier{},
		Type:       nil,
		Parameters: nil,
		Source:     ts.location(obj),
	}

	allTypes := make([]bindings.ExpressionType, 0, st.Len())
	for i := 0; i < st.Len(); i++ {
		term := st.Term(i)
		scriptType, err := ts.typescriptType(term.Type())
		if err != nil {
			return alias, xerrors.Errorf("union %q for %q failed to get type: %w", st.String(), obj.Name(), err)
		}
		// TODO: Generics
		// scriptType.TypeParameters
		allTypes = append(allTypes, scriptType.Value)
	}

	alias.Type = bindings.Union(allTypes...)
	return alias, nil
}

// typeParametersParameters extracts the generic parameters from a named type.
func (ts *Typescript) typeParametersParameters(obj interface{ TypeParams() *types.TypeParamList }) ([]*bindings.TypeParameter, error) {
	args := obj.TypeParams()
	if args == nil || args.Len() == 0 {
		return []*bindings.TypeParameter{}, nil
	}

	params := make([]*bindings.TypeParameter, 0, args.Len())
	for i := 0; i < args.Len(); i++ {
		arg := args.At(i)
		argType, err := ts.typescriptType(arg)
		if err != nil {
			return nil, xerrors.Errorf("type parameter %q: %w", arg.String(), err)
		}

		params = append(params, argType.TypeParameters...)
	}
	return params, nil
}

func (ts *Typescript) typeParametersArgs(obj *types.Named) ([]parsedType, error) {
	args := obj.TypeArgs()
	if args == nil || args.Len() == 0 {
		return []parsedType{}, nil
	}

	params := make([]parsedType, 0, args.Len())
	for i := 0; i < args.Len(); i++ {
		arg := args.At(i)
		argType, err := ts.typescriptType(arg)
		if err != nil {
			return nil, xerrors.Errorf("type parameter %q: %w", arg.String(), err)
		}
		params = append(params, argType)
	}
	return params, nil
}

func (p *GoParser) lookupNamedReference(n *types.Named) (types.Object, bool) {
	if n.Obj().Pkg() == nil {
		return nil, false
	}
	lookupPkg := n.Obj().Pkg().Path()
	pkg, ok := p.Pkgs[lookupPkg]
	if !ok {
		return nil, false
	}

	lookupName := n.Obj().Name()
	obj := pkg.Types.Scope().Lookup(lookupName)
	if obj == nil {
		return nil, false
	}

	// Mark type as referenced
	p.referencedTypes.MarkReferenced(obj)

	return obj, true
}

// ObjectName returns the name of the object including any prefixes defined by
// the config.
func (p *GoParser) Identifier(obj types.Object) bindings.Identifier {
	name := obj.Name()
	prefix := p.Prefix[obj.Pkg().Path()]
	return bindings.Identifier{
		Name:    name,
		Prefix:  prefix,
		Package: obj.Pkg(),
	}
}

func ptr[T any](v T) *T {
	return &v
}