feat: enhance project type detection for Terraform and Ansible

- Added support for detecting Terraform projects based on main.tf and related files.
- Implemented detection for Ansible projects by checking for YAML playbooks in an 'ansible' directory.
- Updated tests to verify detection functionality for both project types.

Author: NeuralEmpowerment

Cargo.lock

@@ -9,6 +9,8 @@ pub enum ProjectType {
     Go,
     Java,
     DotNet,
+    Terraform,
+    Ansible,
     Other,
 }
 
@@ -21,6 +23,8 @@ impl ProjectType {
             ProjectType::Go => "go",
             ProjectType::Java => "java",
             ProjectType::DotNet => ".net",
+            ProjectType::Terraform => "terraform",
+            ProjectType::Ansible => "ansible",
             ProjectType::Other => "other",
         }
     }
@@ -35,6 +39,8 @@ pub fn detect_project_type(dir: &Path) -> Option<ProjectType> {
         (ProjectType::Go, &["go.mod"][..]),
         (ProjectType::Java, &["pom.xml", "build.gradle", "gradlew"]),
         (ProjectType::DotNet, &["global.json"][..]),
+        (ProjectType::Terraform, &["main.tf", "variables.tf", "outputs.tf"][..]),
+        (ProjectType::Ansible, &[]), // Special case - handled below
     ];
 
     for (ptype, files) in candidates.iter() {
@@ -54,6 +60,21 @@ pub fn detect_project_type(dir: &Path) -> Option<ProjectType> {
             }
         }
     }
+
+    // Special detection for Ansible
+    let ansible_dir = dir.join("ansible");
+    if ansible_dir.is_dir() {
+        if let Ok(entries) = fs::read_dir(&ansible_dir) {
+            for entry in entries.flatten() {
+                if let Some(ext) = entry.path().extension() {
+                    if ext == "yml" || ext == "yaml" {
+                        return Some(ProjectType::Ansible);
+                    }
+                }
+            }
+        }
+    }
+
     None
 }
 

crates/indexer/src/detect.rs

@@ -1,4 +1,4 @@
-use indexer::{scan_roots, AppConfig, Db, ScanOptions};
+use indexer::{scan_roots, AppConfig, Db, ScanOptions, detect::{detect_project_type, ProjectType}};
 use std::fs;
 use std::io::Write;
 
@@ -29,3 +29,30 @@ fn scans_minimal_node_project() {
     assert_eq!(rows[0].project_type.as_deref(), Some("node"));
     assert!(rows[0].size_bytes.unwrap_or(0) > 0);
 }
+
+#[test]
+fn detects_terraform_project() {
+    let dir = tempfile::tempdir().unwrap();
+    let proj = dir.path().join("terraform-project");
+    fs::create_dir_all(&proj).unwrap();
+
+    // Create a main.tf file
+    fs::write(proj.join("main.tf"), "resource \"aws_instance\" \"example\" {}").unwrap();
+
+    let project_type = detect_project_type(&proj);
+    assert_eq!(project_type, Some(ProjectType::Terraform));
+}
+
+#[test]
+fn detects_ansible_project() {
+    let dir = tempfile::tempdir().unwrap();
+    let proj = dir.path().join("ansible-project");
+    fs::create_dir_all(&proj).unwrap();
+
+    // Create an ansible directory with a playbook
+    fs::create_dir_all(proj.join("ansible")).unwrap();
+    fs::write(proj.join("ansible").join("playbook.yml"), "- hosts: all\n  tasks:\n    - debug: msg=hello").unwrap();
+
+    let project_type = detect_project_type(&proj);
+    assert_eq!(project_type, Some(ProjectType::Ansible));
+}

crates/indexer/tests/basic.rs

@@ -0,0 +1,20 @@
+use std::path::Path;
+use indexer::detect::{detect_project_type, ProjectType};
+
+fn main() {
+    let test_dirs = vec![
+        "test-repo",
+        "test-repo/terraform",
+        "test-repo/terraform/aws",
+        "test-repo/ansible",
+    ];
+
+    for dir in test_dirs {
+        let path = Path::new(dir);
+        if let Some(project_type) = detect_project_type(path) {
+            println!("{}: {:?}", dir, project_type.as_str());
+        } else {
+            println!("{}: None", dir);
+        }
+    }
+}