pass the diff to codeowners-rs

Author: martinemde

ext/code_ownership/src/lib.rs

@@ -74,15 +74,17 @@ fn version() -> String {
    runner::version()
 }
 
-fn validate() -> Result<Value, Error> {
+fn validate(files: Option<Vec<String>>) -> Result<Value, Error> {
     let run_config = build_run_config();
-    let run_result = runner::validate(&run_config, vec![]);
+    let files_vec = files.unwrap_or_default();
+    let run_result = runner::validate(&run_config, files_vec);
     validate_result(&run_result)
 }
 
-fn generate_and_validate(skip_stage: bool) -> Result<Value, Error> {
+fn generate_and_validate(files: Option<Vec<String>>, skip_stage: bool) -> Result<Value, Error> {
     let run_config = build_run_config();
-    let run_result = runner::generate_and_validate(&run_config, vec![], skip_stage);
+    let files_vec = files.unwrap_or_default();
+    let run_result = runner::generate_and_validate(&run_config, files_vec, skip_stage);
     validate_result(&run_result)
 }
 
@@ -122,12 +124,11 @@ fn build_run_config() -> RunConfig {
 fn init(ruby: &Ruby) -> Result<(), Error> {
     let module = ruby.define_module("RustCodeOwners")?;
     module.define_singleton_method("for_file", function!(for_file, 1))?;
-    module.define_singleton_method("generate_and_validate", function!(generate_and_validate, 1))?;
-    module.define_singleton_method("validate", function!(validate, 0))?;
+    module.define_singleton_method("generate_and_validate", function!(generate_and_validate, 2))?;
+    module.define_singleton_method("validate", function!(validate, 1))?;
     module.define_singleton_method("for_team", function!(for_team, 1))?;
     module.define_singleton_method("version", function!(version, 0))?;
     module.define_singleton_method("teams_for_files", function!(teams_for_files, 1))?;
 
     Ok(())
 }
-

lib/code_ownership.rb

@@ -233,9 +233,9 @@ def validate!(
     files: nil
   )
     if autocorrect
-      ::RustCodeOwners.generate_and_validate(!stage_changes)
+      ::RustCodeOwners.generate_and_validate(files, !stage_changes)
     else
-      ::RustCodeOwners.validate
+      ::RustCodeOwners.validate(files)
     end
   end