Merge pull request #14041 from lpichler/add_rbac_for_rss_feed

Add RBAC for rss feeds

app/models/miq_widget/rss_content.rb

@@ -9,22 +9,22 @@ def external?
     resource.nil?
   end
 
-  def generate(_user_or_group)
+  def generate(user_or_group)
     opts = {:tz => timezone}
 
     data = if external?
              opts[:limit_to_count] = widget_options[:row_count] || 5
              external_feed
            else
-             internal_feed
+             internal_feed(user_or_group)
            end
 
     RssFeed.to_html(data, opts)
   end
 
-  def internal_feed
+  def internal_feed(user_or_group)
     resource.options[:limit_to_count] = widget_options[:row_count] || 5
-    SimpleRSS.parse(resource.generate)
+    SimpleRSS.parse(resource.generate(nil, nil, nil, user_or_group))
   end
 
   def external_feed

app/models/rss_feed.rb

@@ -17,7 +17,7 @@ def url(host = nil)
     "#{host_url}#{link}"
   end
 
-  def generate(host = nil, local = false, proto = nil)
+  def generate(host = nil, local = false, proto = nil, user_or_group = nil)
     proto ||= ::Settings.webservices.consume_protocol
     host_url = host.nil? ? "#{proto}://localhost:3000" : "#{proto}://" + host
 
@@ -35,7 +35,14 @@ def generate(host = nil, local = false, proto = nil)
       }
     }
 
-    feed = Rss.rss_feed_for(find_items, options)
+    rbac_options = {}
+    if user_or_group
+      user_or_group_key = user_or_group.kind_of?(User) ? :user : :miq_group
+      rbac_options[user_or_group_key] = user_or_group
+    end
+
+    filtered_items = Rbac::Filterer.filtered(find_items, rbac_options)
+    feed = Rss.rss_feed_for(filtered_items, options)
     local ? feed : {:text => feed, :content_type => Mime[:rss]}
   end
 

product/alerts/rss/newest_vms.yml

@@ -27,7 +27,7 @@ item_class: Vm
 search_method:
 limit_to_time:
 limit_to_count:
-orderby: "created_on DESC"
+orderby: "vms.created_on DESC"
 
 # Included tables and columns for query performance
 include:

spec/models/rss_feed/data/newest_vms.yml

@@ -27,7 +27,7 @@ item_class: Vm
 search_method:
 limit_to_time:
 limit_to_count:
-orderby: "created_on DESC"
+orderby: "vms.created_on DESC"
 
 # Included tables and columns for query performance
 include:

spec/models/rss_feed/rss_feed_spec.rb

@@ -3,6 +3,51 @@
 
   before(:each) { Kernel.silence_warnings { RssFeed.const_set(:YML_DIR, Y_DIR) } }
 
+  context "with vms" do
+    before do
+      Tenant.seed
+      allow(User).to receive_messages(:server_timezone => "UTC")
+      RssFeed.sync_from_yml_file("newest_vms")
+    end
+
+    let(:owner_tenant)    { FactoryGirl.create(:tenant) }
+    let(:owner_group)     { FactoryGirl.create(:miq_group, :tenant => owner_tenant) }
+    let(:owner_user)      { FactoryGirl.create(:user, :miq_groups => [owner_group]) }
+    let!(:owned_vm)       { FactoryGirl.create(:vm_vmware, :tenant => owner_tenant) }
+    let!(:tenant_root_vm) { FactoryGirl.create(:vm_vmware, :tenant => Tenant.root_tenant) }
+    let(:rss_feed)        { RssFeed.find_by(:name => "newest_vms") }
+
+    it "#generate 1 vms with owner_tenant tenant in newest_vms rss" do
+      [owner_group, owner_user].each do |user_or_group|
+        User.with_user(owner_user) do
+          feed_container = rss_feed.generate(nil, nil, nil, user_or_group)
+
+          expect(feed_container[:text]).to eq <<-EOXML
+<?xml version=\"1.0\" encoding=\"UTF-8\"?>
+<rss version=\"2.0\">
+  <channel>
+    <title>Recently Discovered VMs</title>
+    <link>https://localhost:3000/alert/rss?feed=newest_vms</link>
+    <description>Virtual machines added</description>
+    <language>en-us</language>
+    <ttl>40</ttl>
+    <item>
+      <title>#{owned_vm.name} - location unknown</title>
+      <description>#{owned_vm.name} is a #{owned_vm.vendor_display} VM located at "#{owned_vm.location}"</description>
+      <pubDate>#{owned_vm.created_on.rfc2822}</pubDate>
+      <guid>https://localhost:3000/vm/show/#{owned_vm.id}</guid>
+      <link>https://localhost:3000/vm/show/#{owned_vm.id}</link>
+    </item>
+  </channel>
+</rss>
+          EOXML
+
+          expect(feed_container[:content_type]).to eq('application/rss+xml')
+        end
+      end
+    end
+  end
+
   context "with 2 hosts" do
     before(:each) do
       @host1 = FactoryGirl.create(:host,           :created_on => Time.utc(2013, 1, 1, 0, 0, 0))