wip

wip 1hit

Author: Taka-S-dev

api/handlers_analysis.go

@@ -107,7 +107,12 @@ func (h *Handler) handleDefinition(w http.ResponseWriter, r *http.Request) {
 			if len(h) == 0 && e == nil {
 				slog.Debug("definition gtags miss, fallback to rg", "word", word)
 				t0 = time.Now()
-				h, e = search.FindDefinitions(r.Context(), word, dir, glob)
+				currentFile := q.Get("file")
+				if currentFile != "" {
+					h, e = search.FindDefinitionsSmart(r.Context(), word, currentFile, hroot, glob)
+				} else {
+					h, e = search.FindDefinitions(r.Context(), word, dir, glob)
+				}
 				eng = "rg"
 				slog.Debug("definition rg fallback result", "word", word, "hits", len(h), "elapsed", time.Since(t0))
 			}

search/definition.go

@@ -2,6 +2,7 @@ package search
 
 import (
 	"context"
+	"errors"
 	"fmt"
 	"log/slog"
 	"os"
@@ -10,6 +11,8 @@ import (
 	"regexp"
 	"strings"
 	"time"
+
+	"grepnavi/graph"
 )
 
 // DefHit は定義箇所の1件。
@@ -109,7 +112,7 @@ func FindDefinitionsSmart(ctx context.Context, word, currentFile, root, glob str
 			var hits []DefHit
 			var err error
 			if dir == root {
-				hits, err = FindDefinitionsN(innerCtx, word, root, glob, 50)
+				hits, err = FindDefinitionsN(innerCtx, word, root, glob, 1)
 			} else {
 				files := listFilesInDir(dir, glob)
 				if len(files) > 0 {
@@ -279,27 +282,37 @@ func FindDefinitionsN(ctx context.Context, word, dir, glob string, maxPerQuery i
 		Regex:         true,
 		CaseSensitive: true,
 		ContextLines:  -1,
-		MaxResults:    maxPerQuery,
-	}
-	matches, err := Search(ctx, opts)
-	if err != nil {
-		return nil, err
 	}
 
+	// SearchStream を使い、実態（定義）を maxPerQuery 件見つけた時点で rg を即 kill する。
+	// 宣言（行末 ; など）はカウントせず収集のみ行う。
+	// これにより「.h の宣言を先に見つけても続行し、.c の定義で即終了」が実現できる。
 	seen := map[string]bool{}
 	var results []DefHit
-	for _, m := range matches {
+	defCount := 0
+	errDone := errors.New("done")
+	if err := SearchStream(ctx, opts, func(m graph.Match) error {
 		key := fmt.Sprintf("%s:%d", m.File, m.Line)
 		if seen[key] {
-			continue
+			return nil
 		}
 		seen[key] = true
-		results = append(results, DefHit{
+		hit := DefHit{
 			File: m.File,
 			Line: m.Line,
 			Text: strings.TrimSpace(m.Text),
 			Kind: classifyDefKind(m.Text, word),
-		})
+		}
+		results = append(results, hit)
+		if isDefinitionHit(hit) {
+			defCount++
+			if defCount >= maxPerQuery {
+				return errDone // 実態を maxPerQuery 件確認した時点で rg を kill
+			}
+		}
+		return nil
+	}); err != nil {
+		return nil, err
 	}
 
 	// K&R スタイル: 関数名と ( が別行のケース（例: "void func\n(\n..."）

search/ripgrep.go

@@ -235,6 +235,8 @@ func matchID(file string, line, col int) string {
 func SearchStream(ctx context.Context, opts Options, callback func(graph.Match) error) error {
 	if opts.ContextLines == 0 {
 		opts.ContextLines = 6
+	} else if opts.ContextLines < 0 {
+		opts.ContextLines = 0
 	}
 
 	innerCtx, cancel := context.WithCancel(ctx)