types

type AnalysisState struct {
	// Symbol tables by file
	SymbolTables map[string]*SymbolTable `json:"symbol_tables"`

	// All discovered sources
	Sources []FlowNode `json:"sources"`

	// All discovered carriers
	Carriers []FlowNode `json:"carriers"`

	// Tainted variables by scope
	TaintedVars map[string]map[string]*TaintInfo `json:"tainted_vars"` // scope -> name -> info

	// Object instances being tracked
	ObjectInstances map[string]*ObjectInstance `json:"object_instances"`

	// Call graph
	CallGraph map[string][]string `json:"call_graph"`

	// File dependencies
	FileDependencies map[string][]string `json:"file_dependencies"`

	// Current context
	CurrentFile   string `json:"current_file"`
	CurrentClass  string `json:"current_class"`
	CurrentMethod string `json:"current_method"`
	CurrentScope  string `json:"current_scope"`

	// Analysis depth tracking
	Depth    int `json:"depth"`
	MaxDepth int `json:"max_depth"`

	// Visited tracking (prevent infinite loops)
	Visited map[string]bool `json:"-"`
}

type AnnotationDef struct {
	Name      string                 `json:"name"`
	Arguments map[string]interface{} `json:"arguments,omitempty"`
	Line      int                    `json:"line"`
}

type Assignment struct {
	Target      string `json:"target"`      // Variable being assigned to
	TargetType  string `json:"target_type"` // "variable", "property", "array_element"
	Source      string `json:"source"`      // Expression being assigned
	SourceType  string `json:"source_type"` // Type of source expression
	Line        int    `json:"line"`
	Column      int    `json:"column"`
	FilePath    string `json:"file_path"`
	Scope       string `json:"scope"`
	IsTainted   bool   `json:"is_tainted"`
	TaintSource string `json:"taint_source,omitempty"`

	// For compound assignments
	Operator string `json:"operator,omitempty"` // =, +=, .=, etc.

	// For array/object access
	Keys []string `json:"keys,omitempty"` // Access path: ["input", "thumbnail"]
}

type BackwardPath struct {
	// Source information
	Source SourceInfo `json:"source"`

	// Steps from source to target (in forward order for readability)
	Steps []BackwardStep `json:"steps"`

	// Whether path crosses file boundaries
	CrossFile bool `json:"cross_file"`
}

type BackwardStep struct {
	StepNumber  int    `json:"step_number"`
	Expression  string `json:"expression"` // The code at this step
	FilePath    string `json:"file_path"`
	Line        int    `json:"line"`
	StepType    string `json:"step_type"` // "source", "assignment", "parameter", "return", "property"
	Description string `json:"description"`
}

type BackwardTraceResult struct {
	// Target expression being traced
	TargetExpression string `json:"target_expression"`
	TargetFile       string `json:"target_file"`
	TargetLine       int    `json:"target_line"`

	// All paths from sources to this target
	Paths []BackwardPath `json:"paths"`

	// Summary of all sources found
	Sources []SourceInfo `json:"sources"`

	// Analysis metadata
	AnalyzedFiles int           `json:"analyzed_files"`
	Duration      time.Duration `json:"duration"`
}

type BatchTraceResult struct {
	// Whether ANY variable traces back to user input
	HasUserInput bool `json:"has_user_input"`

	// Results for each variable traced
	PerVariable map[string]*BackwardTraceResult `json:"per_variable"`

	// Analysis metadata
	TotalDuration  time.Duration `json:"total_duration"`
	AnalyzedFiles  int           `json:"analyzed_files"`
	VariablesFound int           `json:"variables_found"`
}

type CallArg struct {
	Index       int         `json:"index"`
	Value       string      `json:"value"`
	Type        string      `json:"type,omitempty"`
	IsTainted   bool        `json:"is_tainted"`
	TaintSource string      `json:"taint_source,omitempty"`
	TaintChain  *TaintChain `json:"taint_chain,omitempty"`
}

type CallSite struct {
	FunctionName string    `json:"function_name"`
	ClassName    string    `json:"class_name,omitempty"`
	MethodName   string    `json:"method_name,omitempty"`
	Arguments    []CallArg `json:"arguments"`
	Line         int       `json:"line"`
	Column       int       `json:"column"`
	FilePath     string    `json:"file_path"`
	Scope        string    `json:"scope"`

	// Result assignment
	ResultVar string `json:"result_var,omitempty"`

	// Call type
	IsStatic      bool `json:"is_static"`
	IsConstructor bool `json:"is_constructor"`

	// Taint info
	HasTaintedArgs    bool  `json:"has_tainted_args"`
	TaintedArgIndices []int `json:"tainted_arg_indices,omitempty"`
}

type CarrierChain struct {
	ClassName        string   `json:"class_name"`
	PropertyName     string   `json:"property_name"`
	Initialization   string   `json:"initialization"`
	PopulationMethod string   `json:"population_method,omitempty"`
	PopulationCalls  []string `json:"population_calls,omitempty"`
	Framework        string   `json:"framework,omitempty"`
}

type CarrierInfo struct {
	PropertyName      string   `json:"property_name"`
	SourceTypes       []string `json:"source_types"`       // Which source types it carries
	PopulationMethod  string   `json:"population_method"`  // Method that populates it
	PopulationPattern string   `json:"population_pattern"` // Pattern used
	AccessPattern     string   `json:"access_pattern"`     // How to access: "array", "method", "property"
}

type ClassDef struct {
	Name     string `json:"name"`
	FilePath string `json:"file_path"`
	Line     int    `json:"line"`
	EndLine  int    `json:"end_line"`

	// Inheritance
	Extends    string   `json:"extends,omitempty"`
	Implements []string `json:"implements,omitempty"`
	Traits     []string `json:"traits,omitempty"` // PHP traits

	// Members
	Properties  map[string]*PropertyDef `json:"properties"`
	Methods     map[string]*MethodDef   `json:"methods"`
	Constructor *MethodDef              `json:"constructor,omitempty"`

	// For framework detection
	IsCarrier   bool         `json:"is_carrier"`
	CarrierInfo *CarrierInfo `json:"carrier_info,omitempty"`

	// Visibility
	Visibility string `json:"visibility"` // public, private, protected
	IsAbstract bool   `json:"is_abstract"`
	IsFinal    bool   `json:"is_final"`

	// Namespace/package
	Namespace string `json:"namespace,omitempty"`
}

type ConstantDef struct {
	Name  string `json:"name"`
	Value string `json:"value"`
	Type  string `json:"type,omitempty"`
	Line  int    `json:"line"`
}

type FlowEdge struct {
	ID   string       `json:"id"`
	From string       `json:"from"` // Source node ID
	To   string       `json:"to"`   // Target node ID
	Type FlowEdgeType `json:"type"`

	// Location where flow occurs
	FilePath string `json:"file_path"`
	Line     int    `json:"line"`

	// Human-readable description
	Description string `json:"description"`

	// Code causing the flow
	Code string `json:"code,omitempty"`

	// Additional context
	Metadata map[string]interface{} `json:"metadata,omitempty"`
}

type FlowEdgeType = constants.FlowEdgeType

type FlowMap struct {
	// Target expression being traced
	Target FlowTarget `json:"target"`

	// Ultimate sources (where data originally comes from)
	Sources []FlowNode `json:"sources"`

	// Complete paths from sources to target
	Paths []FlowPath `json:"paths"`

	// All intermediate carriers
	Carriers []FlowNode `json:"carriers"`

	// All nodes in the flow graph
	AllNodes []FlowNode `json:"all_nodes"`

	// All edges in the flow graph
	AllEdges []FlowEdge `json:"all_edges"`

	// Usage locations (where the data is used)
	Usages []FlowNode `json:"usages"`

	// Carrier chain information
	CarrierChain *CarrierChain `json:"carrier_chain,omitempty"`

	// Call graph relevant to this flow
	CallGraph map[string][]string `json:"call_graph,omitempty"`

	// Analysis metadata
	Metadata FlowMapMetadata `json:"metadata"`
	// contains filtered or unexported fields
}

type FlowMapMetadata struct {
	AnalyzedAt    time.Time `json:"analyzed_at"`
	Duration      string    `json:"duration"`
	FilesAnalyzed int       `json:"files_analyzed"`
	Language      string    `json:"language"`
	Framework     string    `json:"framework,omitempty"`
	TracerVersion string    `json:"tracer_version"`
}

type FlowNode struct {
	ID       string       `json:"id"`
	Type     FlowNodeType `json:"type"`
	Language string       `json:"language"`

	// Location information
	FilePath  string `json:"file_path"`
	Line      int    `json:"line"`
	Column    int    `json:"column"`
	EndLine   int    `json:"end_line,omitempty"`
	EndColumn int    `json:"end_column,omitempty"`

	// Semantic information
	Name       string `json:"name"`                  // Variable/function/property name
	ClassName  string `json:"class_name,omitempty"`  // If part of a class
	MethodName string `json:"method_name,omitempty"` // If inside a method
	Scope      string `json:"scope,omitempty"`       // Scope identifier

	// Type information
	TypeInfo *TypeInfo `json:"type_info,omitempty"`

	// Source information (if this is a source node)
	SourceType SourceType `json:"source_type,omitempty"`
	SourceKey  string     `json:"source_key,omitempty"` // Parameter name

	// Carrier information
	CarrierType string `json:"carrier_type,omitempty"` // "array", "object_property", etc.

	// Code snippet
	Snippet string `json:"snippet"`

	// Metadata
	Metadata map[string]interface{} `json:"metadata,omitempty"`
}

type FlowNodeType = constants.FlowNodeType

type FlowPath struct {
	ID          string     `json:"id"`
	Description string     `json:"description"`
	Steps       []FlowStep `json:"steps"`
	Source      *FlowNode  `json:"source"`
	Target      *FlowNode  `json:"target"`
}

type FlowStep struct {
	Node        FlowNode  `json:"node"`
	Edge        *FlowEdge `json:"edge,omitempty"` // Edge to next step
	Description string    `json:"description"`
	StepNumber  int       `json:"step_number"`
}

type FlowTarget struct {
	FilePath   string `json:"file_path"`
	Line       int    `json:"line"`
	Column     int    `json:"column,omitempty"`
	Expression string `json:"expression"`
}

type FrameworkPattern struct {
	ID          string `json:"id"`
	Framework   string `json:"framework"`
	Language    string `json:"language"`
	Name        string `json:"name"`
	Description string `json:"description"`

	// Pattern matching
	ClassPattern    string `json:"class_pattern,omitempty"`    // Regex for class names
	MethodPattern   string `json:"method_pattern,omitempty"`   // Regex for method names
	PropertyPattern string `json:"property_pattern,omitempty"` // Regex for property names
	AccessPattern   string `json:"access_pattern,omitempty"`   // How data is accessed

	// Source mapping
	SourceType SourceType `json:"source_type"`
	SourceKey  string     `json:"source_key,omitempty"` // How to extract the key

	// Flow information
	CarrierClass    string   `json:"carrier_class,omitempty"`
	CarrierProperty string   `json:"carrier_property,omitempty"`
	PopulatedBy     string   `json:"populated_by,omitempty"`   // Method that populates
	PopulatedFrom   []string `json:"populated_from,omitempty"` // Original sources
}

type FrameworkPatternData struct {
	ID              string
	Framework       string
	Language        string
	Name            string
	Description     string
	ClassPattern    string
	MethodPattern   string
	PropertyPattern string
	AccessPattern   string
	SourceType      string
	SourceKey       string
	CarrierClass    string
	CarrierProperty string
	PopulatedBy     string
	PopulatedFrom   []string
}

type FunctionDef struct {
	Name       string         `json:"name"`
	FilePath   string         `json:"file_path"`
	Parameters []ParameterDef `json:"parameters"`
	ReturnType string         `json:"return_type,omitempty"`
	Line       int            `json:"line"`
	EndLine    int            `json:"end_line"`
	IsExported bool           `json:"is_exported"`
	IsAsync    bool           `json:"is_async"`

	// Body information
	BodyStart  int    `json:"body_start"`
	BodyEnd    int    `json:"body_end"`
	BodySource string `json:"body_source,omitempty"`

	// Flow analysis results
	ParamsToReturn   []int               `json:"params_to_return,omitempty"`
	ReturnsInput     bool                `json:"returns_input"`
	CallsExternal    []string            `json:"calls_external,omitempty"`
	ReturnTaintChain *TaintChain         `json:"return_taint_chain,omitempty"`
	ParamTaintChains map[int]*TaintChain `json:"param_taint_chains,omitempty"`
}

type ImportInfo struct {
	Path       string   `json:"path"` // Import path/module name
	Alias      string   `json:"alias,omitempty"`
	Names      []string `json:"names,omitempty"` // Specific imports (from X import a, b)
	IsRelative bool     `json:"is_relative"`
	Line       int      `json:"line"`
	Type       string   `json:"type"` // "import", "require", "include", "use"
}

type Location struct {
	FilePath  string `json:"file_path"`
	Line      int    `json:"line"`
	Column    int    `json:"column"`
	EndLine   int    `json:"end_line,omitempty"`
	EndColumn int    `json:"end_column,omitempty"`
}

type MethodDef struct {
	Name       string         `json:"name"`
	Parameters []ParameterDef `json:"parameters"`
	ReturnType string         `json:"return_type,omitempty"`
	Line       int            `json:"line"`
	EndLine    int            `json:"end_line"`
	Visibility string         `json:"visibility"`
	IsStatic   bool           `json:"is_static"`
	IsAbstract bool           `json:"is_abstract"`
	IsAsync    bool           `json:"is_async"`

	// Body information
	BodyStart  int    `json:"body_start"`
	BodyEnd    int    `json:"body_end"`
	BodySource string `json:"body_source,omitempty"` // Actual source code

	// Flow analysis results
	ParamsToReturn []int          `json:"params_to_return,omitempty"` // Which params flow to return
	ParamsToProps  map[int]string `json:"params_to_props,omitempty"`  // Param -> property flows
	CallsInternal  []string       `json:"calls_internal,omitempty"`   // Internal method calls
	CallsExternal  []string       `json:"calls_external,omitempty"`   // External function calls
	ReturnsInput   bool           `json:"returns_input"`              // Does it return user input?

	// Annotations/decorators
	Annotations []AnnotationDef `json:"annotations,omitempty"`
}

type ObjectInstance struct {
	VariableName string                `json:"variable_name"`
	ClassName    string                `json:"class_name"`
	CreatedAt    Location              `json:"created_at"`
	Properties   map[string]*TaintInfo `json:"properties"`
	Framework    string                `json:"framework,omitempty"`
}

type ParameterDef struct {
	Name         string `json:"name"`
	Type         string `json:"type,omitempty"`
	DefaultValue string `json:"default_value,omitempty"`
	Index        int    `json:"index"`
	IsVariadic   bool   `json:"is_variadic"`
	IsReference  bool   `json:"is_reference"` // PHP &$param

	// Flow analysis
	ReceivesInput bool        `json:"receives_input"`
	InputSource   string      `json:"input_source,omitempty"`
	TaintChain    *TaintChain `json:"taint_chain,omitempty"`
}

type PropertyDef struct {
	Name         string `json:"name"`
	Type         string `json:"type,omitempty"`
	Visibility   string `json:"visibility"` // public, private, protected
	InitialValue string `json:"initial_value,omitempty"`
	Line         int    `json:"line"`
	IsStatic     bool   `json:"is_static"`
	IsReadonly   bool   `json:"is_readonly"`

	// Flow analysis results
	ReceivesInput bool     `json:"receives_input"`
	InputSources  []string `json:"input_sources,omitempty"`
	TaintDepth    int      `json:"taint_depth,omitempty"`
}

type SourceInfo struct {
	Type       SourceType `json:"type"`       // http_get, http_post, etc.
	Expression string     `json:"expression"` // e.g., "$_GET['id']"
	FilePath   string     `json:"file_path"`
	Line       int        `json:"line"`
}

type SourceType = common.SourceType

type SymbolTable struct {
	FilePath  string                  `json:"file_path"`
	Language  string                  `json:"language"`
	Imports   []ImportInfo            `json:"imports"`
	Classes   map[string]*ClassDef    `json:"classes"`
	Functions map[string]*FunctionDef `json:"functions"`
	Variables map[string]*VariableDef `json:"variables"`
	Constants map[string]*ConstantDef `json:"constants"`
	Namespace string                  `json:"namespace,omitempty"`

	// File-level metadata
	Framework string                 `json:"framework,omitempty"`
	Metadata  map[string]interface{} `json:"metadata,omitempty"`
}

type TaintChain struct {
	// Original source information
	OriginalSource string     `json:"original_source"` // e.g., "$_GET['id']"
	OriginalType   SourceType `json:"original_type"`   // e.g., "http_get"
	OriginalFile   string     `json:"original_file"`
	OriginalLine   int        `json:"original_line"`

	// Chain of transformations/assignments
	Steps []TaintStep `json:"steps"`

	// Current state
	CurrentExpression string `json:"current_expression"` // What the taint looks like now
	Depth             int    `json:"depth"`              // How many hops from source
}

type TaintInfo struct {
	Source     *FlowNode  `json:"source"`
	SourceType SourceType `json:"source_type"`
	SourceKey  string     `json:"source_key"`
	Depth      int        `json:"depth"`
	Path       []string   `json:"path"` // How taint reached this var
}

type TaintStep struct {
	StepType    string `json:"step_type"`  // "assignment", "parameter", "return", "property", "method_call"
	Expression  string `json:"expression"` // The code at this step
	FilePath    string `json:"file_path"`
	Line        int    `json:"line"`
	Description string `json:"description"` // Human-readable description
}

type TypeInfo struct {
	Name       string   `json:"name"`
	Kind       string   `json:"kind"` // "class", "interface", "primitive", "array", "map"
	Package    string   `json:"package,omitempty"`
	Generics   []string `json:"generics,omitempty"`
	IsNullable bool     `json:"is_nullable,omitempty"`
}

type VariableDef struct {
	Name         string `json:"name"`
	Type         string `json:"type,omitempty"`
	InitialValue string `json:"initial_value,omitempty"`
	Line         int    `json:"line"`
	Scope        string `json:"scope"`
	IsGlobal     bool   `json:"is_global"`
	IsConstant   bool   `json:"is_constant"`

	// Flow analysis
	IsTainted   bool        `json:"is_tainted"`
	TaintSource string      `json:"taint_source,omitempty"`
	TaintDepth  int         `json:"taint_depth,omitempty"`
	TaintChain  *TaintChain `json:"taint_chain,omitempty"`
}