feat: Highlight and prioritize code definitions in grep results

Author: yashksaini-coder

crates/fff-core/src/grep.rs

@@ -1150,6 +1150,12 @@ fn collect_grep_results<'a>(
         }
     }
 
+    // Prioritize definition matches by sorting them to the top.
+    // Use a stable sort to maintain relative order within each category.
+    if options.classify_definitions {
+        crate::sort_buffer::sort_by_key_with_buffer(&mut all_matches, |m| !m.is_definition);
+    }
+
     // If no file had any match, we searched the entire slice.
     if result_files.is_empty() {
         files_consumed = files_to_search_len;

crates/fff-nvim/src/lib.rs

@@ -311,6 +311,7 @@ pub fn live_grep(
         smart_case,
         grep_mode,
         time_budget_ms,
+        classify_definitions,
     ): (
         String,
         Option<usize>,
@@ -320,6 +321,7 @@ pub fn live_grep(
         Option<bool>,
         Option<String>,
         Option<u64>,
+        Option<bool>,
     ),
 ) -> LuaResult<LuaValue> {
     let file_picker_guard = FILE_PICKER
@@ -348,7 +350,7 @@ pub fn live_grep(
         time_budget_ms: time_budget_ms.unwrap_or(0),
         before_context: 0,
         after_context: 0,
-        classify_definitions: false,
+        classify_definitions: classify_definitions.unwrap_or(false),
     };
 
     let result = fff_core::grep::grep_search(picker.get_files(), &parsed, &options);

crates/fff-nvim/src/lua_types.rs

@@ -141,6 +141,7 @@ impl IntoLua for GrepResultLua<'_> {
             item.set("col", m.col)?;
             item.set("byte_offset", m.byte_offset)?;
             item.set("line_content", m.line_content.as_str())?;
+            item.set("is_definition", m.is_definition)?;
 
             // Match byte ranges within line_content
             let ranges = lua.create_table()?;

lua/fff/conf.lua

@@ -54,6 +54,7 @@ local M = {}
 --- @field smart_case boolean
 --- @field time_budget_ms number
 --- @field modes string[]
+--- @field classify_definitions boolean
 
 --- @class FffConfig
 --- @field base_path string
@@ -332,6 +333,7 @@ local function init()
       smart_case = true, -- Case-insensitive unless query has uppercase
       time_budget_ms = 150, -- Max search time in ms per call (prevents UI freeze, 0 = no limit)
       modes = { 'plain', 'regex', 'fuzzy' }, -- Available grep modes and their cycling order
+      classify_definitions = false, -- Mark definition lines like fn/struct/class
     },
   }
 

lua/fff/grep/grep_renderer.lua

@@ -47,6 +47,7 @@ end
 local function render_match_line(item, ctx)
   local location = string.format(':%d:%d', item.line_number or 0, (item.col or 0) + 1)
   local separator = '  '
+  
   -- vim.json.decode may return Blobs for strings with NUL bytes; coerce to string.
   local raw_content = item.line_content
   if type(raw_content) ~= 'string' then raw_content = raw_content and tostring(raw_content) or '' end
@@ -193,6 +194,15 @@ local function apply_match_highlights(item, ctx, item_idx, buf, ns_id, row, line
       })
     end
   end
+
+  -- 7. Definition indicator as virtual text
+  if item.is_definition then
+    pcall(vim.api.nvim_buf_set_extmark, buf, ns_id, row, 0, {
+      virt_text = { { ' [def]', config.hl.combo_header or 'Number' } },
+      virt_text_pos = 'eol',
+      priority = 250,
+    })
+  end
 end
 
 --- Render a single item's lines (called by list_renderer's generate_item_lines).

lua/fff/grep/init.lua

@@ -33,7 +33,8 @@ function M.search(query, file_offset, page_size, config, grep_mode)
     conf.max_matches_per_file,
     conf.smart_case,
     grep_mode or 'plain',
-    conf.time_budget_ms
+    conf.time_budget_ms,
+    conf.classify_definitions
   )
   return last_result
 end