Function fetchFileSuggestions

pkg/suggestion/suggestion.go:360–489

(ctx context.Context, data wshrpc.FetchSuggestionsData)

}

func fetchFileSuggestions(ctx context.Context, data wshrpc.FetchSuggestionsData) (*wshrpc.FetchSuggestionsResponse, error) {
	// Only support file suggestions.
	if data.SuggestionType != "file" {
		return nil, fmt.Errorf("unsupported suggestion type: %q", data.SuggestionType)
	}

	// Resolve the base directory, query prefix (for display) and search term.
	baseDir, queryPrefix, searchTerm, err := resolveFileQuery(data.FileCwd, data.Query)
	if err != nil {
		return nil, fmt.Errorf("error resolving base dir: %w", err)
	}

	// Use a cancellable context for directory listing.
	listingCtx, cancelFn := context.WithCancel(ctx)
	defer cancelFn()

	entriesCh, err := listDirectory(listingCtx, data.WidgetId, baseDir, 1000)
	if err != nil {
		return nil, fmt.Errorf("error listing directory: %w", err)
	}

	const maxEntries = MaxSuggestions // top-k entries

	// Always use a heap.
	var topHeap scoredEntryHeap
	heap.Init(&topHeap)

	var patternRunes []rune
	if searchTerm != "" {
		patternRunes = []rune(strings.ToLower(searchTerm))
	}

	var slab util.Slab
	var index int // used for ordering when searchTerm is empty

	// Process each directory entry.
	for result := range entriesCh {
		if result.Err != nil {
			return nil, fmt.Errorf("error reading directory: %w", result.Err)
		}
		de := result.Entry
		fileName := de.Name()
		var score int
		var candidatePositions []int

		if searchTerm != "" {
			// Perform fuzzy matching.
			candidate := strings.ToLower(fileName)
			text := util.ToChars([]byte(candidate))
			matchResult, positions := algo.FuzzyMatchV2(false, true, true, &text, patternRunes, true, &slab)
			if matchResult.Score <= 0 {
				index++
				continue
			}
			score = matchResult.Score
			if positions != nil {
				candidatePositions = *positions
			}