Fix cargo fmt and clippy issues - format code according to style guidelines

Co-authored-by: 0xrinegade <[email protected]>

Author: Copilot

src/main.rs

@@ -4,15 +4,11 @@
 use {
     crate::config::Config,
     crate::utils::diagnostics::DiagnosticCoordinator,
-    crate::utils::{dashboard, ebpf_deploy, examples, nodes, ssh_deploy, svm_info},
     crate::utils::markdown_renderer::MarkdownRenderer,
+    crate::utils::{dashboard, ebpf_deploy, examples, nodes, ssh_deploy, svm_info},
     clparse::parse_command_line,
     solana_client::rpc_client::RpcClient,
-    solana_sdk::{
-        native_token::Sol, 
-        pubkey::Pubkey, 
-        signature::Signer
-    },
+    solana_sdk::{native_token::Sol, pubkey::Pubkey, signature::Signer},
     std::{process::exit, str::FromStr},
 };
 
@@ -64,18 +60,18 @@ async fn handle_ai_query(
     // For external subcommands, clap collects additional arguments in subcommand_value
     // This is the proper way to handle external subcommands with clap
     let mut query_parts = vec![sub_command.to_string()];
-    
+
     // Get additional arguments from clap's external subcommand handling
     // External subcommands store arguments as OsString, not String
     if let Some(external_args) = sub_matches.get_many::<std::ffi::OsString>("") {
         query_parts.extend(external_args.map(|os_str| os_str.to_string_lossy().to_string()));
     }
-    
+
     // If clap doesn't provide args (fallback), parse from environment
     // This maintains compatibility while documenting the limitation
     if query_parts.len() == 1 {
         let args: Vec<String> = std::env::args().collect();
-        
+
         // Collect non-flag arguments starting from the subcommand
         let mut found_subcommand = false;
         for arg in args.iter().skip(1) {

src/services/ai_service.rs

@@ -108,7 +108,9 @@ impl AiService {
         let mut template_manager = PromptTemplateManager::new();
 
         // Initialize template manager
-        if let Err(e) = template_manager.load_from_directory_with_debug("./templates/ai_prompts", debug_mode) {
+        if let Err(e) =
+            template_manager.load_from_directory_with_debug("./templates/ai_prompts", debug_mode)
+        {
             if debug_mode {
                 println!("⚠️  Failed to load AI prompt templates: {}", e);
             }

src/utils/ast_analyzer.rs

@@ -294,8 +294,9 @@ impl AstAnalyzer {
             let after_let = &content[start + 4..];
             if let Some(equals_pos) = after_let.find(" = ") {
                 let var_name = after_let[..equals_pos].trim();
-                if content.contains(&format!("{}.balance", var_name)) || 
-                   content.contains(&format!("{}.", var_name)) {
+                if content.contains(&format!("{}.balance", var_name))
+                    || content.contains(&format!("{}.", var_name))
+                {
                     return Some(var_name.to_string());
                 }
             }
@@ -309,7 +310,9 @@ impl AstAnalyzer {
 
         match finding.category.as_str() {
             "Authentication & Authorization" => {
-                if !content.contains("is_signer") && (content.contains("AccountInfo") || content.contains("ctx.accounts")) {
+                if !content.contains("is_signer")
+                    && (content.contains("AccountInfo") || content.contains("ctx.accounts"))
+                {
                     // Extract account variable name from the content
                     if let Some(account_var) = self.extract_account_variable_name(content) {
                         fixed_content = format!(
@@ -419,8 +422,9 @@ impl SolanaSecurityVisitor {
 
         // Check for missing signer validation
         // Token stream format: "ctx . accounts" instead of "ctx.accounts"
-        if (func_str.contains("ctx . accounts") || func_str.contains("ctx.accounts")) && 
-           !func_str.contains("is_signer") {
+        if (func_str.contains("ctx . accounts") || func_str.contains("ctx.accounts"))
+            && !func_str.contains("is_signer")
+        {
             self.context.security_issues.push(SecurityIssue {
                 issue_type: SecurityIssueType::MissingSignerCheck,
                 location: CodeLocation {

src/utils/audit_modular.rs

@@ -1423,7 +1423,7 @@ mod tests {
         let findings_hardcoded = check
             .check_content(code_with_hardcoded_key, "test.rs")
             .unwrap();
-        
+
         let hardcoded_findings: Vec<_> = findings_hardcoded
             .iter()
             .filter(|f| f.title.contains("hardcoded Solana public key"))

src/utils/audit_parser.rs

@@ -322,11 +322,11 @@ impl SecurityVisitor {
     fn analyze_field_access(&mut self, expr: &Expr, field_expr: &syn::ExprField) {
         let field_name = field_expr.member.to_token_stream().to_string();
         let receiver_str = quote::quote!(#expr).to_string();
-        
+
         // Look for Solana account field access patterns
         if receiver_str.contains("account") {
             let line_num = self.get_line_number_for_pattern(&format!(".{}", field_name));
-            
+
             match field_name.as_str() {
                 "owner" => {
                     let mut solana_op = SolanaOperation {

src/utils/audit_tests.rs

@@ -75,7 +75,7 @@ mod tests {
 
             // With empty API key, this should succeed and use internal OSVM AI service
             match AuditService::validate_environment(&request) {
-                Ok(()) => true,  // Success expected when using internal service
+                Ok(()) => true, // Success expected when using internal service
                 Err(_) => false,
             }
         });
@@ -119,7 +119,7 @@ mod tests {
 
             // With whitespace API key, this should succeed and use internal OSVM AI service
             match AuditService::validate_environment(&request) {
-                Ok(()) => true,  // Success expected when using internal service
+                Ok(()) => true, // Success expected when using internal service
                 Err(_) => false,
             }
         });

src/utils/code_extractor.rs

@@ -215,7 +215,9 @@ impl CodeExtractor {
 
         match finding.category.as_str() {
             "Authentication & Authorization" => {
-                if !content.contains("is_signer") && (content.contains("AccountInfo") || content.contains("ctx.accounts")) {
+                if !content.contains("is_signer")
+                    && (content.contains("AccountInfo") || content.contains("ctx.accounts"))
+                {
                     fixed_content = fixed_content.replace(
                         "let user_account = &ctx.accounts.user_account;",
                         "let user_account = &ctx.accounts.user_account;\n    require!(user_account.is_signer, ErrorCode::MissingSignature);"

src/utils/markdown_renderer.rs

@@ -1,4 +1,4 @@
-use termimad::{MadSkin};
+use termimad::MadSkin;
 
 /// A markdown renderer for CLI output
 #[derive(Debug, Clone)]
@@ -65,4 +65,4 @@ mod tests {
         let result = renderer.render_to_string("# Themed Test");
         assert!(result.contains("Themed Test"));
     }
-}
+}

src/utils/prompt_templates.rs

@@ -138,7 +138,11 @@ impl PromptTemplateManager {
         self.load_from_directory_with_debug(dir_path, true)
     }
 
-    pub fn load_from_directory_with_debug(&mut self, dir_path: &str, debug_mode: bool) -> Result<usize> {
+    pub fn load_from_directory_with_debug(
+        &mut self,
+        dir_path: &str,
+        debug_mode: bool,
+    ) -> Result<usize> {
         self.template_dirs.push(dir_path.to_string());
 
         let dir = Path::new(dir_path);
@@ -579,7 +583,7 @@ mod tests {
     fn test_template_loading() {
         let temp_dir = TempDir::new().unwrap();
         let temp_path = temp_dir.path();
-        
+
         // Create a simple template file for testing
         let template_content = r#"id: test_template
 name: Test Template
@@ -603,10 +607,10 @@ version: 1.0.0
 enabled: true
 metadata: {}
 "#;
-        
+
         let template_file = temp_path.join("test_template.yaml");
         std::fs::write(&template_file, template_content).unwrap();
-        
+
         let mut manager = PromptTemplateManager::new();
 
         let loaded = manager
@@ -621,16 +625,16 @@ metadata: {}
     #[test]
     fn test_template_rendering() {
         let mut manager = PromptTemplateManager::new();
-        
+
         // Load from the actual templates directory
         let templates_dir = "templates/ai_prompts";
         let loaded = manager.load_from_directory(templates_dir).unwrap_or(0);
-        
+
         // If no templates in the actual directory, create a test template
         if loaded == 0 {
             let temp_dir = TempDir::new().unwrap();
             let temp_path = temp_dir.path();
-            
+
             // Create the specific template the test expects
             let template_content = r#"id: deeplogic_economic_exploit
 name: DeepLogic Economic Exploit Analysis
@@ -659,11 +663,13 @@ version: 1.0.0
 enabled: true
 metadata: {}
 "#;
-            
+
             let template_file = temp_path.join("deeplogic_economic_exploit.yaml");
             std::fs::write(&template_file, template_content).unwrap();
-            
-            manager.load_from_directory(temp_dir.path().to_str().unwrap()).unwrap();
+
+            manager
+                .load_from_directory(temp_dir.path().to_str().unwrap())
+                .unwrap();
         }
 
         let mut variables = HashMap::new();

tests/debug_flag_tests.rs

@@ -9,15 +9,15 @@ fn test_ai_query_without_debug_flag() {
         .expect("Failed to execute command");
 
     let stdout = String::from_utf8_lossy(&output.stdout);
-    
+
     // Should not contain debug information
     assert!(!stdout.contains("🔍 Interpreting as AI query"));
     assert!(!stdout.contains("✅ Loaded"));
     assert!(!stdout.contains("📤 OSVM AI Request"));
     assert!(!stdout.contains("📥 OSVM AI Response"));
     assert!(!stdout.contains("🔍 AI Response received"));
     assert!(!stdout.contains("🤖 AI Response:"));
-    
+
     // Should contain the actual response (without debug prefix)
     assert!(!stdout.is_empty());
 }
@@ -30,7 +30,7 @@ fn test_ai_query_with_debug_flag() {
         .expect("Failed to execute command");
 
     let stdout = String::from_utf8_lossy(&output.stdout);
-    
+
     // Should contain debug information
     assert!(stdout.contains("🔍 Interpreting as AI query"));
     assert!(stdout.contains("✅ Loaded"));
@@ -48,8 +48,8 @@ fn test_debug_flag_appears_in_help() {
         .expect("Failed to execute command");
 
     let stdout = String::from_utf8_lossy(&output.stdout);
-    
+
     // Should contain debug flag in help
     assert!(stdout.contains("--debug"));
     assert!(stdout.contains("Show debug information"));
-}
+}