chore: bump version to 0.2.4

Author: codeitlikemiley

Cargo.toml

@@ -36,7 +36,7 @@ chrono = { version = "0.4", features = ["serde"] }
 fuzzy-matcher = "0.3"
 
 # Validation system dependencies
-raz-validation = { path = "raz-validation", version = "0.2.3" }
+raz-validation = { path = "raz-validation", version = "0.2.4" }
 
 [profile.release]
 opt-level = 3

raz-adapters/cli/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "raz-cli"
-version = "0.2.3"
+version = "0.2.4"
 edition = "2024"
 description = "Universal command runner for Rust - Run any Rust file from anywhere with cursor-aware test targeting and override persistence"
 license = "MIT"
@@ -18,10 +18,10 @@ name = "raz"
 path = "src/main.rs"
 
 [dependencies]
-raz-common = { path = "../../raz-common", version = "0.2.3" }
-raz-core = { path = "../../raz-core", version = "0.2.3" }
-raz-config = { path = "../../raz-config", version = "0.2.3" }
-raz-override = { path = "../../raz-override", version = "0.2.3" }
+raz-common = { path = "../../raz-common", version = "0.2.4" }
+raz-core = { path = "../../raz-core", version = "0.2.4" }
+raz-config = { path = "../../raz-config", version = "0.2.4" }
+raz-override = { path = "../../raz-override", version = "0.2.4" }
 clap = { workspace = true }
 tokio = { workspace = true }
 serde_json = { workspace = true }

raz-adapters/vscode/package-lock.json

@@ -5,7 +5,7 @@
 	"author": "Uriah Galang <[email protected]>",
 	"icon": "images/raz-logo.gif",
 	"license": "MIT",
-	"version": "0.2.3",
+	"version": "0.2.4",
 	"repository": {
 		"type": "git",
 		"url": "https://github.com/codeitlikemiley/raz"

raz-adapters/vscode/package.json

@@ -1,6 +1,6 @@
 [package]
 name = "raz-common"
-version = "0.2.3"
+version = "0.2.4"
 edition = "2024"
 authors = ["raz contributors"]
 description = "Common utilities and shared types for the raz project"

raz-common/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "raz-config"
-version = "0.2.3"
+version = "0.2.4"
 edition = "2024"
 description = "Configuration management for RAZ - handles loading, saving, validation, and inheritance"
 license = "MIT"

raz-config/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "raz-core"
-version = "0.2.3"
+version = "0.2.4"
 edition = "2024"
 description = "Universal command generator for Rust projects - Core library with stateless file analysis and cursor-aware execution"
 license = "MIT"
@@ -14,9 +14,9 @@ categories = ["command-line-utilities", "development-tools", "development-tools:
 exclude = ["tests/fixtures/", "benches/"]
 
 [dependencies]
-raz-common = { path = "../raz-common", version = "0.2.3" }
-raz-config = { path = "../raz-config", version = "0.2.3" }
-raz-override = { path = "../raz-override", version = "0.2.3" }
+raz-common = { path = "../raz-common", version = "0.2.4" }
+raz-config = { path = "../raz-config", version = "0.2.4" }
+raz-override = { path = "../raz-override", version = "0.2.4" }
 serde = { workspace = true }
 serde_json = { workspace = true }
 toml = { workspace = true }

raz-core/Cargo.toml

@@ -366,8 +366,7 @@ impl FileDetector {
         #[cfg(feature = "tree-sitter-support")]
         {
             if let Ok(mut detector) = TreeSitterTestDetector::new() {
-                if let Ok(mut tree_sitter_entries) =
-                    detector.detect_test_entry_points(&content, cursor)
+                if let Ok(mut tree_sitter_entries) = detector.detect_entry_points(&content, cursor)
                 {
                     // Build module path from file location for cargo projects
                     let file_module_path =

raz-core/src/file_detection.rs

@@ -29,8 +29,8 @@ impl TreeSitterTestDetector {
         Ok(Self { parser })
     }
 
-    /// Detect all test-related entry points in the file
-    pub fn detect_test_entry_points(
+    /// Detect all entry points in the file (tests, main, benchmarks, etc.)
+    pub fn detect_entry_points(
         &mut self,
         source: &str,
         cursor: Option<Position>,
@@ -339,12 +339,11 @@ impl TreeSitterTestDetector {
             }
             "function_item" => {
                 let fn_name = self.get_function_name(&node, source)?;
+                let start_line = node.start_position().row as u32 + 1;
+                let end_line = node.end_position().row as u32 + 1;
 
                 // Check if this is a test function
                 if self.is_test_function(&node, source) {
-                    let start_line = node.start_position().row as u32 + 1;
-                    let end_line = node.end_position().row as u32 + 1;
-
                     // Build full path including module hierarchy
                     let full_path = if module_stack.is_empty() {
                         fn_name.clone()
@@ -368,6 +367,26 @@ impl TreeSitterTestDetector {
                         line_range: (start_line, end_line),
                         full_path: Some(full_path),
                     });
+                } else if fn_name == "main" {
+                    // Detect main function with proper line range
+                    entry_points.push(EntryPoint {
+                        name: "main".to_string(),
+                        entry_type: EntryPointType::Main,
+                        line: start_line,
+                        column: node.start_position().column as u32,
+                        line_range: (start_line, end_line),
+                        full_path: None,
+                    });
+                } else if self.is_bench_function(&node, source) {
+                    // Detect benchmark function with proper line range
+                    entry_points.push(EntryPoint {
+                        name: fn_name,
+                        entry_type: EntryPointType::Benchmark,
+                        line: start_line,
+                        column: node.start_position().column as u32,
+                        line_range: (start_line, end_line),
+                        full_path: None,
+                    });
                 }
             }
             _ => {}
@@ -419,7 +438,24 @@ impl TreeSitterTestDetector {
                 let attr_text = prev_sibling.utf8_text(source.as_bytes());
                 if let Ok(text) = attr_text {
                     // Match various test attributes
-                    if text.contains("test") {
+                    if text.contains("test") && !text.contains("bench") {
+                        return true;
+                    }
+                }
+            }
+        }
+
+        false
+    }
+
+    /// Check if a function node is a benchmark function (#[bench])
+    fn is_bench_function(&self, node: &Node, source: &str) -> bool {
+        // Check previous sibling for bench attributes
+        if let Some(prev_sibling) = node.prev_sibling() {
+            if prev_sibling.kind() == "attribute_item" {
+                let attr_text = prev_sibling.utf8_text(source.as_bytes());
+                if let Ok(text) = attr_text {
+                    if text.contains("bench") {
                         return true;
                     }
                 }

raz-core/src/tree_sitter_test_detector.rs

@@ -1695,11 +1695,11 @@ impl UniversalCommandGenerator {
     fn find_main_at_cursor(entry_points: &[EntryPoint], cursor: Position) -> Option<&EntryPoint> {
         let cursor_line = cursor.line + 1; // Convert to 1-based
 
-        // Check if cursor is on or near main function
+        // Check if cursor is within main function scope
         for ep in entry_points {
             if ep.entry_type == EntryPointType::Main {
-                // Check if cursor is on the main function line or within 2 lines
-                if cursor_line >= ep.line.saturating_sub(1) && cursor_line <= ep.line + 2 {
+                // Check if cursor is within the full scope of the main function
+                if cursor_line >= ep.line_range.0 && cursor_line <= ep.line_range.1 {
                     return Some(ep);
                 }
             }