Refs #7057 Add some improvements to select type view & compliance control table builder on subclass column

1 files changed, 11 insertions, 0 deletions

diff --git a/app/models/compliance_control.rb b/app/models/compliance_control.rb
index 59c359bd5..4a2a6e91c 100644
--- a/app/models/compliance_control.rb
+++ b/app/models/compliance_control.rb
@@ -20,6 +20,17 @@ class ComplianceControl < ApplicationModel
       }
     end
 
+    def subclasses_to_hash
+      ComplianceControl.subclasses.inject(Hash.new{ |h, k| h[k] = [] } ) do |h, klass|
+        h[ComplianceControl.subclass_patterns.key(klass.object_type)] << klass
+        h 
+      end
+    end
+
+    def translated_subclass
+      I18n.t("compliance_controls.filters.subclasses.#{subclass_patterns.key(self.object_type)}")
+    end
+
     def object_type
       self.default_code.match(/^\d+-(?'object_type'\w+)-\d+$/)[:object_type]
     end