Support script file mode

Author: AmrDeveloper

src/arguments.rs

@@ -27,6 +27,7 @@ impl Arguments {
 pub enum Command {
     ReplMode(Arguments),
     QueryMode(String, Arguments),
+    ScriptMode(String, Arguments),
     Help,
     Version,
     Error(String),
@@ -44,6 +45,7 @@ pub fn parse_arguments(args: &[String]) -> Command {
     }
 
     let mut optional_query: Option<String> = None;
+    let mut optional_script_file: Option<String> = None;
     let mut arguments = Arguments::new();
 
     let mut arg_index = 1;
@@ -95,6 +97,16 @@ pub fn parse_arguments(args: &[String]) -> Command {
                 optional_query = Some(args[arg_index].to_string());
                 arg_index += 1;
             }
+            "--script" | "-s" => {
+                arg_index += 1;
+                if arg_index >= args_len {
+                    let message = format!("Argument {} must be followed by the file", arg);
+                    return Command::Error(message);
+                }
+
+                optional_script_file = Some(args[arg_index].to_string());
+                arg_index += 1;
+            }
             "--analysis" | "-a" => {
                 arguments.analysis = true;
                 arg_index += 1;
@@ -147,7 +159,9 @@ pub fn parse_arguments(args: &[String]) -> Command {
         return Command::Error("Must provide one or more C/C++ files".to_string());
     }
 
-    if let Some(query) = optional_query {
+    if let Some(script_file) = optional_script_file {
+        Command::ScriptMode(script_file, arguments)
+    } else if let Some(query) = optional_query {
         Command::QueryMode(query, arguments)
     } else {
         Command::ReplMode(arguments)

src/main.rs

@@ -1,3 +1,4 @@
+use std::fs;
 use std::io::IsTerminal;
 use std::path::Path;
 
@@ -34,6 +35,23 @@ fn main() {
     let command = arguments::parse_arguments(&args);
     match command {
         Command::ReplMode(arguments) => launch_clangql_repl(arguments),
+        Command::ScriptMode(script_file, arguments) => {
+            let mut reporter = DiagnosticReporter::default();
+            let files = &arguments.files;
+            if let Err(error) = validate_files_paths(files) {
+                reporter.report_diagnostic("", Diagnostic::error(error.as_str()));
+                return;
+            }
+
+            let mut env = create_clang_ql_environment();
+            let query =
+                fs::read_to_string(script_file).expect("Should have been able to read the file");
+
+            let compilation_units = parse_files(files);
+            let provider: Box<dyn DataProvider> =
+                Box::new(ClangDataProvider::new(compilation_units));
+            execute_clang_ql_query(query, &arguments, &mut env, &provider, &mut reporter);
+        }
         Command::QueryMode(query, arguments) => {
             let mut reporter = DiagnosticReporter::default();
             let files = &arguments.files;