ingest

type AnnotatingParser interface {
	Parser
	ParseWithComments(ctx context.Context, filePath string, content []byte) ([]graph.Node, []graph.Edge, []CommentBlock, error)
	Language() string
}

type BatchIncrementalSyncer interface {
	SyncBatchesWithExisting(ctx context.Context, source FileBatchSource, deletedFiles []string) (*SyncStats, error)
}

type CommentBlock struct {
	StartLine      int
	EndLine        int
	Text           string
	IsDocstring    bool
	OwnerStartLine int
}

type FileBatchSource func(visitor FileBatchVisitor) error

type FileBatchVisitor func(files map[string]FileInfo) error

type FileInfo struct {
	Hash    string
	Content []byte
	Force   bool
}

type GraphStore interface {
	GetNodesByIDs(ctx context.Context, ids []uint) ([]graph.Node, error)
	GetNodesByFile(ctx context.Context, filePath string) ([]graph.Node, error)
	GetNodesByFiles(ctx context.Context, filePaths []string) (map[string][]graph.Node, error)
	GetNodesByQualifiedNames(ctx context.Context, names []string) (map[string][]graph.Node, error)
	ListFileNodes(ctx context.Context) ([]graph.Node, error)
	ListImportFileNodes(ctx context.Context) ([]graph.Node, error)
	GetFileNodesByPathSuffix(ctx context.Context, suffix string) ([]graph.Node, error)
	GetEdgesFromNodes(ctx context.Context, nodeIDs []uint) ([]graph.Edge, error)
	GetEdgesToNodes(ctx context.Context, nodeIDs []uint) ([]graph.Edge, error)
	UpsertNodes(ctx context.Context, nodes []graph.Node) error
	UpsertEdges(ctx context.Context, edges []graph.Edge) error
	UpsertAnnotations(ctx context.Context, annotations []*graph.Annotation) error
	DeleteNodesByFiles(ctx context.Context, filePaths []string) error
	DeleteEdgesByFile(ctx context.Context, filePath string) error
	DeletePackageSemanticEdges(ctx context.Context, anchorFiles []string) error
	DeleteGraph(ctx context.Context) error
}

type IncrementalSyncer interface {
	SyncWithExisting(ctx context.Context, files map[string]FileInfo, existingFiles []string) (*SyncStats, error)
}

type MetadataParser interface {
	AnnotatingParser
	ParseWithCommentsAndMetadata(ctx context.Context, filePath string, content []byte) ([]graph.Node, []graph.Edge, []CommentBlock, ParseMetadata, error)
}

type PackageContext struct {
	Package        string
	Language       string
	Files          []string
	Nodes          []graph.Node
	Interfaces     []PackageInterfaceInfo
	ImportPackages map[string]string
}

type PackageDiscoverer interface {
	DiscoverPackages(ctx context.Context, opts PackageDiscoveryOptions) (map[string]PackageInfo, error)
}

type PackageDiscoveryOptions struct {
	RootDir   string
	WalkFiles func(func(path, relPath string) error) error
	HasParser func(ext string) bool
}

type PackageEdgeBuilder interface {
	PackageEdges(ctx PackageContext) []graph.Edge
}

type PackageInfo struct {
	ImportPath string
	Name       string
	Dir        string
	Language   string
	Files      []string
}

type PackageInterfaceInfo struct {
	Name    string
	Methods []string
}

type ParseCache interface {
	LoadParseResult(ctx context.Context, key ParseCacheKey) ([]byte, bool, error)
	StoreParseResult(ctx context.Context, key ParseCacheKey, payload []byte) error
}

type ParseCacheKey struct {
	FilePath      string
	SourceHash    string
	ParserVersion string
	ContextHash   string
}

type ParseMetadata struct {
	Package    string
	Interfaces []PackageInterfaceInfo
}

type Parser interface {
	Parse(filePath string, content []byte) ([]graph.Node, []graph.Edge, error)
	ParseWithContext(ctx context.Context, filePath string, content []byte) ([]graph.Node, []graph.Edge, error)
}

type SearchWriter interface {
	RebuildAll(ctx context.Context) error
	RebuildNodes(ctx context.Context, nodeIDs []uint) error
}

type SyncStats struct {
	Added      int
	Modified   int
	Skipped    int
	Deleted    int
	Unresolved resolve.FilterResolvedDiagnostics
}

type Transaction interface {
	Graph() GraphStore
	Search() SearchWriter
}

type TransactionalBatchIncrementalSyncer interface {
	SyncBatchesWithExistingStore(ctx context.Context, graphStore GraphStore, source FileBatchSource, deletedFiles []string) (*SyncStats, error)
}

type TransactionalIncrementalSyncer interface {
	SyncWithExistingStore(ctx context.Context, graphStore GraphStore, files map[string]FileInfo, existingFiles []string) (*SyncStats, error)
}

type UnitOfWork interface {
	WithinTransaction(ctx context.Context, fn func(Transaction) error) error
}

type UnresolvedEdgeStore interface {
	UpsertUnresolvedEdges(ctx context.Context, candidates []graph.UnresolvedEdgeCandidate) error
	FindUnresolvedEdgesByLookupKeys(ctx context.Context, keys []string) ([]graph.Edge, error)
	FindUnresolvedEdgesByFiles(ctx context.Context, filePaths []string) ([]graph.Edge, error)
	DeleteUnresolvedEdgesByFingerprints(ctx context.Context, fingerprints []string) error
	UnresolvedIndexReady(ctx context.Context, version string) (bool, error)
	MarkUnresolvedIndexReady(ctx context.Context, version string) error
}

type VersionedParser interface {
	Parser
	ParseCacheVersion() string
}