From 7110330a7e154fb8d746f6b15e12c7b42c6c00dc Mon Sep 17 00:00:00 2001
From: Xinhui
Date: Tue, 19 Sep 2017 12:33:36 +0200
Subject: Compliance Control decorator

Refs #4543
---
 app/controllers/compliance_controls_controller.rb | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

(limited to 'app/controllers/compliance_controls_controller.rb')

diff --git a/app/controllers/compliance_controls_controller.rb b/app/controllers/compliance_controls_controller.rb
index d198f2cdb..dad9b935a 100644
--- a/app/controllers/compliance_controls_controller.rb
+++ b/app/controllers/compliance_controls_controller.rb
@@ -1,7 +1,16 @@
 class ComplianceControlsController < BreadcrumbController
+  include PolicyChecker
   defaults resource_class: ComplianceControl
   belongs_to :compliance_control_set
 
+  def index
+    index! do |format|
+      format.html {
+        @compliance_controls = decorate_compliance_controls(@compliance_controls)
+      }
+    end
+  end
+
   def create
     create!(notice: t('notice.compliance_control.created'))
   end
@@ -16,6 +25,13 @@ class ComplianceControlsController < BreadcrumbController
   end
 
   private
+  def decorate_compliance_controls(compliance_controls)
+    ModelDecorator.decorate(
+      compliance_controls,
+      with: ComplianceControlDecorator,
+    )
+  end
+
   def compliance_control_params
     params.require(:compliance_control).permit(:name, :code, :criticity, :comment, :control_attributes)
   end
-- 
cgit v1.2.3