incremental_commit.go

package codetrip

import (
	"context"
	"fmt"

	"github.com/mengshi02/codetrip/internal/graph"
	inc "github.com/mengshi02/codetrip/internal/incremental"
	"github.com/mengshi02/codetrip/internal/search/semantic"
	"github.com/mengshi02/codetrip/internal/search/source"
	"github.com/mengshi02/codetrip/internal/search/symbol"
	"github.com/mengshi02/codetrip/internal/store"
)

const indexStateMetadata = "index-state"

func newRepositoryCommitter(
	database *store.Store,
	graphStore *graph.GraphStore,
	lexical *symbol.LexicalIndex,
	content *source.Index,
	vector *semantic.VectorSearch,
	repoRoot string,
) *inc.RecoverableCommitter {
	state := inc.ComponentFunc{ComponentName: "index-state", ApplyDelta: func(_ context.Context, delta *inc.Delta) error {
		encoded := delta.Metadata[indexStateMetadata]
		if len(encoded) == 0 {
			return fmt.Errorf("incremental delta has no index state")
		}
		return writeIndexStateBytes(repoRoot, graphStore.Repo(), encoded)
	}}
	components := []inc.Component{
		inc.NewSymbolComponent(lexical),
		inc.NewSourceComponent(content),
	}
	if vector != nil {
		components = append(components, inc.NewVectorComponent(vector))
	}
	components = append(components, state)
	return inc.NewRecoverableCommitter(
		inc.NewStoreJournal(database), inc.NewGraphAuthority(graphStore), components...,
	)
}