Help us shape the future of Go! Take the 2026 Go Developer Survey

code

package
v0.609.1 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Jul 13, 2026 License: MIT Imports: 22 Imported by: 0

Documentation

Overview

Package code provides tree-sitter based code indexing with semantic search.

Package code provides code indexing with tree-sitter and semantic search.

Index

Constants

View Source 
const (
	EdgeImports = treesitter.EdgeImports
	EdgeCalls   = treesitter.EdgeCalls
	EdgeTypeRef = treesitter.EdgeTypeRef
)

Re-export edge-type constants.

View Source 
const (
	IndexingStatusPending    = treesitter.IndexingStatusPending
	IndexingStatusInProgress = treesitter.IndexingStatusInProgress
	IndexingStatusCompleted  = treesitter.IndexingStatusCompleted
	IndexingStatusFailed     = treesitter.IndexingStatusFailed
	IndexingStatusCancelled  = treesitter.IndexingStatusCancelled
)

Re-export constants

Variables

This section is empty.

Functions

This section is empty.

Types

type CodeEdge added in v0.608.1 

type CodeEdge = treesitter.CodeEdge

Re-export edge types for convenience.

type CodeFile 

type CodeFile = treesitter.CodeFile

Re-export treesitter types for convenience

type CodeIndexer 

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

CodeIndexer manages code indexing projects using tree-sitter for parsing and embeddings for semantic search.

func NewCodeIndexer 

func NewCodeIndexer(db *sql.DB, embedder embeddings.Embedder, workers int) *CodeIndexer

NewCodeIndexer creates a new CodeIndexer with the given number of concurrent workers. If workers is <= 0, the defaultWorkers value is used.

func (*CodeIndexer) DeleteFile added in v0.320.0 

func (c *CodeIndexer) DeleteFile(ctx context.Context, projectID, filePath string) error

DeleteFile removes a single indexed file and all related indexed symbols.

func (*CodeIndexer) DeleteFileDirect added in v0.326.0 

func (c *CodeIndexer) DeleteFileDirect(ctx context.Context, projectID, filePath string) error

DeleteFileDirect removes a single indexed file and all related symbols. Called by the primary IPC dispatcher when a secondary forwards a CodeDeleteFile write.

func (*CodeIndexer) DeleteProject added in v0.162.0 

func (c *CodeIndexer) DeleteProject(ctx context.Context, projectID string) error

DeleteProject removes an indexed project and all related indexed data.

func (*CodeIndexer) DeleteProjectDirect added in v0.326.0 

func (c *CodeIndexer) DeleteProjectDirect(ctx context.Context, projectID string) error

DeleteProjectDirect removes an indexed project and all related data. Called by the primary IPC dispatcher when a secondary forwards a CodeDeleteProject write.

func (*CodeIndexer) FindReferences 

func (c *CodeIndexer) FindReferences(ctx context.Context, projectID, symbolID, symbolName string, limit int) ([]*CodeSymbol, error)

FindReferences finds all symbols that reference the given symbol by name.

func (*CodeIndexer) FindSymbol 

func (c *CodeIndexer) FindSymbol(ctx context.Context, query SymbolQuery) ([]*CodeSymbol, error)

FindSymbol finds symbols matching the given query criteria.

func (*CodeIndexer) GetFileSymbols 

func (c *CodeIndexer) GetFileSymbols(ctx context.Context, projectID, filePath string, includeBody bool) ([]*CodeSymbol, error)

GetFileSymbols returns all symbols for a file.

func (*CodeIndexer) GetJob 

func (c *CodeIndexer) GetJob(jobID string) (*IndexingJob, bool)

GetJob returns the current status of an indexing job.

func (*CodeIndexer) GetProjectStats 

func (c *CodeIndexer) GetProjectStats(ctx context.Context, projectID string) (map[string]interface{}, error)

GetProjectStats returns statistics for a project.

func (*CodeIndexer) GetSymbolsOverview 

func (c *CodeIndexer) GetSymbolsOverview(ctx context.Context, projectID, filePath string, maxResults int) ([]*CodeSymbol, error)

GetSymbolsOverview returns a high-level overview of symbols in a file.

func (*CodeIndexer) GraphEnabled added in v0.608.1 

func (c *CodeIndexer) GraphEnabled() bool

GraphEnabled reports whether property-graph edge extraction is active.

func (*CodeIndexer) HasProject added in v0.320.0 

func (c *CodeIndexer) HasProject(ctx context.Context, projectID, rootPath string) (bool, error)

HasProject reports whether a project is already registered for code indexing and whether the stored root path matches the requested one.

func (*CodeIndexer) HybridSearch 

func (c *CodeIndexer) HybridSearch(ctx context.Context, projectID, query string, limit int, langs []Language, symbolTypes []SymbolType) ([]HybridSearchResult, error)

HybridSearch performs hybrid vector + FTS search over code symbols.

func (*CodeIndexer) ImpactAnalysis added in v0.608.1 

func (c *CodeIndexer) ImpactAnalysis(ctx context.Context, projectID, symbolName string, maxDepth, limit int) (*ImpactResult, error)

ImpactAnalysis returns the symbols that call (directly or transitively, up to maxDepth) a symbol named symbolName — i.e. "what breaks if I change this".

Resolution is name-based via the `calls` edges, which is cheap and works cross-file/cross-package but is approximate when a name is shared by unrelated symbols. Results are deduplicated by caller symbol and capped at limit.

func (*CodeIndexer) IndexFileDirect added in v0.326.0 

func (c *CodeIndexer) IndexFileDirect(ctx context.Context, projectID, filePath, language, fileHash string, symbolsJSON, edgesJSON json.RawMessage) error

IndexFileDirect inserts or updates a file and its symbols, then generates embeddings. Called by the primary IPC dispatcher when a secondary forwards a CodeIndexFile write. symbolsJSON is the JSON-encoded []*treesitter.CodeSymbol payload from the RPC request. Symbol embeddings are generated on the primary using its configured embedder.

func (*CodeIndexer) IndexProject 

func (c *CodeIndexer) IndexProject(ctx context.Context, projectID, projectPath string, languages []Language) (string, error)

IndexProject indexes all supported source files in a project directory. It runs asynchronously and updates the job status in the jobs map. Returns the job ID immediately.

func (*CodeIndexer) ListIndexedFiles added in v0.418.6 

func (c *CodeIndexer) ListIndexedFiles(ctx context.Context, rootPath, query string, limit int) ([]string, error)

ListIndexedFiles returns up to limit indexed file paths (relative to the project root) for the project whose root_path matches rootPath. When query is non-empty it is used as a substring pre-filter (SQL LIKE) to keep the result small. It returns (nil, nil) when no project is indexed for rootPath, so the caller can transparently fall back to a filesystem scan.

func (*CodeIndexer) ListProjects 

func (c *CodeIndexer) ListProjects(ctx context.Context) ([]*CodeProject, error)

ListProjects returns all indexed projects.

func (*CodeIndexer) ReindexFile 

func (c *CodeIndexer) ReindexFile(ctx context.Context, projectID, filePath string) error

ReindexFile re-indexes a single file in a project.

func (*CodeIndexer) RelatedFiles added in v0.608.1 

func (c *CodeIndexer) RelatedFiles(ctx context.Context, projectID, filePath string, limit int) (*RelatedResult, error)

RelatedFiles returns files related to filePath, ranked by a weighted blend of resolved imports and bidirectional call coupling. Import resolution is best-effort: relative ESM specifiers are resolved against the project's file list; bare/package import paths rely on call coupling instead.

func (*CodeIndexer) SearchPattern 

func (c *CodeIndexer) SearchPattern(ctx context.Context, projectID, pattern string, caseSensitive bool, isRegex bool, limit int, langs []Language, symbolTypes []SymbolType) ([]*CodeSymbol, error)

SearchPattern searches for text patterns in code symbols.

func (*CodeIndexer) SetGraphEnabled added in v0.608.1 

func (c *CodeIndexer) SetGraphEnabled(enabled bool)

SetGraphEnabled toggles code property-graph edge extraction during indexing (lean-ctx Phase 4). When false, no `code_edges` rows are written, leaving the impact/related tools to operate on whatever edges already exist.

func (*CodeIndexer) SetProjectStatusDirect added in v0.326.0 

func (c *CodeIndexer) SetProjectStatusDirect(ctx context.Context, projectID, status string, lastIndexedAt *time.Time, updatedAt time.Time) error

SetProjectStatusDirect updates the indexing status of an existing project. Called by the primary IPC dispatcher when a secondary forwards a CodeSetProjectStatus write.

func (*CodeIndexer) SetWriteProxy added in v0.326.0 

func (c *CodeIndexer) SetWriteProxy(proxy *dbproxy.DBProxy)

SetWriteProxy configures a DB proxy for mutating operations.

func (*CodeIndexer) UpdateLanguageStatsDirect added in v0.326.0 

func (c *CodeIndexer) UpdateLanguageStatsDirect(ctx context.Context, projectID string) error

UpdateLanguageStatsDirect recomputes and persists language statistics for a project. Called by the primary IPC dispatcher when a secondary forwards a CodeUpdateLanguageStats write.

func (*CodeIndexer) UpsertProjectDirect added in v0.326.0 

func (c *CodeIndexer) UpsertProjectDirect(ctx context.Context, projectID, name, rootPath, status string, createdAt, updatedAt time.Time) error

UpsertProjectDirect inserts or updates a code project record. Called by the primary IPC dispatcher when a secondary forwards a CodeUpsertProject write.

type CodeProject 

type CodeProject = treesitter.CodeProject

Re-export treesitter types for convenience

type CodeSymbol 

type CodeSymbol = treesitter.CodeSymbol

Re-export treesitter types for convenience

type EdgeType added in v0.608.1 

type EdgeType = treesitter.EdgeType

Re-export edge types for convenience.

type HybridSearchResult 

type HybridSearchResult struct {
	Symbol      *CodeSymbol `json:"symbol"`
	Score       float64     `json:"score"`
	VectorScore float64     `json:"vector_score,omitempty"`
	FTSScore    float64     `json:"fts_score,omitempty"`
	Rank        int         `json:"rank"`
}

HybridSearchResult represents a result from hybrid search

type ImpactCaller added in v0.608.1 

type ImpactCaller struct {
	Name       string `json:"name"`
	NamePath   string `json:"name_path,omitempty"`
	FilePath   string `json:"file_path"`
	StartLine  int    `json:"start_line"`
	SymbolType string `json:"symbol_type"`
	Depth      int    `json:"depth"`
}

ImpactCaller is a symbol that (transitively) depends on the analyzed symbol.

type ImpactResult added in v0.608.1 

type ImpactResult struct {
	Symbol    string         `json:"symbol"`
	Callers   []ImpactCaller `json:"callers"`
	Truncated bool           `json:"truncated"`
}

ImpactResult is the outcome of a reverse-call impact analysis.

type IndexingJob 

type IndexingJob = treesitter.IndexingJob

Re-export treesitter types for convenience

type IndexingStatus 

type IndexingStatus = treesitter.IndexingStatus

Re-export treesitter types for convenience

type Language 

type Language = treesitter.Language

Re-export treesitter types for convenience

type ParseResult 

type ParseResult = treesitter.ParseResult

Re-export treesitter types for convenience

type RelatedFile added in v0.608.1 

type RelatedFile struct {
	FilePath string   `json:"file_path"`
	Score    float64  `json:"score"`
	Reasons  []string `json:"reasons"`
}

RelatedFile is a file related to the queried file with a weighted score.

type RelatedResult added in v0.608.1 

type RelatedResult struct {
	File      string        `json:"file"`
	Related   []RelatedFile `json:"related"`
	Truncated bool          `json:"truncated"`
}

RelatedResult is the outcome of a related-files query.

type SemanticSearchQuery 

type SemanticSearchQuery = treesitter.SemanticSearchQuery

Re-export treesitter types for convenience

type SemanticSearchResult 

type SemanticSearchResult = treesitter.SemanticSearchResult

Re-export treesitter types for convenience

type SymbolQuery 

type SymbolQuery = treesitter.SymbolQuery

Re-export treesitter types for convenience

type SymbolType 

type SymbolType = treesitter.SymbolType

Re-export treesitter types for convenience

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.