metadata

type ArgumentProcessor struct {
	// contains filtered or unexported fields
}

type ArgumentType int

type Assignment struct {
	VariableName int          `yaml:"variable_name,omitempty"`
	Pkg          int          `yaml:"pkg,omitempty"`
	ConcreteType int          `yaml:"concrete_type,omitempty"`
	Position     int          `yaml:"position,omitempty"`
	Scope        int          `yaml:"scope,omitempty"`
	Value        CallArgument `yaml:"value,omitempty"`
	Lhs          CallArgument `yaml:"lhs,omitempty"`
	Func         int          `yaml:"func,omitempty"`

	// For assignments from function calls
	CalleeFunc  string `yaml:"callee_func,omitempty"`
	CalleePkg   string `yaml:"callee_pkg,omitempty"`
	ReturnIndex int    `yaml:"return_index,omitempty"`
}

type AssignmentKey struct {
	Name      string
	Pkg       string
	Type      string
	Container string
}

type AssignmentLink struct {
	AssignmentKey AssignmentKey
	Assignment    *Assignment
	Edge          *CallGraphEdge
}

type Call struct {
	Meta *Metadata      `yaml:"-"`
	Edge *CallGraphEdge `yaml:"-"`

	// Keep existing fields for serialization
	Name     int `yaml:"name,omitempty"`
	Pkg      int `yaml:"pkg,omitempty"`
	Position int `yaml:"position,omitempty"`
	RecvType int `yaml:"recv_type,omitempty"`
	// contains filtered or unexported fields
}

type CallArgument struct {
	Kind     int                    `yaml:"kind"`            // ident, literal, selector, call, raw
	Name     int                    `yaml:"name,omitempty"`  // for ident
	Value    int                    `yaml:"value,omitempty"` // for literal
	X        *CallArgument          `yaml:"x,omitempty"`     // for selector/call
	Sel      *CallArgument          `yaml:"sel,omitempty"`   // for selector
	Fun      *CallArgument          `yaml:"fun,omitempty"`   // for call
	Args     []CallArgument         `yaml:"args,omitempty"`  // for call
	Raw      int                    `yaml:"raw,omitempty"`   // fallback
	Extra    map[string]interface{} `yaml:"extra,omitempty"` // extensibility
	Pkg      int                    `yaml:"pkg,omitempty"`   // for ident
	Type     int                    `yaml:"type,omitempty"`  // for ident
	Position int                    `yaml:"position,omitempty"`

	// Callee edge for the same call if it's kind is call
	Edge *CallGraphEdge `yaml:"-"`

	// fields for argument-to-parameter and type parameter mapping
	ParamArgMap  map[string]CallArgument `yaml:"-"` // parameter name -> argument
	TypeParamMap map[string]string       `yaml:"-"` // type parameter name -> concrete type

	// Type parameter resolution information
	ResolvedType    int  `yaml:"resolved_type,omitempty"`     // The concrete type after type parameter resolution
	IsGenericType   bool `yaml:"is_generic_type,omitempty"`   // Whether this argument represents a generic type
	GenericTypeName int  `yaml:"generic_type_name,omitempty"` // The generic type parameter name (e.g., "TRequest", "TData")

	// Reference to metadata for StringPool access
	Meta *Metadata `yaml:"-"`
	// contains filtered or unexported fields
}

type CallGraphEdge struct {
	Caller        Call                    `yaml:"caller,omitempty"`
	Callee        Call                    `yaml:"callee,omitempty"`
	Position      int                     `yaml:"position,omitempty"`
	Args          []CallArgument          `yaml:"args,omitempty"`
	AssignmentMap map[string][]Assignment `yaml:"assignments,omitempty"`

	// New fields for argument-to-parameter and type parameter mapping
	ParamArgMap  map[string]CallArgument `yaml:"param_arg_map,omitempty"`  // parameter name -> argument
	TypeParamMap map[string]string       `yaml:"type_param_map,omitempty"` // type parameter name -> concrete type

	CalleeVarName     string `yaml:"callee_var_name,omitempty"`
	CalleeRecvVarName string `yaml:"callee_recv_var_name,omitempty"`

	// Chain tracking for chained method calls like app.Group().Use()
	ChainParent *CallGraphEdge `yaml:"-"`                     // Reference to parent call in chain
	ChainRoot   string         `yaml:"chain_root,omitempty"`  // Root variable name (e.g., "app")
	ChainDepth  int            `yaml:"chain_depth,omitempty"` // Depth in chain (0 = root)

	// NEW: Parent function tracking for function literals
	ParentFunction *Call `yaml:"parent_function,omitempty"` // The function that contains this call (for function literals)
	// contains filtered or unexported fields
}

type CallIdentifier struct {
	// contains filtered or unexported fields
}

type CallIdentifierType int

type Field struct {
	Name     int `yaml:"name,omitempty"`
	Type     int `yaml:"type,omitempty"`
	Tag      int `yaml:"tag,omitempty"`
	Scope    int `yaml:"scope,omitempty"`
	Comments int `yaml:"comments,omitempty"`

	// For nested struct types, store the nested type definition
	NestedType *Type `yaml:"nested_type,omitempty"`
}

type File struct {
	Types           map[string]*Type     `yaml:"types,omitempty"`
	Functions       map[string]*Function `yaml:"functions,omitempty"`
	Variables       map[string]*Variable `yaml:"variables,omitempty"`
	StructInstances []StructInstance     `yaml:"struct_instances,omitempty"`
	// Selectors       []Selector           `yaml:"selectors"`
	Imports map[int]int `yaml:"imports"` // alias -> path
}

type Function struct {
	Name      int          `yaml:"name,omitempty"`
	Pkg       int          `yaml:"pkg,omitempty"`
	Signature CallArgument `yaml:"signature,omitempty"`
	Position  int          `yaml:"position,omitempty"`
	Scope     int          `yaml:"scope,omitempty"`
	Comments  int          `yaml:"comments,omitempty"`
	Tags      []int        `yaml:"tags,omitempty"`

	// Type parameter names for generics
	TypeParams []string `yaml:"type_params,omitempty"`

	// Return value origins for tracing through return values
	ReturnVars []CallArgument `yaml:"return_vars,omitempty"`

	// map of variable name to all assignments (for alias/reassignment tracking)
	AssignmentMap map[string][]Assignment `yaml:"assignments,omitempty"`
}

type GenericTypeResolver struct {
	// contains filtered or unexported fields
}

type GlobalAssignment struct {
	ConcreteType string `yaml:"-"`
	PkgName      string `yaml:"-"`
}

type InterfaceResolution struct {
	InterfaceType string `yaml:"interface_type,omitempty"`
	StructType    string `yaml:"struct_type,omitempty"`
	Pkg           string `yaml:"pkg,omitempty"`
	ConcreteType  string `yaml:"concrete_type,omitempty"`
	Position      string `yaml:"position,omitempty"`
}

type InterfaceResolutionKey struct {
	InterfaceType string
	StructType    string
	Pkg           string
}

type Metadata struct {
	StringPool *StringPool         `yaml:"string_pool,omitempty"`
	Packages   map[string]*Package `yaml:"packages,omitempty"`
	CallGraph  []CallGraphEdge     `yaml:"call_graph,omitempty"`

	Callers         map[string][]*CallGraphEdge `yaml:"-"`
	ParentFunctions map[string][]*CallGraphEdge `yaml:"-"`
	Callees         map[string][]*CallGraphEdge `yaml:"-"`
	Args            map[string][]*CallGraphEdge `yaml:"-"`
	// contains filtered or unexported fields
}

type Method struct {
	Name         int          `yaml:"name,omitempty"`
	Receiver     int          `yaml:"receiver,omitempty"`
	Signature    CallArgument `yaml:"signature,omitempty"`
	SignatureStr int          `yaml:"signature_str,omitempty"`
	Position     int          `yaml:"position,omitempty"`
	Scope        int          `yaml:"scope,omitempty"`
	Comments     int          `yaml:"comments,omitempty"`
	Tags         []int        `yaml:"tags,omitempty"`
	Filename     int          `yaml:"filename,omitempty"`

	// Type parameter names for generics
	TypeParams []string `yaml:"type_params,omitempty"`

	// Return value origins for tracing through return values
	ReturnVars []CallArgument `yaml:"return_vars,omitempty"`

	// map of variable name to all assignments (for alias/reassignment tracking)
	AssignmentMap map[string][]Assignment `yaml:"assignments,omitempty"`
}

type Package struct {
	Files map[string]*File `yaml:"files,omitempty"`
	Types map[string]*Type `yaml:"types,omitempty"`
}

type ParamKey struct {
	Name      string
	Pkg       string
	Container string
}

type ProcessedArgument struct {
	Argument   *CallArgument
	Edge       *CallGraphEdge
	ArgType    ArgumentType
	ArgIndex   int
	ArgContext string
	Children   []*ProcessedArgument
}

type Selector struct {
	Expr     CallArgument `yaml:"expr,omitempty"`
	Kind     int          `yaml:"kind,omitempty"`
	Position int          `yaml:"position,omitempty"`
}

type StringPool struct {
	// contains filtered or unexported fields
}

type StructInstance struct {
	Type     int         `yaml:"type,omitempty"`
	Pkg      int         `yaml:"pkg,omitempty"`
	Position int         `yaml:"position,omitempty"`
	Fields   map[int]int `yaml:"fields,omitempty"`
}

type TraceVariableResult struct {
	OriginVar      string
	OriginPkg      string
	OriginType     *CallArgument
	CallerFuncName string
}

type TrackerLimits struct {
	MaxNodesPerTree    int
	MaxChildrenPerNode int
	MaxArgsPerFunction int
	MaxNestedArgsDepth int
	MaxRecursionDepth  int
}

type Type struct {
	Name          int      `yaml:"name,omitempty"`
	Pkg           int      `yaml:"pkg,omitempty"`
	Kind          int      `yaml:"kind,omitempty"`
	Target        int      `yaml:"target,omitempty"`
	Implements    []int    `yaml:"implements,omitempty"`
	ImplementedBy []int    `yaml:"implemented_by,omitempty"`
	Embeds        []int    `yaml:"embeds,omitempty"`
	Fields        []Field  `yaml:"fields,omitempty"`
	Scope         int      `yaml:"scope,omitempty"`
	Methods       []Method `yaml:"methods,omitempty"`
	Comments      int      `yaml:"comments,omitempty"`
	Tags          []int    `yaml:"tags,omitempty"`
}

type Variable struct {
	Name          int         `yaml:"name,omitempty"`
	Tok           int         `yaml:"tok,omitempty"`
	Pkg           int         `yaml:"pkg,omitempty"`
	Type          int         `yaml:"type,omitempty"`
	ResolvedType  int         `yaml:"resolved_type,omitempty"` // underlying type from types.Info
	Value         int         `yaml:"value,omitempty"`
	ComputedValue interface{} `yaml:"computed_value,omitempty"` // actual value from types.Info (for constants)
	Position      int         `yaml:"position,omitempty"`
	Comments      int         `yaml:"comments,omitempty"`
	GroupIndex    int         `yaml:"group_index,omitempty"` // which const group this belongs to
}

type VariableLink struct {
	ParamKey   ParamKey
	OriginVar  string
	OriginPkg  string
	OriginFunc string
	Edge       *CallGraphEdge
	Argument   *CallArgument
}

type VariableOrigin struct {
	OriginVar  string
	OriginPkg  string
	OriginFunc string
	OriginArg  *CallArgument
}