[Rust] Add a command for installing rules/rules.md into a repository (#17405)

Author: AutomatedTester

rust/BUILD.bazel

@@ -91,7 +91,10 @@ rust_library(
         ["src/**/*.rs"],
         exclude = ["main.rs"],
     ),
-    compile_data = ["src/resources/skills.md"],
+    compile_data = [
+        "src/resources/rules.md",
+        "src/resources/skills.md",
+    ],
     crate_features = select({
         "//common:stamp": [],
         "//conditions:default": ["avoid_stats"],
@@ -106,6 +109,7 @@ filegroup(
     srcs = [
         "Cargo.lock",
         "Cargo.toml",
+        "src/resources/rules.md",
         "src/resources/skills.md",
         ":selenium_manager_rs_srcs",
     ],

rust/src/lib.rs

@@ -70,6 +70,7 @@ pub mod lock;
 pub mod logger;
 pub mod metadata;
 pub mod mirror;
+pub mod rules;
 pub mod safari;
 pub mod safaritp;
 pub mod shell;

rust/src/main.rs

@@ -27,6 +27,7 @@ use selenium_manager::jre::ensure_jre;
 use selenium_manager::lock::clear_lock_if_required;
 use selenium_manager::logger::{BROWSER_PATH, DRIVER_PATH, Logger};
 use selenium_manager::metadata::clear_metadata;
+use selenium_manager::rules::write_rules_file;
 use selenium_manager::skills::write_skills_file;
 use selenium_manager::{REQUEST_TIMEOUT_SEC, SM_BETA_LABEL};
 use selenium_manager::{
@@ -162,6 +163,10 @@ struct Cli {
     /// Add a skills file with Selenium best practices to the repository
     #[clap(long, value_name = "FILE_NAME", num_args = 0..=1, default_missing_value = "")]
     init_skills: Option<String>,
+
+    /// Add a rules file with Selenium guidance for LLM coding assistants to the repository
+    #[clap(long, value_name = "FILE_NAME", num_args = 0..=1, default_missing_value = "")]
+    init_rules: Option<String>,
 }
 
 fn main() {
@@ -189,6 +194,23 @@ fn main() {
         flush_and_exit(OK, &log, None);
     }
 
+    if let Some(mut rules_file) = cli.init_rules {
+        if rules_file.is_empty() {
+            let default_path = Path::new("rules").join("selenium.md");
+            if default_path.exists() {
+                rules_file = "selenium-rules.md".to_string();
+            } else {
+                rules_file = default_path.to_string_lossy().to_string();
+            }
+        }
+        if let Err(err) = write_rules_file(Path::new(&rules_file), &log) {
+            log.error(format!("Error creating {}: {}", rules_file, err));
+            flush_and_exit(DATAERR, &log, Some(err));
+        }
+        log.info(format!("{} file successfully created", rules_file));
+        flush_and_exit(OK, &log, None);
+    }
+
     let cache_path =
         StringKey(vec![CACHE_PATH_KEY], &cli.cache_path.unwrap_or_default()).get_value();
     let grid = cli.grid;