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
	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
	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
	UpdateSymbolChurn(ctx context.Context, path string, score float64) error
	UpdateSymbolPagerank(ctx context.Context, name, path string, score 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)
	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"`
}