package task
import "strings"
// Operation represents the type of task operation.
type Operation string
// Operation values for the task queue system.
const (
OperationRoot Operation = "kodit.root"
OperationCreateIndex Operation = "kodit.index.create"
OperationRunIndex Operation = "kodit.index.run"
OperationRefreshWorkingCopy Operation = "kodit.index.run.refresh_working_copy"
OperationDeleteOldSnippets Operation = "kodit.index.run.delete_old_snippets"
OperationExtractSnippets Operation = "kodit.index.run.extract_snippets"
OperationCreateBM25Index Operation = "kodit.index.run.create_bm25_index"
OperationCreateCodeEmbeddings Operation = "kodit.index.run.create_code_embeddings"
OperationEnrichSnippets Operation = "kodit.index.run.enrich_snippets"
OperationCreateTextEmbeddings Operation = "kodit.index.run.create_text_embeddings"
OperationUpdateIndexTimestamp Operation = "kodit.index.run.update_index_timestamp"
OperationClearFileProcessingStatuses Operation = "kodit.index.run.clear_file_processing_statuses"
OperationRepository Operation = "kodit.repository"
OperationCreateRepository Operation = "kodit.repository.create"
OperationDeleteRepository Operation = "kodit.repository.delete"
OperationCloneRepository Operation = "kodit.repository.clone"
OperationSyncRepository Operation = "kodit.repository.sync"
OperationCommit Operation = "kodit.commit"
OperationExtractSnippetsForCommit Operation = "kodit.commit.extract_snippets"
OperationCreateBM25IndexForCommit Operation = "kodit.commit.create_bm25_index"
OperationCreateCodeEmbeddingsForCommit Operation = "kodit.commit.create_code_embeddings"
OperationCreateSummaryEnrichmentForCommit Operation = "kodit.commit.create_summary_enrichment"
OperationCreateSummaryEmbeddingsForCommit Operation = "kodit.commit.create_summary_embeddings"
OperationCreateArchitectureEnrichmentForCommit Operation = "kodit.commit.create_architecture_enrichment"
OperationCreatePublicAPIDocsForCommit Operation = "kodit.commit.create_public_api_docs"
OperationCreateCommitDescriptionForCommit Operation = "kodit.commit.create_commit_description"
OperationCreateDatabaseSchemaForCommit Operation = "kodit.commit.create_database_schema"
OperationCreateCookbookForCommit Operation = "kodit.commit.create_cookbook"
OperationExtractExamplesForCommit Operation = "kodit.commit.extract_examples"
OperationCreateExampleSummaryForCommit Operation = "kodit.commit.create_example_summary"
OperationCreateExampleCodeEmbeddingsForCommit Operation = "kodit.commit.create_example_code_embeddings"
OperationCreateExampleSummaryEmbeddingsForCommit Operation = "kodit.commit.create_example_summary_embeddings"
OperationScanCommit Operation = "kodit.commit.scan"
OperationRescanCommit Operation = "kodit.commit.rescan"
)
// String returns the string representation of the operation.
func (o Operation) String() string {
return string(o)
}
// IsRepositoryOperation returns true if this is a repository-level operation.
func (o Operation) IsRepositoryOperation() bool {
return strings.HasPrefix(string(o), "kodit.repository.")
}
// IsCommitOperation returns true if this is a commit-level operation.
func (o Operation) IsCommitOperation() bool {
return strings.HasPrefix(string(o), "kodit.commit.")
}
// PrescribedOperations provides predefined operation sequences for common workflows.
type PrescribedOperations struct{}
// All returns every operation that appears in any prescribed workflow.
// Used at startup to validate that all required handlers are registered.
func (PrescribedOperations) All() []Operation {
seen := make(map[Operation]struct{})
var all []Operation
for _, ops := range [][]Operation{
PrescribedOperations{}.CreateNewRepository(),
PrescribedOperations{}.SyncRepository(),
PrescribedOperations{}.ScanAndIndexCommit(),
PrescribedOperations{}.IndexCommit(),
PrescribedOperations{}.RescanCommit(),
} {
for _, op := range ops {
if _, ok := seen[op]; !ok {
seen[op] = struct{}{}
all = append(all, op)
}
}
}
return all
}
// CreateNewRepository returns the operations needed to create a new repository.
func (PrescribedOperations) CreateNewRepository() []Operation {
return []Operation{
OperationCloneRepository,
}
}
// SyncRepository returns the operations needed to sync a repository.
func (PrescribedOperations) SyncRepository() []Operation {
return []Operation{
OperationCloneRepository,
OperationSyncRepository,
}
}
// ScanAndIndexCommit returns the full operation sequence for scanning and indexing a commit.
func (PrescribedOperations) ScanAndIndexCommit() []Operation {
return []Operation{
OperationScanCommit,
OperationExtractSnippetsForCommit,
OperationExtractExamplesForCommit,
OperationCreateBM25IndexForCommit,
OperationCreateCodeEmbeddingsForCommit,
OperationCreateExampleCodeEmbeddingsForCommit,
OperationCreateSummaryEnrichmentForCommit,
OperationCreateExampleSummaryForCommit,
OperationCreateSummaryEmbeddingsForCommit,
OperationCreateExampleSummaryEmbeddingsForCommit,
OperationCreateArchitectureEnrichmentForCommit,
OperationCreatePublicAPIDocsForCommit,
OperationCreateCommitDescriptionForCommit,
OperationCreateDatabaseSchemaForCommit,
OperationCreateCookbookForCommit,
}
}
// IndexCommit returns the operation sequence for indexing an already-scanned commit.
func (PrescribedOperations) IndexCommit() []Operation {
return []Operation{
OperationExtractSnippetsForCommit,
OperationCreateBM25IndexForCommit,
OperationCreateCodeEmbeddingsForCommit,
OperationCreateSummaryEnrichmentForCommit,
OperationCreateSummaryEmbeddingsForCommit,
OperationCreateArchitectureEnrichmentForCommit,
OperationCreatePublicAPIDocsForCommit,
OperationCreateCommitDescriptionForCommit,
OperationCreateDatabaseSchemaForCommit,
OperationCreateCookbookForCommit,
}
}
// RescanCommit returns the operation sequence for rescanning a commit (full reindex).
func (PrescribedOperations) RescanCommit() []Operation {
return []Operation{
OperationRescanCommit,
OperationExtractSnippetsForCommit,
OperationExtractExamplesForCommit,
OperationCreateBM25IndexForCommit,
OperationCreateCodeEmbeddingsForCommit,
OperationCreateExampleCodeEmbeddingsForCommit,
OperationCreateSummaryEnrichmentForCommit,
OperationCreateExampleSummaryForCommit,
OperationCreateSummaryEmbeddingsForCommit,
OperationCreateExampleSummaryEmbeddingsForCommit,
OperationCreateArchitectureEnrichmentForCommit,
OperationCreatePublicAPIDocsForCommit,
OperationCreateCommitDescriptionForCommit,
OperationCreateDatabaseSchemaForCommit,
OperationCreateCookbookForCommit,
}
}
