Renamed file.rs to ast.rs to more accurately represent what that code does

src/ast.rs

@@ -0,0 +1,70 @@
+use std::cell::RefCell;
+use std::path::Path;
+use tree_sitter::{Language, QueryCursor, StreamingIterator};
+use crate::commands::scan::Diag;
+use crate::errors::{NyxResult, NyxError};
+use crate::utils::{query_cache, Config};
+use crate::utils::ext::lowercase_ext;
+
+thread_local! {
+    static PARSER: RefCell<tree_sitter::Parser> = RefCell::new(tree_sitter::Parser::new());
+}
+
+pub(crate) fn run_rules_on_file(
+  path: &Path,
+  cfg: &Config,
+) -> NyxResult<Vec<Diag>> {
+  tracing::debug!("Running rules on: {}", path.display());
+  let bytes = std::fs::read(path)?;
+
+  // Fast binary-file guard (skip if >1% NULs)
+  if bytes.iter().filter(|b| **b == 0).count() * 100 / bytes.len().max(1) > 1 {
+    return Ok(vec![]);
+  }
+
+  let (ts_lang, lang_slug) = match lowercase_ext(path) {
+    Some("rs")  => (Language::from(tree_sitter_rust::LANGUAGE),                   "rust"),
+    Some("c")   => (Language::from(tree_sitter_c::LANGUAGE),                      "c"),
+    Some("cpp") => (Language::from(tree_sitter_cpp::LANGUAGE),                    "cpp"),
+    Some("java")=> (Language::from(tree_sitter_java::LANGUAGE),                   "java"),
+    Some("go")  => (Language::from(tree_sitter_go::LANGUAGE),                     "go"),
+    Some("php") => (Language::from(tree_sitter_php::LANGUAGE_PHP),                "php"),
+    Some("py")  => (Language::from(tree_sitter_python::LANGUAGE),                 "python"),
+    Some("ts")  => (Language::from(tree_sitter_typescript::LANGUAGE_TYPESCRIPT),  "typescript"),
+    Some("js")  => (Language::from(tree_sitter_javascript::LANGUAGE),             "javascript"),
+    _           => return Ok(vec![]),
+  };
+
+  let _tree = PARSER.with(|cell| {
+    let mut parser = cell.borrow_mut();
+    parser.set_language(&ts_lang)?;
+    parser.parse(&*bytes, None)
+      .ok_or_else(|| NyxError::Other("tree-sitter failed".into()))
+  })?;
+
+  let root = _tree.root_node();
+
+  let compiled = query_cache::for_lang(lang_slug, ts_lang);
+  let mut cursor = QueryCursor::new();
+  let mut out = Vec::new();
+
+  for cq in compiled.iter() {
+    if cfg.scanner.min_severity <= cq.meta.severity {
+      continue;
+    }
+    let mut matches = cursor.matches(&cq.query, root, &*bytes);
+    while let Some(m) = matches.next() {
+      if let Some(cap) = m.captures.iter().find(|c| c.index == 0) {
+        let point = cap.node.start_position();
+        out.push(Diag {
+          path: path.to_string_lossy().into_owned(),
+          line: point.row + 1,
+          col:  point.column + 1,
+          severity: cq.meta.severity,
+          id: cq.meta.id.to_owned(),
+        });
+      }
+    }
+}
+  Ok(out)
+}