controller.rb in aptible-rails-0.6.7

- old
+ new

@@ -9,11 +9,12 @@
 
       included do
         helper_method :current_user, :current_organization, :user_url,
                       :organization_url, :criterion_by_handle, :auth_url,
                       :risk_criterion, :policy_criterion, :security_criterion,
-                      :training_criterion, :url_helpers
+                      :training_criterion, :url_helpers, :compliance_alerts,
+                      :criteria
       end
 
       def current_user
         return unless current_user_url
         @current_user ||= Aptible::Auth::User.find_by_url(current_user_url,
@@ -75,9 +76,25 @@
       def ensure_auth_key
         return if Fridge.configuration.public_key
         Fridge.configure do |config|
           config.public_key = Aptible::Auth.public_key unless ::Rails.env.test?
         end
+      end
+
+      def criteria
+        @criteria ||= Aptible::Gridiron::Criterion.where(
+          token: service_token,
+          organization: current_organization
+        )
+      end
+
+      def compliance_alerts
+        return @compliance_alerts if @compliance_alerts
+        @apps = Aptible::Api::App.all(token: service_token)
+        @users = current_organization.users
+        @compliance_alerts = ComplianceAlertCollection.new(
+                               @criteria, @apps, @users
+                             ).all
       end
 
       def service_token
         return unless session_token && session_token.session
         return @service_token if @service_token