Fix missing RepoURLs and LineFragments for result subrepositories

RepoURLs for subrepositories are currently filtered out when results
are streamed per-repo from shards.

Author: H3mul

search/shards.go

@@ -816,15 +816,33 @@ func sendByRepository(result *zoekt.SearchResult, opts *zoekt.SearchOptions, sen
 
 	send := func(repoName string, a, b int, stats zoekt.Stats) {
 		index.SortFiles(result.Files[a:b])
+
+		// Filter RepoURLs and LineFragments to only those of repoName and its subRepositories
+		repoNames := []string{repoName}
+		for _, file := range result.Files[a:b] {
+			repoNames = append(repoNames, file.SubRepositoryName)
+		}
+
+		filteredRepoURLs := make(map[string]string)
+		filteredLineFragments := make(map[string]string)
+		for _, name := range repoNames {
+			if url, ok := result.RepoURLs[name]; ok {
+				filteredRepoURLs[name] = url
+			}
+			if frag, ok := result.LineFragments[name]; ok {
+				filteredLineFragments[name] = frag
+			}
+		}
+
 		sender.Send(&zoekt.SearchResult{
 			Stats: stats,
 			Progress: zoekt.Progress{
 				Priority:           result.Files[a].RepositoryPriority,
 				MaxPendingPriority: result.MaxPendingPriority,
 			},
 			Files:         result.Files[a:b],
-			RepoURLs:      map[string]string{repoName: result.RepoURLs[repoName]},
-			LineFragments: map[string]string{repoName: result.LineFragments[repoName]},
+			RepoURLs:      filteredRepoURLs,
+			LineFragments: filteredLineFragments,
 		})
 	}