Checker error template overrides (#2)

* adding checker error template overrides

* bumping version

* adding error template overrides description

Author: perryqh

Cargo.toml

@@ -2,7 +2,7 @@
 
 [package]
 name = "pks"
-version = "0.2.16"
+version = "0.2.17"
 edition = "2021"
 description = "Welcome! Please see https://github.com/rubyatscale/pks for more information!"
 license = "MIT"

README.md

@@ -82,6 +82,7 @@ Instructions:
 - Confirm the output of `git diff` is empty
 - Please file an issue if it's not!
 
+
 # New to Rust?
 Me too! This is my first Rust project, so I'd love to have feedback, advice, and contributions!
 
@@ -155,6 +156,32 @@ enforcement_globs_ignore:
   reason: "The other dependency violations are fine as those packs will be absorbed into this one."
 ```
 
+## "check" error messages
+The error messages resulting from running `pks check` can be customized with mustache-style interpolation. The available
+variables are:
+- violation_name
+- referencing_pack_name
+- defining_pack_name
+- constant_name
+- reference_location
+- referencing_pack_relative_yml
+
+Layer violations also have 
+- defining_layer
+- referencing_layer
+
+Example:
+packwerk.yml
+```yml
+checker_overrides:
+  folder_privacy_error_template: "{{reference_location}} {{violation_name}} / Product Service Privacy Violation: `{{constant_name}}` belongs to the `{{defining_pack_name}}` product service, which is not visible to `{{referencing_pack_name}}` as it is a different product service. See https://go/pks-folder-privacy"
+  layer_error_template: "{{reference_location}}Layer violation: `{{constant_name}}` belongs to `{{defining_pack_name}}` (whose layer is `{{defining_layer}}`) cannot be accessed from `{{referencing_pack_name}}` (whose layer is `{{referencing_layer}}`). See https://go/pks-layer"
+  visibility_error_template: "{{reference_location}}Visibility violation: `{{constant_name}}` belongs to `{{defining_pack_name}}`, which is not visible to `{{referencing_pack_name}}`. See https://go/pks-visibility"
+  privacy_error_template: "{{reference_location}}Privacy violation: `{{constant_name}}` is private to `{{defining_pack_name}}`, but referenced from `{{referencing_pack_name}}`. See https://go/pks-privacy"
+  dependency_error_template: "{{reference_location}}Dependency violation: `{{constant_name}}` belongs to `{{defining_pack_name}}`, but `{{referencing_pack_relative_yml}}` does not specify a dependency on `{{defining_pack_name}}`. See https://go/pks-dependency"
+```
+
+
 # Benchmarks
 See [BENCHMARKS.md](https://github.com/rubyatscale/pks/blob/main/BENCHMARKS.md)
 

src/packs/configuration.rs

@@ -180,35 +180,64 @@ fn build_violation_checker_configuration(
     violation_checker_overrides: Option<&CheckerOverrides>,
 ) -> HashMap<CheckerType, CheckerConfiguration> {
     let mut checker_configurations = HashMap::new();
-    checker_configurations.insert(
-        CheckerType::Dependency,
-        CheckerConfiguration::new(CheckerType::Dependency),
-    );
-    checker_configurations.insert(
-        CheckerType::Privacy,
-        CheckerConfiguration::new(CheckerType::Privacy),
-    );
-    checker_configurations.insert(
-        CheckerType::Layer,
-        CheckerConfiguration::new(CheckerType::Layer),
-    );
-    checker_configurations.insert(
-        CheckerType::Visibility,
-        CheckerConfiguration::new(CheckerType::Visibility),
-    );
-    let mut checker_configuration =
+    let mut folder_privacy_checker_configuration =
         CheckerConfiguration::new(CheckerType::FolderPrivacy);
+    let mut privacy_checker_configuration =
+        CheckerConfiguration::new(CheckerType::Privacy);
+    let mut dependency_checker_configuration =
+        CheckerConfiguration::new(CheckerType::Dependency);
+    let mut layer_checker_configuration =
+        CheckerConfiguration::new(CheckerType::Layer);
+    let mut visibility_checker_configuration =
+        CheckerConfiguration::new(CheckerType::Visibility);
+
     if let Some(violation_checker_overrides) = violation_checker_overrides {
         if let Some(error_template) = violation_checker_overrides
             .folder_privacy_error_template
             .clone()
         {
-            checker_configuration.override_error_template =
+            folder_privacy_checker_configuration.override_error_template =
+                Some(error_template);
+        }
+        if let Some(error_template) =
+            violation_checker_overrides.privacy_error_template.clone()
+        {
+            privacy_checker_configuration.override_error_template =
+                Some(error_template);
+        }
+        if let Some(error_template) =
+            violation_checker_overrides.layer_error_template.clone()
+        {
+            layer_checker_configuration.override_error_template =
+                Some(error_template);
+        }
+        if let Some(error_template) = violation_checker_overrides
+            .visibility_error_template
+            .clone()
+        {
+            visibility_checker_configuration.override_error_template =
+                Some(error_template);
+        }
+        if let Some(error_template) = violation_checker_overrides
+            .dependency_error_template
+            .clone()
+        {
+            dependency_checker_configuration.override_error_template =
                 Some(error_template);
         }
     }
+    checker_configurations.insert(
+        CheckerType::FolderPrivacy,
+        folder_privacy_checker_configuration,
+    );
+    checker_configurations
+        .insert(CheckerType::Dependency, dependency_checker_configuration);
+    checker_configurations
+        .insert(CheckerType::Privacy, privacy_checker_configuration);
+    checker_configurations
+        .insert(CheckerType::Layer, layer_checker_configuration);
     checker_configurations
-        .insert(CheckerType::FolderPrivacy, checker_configuration);
+        .insert(CheckerType::Visibility, visibility_checker_configuration);
 
     checker_configurations
 }

src/packs/raw_configuration.rs

@@ -80,6 +80,10 @@ pub(crate) struct RawConfiguration {
 pub struct CheckerOverrides {
     /// Mustache style error message template
     pub folder_privacy_error_template: Option<String>,
+    pub privacy_error_template: Option<String>,
+    pub layer_error_template: Option<String>,
+    pub visibility_error_template: Option<String>,
+    pub dependency_error_template: Option<String>,
 }
 
 pub(crate) fn get(absolute_root: &Path) -> anyhow::Result<RawConfiguration> {

tests/check_test.rs

@@ -8,6 +8,29 @@ pub fn stripped_output(output: Vec<u8>) -> String {
     String::from_utf8_lossy(&strip_ansi_escapes::strip(output)).to_string()
 }
 
+#[test]
+fn test_check_with_privacy_dependency_error_template_overrides(
+) -> Result<(), Box<dyn Error>> {
+    let output = Command::cargo_bin("pks")?
+        .arg("--project-root")
+        .arg("tests/fixtures/privacy_violation_overrides")
+        .arg("--debug")
+        .arg("check")
+        .assert()
+        .failure()
+        .get_output()
+        .stdout
+        .clone();
+
+    let stripped_output = stripped_output(output);
+
+    assert!(stripped_output.contains("2 violation(s) detected:"));
+    assert!(stripped_output.contains("packs/foo/app/services/foo.rb:3:4\nDependency violation: `::Bar` belongs to `packs/bar`, but `packs/foo/package.yml` does not specify a dependency on `packs/bar`. See https://go/pks-dependency"));
+    assert!(stripped_output.contains("packs/foo/app/services/foo.rb:3:4\nPrivacy violation: `::Bar` is private to `packs/bar`, but referenced from `packs/foo`. See https://go/pks-privacy"));
+
+    common::teardown();
+    Ok(())
+}
 #[test]
 fn test_check() -> Result<(), Box<dyn Error>> {
     let output = Command::cargo_bin("pks")?

tests/fixtures/layer_violations_with_overrides/config/initializers/inflections.rb

@@ -0,0 +1,6 @@
+ActiveSupport::Inflector.inflections do |do_not_couple_implementation_to_this_string|
+  do_not_couple_implementation_to_this_string.acronym 'API'
+
+  # Using single vs double quotes inconsistently
+  do_not_couple_implementation_to_this_string.acronym "CSV"
+end

tests/fixtures/layer_violations_with_overrides/package.yml

@@ -0,0 +1,2 @@
+# feature_flags, a utility pack, should not rely on Payments, a product pack
+Payments

tests/fixtures/layer_violations_with_overrides/packs/feature_flags/app/services/feature_flags.rb

@@ -0,0 +1,2 @@
+enforce_layers: true
+layer: utilities

tests/fixtures/layer_violations_with_overrides/packs/feature_flags/package.yml

@@ -0,0 +1,2 @@
+module Payments
+end