antlrhelpers

type Function struct {
	RequiredArgs         int
	Variadic             bool
	Args                 []string
	ReturnType           string
	CalcReturnType       func(...string) string // If present, will be used to calculate the return type
	ShouldArgsBeNullable bool
	CalcNullable         func(...func() bool) func() bool // will be provided with the nullability of the args
}

type Functions map[string]Function

type Node interface {
	GetStart() antlr.Token
	GetStop() antlr.Token
	GetText() string
	GetParser() antlr.Parser
}

type NodeInfo struct {
	Node                 Node
	ExprDescription      string
	Type                 NodeTypes
	ExprRef              Node
	IgnoreRefNullability bool

	// Go Info
	ArgKey         string // Positional or named arg in the query
	EditedPosition [2]int
	CanBeMultiple  bool
}

type NodeKey struct {
	Start int
	Stop  int
}

type NodeType struct {
	DBType    string
	NullableF func() bool
}

type NodeTypes []NodeType

type QuerySource struct {
	Schema  string
	Name    string
	Columns []ReturnColumn
	CTE     bool
}

type QuerySources = []QuerySource

type ReturnColumn struct {
	Name   string
	Type   NodeTypes
	Config drivers.QueryCol
}

type StmtInfo struct {
	Node      Node
	QueryType bob.QueryType
	Comment   string
	Columns   []ReturnColumn
	EditRules []internal.EditRule
	Mods      *strings.Builder
	Imports   [][]string
}

type Visitor[C, I any] struct {
	Err       error
	DB        drivers.Tables[C, I]
	Args      []NodeKey
	Groups    []NodeKey
	Names     map[NodeKey]string
	Infos     map[NodeKey]NodeInfo
	Sources   []QuerySource
	Functions Functions
	BaseRules []internal.EditRule

	// Refresh these for each statement
	StmtRules []internal.EditRule
	Atom      *atomic.Int64
}