lib/aptible/rails/controller.rb in aptible-rails-0.4.6 vs lib/aptible/rails/controller.rb in aptible-rails-0.4.7

@@ -43,9 +43,17 @@
       # before_action :ensure_service_token
       def ensure_service_token
         redirect_to aptible_login_url unless service_token
       end
 
+      # before_action :ensure_compliance_plan
+      def ensure_compliance_plan
+        unless current_organization &&
+               current_organization.can_manage_compliance?
+          redirect_to compliance_url
+        end
+      end
+
       # before_action :ensure_auth_key
       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?