model

type AiContextFile struct {
	Language Language `json:"language"`
	Tags     []Tag    `json:"tags"`
}

type AiContextProject struct {
	// for each language, list the files for this language
	Languages map[Language]map[string]struct{} `json:"languages"`

	// the file context for each file
	FileContext map[string]AiContextFile `json:"file_context"`
	// list of tags definition for each tag
	Tags map[string][]Tag `json:"tags"`
}

type AnalysisOptions struct {
	Directory         string
	DetectionModel    Model
	ValidationModel   Model
	Debug             bool
	OpenAIBaseURL     string
	RequestTimeoutSec int
	FileConcurrency   int
	WritePrompts      bool
	IsAIGateway       bool
	AIGuardEnabled    bool
	Rules             []api.AiPrompt
	OrgID             int64
	RepositoryID      string
	SkipIndexing      bool
	DatadogDriver     *DatadogDriverConfig
}

type DatadogDriverConfig struct {
	Files map[string][]string `json:"files"`
}

type DetectionContext struct {
	ProjectContext      AiContextProject
	Language            Language
	RepositoryDirectory string
	Path                string
	Code                string
	RelatedFiles        []DetectionContextRelatedFile
	WritePrompts        bool
	Rule                api.AiPrompt

	// StrippedCode is the lowercased code with comments/docstrings stripped.
	// Pre-compute this once per file and reuse across multiple rule checks
	// to avoid redundant regex operations. If empty, it will be computed on demand.
	StrippedCode string
}

type DetectionContextRelatedFile struct {
	Path    string
	Content string
}

type DetectionPromptSystem struct {
	Language   Language
	Frameworks []string
}

type DetectionPromptUser struct {
	Language   Language
	Frameworks []string
	Libraries  []string
	Path       string
	Code       string
}

type File struct {
	Path string
	Hash string
}

type FileContent struct {
	Text     string
	Numbered string
}

type FileResult struct {
	Path           string
	Violations     []Violation
	InputTokens    int32
	OutputTokens   int32
	LLMCalls       int32
	RulesSucceeded []string
	RulesFailed    []string
}

type FunctionCall struct {
	Name     string
	Language Language
	Path     string
	Line     uint
}

type FunctionDefinition struct {
	Name      string
	Language  Language
	Path      string
	StartLine uint
	EndLine   uint
}

type LLMResult struct {
	Violations []LLMResultViolation `json:"violations"`
}

type LLMResultViolation struct {
	StartLine   uint   `json:"startLine"`
	StartColumn uint   `json:"startColumn"`
	EndLine     uint   `json:"endLine"`
	EndColumn   uint   `json:"endColumn"`
	Reason      string `json:"reason"`
}

type Language int

type MemoryType string

type Model struct {
	ID          int
	Name        string
	CLIName     string // user-facing CLI name; falls back to Name if empty
	Provider    string
	RawAPIModel string // When set, this is used directly for API calls (for custom AI Gateway models)
}

type PromptMemory struct {
	Language       Language
	TruePositives  []PromptMemoryItem
	FalsePositives []PromptMemoryItem
}

type PromptMemoryItem struct {
	MemoryType MemoryType
	Language   Language
	Code       string
	Reason     string
}

type ScanData struct {
	Model Model

	UserPrompt   string
	SystemPrompt string

	EngineVersion    string
	RelativeFilePath string
	FileHash         string

	// FileContent is shared across all ScanData built for the same file.
	FileContent *FileContent

	// Rule
	Rule *api.AiPrompt
}

type Tag struct {
	Name     string
	Path     string
	Type     TagType
	Language Language
}

type TagType int

type Violation struct {
	Rule        string
	Cwe         *string
	Path        string
	FileHash    string
	StartLine   uint
	StartColumn uint
	EndLine     uint
	EndColumn   uint
	Message     string
	Fingerprint string
}

type Vulnerability int

type VulnerabilityMemory struct {
	RuleID     string
	File       string
	Line       uint
	MemoryType MemoryType
	Reason     string
}