store

type BatchItem struct {
	Index      *FileIndex
	Symbols    []Symbol
	Calls      []Call
	Flows      []Flow
	Violations []Violation
}

type Call struct {
	ID         int    `json:"id"`
	CallerName string `json:"caller_name"`
	CalleeName string `json:"callee_name"`
	Path       string `json:"path"`
	Line       int    `json:"line"`
	CalleePath string `json:"callee_path"`
	LinkType   string `json:"link_type"`
	Indegree   int    `json:"indegree"`
	Body       string `json:"body"`
}

type CriticalSymbol struct {
	Symbol
	Centrality int `json:"centrality"`
	Fragility  int `json:"fragility"`
}

type DBTX interface {
	ExecContext(context.Context, string, ...interface{}) (sql.Result, error)
	PrepareContext(context.Context, string) (*sql.Stmt, error)
	QueryContext(context.Context, string, ...interface{}) (*sql.Rows, error)
	QueryRowContext(context.Context, string, ...interface{}) *sql.Row
}

type DiagnosticStore interface {
	GetAllFailedTests(ctx context.Context) iter.Seq2[types.TestResult, error]
	SaveTestResult(ctx context.Context, res *types.TestResult) error
	GetHealthReport(ctx context.Context, symbol string, failuresOnly bool) iter.Seq2[types.TestResult, error]
	ClearTestResults(ctx context.Context) error
	SaveViolation(ctx context.Context, v *types.ASTRuleMatch) error
	GetViolationsByFile(ctx context.Context, path string) ([]Violation, error)
}

type FileIndex struct {
	Path      string `json:"path"`
	Mtime     int    `json:"mtime"`
	Hash      string `json:"hash"`
	AstJson   string `json:"ast_json"`
	Project   string `json:"project"`
	Freshness int    `json:"freshness"`
}

type Flow struct {
	Source string `json:"source"`
	Sink   string `json:"sink"`
	Type   string `json:"type"` // assignment, parameter, return
	Path   string `json:"path"`
	Line   int    `json:"line"`
}

type GetCallersParams struct {
	CalleeName string `json:"callee_name"`
	Limit      int64  `json:"limit"`
	Offset     int64  `json:"offset"`
}

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

type SaveCallParams struct {
	CallerName string `json:"caller_name"`
	CalleeName string `json:"callee_name"`
	Path       string `json:"path"`
	Line       int    `json:"line"`
	CalleePath string `json:"callee_path"`
	LinkType   string `json:"link_type"`
	Indegree   int    `json:"indegree"`
	Body       string `json:"body"`
}

type SaveFileIndexParams struct {
	Path      string `json:"path"`
	Mtime     int    `json:"mtime"`
	Hash      string `json:"hash"`
	AstJson   string `json:"ast_json"`
	Project   string `json:"project"`
	Freshness int    `json:"freshness"`
}

type SaveSymbolParams struct {
	Name           string  `json:"name"`
	Type           string  `json:"type"`
	PackagePath    string  `json:"package_path"`
	ReceiverType   string  `json:"receiver_type"`
	Signature      string  `json:"signature"`
	Doc            string  `json:"doc"`
	Path           string  `json:"path"`
	StartByte      int     `json:"start_byte"`
	EndByte        int     `json:"end_byte"`
	StartLine      int     `json:"start_line"`
	StartCol       int     `json:"start_col"`
	EndLine        int     `json:"end_line"`
	StructuralHash string  `json:"structural_hash"`
	Indegree       int     `json:"indegree"`
	Relevance      float64 `json:"relevance"`
	Pagerank       float64 `json:"pagerank"`
	ChurnScore     float64 `json:"churn_score"`
	RuntimeHits    int     `json:"runtime_hits"`
	AiSummary      string  `json:"ai_summary"`
}

type SearchSymbolsParams struct {
	Column1 sql.NullString `json:"column_1"`
	Column2 sql.NullString `json:"column_2"`
	Limit   int64          `json:"limit"`
	Offset  int64          `json:"offset"`
}

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

type SovereignDelta struct {
	Path    string   `json:"path"`
	Hash    string   `json:"hash"`
	Symbols []Symbol `json:"symbols"`
	Calls   []Call   `json:"calls"`
	Flows   []Flow   `json:"flows"`
}

type Store interface {
	SymbolRegistry
	StructuralGraph
	DiagnosticStore
	SyncEngine
	TransactionManager
}

type StructuralGraph interface {
	GetAllCalls(ctx context.Context) iter.Seq2[Call, error]
	SaveCall(ctx context.Context, call Call) error
	SaveCallsBulk(ctx context.Context, calls []Call) error
	GetCallers(ctx context.Context, calleeName string, limit, offset int) ([]Call, error)
	GetCallees(ctx context.Context, callerName string) ([]Call, error)
	GetCallersRecursive(ctx context.Context, name, path string, maxDepth int) ([]Call, error)
	GetAffectedTestsRecursive(ctx context.Context, name, path string) ([]Symbol, error)
	ClearCalls(ctx context.Context, path string) error

	// Data Flow
	SaveFlow(ctx context.Context, flow Flow) error
	GetFlows(ctx context.Context, sink string) ([]Flow, error)
	ClearFlows(ctx context.Context, path string) error
}

type Symbol struct {
	ID             int                    `json:"id"`
	Name           string                 `json:"name"`
	Type           string                 `json:"type"`
	PackagePath    string                 `json:"package_path"`
	ReceiverType   string                 `json:"receiver_type"`
	Signature      string                 `json:"signature"`
	Doc            string                 `json:"doc"`
	Path           string                 `json:"path"`
	StartByte      int                    `json:"start_byte"`
	EndByte        int                    `json:"end_byte"`
	StartLine      int                    `json:"start_line"`
	StartCol       int                    `json:"start_col"`
	EndLine        int                    `json:"end_line"`
	StructuralHash string                 `json:"structural_hash"`
	Indegree       int                    `json:"indegree"`
	Relevance      float64                `json:"relevance"`
	Pagerank       float64                `json:"pagerank"`
	ChurnScore     float64                `json:"churn_score"`
	RuntimeHits    int                    `json:"runtime_hits"`
	AiSummary      string                 `json:"ai_summary"`
	Metrics        *types.SemanticMetrics `json:"metrics,omitempty"`
}

type SymbolRegistry interface {
	GetFileIndex(ctx context.Context, path string) (*FileIndex, error)
	SaveFileIndex(ctx context.Context, idx *FileIndex) error
	SaveFileIndexBatch(ctx context.Context, items []BatchItem) error
	GetDirectoryHash(ctx context.Context, path string) (string, int64, error)
	SaveDirectoryHash(ctx context.Context, path string, hash string, mtime int64) error
	ClearSymbols(ctx context.Context, path string) error
	SaveSymbol(ctx context.Context, sym *Symbol) error
	SaveSymbolsBulk(ctx context.Context, symbols []*Symbol) error
	SearchSymbols(ctx context.Context, query string, symType string, limit, offset int) ([]Symbol, error)
	GetSymbolsByNameInFile(ctx context.Context, name, path string) ([]Symbol, error)
	GetSymbolsByStructuralHash(ctx context.Context, hash string) ([]Symbol, error)
	SearchSymbolsWeighted(ctx context.Context, query string, symType string) iter.Seq2[Symbol, error]
	GetSymbolsByRange(ctx context.Context, path string, startLine, endLine int) ([]Symbol, error)
	GetSymbolsByPathPrefix(ctx context.Context, pathPrefix string) ([]Symbol, error)
	GetSymbolsByType(ctx context.Context, symType string) ([]Symbol, error)
	GetInterfaces(ctx context.Context) ([]Symbol, error)
	GetAllSymbols(ctx context.Context) iter.Seq2[Symbol, error]
	UpdateSymbolCentrality(ctx context.Context, name, path string, centrality int) error
	UpdateSymbolCentralitiesBulk(ctx context.Context, updates map[string]int) error
	RecomputeIndegrees(ctx context.Context) error
	UpdateSymbolChurn(ctx context.Context, path string, score float64) error
	UpdateSymbolPagerank(ctx context.Context, name, path string, score float64) error
	UpdateSymbolPageranksBulk(ctx context.Context, updates map[string]float64) error
	UpdateSymbolRuntimeHits(ctx context.Context, name, path string, hits int) error
	RecordSymbolUsage(ctx context.Context, env string, usages []UsageRecord) error
	GetStats(ctx context.Context) (int, int, error)
	GetAllFilePaths(ctx context.Context) ([]string, error)
	DeleteFileIndex(ctx context.Context, path string) error
	InsertSemanticVector(ctx context.Context, symbolID int64, embedding []float32) error
	SearchSemantic(ctx context.Context, queryEmbedding []float32, limit int) ([]Symbol, error)
	SearchHybrid(ctx context.Context, textQuery string, queryEmbedding []float32, limit int) ([]Symbol, error)
	SaveDependency(ctx context.Context, dep *types.Dependency) error
	GetDependencies(ctx context.Context) ([]types.Dependency, error)
	ClearDependencies(ctx context.Context) error
	GetUnusedSymbols(ctx context.Context, includeExported bool) ([]Symbol, error)
}

type SyncEngine interface {
	ExportDelta(ctx context.Context, syncDir string) error
	ImportDelta(ctx context.Context, syncDir string) error
}

type TransactionManager interface {
	WithTransaction(ctx context.Context, fn func(context.Context, Store) error) error
	Close() error
}

type UsageRecord struct {
	SymbolName string `json:"symbol_name"`
	SymbolPath string `json:"symbol_path"`
	HitCount   int    `json:"hit_count"`
	LastUsed   int64  `json:"last_used"`
}

type Violation struct {
	ID        int    `json:"id"`
	RuleID    string `json:"rule_id"`
	FilePath  string `json:"file_path"`
	Message   string `json:"message"`
	Severity  string `json:"severity"`
	StartLine int    `json:"start_line"`
	StartCol  int    `json:"start_col"`
	Text      string `json:"text"`
}