Implement Erwan's feedback #853

This commit is contained in:
ethicalhack3r
2015-09-05 13:49:03 +02:00
parent e6d29f6f18
commit db528b27f4

View File

@@ -172,11 +172,11 @@ class WpItems < Array
case type case type
when :vulnerable when :vulnerable
items = json.select { |item| !json[item]['vulnerabilities'].empty? } items = json.select { |item| !json[item]['vulnerabilities'].empty? }.keys
when :popular when :popular
items = json.select { |item| json[item]['popular'] == true } items = json.select { |item| json[item]['popular'] == true }.keys
when :all when :all
items = [*json] items = json.keys
else else
raise "Unknown type #{type}" raise "Unknown type #{type}"
end end
@@ -184,7 +184,7 @@ class WpItems < Array
items.each do |item| items.each do |item|
targets << create_item( targets << create_item(
item_class, item_class,
item[0], item,
wp_target, wp_target,
vulns_file vulns_file
) )